return args moved to .Run, remove duplicate

nais · Jun 5, 2024 · 5bfaba8 · 5bfaba8
1 parent 693162b
commit 5bfaba8
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/internal/device-agent/state_machine_test.go b/internal/device-agent/state_machine_test.go
@@ -34,11 +34,7 @@ func TestStateMachine(t *testing.T) {
 		rc.EXPECT().BuildHelperConfiguration(mock.Anything).Return(&pb.Configuration{})
 
 		mockGetDeviceConfigclient := pb.NewMockAPIServer_GetDeviceConfigurationClient(t)
-		recv := mockGetDeviceConfigclient.EXPECT().Recv().After(20*time.Millisecond).Return(
-			&pb.GetDeviceConfigurationResponse{
-				Status:   pb.DeviceConfigurationStatus_DeviceHealthy,
-				Gateways: []*pb.Gateway{},
-			}, nil)
+		recv := mockGetDeviceConfigclient.EXPECT().Recv().After(20 * time.Millisecond)
 		var streamContext context.Context
 		recv.Run(func(args mock.Arguments) {
 			if streamContext != nil && streamContext.Err() != nil {