Skip to content

Commit 094ecd7

Browse files
committed
fix unit tests
1 parent 7790669 commit 094ecd7

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

pkg/workloads/manager_test.go

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1252,6 +1252,8 @@ func TestDefaultManager_updateSingleWorkload(t *testing.T) {
12521252
State: "running",
12531253
Labels: map[string]string{"toolhive-basename": "test-workload"},
12541254
}, nil)
1255+
// Mock GetWorkloadPID call from stopProcess
1256+
sm.EXPECT().GetWorkloadPID(gomock.Any(), "test-workload").Return(1234, nil)
12551257
rt.EXPECT().StopWorkload(gomock.Any(), "test-workload").Return(nil)
12561258
sm.EXPECT().ResetWorkloadPID(gomock.Any(), "test-workload").Return(nil)
12571259

@@ -1288,6 +1290,8 @@ func TestDefaultManager_updateSingleWorkload(t *testing.T) {
12881290
State: "running",
12891291
Labels: map[string]string{"toolhive-basename": "test-workload"},
12901292
}, nil)
1293+
// Mock GetWorkloadPID call from stopProcess
1294+
sm.EXPECT().GetWorkloadPID(gomock.Any(), "test-workload").Return(1234, nil)
12911295
rt.EXPECT().StopWorkload(gomock.Any(), "test-workload").Return(nil)
12921296
sm.EXPECT().ResetWorkloadPID(gomock.Any(), "test-workload").Return(nil)
12931297

0 commit comments

Comments
 (0)