fix unit test

aws · May 16, 2023 · 888d2d9 · 888d2d9
1 parent efa4974
commit 888d2d9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/agent/app/agent_unix_test.go b/agent/app/agent_unix_test.go
@@ -721,6 +721,7 @@ func TestDoStartGPUManagerInitError(t *testing.T) {
 	mockServiceConnectManager.EXPECT().GetLoadedAppnetVersion().AnyTimes()
 	mockServiceConnectManager.EXPECT().GetCapabilitiesForAppnetInterfaceVersion("").AnyTimes()
 	mockServiceConnectManager.EXPECT().SetECSClient(gomock.Any(), gomock.Any()).AnyTimes()
+	client.EXPECT().GetHostResources().Return(testHostResource, nil).Times(1)
 
 	cfg := getTestConfig()
 	cfg.GPUSupportEnabled = true