File tree Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Original file line number Diff line number Diff line change @@ -1252,6 +1252,8 @@ func TestDefaultManager_updateSingleWorkload(t *testing.T) {
1252
1252
State : "running" ,
1253
1253
Labels : map [string ]string {"toolhive-basename" : "test-workload" },
1254
1254
}, nil )
1255
+ // Mock GetWorkloadPID call from stopProcess
1256
+ sm .EXPECT ().GetWorkloadPID (gomock .Any (), "test-workload" ).Return (1234 , nil )
1255
1257
rt .EXPECT ().StopWorkload (gomock .Any (), "test-workload" ).Return (nil )
1256
1258
sm .EXPECT ().ResetWorkloadPID (gomock .Any (), "test-workload" ).Return (nil )
1257
1259
@@ -1288,6 +1290,8 @@ func TestDefaultManager_updateSingleWorkload(t *testing.T) {
1288
1290
State : "running" ,
1289
1291
Labels : map [string ]string {"toolhive-basename" : "test-workload" },
1290
1292
}, nil )
1293
+ // Mock GetWorkloadPID call from stopProcess
1294
+ sm .EXPECT ().GetWorkloadPID (gomock .Any (), "test-workload" ).Return (1234 , nil )
1291
1295
rt .EXPECT ().StopWorkload (gomock .Any (), "test-workload" ).Return (nil )
1292
1296
sm .EXPECT ().ResetWorkloadPID (gomock .Any (), "test-workload" ).Return (nil )
1293
1297
You can’t perform that action at this time.
0 commit comments