X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/797d78f4e7bc7807d42366054d6d49b4a550d8b9..70ab42cb727b6a1bd1a8b26e2481de7caa0853d5:/apps/workbench/test/test_helper.rb?ds=sidebyside

diff --git a/apps/workbench/test/test_helper.rb b/apps/workbench/test/test_helper.rb
index 695d01ad5b..fdde55d41d 100644
--- a/apps/workbench/test/test_helper.rb
+++ b/apps/workbench/test/test_helper.rb
@@ -36,8 +36,11 @@ class ActiveSupport::TestCase
     auth = api_fixture('api_client_authorizations')[token_name.to_s]
     Thread.current[:arvados_api_token] = auth['api_token']
     if block_given?
-      yield
-      Thread.current[:arvados_api_token] = was
+      begin
+        yield
+      ensure
+        Thread.current[:arvados_api_token] = was
+      end
     end
   end