Fix tests

Author: dhurley

internal/command/command_service_test.go

@@ -237,8 +237,6 @@ func TestCommandService_UpdateClient_requestInProgress(t *testing.T) {
 		wg.Done()
 	}()
 
-	time.Sleep(100 * time.Millisecond)
-	delete(commandService.requestsInProgress, instanceID)
 	wg.Wait()
 
 	require.NoError(t, updateClientErr)

test/types/config.go

@@ -31,6 +31,7 @@ const (
 
 	maxParallelFileOperations = 5
 	reloadMonitoringPeriod    = 400 * time.Millisecond
+	connectionResetTimeout    = 200 * time.Millisecond
 )
 
 // Produces a populated Agent Config for testing usage.
@@ -58,6 +59,7 @@ func AgentConfig() *config.Config {
 				MaxFileSize:               1,
 				FileChunkSize:             1,
 				MaxParallelFileOperations: maxParallelFileOperations,
+				ConnectionResetTimeout:    connectionResetTimeout,
 			},
 			Backoff: &config.BackOff{
 				InitialInterval:     commonInitialInterval,