beenull/crowdsec
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix flakey test in file_tests (#1783)

Browse source 
Signed-off-by: Shivam Sandbhor <shivam.sandbhor@gmail.com>
This commit is contained in:
Shivam Sandbhor 2022-10-05 20:10:09 +05:30 committed by GitHub
parent af30ba0e3b
commit b203b3f444
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
pkg/acquisition/modules/file/file_test.go
View file

@ -213,7 +213,7 @@ filename: test_files/test_delete.log`,
}) })
tomb := tomb.Tomb{} tomb := tomb.Tomb{}
out := make(chan types.Event) out := make(chan types.Event, 100)
f := fileacquisition.FileSource{} f := fileacquisition.FileSource{}
if tc.setup != nil { if tc.setup != nil {
@ -229,22 +229,8 @@ filename: test_files/test_delete.log`,
if tc.afterConfigure != nil { if tc.afterConfigure != nil {
tc.afterConfigure() tc.afterConfigure()
} }
actualLines := 0
if tc.expectedLines != 0 {
go func() {
for {
select {
case <-out:
actualLines++
case <-time.After(2 * time.Second):
return
}
}
}()
}
err = f.OneShotAcquisition(out, &tomb) err = f.OneShotAcquisition(out, &tomb)
actualLines := len(out)
cstest.RequireErrorContains(t, err, tc.expectedErr) cstest.RequireErrorContains(t, err, tc.expectedErr)
if tc.expectedLines != 0 { if tc.expectedLines != 0 {