OpenCHAMI · rainest · May 30, 2025 · May 30, 2025 · May 30, 2025 · May 30, 2025
diff --git a/internal/api/api_helpers.go b/internal/api/api_helpers.go
@@ -34,6 +34,14 @@ import (
 	"github.com/OpenCHAMI/power-control/v2/internal/model"
 )
 
+type IDList struct {
+	IDs []uuid.UUID `json:"ids"`
+}
+
+type IDResp struct {
+	ID uuid.UUID `json:"id"`
+}
+
 // WriteJSON - writes JSON to the open http connection
 func WriteJSON(w http.ResponseWriter, i interface{}) {
 	obj, err := json.Marshal(i)
@@ -57,9 +65,9 @@ func WriteHeaders(w http.ResponseWriter, pb model.Passback) {
 		w.WriteHeader(pb.StatusCode)
 		switch val := pb.Obj.(type) {
 		case []uuid.UUID:
-			WriteJSON(w, model.IDList{IDs: val})
+			WriteJSON(w, IDList{IDs: val})
 		case uuid.UUID:
-			WriteJSON(w, model.IDResp{ID: val})
+			WriteJSON(w, IDResp{ID: val})
 		default:
 			WriteJSON(w, pb.Obj)
 		}

diff --git a/internal/api/api_test_setup_handler.go b/internal/api/api_test_setup_handler.go
diff --git a/internal/model/models.go b/internal/model/models.go
diff --git a/internal/model/models_test.go b/internal/model/models_test.go
diff --git a/internal/storage/power_status.go b/internal/storage/power_status.go
diff --git a/internal/storage/transition_size_test.go b/internal/storage/transition_size_test.go
@@ -64,9 +64,9 @@ func createProviders(t *testing.T, settings *EtcdTestSettings) (StorageProvider,
 		t.Errorf("Storage Init() failed: %v", err)
 	}
 
-	err = ds.Init(nil)
+	ds.Init(nil)
 	if err != nil {
-		t.Errorf("DistributedLockProvider Init() failed: %v", err)
+		t.Errorf("DistLock Init() failed: %v", err)
 	}
 
 	if e, ok := ms.(*ETCDStorage); ok {