Implemente measurement enabled cache using AtomicReference

1 files changed, 2 insertions, 2 deletions

diff --git a/spec/lib/gitlab/metrics/method_call_spec.rb b/spec/lib/gitlab/metrics/method_call_spec.rb
index f514fdd70ed..eca6a2cb54a 100644
--- a/spec/lib/gitlab/metrics/method_call_spec.rb
+++ b/spec/lib/gitlab/metrics/method_call_spec.rb
@@ -21,7 +21,7 @@ describe Gitlab::Metrics::MethodCall do
       context 'prometheus instrumentation is enabled' do
         before do
           allow(Feature.get(:prometheus_metrics_method_instrumentation)).to receive(:enabled?).and_call_original
-          described_class.call_measurement_enabled_cache_expire
+          described_class::MEASUREMENT_ENABLED_CACHE.set({enabled: false, expires_at: Time.now - 1.second})
           Feature.get(:prometheus_metrics_method_instrumentation).enable
         end
 
@@ -62,7 +62,7 @@ describe Gitlab::Metrics::MethodCall do
 
       context 'prometheus instrumentation is disabled' do
         before do
-          described_class.call_measurement_enabled_cache_expire
+          described_class::MEASUREMENT_ENABLED_CACHE.set({enabled: false, expires_at: Time.now})
           Feature.get(:prometheus_metrics_method_instrumentation).disable
         end