Use new mock behaviour in specs

appbot · Aug 18, 2021 · 97505b9 · 97505b9
1 parent ab60c70
commit 97505b9
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/spec/spec_helper.rb b/spec/spec_helper.rb
@@ -38,9 +38,6 @@
 # Set up codebase
 require 'kms_rails'
 require 'kms_rails/kms_client_mock'
-KmsRails.configure do |config|
-  config.fake_kms_api = true
-end
 
 # Disable ActiveJob noise
 ActiveJob::Base.logger = Logger.new(nil)
@@ -146,6 +143,12 @@
   # end
 
   # Ensure configuration is reset after each test
+  config.before(:each) do
+    KmsRails.configure do |c|
+      c.kms_client = KmsRails::KmsClientMock.new
+    end
+  end
+
   config.after(:each) do
     KmsRails.reset_config
   end