Fix integration

metal-stack · Oct 11, 2024 · 44334a3 · 44334a3
1 parent 94910c3
commit 44334a3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/cmd/metal-api/internal/datastore/switch.go b/cmd/metal-api/internal/datastore/switch.go
@@ -138,6 +138,9 @@ func (rs *RethinkStore) SetVrfAtSwitches(m *metal.Machine, vrf string) (metal.Sw
 }
 
 func (rs *RethinkStore) ConnectMachineWithSwitches(m *metal.Machine) error {
+	if m.PartitionID == "" {
+		return fmt.Errorf("partitionID is empty in machine:%s", m.ID)
+	}
 	var switches metal.Switches
 	err := rs.SearchSwitches(&SwitchSearchQuery{PartitionID: &m.PartitionID}, &switches)
 	if err != nil {

diff --git a/cmd/metal-api/internal/service/machine-service_integration_test.go b/cmd/metal-api/internal/service/machine-service_integration_test.go
@@ -47,7 +47,8 @@ func TestMachineAllocationIntegrationFullCycle(t *testing.T) {
 
 	// Register a machine
 	mrr := &grpcv1.BootServiceRegisterRequest{
-		Uuid: "test-uuid",
+		Uuid:        "test-uuid",
+		PartitionId: "test-partition",
 		Bios: &grpcv1.MachineBIOS{
 			Version: "a",
 			Vendor:  "metal",