append tuid to metadata (#451)

CDCgov · Aug 19, 2024 · 5ce6a95 · 5ce6a95
1 parent a3a18f9
commit 5ce6a95
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/upload-server/internal/metadata/metadata.go b/upload-server/internal/metadata/metadata.go
@@ -380,8 +380,8 @@ func WithPreCreateManifestTransforms(event handler.HookEvent, resp hooks.HookRes
 	logger.Info("adding global timestamp", "timestamp", timestamp)
 
 	manifest := event.Upload.MetaData
-	fieldname := "dex_ingest_datetime"
-	manifest[fieldname] = timestamp
+	manifest["dex_ingest_datetime"] = timestamp
+	manifest["upload_id"] = tuid
 	resp.ChangeFileInfo.MetaData = manifest
 
 	report := reports.NewBuilderWithManifest[reports.BulkMetadataTransformReportContent](
@@ -399,8 +399,12 @@ func WithPreCreateManifestTransforms(event handler.HookEvent, resp hooks.HookRes
 				Field: "ID",
 				Value: tuid}, {
 				Action: "append",
-				Field:  fieldname,
+				Field:  "dex_ingest_datetime",
 				Value:  timestamp,
+			}, {
+				Action: "append",
+				Field:  "upload_id",
+				Value:  tuid,
 			}},
 	}).Build()
 

diff --git a/upload-server/testing/cli_test.go b/upload-server/testing/cli_test.go
@@ -184,6 +184,13 @@ func TestTus(t *testing.T) {
 						}
 					}
 				}
+
+				appendedUid, ok := processedMeta["upload_id"]
+				if !ok {
+					t.Error("upload ID not appended to file metadata")
+				} else if appendedUid != tuid {
+					t.Error("appended upload ID did not match upload ID", appendedUid, tuid)
+				}
 			}
 
 			t.Log("test case", name, "passed", tuid)