1 files changed, 46 insertions, 0 deletions
diff --git a/spec/migrations/disable_expiration_policies_linked_to_no_container_images_spec.rb b/spec/migrations/disable_expiration_policies_linked_to_no_container_images_spec.rb
new file mode 100644
index 00000000000..f2be06f1ed6
--- /dev/null
+++ b/spec/migrations/disable_expiration_policies_linked_to_no_container_images_spec.rb
@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+require_migration!
+
+RSpec.describe DisableExpirationPoliciesLinkedToNoContainerImages do
+  let(:projects) { table(:projects) }
+  let(:container_expiration_policies) { table(:container_expiration_policies) }
+  let(:container_repositories) { table(:container_repositories) }
+  let(:namespaces) { table(:namespaces) }
+
+  let!(:namespace) { namespaces.create!(name: 'test', path: 'test') }
+  let!(:project) { projects.create!(id: 1, namespace_id: namespace.id, name: 'gitlab1') }
+  let!(:container_expiration_policy) { container_expiration_policies.create!(project_id: project.id, enabled: true) }
+
+  before do
+    projects.create!(id: 2, namespace_id: namespace.id, name: 'gitlab2')
+    container_expiration_policies.create!(project_id: 2, enabled: true)
+    container_repositories.create!(id: 1, project_id: 2, name: 'image2')
+
+    projects.create!(id: 3, namespace_id: namespace.id, name: 'gitlab3')
+    container_expiration_policies.create!(project_id: 3, enabled: false)
+    container_repositories.create!(id: 2, project_id: 3, name: 'image3')
+  end
+
+  it 'correctly disable expiration policies linked to no container images' do
+    expect(enabled_policies.count).to eq 2
+    expect(disabled_policies.count).to eq 1
+    expect(container_expiration_policy.enabled).to eq true
+
+    migrate!
+
+    expect(enabled_policies.count).to eq 1
+    expect(disabled_policies.count).to eq 2
+    expect(container_expiration_policy.reload.enabled).to eq false
+  end
+
+  def enabled_policies
+    container_expiration_policies.where(enabled: true)
+  end
+
+  def disabled_policies
+    container_expiration_policies.where(enabled: false)
+  end
+end