databasefake

mtojek · mtojek · commit 2af41331ccc2 · 2022-11-22T16:02:08.000+01:00
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
 			}
 

Original file line number	Diff line number	Diff line change
`@@ -904,9 +904,12 @@ func (q *fakeQuerier) GetAuthorizedWorkspaces(ctx context.Context, arg database.`
`904`	`904`	`hasAgentValid = wa.LastConnectedAt.Valid`
`905`	`905`	`case "connecting":`
`906`	`906`	`hasAgentValid = !wa.FirstConnectedAt.Valid`
	`907`	`+ case "disconnected":`
	`908`	`+ hasAgentValid = wa.DisconnectedAt.Valid && wa.DisconnectedAt.Time.After(wa.LastConnectedAt.Time)`
	`909`	`+ case "timeout":`
	`910`	`+ hasAgentValid = !wa.FirstConnectedAt.Valid &&`
	`911`	`+ wa.CreatedAt.Add(time.Duration(wa.ConnectionTimeoutSeconds)*time.Second).Before(database.Now())`
`907`	`912`	`}`
`908`		`- // TODO disconnected`
`909`		`- // TODO timeout`
`910`	`913`	`break // only 1 agent is expected`
`911`	`914`	`}`
`912`	`915`