1 files changed, 13 insertions, 3 deletions
diff --git a/spec/support/shared_examples/workers/batched_background_migration_worker_shared_examples.rb b/spec/support/shared_examples/workers/batched_background_migration_worker_shared_examples.rb
index d202c4e00f0..26731f34ed6 100644
--- a/spec/support/shared_examples/workers/batched_background_migration_worker_shared_examples.rb
+++ b/spec/support/shared_examples/workers/batched_background_migration_worker_shared_examples.rb
@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-RSpec.shared_examples 'it runs batched background migration jobs' do |tracking_database|
+RSpec.shared_examples 'it runs batched background migration jobs' do |tracking_database, feature_flag:|
   include ExclusiveLeaseHelpers
 
   describe 'defining the job attributes' do
@@ -39,6 +39,16 @@ RSpec.shared_examples 'it runs batched background migration jobs' do |tracking_d
     end
   end
 
+  describe '.enabled?' do
+    it 'does not raise an error' do
+      expect { described_class.enabled? }.not_to raise_error
+    end
+
+    it 'returns true' do
+      expect(described_class.enabled?).to be_truthy
+    end
+  end
+
   describe '#perform' do
     subject(:worker) { described_class.new }
 
@@ -76,7 +86,7 @@ RSpec.shared_examples 'it runs batched background migration jobs' do |tracking_d
 
       context 'when the feature flag is disabled' do
         before do
-          stub_feature_flags(execute_batched_migrations_on_schedule: false)
+          stub_feature_flags(feature_flag => false)
         end
 
         it 'does nothing' do
@@ -89,7 +99,7 @@ RSpec.shared_examples 'it runs batched background migration jobs' do |tracking_d
 
       context 'when the feature flag is enabled' do
         before do
-          stub_feature_flags(execute_batched_migrations_on_schedule: true)
+          stub_feature_flags(feature_flag => true)
 
           allow(Gitlab::Database::BackgroundMigration::BatchedMigration).to receive(:active_migration).and_return(nil)
         end