fixup! chore: refactor agent stats streaming

coadler · coadler · commit 8b793c9b0f08 · 2022-11-17T19:24:36.000-06:00
diff --git a/agent/agent_test.go b/agent/agent_test.go
@@ -69,6 +69,7 @@ func TestAgent(t *testing.T) {
 			session, err := sshClient.NewSession()
 			require.NoError(t, err)
 			defer session.Close()
+			require.NoError(t, session.Run("echo test"))
 
 			var s *codersdk.AgentStats
 			require.Eventuallyf(t, func() bool {
@@ -680,7 +681,7 @@ func setupAgent(t *testing.T, metadata codersdk.WorkspaceAgentMetadata, ptyTimeo
 	}
 	coordinator := tailnet.NewCoordinator()
 	agentID := uuid.New()
-	statsCh := make(chan *codersdk.AgentStats)
+	statsCh := make(chan *codersdk.AgentStats, 50)
 	fs := afero.NewMemMapFs()
 	closer := agent.New(agent.Options{
 		Client: &client{
@@ -787,7 +788,7 @@ func (c *client) AgentReportStats(ctx context.Context, _ slog.Logger, stats func
 	go func() {
 		defer close(doneCh)
 
-		t := time.NewTicker(time.Millisecond * 100)
+		t := time.NewTicker(500 * time.Millisecond)
 		defer t.Stop()
 		for {
 			select {
