From 07e3f7e804a372d95622da46d9bdc77c9f0f98e1 Mon Sep 17 00:00:00 2001 From: sophia Date: Wed, 4 May 2022 09:45:51 -0500 Subject: [PATCH] Fix machine state tests --- internal/core/machine_test.go | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/internal/core/machine_test.go b/internal/core/machine_test.go index b5ca2e7b6..9a136a9a5 100644 --- a/internal/core/machine_test.go +++ b/internal/core/machine_test.go @@ -94,6 +94,7 @@ func TestMachineGetNonExistentBox(t *testing.T) { require.NoError(t, err) require.Empty(t, metaurl) } + func TestMachineGetExistentBox(t *testing.T) { tp := TestMinimalProject(t) tm, _ := TestMachine(t, tp, @@ -238,11 +239,8 @@ func TestMachineSetState(t *testing.T) { // Set MachineState desiredState := &core.MachineState{ID: tc.id} tm.SetMachineState(desiredState) - newState, err := tm.MachineState() - if err != nil { - t.Errorf("Failed to get id") - } - require.Equal(t, newState, desiredState) + require.Equal(t, tm.machine.State.Id, tc.id) + require.Equal(t, tm.target.State, tc.state) // Ensure new id is save to db dbTarget, err := tm.Client().GetTarget(tm.ctx,