Function TestGetWorkspaceConn_StatusCheck

coderd/x/chatd/chatd_internal_test.go:4777–4923

(t *testing.T)

}

func TestGetWorkspaceConn_StatusCheck(t *testing.T) {
	// The cache-hit status check re-fetches the agent row for a fresh
	// heartbeat timestamp. Healthy, timed-out, and DB-error paths return
	// the cached connection. Disconnected agents are covered separately
	// because they now trigger a fresh dial before recovery.
	t.Parallel()

	type testCase struct {
		name       string
		buildAgent func(now time.Time) database.WorkspaceAgent
		dbError    bool
	}

	tests := []testCase{
		{
			// Agent never connected and the connection timeout
			// has elapsed. This should not trigger lifecycle
			// recovery because the agent did not connect and
			// then disconnect.
			name: "TimedOutAgentCacheHit",
			buildAgent: func(now time.Time) database.WorkspaceAgent {
				return database.WorkspaceAgent{
					CreatedAt:                now.Add(-10 * time.Minute),
					ConnectionTimeoutSeconds: 60,
				}
			},
		},
		{
			name: "CacheHitHealthyAgent",
			buildAgent: func(now time.Time) database.WorkspaceAgent {
				return database.WorkspaceAgent{
					FirstConnectedAt: sql.NullTime{
						Time:  now.Add(-5 * time.Minute),
						Valid: true,
					},
					LastConnectedAt: sql.NullTime{
						Time:  now,
						Valid: true,
					},
				}
			},
		},
		{
			// When GetWorkspaceAgentByID returns an error on
			// cache hit, the cached connection should be returned.
			name: "CacheHitDBError",
			buildAgent: func(now time.Time) database.WorkspaceAgent {
				return database.WorkspaceAgent{
					FirstConnectedAt: sql.NullTime{
						Time:  now.Add(-5 * time.Minute),
						Valid: true,
					},
					LastConnectedAt: sql.NullTime{
						Time:  now,
						Valid: true,
					},
				}
			},