Fix a bug migrating CI job artifact registry entries to a separate table

ee/changelogs/unreleased/5646-bad-job-artifact-registry-migration.yml

+---
+title: Fix a bug migrating CI job artifact registry entries to a separate table
+merge_request: 5345
+author:
+type: fixed

ee/db/geo/migrate/20180322062741_migrate_ci_job_artifacts_to_separate_registry.rb

@@ -25,7 +25,10 @@ class MigrateCiJobArtifactsToSeparateRegistry < ActiveRecord::Migration
           $BODY$
           BEGIN
               IF (TG_OP = 'UPDATE') THEN
-                  UPDATE job_artifact_registry SET (retry_at, bytes, retry_count, success, sha256) = (NEW.retry_at, NEW.bytes, NEW.retry_count, NEW.success, NEW.sha256);
+                  UPDATE job_artifact_registry
+                  SET (retry_at, bytes, retry_count, success, sha256) =
+                      (NEW.retry_at, NEW.bytes, NEW.retry_count, NEW.success, NEW.sha256)
+                  WHERE artifact_id = NEW.file_id;
               ELSEIF (TG_OP = 'INSERT') THEN
                   INSERT INTO job_artifact_registry (created_at, retry_at, artifact_id, bytes, retry_count, success, sha256)
                   VALUES (NEW.created_at, NEW.retry_at, NEW.file_id, NEW.bytes, NEW.retry_count, NEW.success, NEW.sha256);

ee/spec/migrations/geo/migrate_ci_job_artifacts_to_separate_registry_spec.rb

@@ -47,6 +47,8 @@ describe MigrateCiJobArtifactsToSeparateRegistry, :geo, :migration do
       entry.update_attributes(success: false, bytes: 10240, sha256: '10' * 64)
 
       expect(job_artifact_registry.where(artifact_id: 1, success: false, bytes: 10240, sha256: '10' * 64).count).to eq(1)
+      # Ensure that *only* the correct job artifact is updated
+      expect(job_artifact_registry.find_by(artifact_id: 2).bytes).to eq(2048)
     end
 
     it 'creates a new artifact using the next ID' do