@@ -11,6 +11,7 @@ import (
1111"time"
1212
1313"github.com/google/uuid"
14+ "github.com/stretchr/testify/assert"
1415"github.com/stretchr/testify/require"
1516"go.uber.org/goleak"
1617"golang.org/x/exp/slices"
@@ -183,20 +184,29 @@ func TestDeleteOldWorkspaceAgentLogs(t *testing.T) {
183184// given
184185agent := mustCreateAgentWithLogs (ctx ,t ,db ,user ,org ,tmpl ,tv ,now .Add (- 8 * 24 * time .Hour ),t .Name ())
185186
187+ // Make sure that agent logs have been collected.
188+ agentLogs ,err := db .GetWorkspaceAgentLogsAfter (ctx , database.GetWorkspaceAgentLogsAfterParams {
189+ AgentID :agent ,
190+ })
191+ require .NoError (t ,err )
192+ require .NotZero (t ,agentLogs ,"agent logs must be present" )
193+
186194// when
187195closer := dbpurge .New (ctx ,logger ,db )
188196defer closer .Close ()
189197
190198// then
191- require .Eventually (t ,func ()bool {
192- agentLogs ,err : =db .GetWorkspaceAgentLogsAfter (ctx , database.GetWorkspaceAgentLogsAfterParams {
199+ assert .Eventually (t ,func ()bool {
200+ agentLogs ,err = db .GetWorkspaceAgentLogsAfter (ctx , database.GetWorkspaceAgentLogsAfterParams {
193201AgentID :agent ,
194202})
195203if err != nil {
196204return false
197205}
198206return ! containsAgentLog (agentLogs ,t .Name ())
199207},testutil .WaitShort ,testutil .IntervalFast )
208+ require .NoError (t ,err )
209+ require .NotContains (t ,agentLogs ,t .Name ())
200210})
201211
202212t .Run ("AgentConnectedSixDaysAgo_LogsValid" ,func (t * testing.T ) {