1 files changed, 32 insertions, 0 deletions
diff --git a/spec/support/shared_examples/services/measurable_service_shared_examples.rb b/spec/support/shared_examples/services/measurable_service_shared_examples.rb
new file mode 100644
index 00000000000..206c25e49af
--- /dev/null
+++ b/spec/support/shared_examples/services/measurable_service_shared_examples.rb
@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+RSpec.shared_examples 'measurable service' do
+  context 'when measurement is enabled' do
+    let!(:measuring) { Gitlab::Utils::Measuring.new(base_log_data) }
+
+    before do
+      stub_feature_flags(feature_flag => true)
+    end
+
+    it 'measure service execution with Gitlab::Utils::Measuring', :aggregate_failures do
+      expect(Gitlab::Utils::Measuring).to receive(:new).with(base_log_data).and_return(measuring)
+      expect(measuring).to receive(:with_measuring).and_call_original
+    end
+  end
+
+  context 'when measurement is disabled' do
+    it 'does not measure service execution' do
+      stub_feature_flags(feature_flag => false)
+
+      expect(Gitlab::Utils::Measuring).not_to receive(:new)
+    end
+  end
+
+  def feature_flag
+    "gitlab_service_measuring_#{described_class_name}"
+  end
+
+  def described_class_name
+    described_class.name.underscore.tr('/', '_')
+  end
+end