databasefake

coder · mtojek · Nov 24, 2022 · Nov 21, 2022 · Nov 22, 2022 · Nov 22, 2022
commit 2af41331ccc2264b02ac82bd6fd4a32c35a490c9
diff --git a/coderd/database/databasefake/databasefake.go b/coderd/database/databasefake/databasefake.go
@@ -904,9 +904,12 @@ func (q *fakeQuerier) GetAuthorizedWorkspaces(ctx context.Context, arg database.
 					hasAgentValid = wa.LastConnectedAt.Valid
 				case "connecting":
 					hasAgentValid = !wa.FirstConnectedAt.Valid
+				case "disconnected":
+					hasAgentValid = wa.DisconnectedAt.Valid && wa.DisconnectedAt.Time.After(wa.LastConnectedAt.Time)
+				case "timeout":
+					hasAgentValid = !wa.FirstConnectedAt.Valid &&
+						wa.CreatedAt.Add(time.Duration(wa.ConnectionTimeoutSeconds)*time.Second).Before(database.Now())
 				}
-				// TODO disconnected
-				// TODO timeout
 				break // only 1 agent is expected
 			}