1 files changed, 75 insertions, 0 deletions
diff --git a/spec/lib/system_check/app/table_truncate_check_spec.rb b/spec/lib/system_check/app/table_truncate_check_spec.rb
new file mode 100644
index 00000000000..673365f3e5e
--- /dev/null
+++ b/spec/lib/system_check/app/table_truncate_check_spec.rb
@@ -0,0 +1,75 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe SystemCheck::App::TableTruncateCheck, feature_category: :cell do
+  context 'when running on single databases' do
+    before do
+      skip_if_database_exists(:ci)
+    end
+
+    describe '#skip?' do
+      subject { described_class.new.skip? }
+
+      it { is_expected.to eq(true) }
+    end
+  end
+
+  context 'when running on multiple databases' do
+    let(:needs_truncation) { true }
+
+    before do
+      skip_if_shared_database(:ci)
+
+      allow_next_instances_of(Gitlab::Database::TablesTruncate, 2) do |instance|
+        allow(instance).to receive(:needs_truncation?).and_return(needs_truncation)
+      end
+    end
+
+    describe '#skip?' do
+      subject { described_class.new.skip? }
+
+      it { is_expected.to eq(false) }
+    end
+
+    describe '#check?' do
+      subject { described_class.new.check? }
+
+      context 'when TableTruncate returns false' do
+        let(:needs_truncation) { false }
+
+        it { is_expected.to eq(true) }
+      end
+
+      context 'when TableTruncate returns true' do
+        let(:needs_truncation) { true }
+
+        it { is_expected.to eq(false) }
+      end
+    end
+
+    describe '#show_error' do
+      let(:needs_truncation) { true }
+      let(:checker) { described_class.new }
+
+      before do
+        checker.check?
+      end
+
+      subject(:show_error) { checker.show_error }
+
+      it 'outputs error information' do
+        expected = %r{
+          Try\sfixing\sit:\s+
+          sudo\s-u\s.+?\s-H\sbundle\sexec\srake\sgitlab:db:truncate_legacy_tables:main\s
+          gitlab:db:truncate_legacy_tables:ci\s+
+          For\smore\sinformation\ssee:\s+
+          doc/development/database/multiple_databases.md\sin\ssection\s'Truncating\stables'\s+
+          Please\sfix\sthe\serror\sabove\sand\srerun\sthe\schecks.\s+
+        }x
+
+        expect { show_error }.to output(expected).to_stdout
+      end
+    end
+  end
+end