try to fix batcher flake again

coadler · coadler · commit 9a08790073a0 · 2023-08-08T00:16:32.000Z
diff --git a/coderd/database/dbfake/dbfake.go b/coderd/database/dbfake/dbfake.go
@@ -2802,14 +2802,14 @@ func (q *FakeQuerier) GetWorkspaceAgentStats(_ context.Context, createdAfter tim
 
 	agentStatsCreatedAfter := make([]database.WorkspaceAgentStat, 0)
 	for _, agentStat := range q.workspaceAgentStats {
-		if agentStat.CreatedAt.After(createdAfter) {
+		if agentStat.CreatedAt.After(createdAfter) || agentStat.CreatedAt.Equal(createdAfter) {
 			agentStatsCreatedAfter = append(agentStatsCreatedAfter, agentStat)
 		}
 	}
 
 	latestAgentStats := map[uuid.UUID]database.WorkspaceAgentStat{}
 	for _, agentStat := range q.workspaceAgentStats {
-		if agentStat.CreatedAt.After(createdAfter) {
+		if agentStat.CreatedAt.After(createdAfter) || agentStat.CreatedAt.Equal(createdAfter) {
 			latestAgentStats[agentStat.AgentID] = agentStat
 		}
 	}

Original file line number	Diff line number	Diff line change
`@@ -2802,14 +2802,14 @@ func (q *FakeQuerier) GetWorkspaceAgentStats(_ context.Context, createdAfter tim`
`2802`	`2802`
`2803`	`2803`	`agentStatsCreatedAfter := make([]database.WorkspaceAgentStat, 0)`
`2804`	`2804`	`for _, agentStat := range q.workspaceAgentStats {`
`2805`		`- if agentStat.CreatedAt.After(createdAfter) {`
	`2805`	`+ if agentStat.CreatedAt.After(createdAfter) \|\| agentStat.CreatedAt.Equal(createdAfter) {`
`2806`	`2806`	`agentStatsCreatedAfter = append(agentStatsCreatedAfter, agentStat)`
`2807`	`2807`	`}`
`2808`	`2808`	`}`
`2809`	`2809`
`2810`	`2810`	`latestAgentStats := map[uuid.UUID]database.WorkspaceAgentStat{}`
`2811`	`2811`	`for _, agentStat := range q.workspaceAgentStats {`
`2812`		`- if agentStat.CreatedAt.After(createdAfter) {`
	`2812`	`+ if agentStat.CreatedAt.After(createdAfter) \|\| agentStat.CreatedAt.Equal(createdAfter) {`
`2813`	`2813`	`latestAgentStats[agentStat.AgentID] = agentStat`
`2814`	`2814`	`}`
`2815`	`2815`	`}`