1 files changed, 16 insertions, 0 deletions
diff --git a/db/post_migrate/20221216150919_validate_foreign_key_on_ci_unit_test_failure_partition_id_build_id.rb b/db/post_migrate/20221216150919_validate_foreign_key_on_ci_unit_test_failure_partition_id_build_id.rb
new file mode 100644
index 00000000000..a886949aea7
--- /dev/null
+++ b/db/post_migrate/20221216150919_validate_foreign_key_on_ci_unit_test_failure_partition_id_build_id.rb
@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+class ValidateForeignKeyOnCiUnitTestFailurePartitionIdBuildId < Gitlab::Database::Migration[2.1]
+  TABLE_NAME = :ci_unit_test_failures
+  FK_NAME = :fk_0f09856e1f_p
+  PARTITION_COLUMN = :partition_id
+  COLUMN = :build_id
+
+  def up
+    validate_foreign_key(TABLE_NAME, [PARTITION_COLUMN, COLUMN], name: FK_NAME)
+  end
+
+  def down
+    # no-op
+  end
+end