Revert errant change in f8e6cb3. refs #3126

[arvados.git] / apps / workbench / test / test_helper.rb

diff --git a/apps/workbench/test/test_helper.rb b/apps/workbench/test/test_helper.rb
index 335fcd0ee51b76fd04077cfbcbfc0bb39d381ced..ade6292efe6d2cd59adbeb4b8021642e1a00584a 100644 (file)
--- a/apps/workbench/test/test_helper.rb
+++ b/apps/workbench/test/test_helper.rb
@@ -32,11 +32,19 @@ class ActiveSupport::TestCase
   # in integration tests -- they do not yet inherit this setting
   fixtures :all
   def use_token token_name
+    was = Thread.current[:arvados_api_token]
     auth = api_fixture('api_client_authorizations')[token_name.to_s]
     Thread.current[:arvados_api_token] = auth['api_token']
+    if block_given?
+      begin
+        yield
+      ensure
+        Thread.current[:arvados_api_token] = was
+      end
+    end
   end
 
-  setup do
+  teardown do
     Thread.current[:arvados_api_token] = nil
     Thread.current[:user] = nil
     Thread.current[:reader_tokens] = nil
@@ -87,27 +95,24 @@ module ApiFixtureLoader
 end
 
 module ApiMockHelpers
-  def self.included base
-    base.class_eval do
-      def stub_api_calls_with_body body, status_code=200
-        resp = mock
-        stubbed_client = ArvadosApiClient.new
-        stubbed_client.instance_eval do
-          resp.responds_like_instance_of HTTP::Message
-          resp.stubs(:content).returns body
-          resp.stubs(:status_code).returns status_code
-          @api_client = HTTPClient.new
-          @api_client.stubs(:post).returns resp
-        end
-        ArvadosApiClient.stubs(:new_or_current).returns(stubbed_client)
-      end
-
-      def stub_api_calls_with_invalid_json
-        stub_api_calls_with_body ']"omg,bogus"['
-      end
+  def stub_api_calls_with_body body, status_code=200
+    resp = mock
+    stubbed_client = ArvadosApiClient.new
+    stubbed_client.instance_eval do
+      resp.responds_like_instance_of HTTP::Message
+      resp.stubs(:content).returns body
+      resp.stubs(:status_code).returns status_code
+      @api_client = HTTPClient.new
+      @api_client.stubs(:post).returns resp
     end
+    ArvadosApiClient.stubs(:new_or_current).returns(stubbed_client)
+  end
+
+  def stub_api_calls_with_invalid_json
+    stub_api_calls_with_body ']"omg,bogus"['
   end
 end
+
 class ActiveSupport::TestCase
   include ApiMockHelpers
 end