Update tests.

Comfy-Org · Jan 4, 2025 · 3b12e6e · 3b12e6e
1 parent 93b3b93
commit 3b12e6e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/integration-tests/registry_integration_test.go b/integration-tests/registry_integration_test.go
@@ -620,6 +620,7 @@ func TestRegistryNodeVersion(t *testing.T) {
 			DownloadUrl:  &downloadUrl,
 			Status:       &nodeVersionStatus,
 			StatusReason: proto.String(""),
+			NodeId:       node.Id,
 		}, resVersions200[0], "should be equal")
 
 		// retrieve node versions with the status reason
@@ -668,6 +669,7 @@ func TestRegistryNodeVersion(t *testing.T) {
 			DownloadUrl:  &downloadUrl,
 			Status:       &status,
 			StatusReason: proto.String(""),
+			NodeId:       node.Id,
 		}
 		assert.Equal(t, updatedNodeVersion, res200[0], "should be equal")
 		createdNodeVersion = res200[0]