1 files changed, 44 insertions, 0 deletions
diff --git a/spec/lib/gitlab/database/count_spec.rb b/spec/lib/gitlab/database/count_spec.rb
index d65413c2a00..e712ad09927 100644
--- a/spec/lib/gitlab/database/count_spec.rb
+++ b/spec/lib/gitlab/database/count_spec.rb
@@ -46,5 +46,49 @@ RSpec.describe Gitlab::Database::Count do
         subject
       end
     end
+
+    context 'default strategies' do
+      subject { described_class.approximate_counts(models) }
+
+      context 'with a read-only database' do
+        before do
+          allow(Gitlab::Database).to receive(:read_only?).and_return(true)
+        end
+
+        it 'only uses the ExactCountStrategy' do
+          allow_next_instance_of(Gitlab::Database::Count::TablesampleCountStrategy) do |instance|
+            expect(instance).not_to receive(:count)
+          end
+          allow_next_instance_of(Gitlab::Database::Count::ReltuplesCountStrategy) do |instance|
+            expect(instance).not_to receive(:count)
+          end
+          expect_next_instance_of(Gitlab::Database::Count::ExactCountStrategy) do |instance|
+            expect(instance).to receive(:count).and_return({})
+          end
+
+          subject
+        end
+      end
+
+      context 'with a read-write database' do
+        before do
+          allow(Gitlab::Database).to receive(:read_only?).and_return(false)
+        end
+
+        it 'uses the available strategies' do
+          [
+            Gitlab::Database::Count::TablesampleCountStrategy,
+            Gitlab::Database::Count::ReltuplesCountStrategy,
+            Gitlab::Database::Count::ExactCountStrategy
+          ].each do |strategy_klass|
+            expect_next_instance_of(strategy_klass) do |instance|
+              expect(instance).to receive(:count).and_return({})
+            end
+          end
+
+          subject
+        end
+      end
+    end
   end
 end