1 files changed, 15 insertions, 0 deletions
diff --git a/spec/lib/gitlab/background_migration/batching_strategies/base_strategy_spec.rb b/spec/lib/gitlab/background_migration/batching_strategies/base_strategy_spec.rb
new file mode 100644
index 00000000000..56ed1f23799
--- /dev/null
+++ b/spec/lib/gitlab/background_migration/batching_strategies/base_strategy_spec.rb
@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Gitlab::BackgroundMigration::BatchingStrategies::BaseStrategy, '#next_batch' do
+  let(:connection) { double(:connection) }
+  let(:base_strategy_class) { Class.new(described_class) }
+  let(:base_strategy) { base_strategy_class.new(connection: connection) }
+
+  describe '#next_batch' do
+    it 'raises an error if not overridden by a subclass' do
+      expect { base_strategy.next_batch }.to raise_error(NotImplementedError, /does not implement next_batch/)
+    end
+  end
+end