4533: Merge branch 'master' into 4533-remote-reset

[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 ab2ac395b4c487b85643d6df5989934acd7e630f..4fd5aafb475c4432abd6b1f6a9c2a02d218b43e1 100644 (file)
--- a/apps/workbench/test/test_helper.rb
+++ b/apps/workbench/test/test_helper.rb
@@ -204,6 +204,35 @@ class ActionController::TestCase
   end
 end
 
+# Test classes can call reset_api_fixtures(:before_suite) or
+# ...(:after_suite)
+class ActiveSupport::TestCase
+  class << self
+    attr_accessor :want_reset_api_fixtures
+  end
+
+  def self.reset_api_fixtures where, t=true
+    raise unless [:before_suite, :after_suite].include? where
+    self.want_reset_api_fixtures ||= {}
+    self.want_reset_api_fixtures[where] = t
+  end
+
+  def self.run *args
+    self.want_reset_api_fixtures ||= {}
+    reset_api_fixtures_now if want_reset_api_fixtures[:before_suite]
+    super
+    reset_api_fixtures_now if want_reset_api_fixtures[:after_suite]
+  end
+
+  protected
+  def self.reset_api_fixtures_now
+    auth = api_fixture('api_client_authorizations')['admin_trustedclient']
+    Thread.current[:arvados_api_token] = auth['api_token']
+    ArvadosApiClient.new.api(nil, '../../database/reset', {})
+    Thread.current[:arvados_api_token] = nil
+  end
+end
+
 # If it quacks like a duck, it must be a HTTP request object.
 class RequestDuck
   def self.host