Fix tests

kylecarbs · kylecarbs · commit 0f06b23ce3a7 · 2023-03-07T16:56:51.000Z
diff --git a/agent/agent_test.go b/agent/agent_test.go
@@ -68,7 +68,10 @@ func TestAgent_Stats_SSH(t *testing.T) {
 	session, err := sshClient.NewSession()
 	require.NoError(t, err)
 	defer session.Close()
-	require.NoError(t, session.Run("echo test"))
+	stdin, err := session.StdinPipe()
+	require.NoError(t, err)
+	err = session.Shell()
+	require.NoError(t, err)
 
 	var s *agentsdk.Stats
 	require.Eventuallyf(t, func() bool {
@@ -78,6 +81,9 @@ func TestAgent_Stats_SSH(t *testing.T) {
 	}, testutil.WaitLong, testutil.IntervalFast,
 		"never saw stats: %+v", s,
 	)
+	_ = stdin.Close()
+	err = session.Wait()
+	require.NoError(t, err)
 }
 
 func TestAgent_Stats_ReconnectingPTY(t *testing.T) {
diff --git a/codersdk/deployment.go b/codersdk/deployment.go
@@ -554,12 +554,12 @@ type WorkspaceConnectionLatencyMS struct {
 type DeploymentStats struct {
 	// AggregatedFrom is the time in which stats are aggregated from.
 	// This might be back in time a specific duration or interval.
-	AggregatedFrom time.Time `json:"aggregated_since"`
+	AggregatedFrom time.Time `json:"aggregated_since" format:"date-time"`
 	// CollectedAt is the time in which stats are collected at.
-	CollectedAt time.Time `json:"collected_at"`
+	CollectedAt time.Time `json:"collected_at" format:"date-time"`
 	// RefreshingAt is the time when the next batch of stats will
 	// be refreshed.
-	RefreshingAt time.Time `json:"refreshing_at"`
+	RefreshingAt time.Time `json:"refreshing_at" format:"date-time"`
 
 	PendingWorkspaces  int64 `json:"pending_workspaces"`
 	BuildingWorkspaces int64 `json:"building_workspaces"`
