try to fix batcher flake again

coadler · coadler · commit 3314b7fcae1c · 2023-08-08T00:17:06.000Z
diff --git a/coderd/database/dbfake/dbfake.go b/coderd/database/dbfake/dbfake.go
@@ -2799,14 +2799,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
`@@ -2799,14 +2799,14 @@ func (q *FakeQuerier) GetWorkspaceAgentStats(_ context.Context, createdAfter tim`
`2799`	`2799`
`2800`	`2800`	`agentStatsCreatedAfter := make([]database.WorkspaceAgentStat, 0)`
`2801`	`2801`	`for _, agentStat := range q.workspaceAgentStats {`
`2802`		`- if agentStat.CreatedAt.After(createdAfter) {`
	`2802`	`+ if agentStat.CreatedAt.After(createdAfter) \|\| agentStat.CreatedAt.Equal(createdAfter) {`
`2803`	`2803`	`agentStatsCreatedAfter = append(agentStatsCreatedAfter, agentStat)`
`2804`	`2804`	`}`
`2805`	`2805`	`}`
`2806`	`2806`
`2807`	`2807`	`latestAgentStats := map[uuid.UUID]database.WorkspaceAgentStat{}`
`2808`	`2808`	`for _, agentStat := range q.workspaceAgentStats {`
`2809`		`- if agentStat.CreatedAt.After(createdAfter) {`
	`2809`	`+ if agentStat.CreatedAt.After(createdAfter) \|\| agentStat.CreatedAt.Equal(createdAfter) {`
`2810`	`2810`	`latestAgentStats[agentStat.AgentID] = agentStat`
`2811`	`2811`	`}`
`2812`	`2812`	`}`