tewsts.

Comfy-Org · Jun 15, 2024 · 3f6f0e4 · 3f6f0e4
1 parent 0a0f583
commit 3f6f0e4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/integration-tests/registry_integration_test.go b/integration-tests/registry_integration_test.go
@@ -630,6 +630,7 @@ func TestRegistry(t *testing.T) {
 				Dependencies: &dependencies,
 				DownloadUrl:  &downloadUrl,
 				Status:       &nodeVersionStatus,
+				StatusReason: proto.String(""),
 			}, resVersions200[0], "should be equal")
 		})
 
@@ -664,6 +665,7 @@ func TestRegistry(t *testing.T) {
 				Dependencies: &dependencies,
 				DownloadUrl:  &downloadUrl,
 				Status:       &status,
+				StatusReason: proto.String(""),
 			}
 			assert.Equal(t, updatedNodeVersion, res200[0], "should be equal")
 			createdNodeVersion = res200[0]

diff --git a/server/implementation/registry.go b/server/implementation/registry.go
@@ -1219,7 +1219,11 @@ func (s *DripStrictServerImplementation) AdminUpdateNodeVersion(
 	}
 
 	dbNodeVersion := mapper.ApiNodeVersionStatusToDbNodeVersionStatus(*request.Body.Status)
-	err = nodeVersion.Update().SetStatus(dbNodeVersion).SetStatusReason(*request.Body.StatusReason).Exec(ctx)
+	statusReason := ""
+	if request.Body.StatusReason != nil {
+		statusReason = *request.Body.StatusReason
+	}
+	err = nodeVersion.Update().SetStatus(dbNodeVersion).SetStatusReason(statusReason).Exec(ctx)
 	if err != nil {
 		log.Ctx(ctx).Error().Msgf("Failed to update node version w/ err: %v", err)
 		return drip.AdminUpdateNodeVersion500JSONResponse{}, err