1 files changed, 65 insertions, 0 deletions
diff --git a/spec/lib/gitlab/database/migrations/squasher_spec.rb b/spec/lib/gitlab/database/migrations/squasher_spec.rb
new file mode 100644
index 00000000000..e7ab5873f73
--- /dev/null
+++ b/spec/lib/gitlab/database/migrations/squasher_spec.rb
@@ -0,0 +1,65 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+RSpec.describe Gitlab::Database::Migrations::Squasher, feature_category: :database do
+  let(:git_output) do
+    <<~FILES
+    db/migrate/misplaced.txt
+    db/migrate/20221003041700_init_schema.rb
+    db/migrate/20221003041800_foo_migrate.rb
+    db/migrate/20221003041900_foo_migrate_two.rb
+    db/migrate/20221003042000_add_name_to_widgets.rb
+    db/migrate/20221003042200_add_enterprise.rb
+    db/post_migrate/20221003042100_post_migrate.rb
+    FILES
+  end
+
+  let(:spec_files) do
+    [
+      'spec/migrations/add_name_to_widgets_spec.rb',
+      'spec/migrations/20221003041800_foo_migrate_spec.rb',
+      'spec/migrations/foo_migrate_three_spec.rb',
+      'spec/migrations/foo_migrate_two_spec.rb',
+      'spec/migrations/post_migrate_spec.rb'
+    ]
+  end
+
+  let(:ee_spec_files) do
+    [
+      'ee/spec/migrations/add_enterprise_spec.rb'
+    ]
+  end
+
+  let(:expected_list) do
+    [
+      'db/migrate/20221003041800_foo_migrate.rb',
+      'db/migrate/20221003041900_foo_migrate_two.rb',
+      'db/migrate/20221003042000_add_name_to_widgets.rb',
+      'spec/migrations/add_name_to_widgets_spec.rb',
+      'spec/migrations/20221003041800_foo_migrate_spec.rb',
+      'spec/migrations/foo_migrate_two_spec.rb',
+      'db/schema_migrations/20221003041800',
+      'db/schema_migrations/20221003041900',
+      'db/schema_migrations/20221003042000',
+      'db/schema_migrations/20221003042100',
+      'db/schema_migrations/20221003042200',
+      'db/post_migrate/20221003042100_post_migrate.rb',
+      'spec/migrations/post_migrate_spec.rb',
+      'ee/spec/migrations/add_enterprise_spec.rb',
+      'db/migrate/20221003042200_add_enterprise.rb'
+    ]
+  end
+
+  describe "#files_to_delete" do
+    before do
+      allow(Dir).to receive(:glob).with(Rails.root.join('spec/migrations/*.rb')).and_return(spec_files)
+      allow(Dir).to receive(:glob).with(Rails.root.join('ee/spec/migrations/*.rb')).and_return(ee_spec_files)
+    end
+
+    let(:squasher) { described_class.new(git_output) }
+
+    it 'only deletes the files we\'re expecting' do
+      expect(squasher.files_to_delete).to match_array expected_list
+    end
+  end
+end