9
9
"testing"
10
10
"time"
11
11
12
+ "github.com/samber/lo"
13
+
12
14
"github.com/rudderlabs/rudder-go-kit/testhelper/docker/resource"
13
15
14
16
"github.com/google/uuid"
@@ -21,7 +23,6 @@ import (
21
23
backendconfig "github.com/rudderlabs/rudder-server/backend-config"
22
24
"github.com/rudderlabs/rudder-server/jobsdb"
23
25
"github.com/rudderlabs/rudder-server/services/fileuploader"
24
- "github.com/rudderlabs/rudder-server/testhelper"
25
26
)
26
27
27
28
var prefix = "proc_error_jobs_"
@@ -35,12 +36,10 @@ func TestStoreErrorsToObjectStorage(t *testing.T) {
35
36
// running minio container on docker
36
37
pool , err := dockertest .NewPool ("" )
37
38
require .NoError (t , err , "Failed to create docker pool" )
38
- cleanup := & testhelper.Cleanup {}
39
- defer cleanup .Run ()
40
39
41
40
minioResource := make ([]* resource.MinioResource , uniqueWorkspaces )
42
41
for i := 0 ; i < uniqueWorkspaces ; i ++ {
43
- minioResource [i ], err = resource .SetupMinio (pool , cleanup )
42
+ minioResource [i ], err = resource .SetupMinio (pool , t )
44
43
require .NoError (t , err )
45
44
}
46
45
@@ -107,7 +106,7 @@ func TestStoreErrorsToObjectStorage(t *testing.T) {
107
106
},
108
107
}
109
108
110
- fileuploaderProvider := fileuploader .NewStaticProvider (storageSettings )
109
+ fileUploaderProvider := fileuploader .NewStaticProvider (storageSettings )
111
110
112
111
jobs := []* jobsdb.JobT {
113
112
{
@@ -137,7 +136,7 @@ func TestStoreErrorsToObjectStorage(t *testing.T) {
137
136
}
138
137
139
138
st := New ()
140
- st .fileuploader = fileuploaderProvider
139
+ st .fileuploader = fileUploaderProvider
141
140
st .logger = logger .NOP
142
141
143
142
jobsCount := countJobsByWorkspace (jobs )
@@ -156,16 +155,18 @@ func TestStoreErrorsToObjectStorage(t *testing.T) {
156
155
return true
157
156
}
158
157
if len (file ) != 1 {
159
- t .Log ("file list: " , file , " err: " , err , "len: " , len (file ))
160
- fm , err = fileuploaderProvider .GetFileManager (workspace )
158
+ t .Logf ("file list: %+v err: %v" , lo .Map (file , func (item * filemanager.FileInfo , _ int ) string {
159
+ return item .Key
160
+ }), err )
161
+ fm , err = fileUploaderProvider .GetFileManager (workspace )
161
162
require .NoError (t , err )
162
163
return false
163
164
}
164
165
return true
165
166
}, 20 * time .Second , 1 * time .Second , "no backup files found in backup store: " , err )
166
167
167
168
if storageSettings [workspace ].Preferences .ProcErrors {
168
- f := downloadFile (t , fm , file [0 ].Key , cleanup )
169
+ f := downloadFile (t , fm , file [0 ].Key )
169
170
jobsFromFile , err := readGzipJobFile (f .Name ())
170
171
require .NoError (t , err )
171
172
require .NotZero (t , jobsCount [workspace ], "jobsCount for workspace: " , workspace , " is zero" )
@@ -203,7 +204,7 @@ func readGzipJobFile(filename string) ([]*jobsdb.JobT, error) {
203
204
if err != nil {
204
205
return []* jobsdb.JobT {}, err
205
206
}
206
- defer gz .Close ()
207
+ defer func () { _ = gz .Close () } ()
207
208
208
209
sc := bufio .NewScanner (gz )
209
210
// default scanner buffer maxCapacity is 64K
@@ -212,12 +213,11 @@ func readGzipJobFile(filename string) ([]*jobsdb.JobT, error) {
212
213
buf := make ([]byte , maxCapacity )
213
214
sc .Buffer (buf , maxCapacity )
214
215
215
- jobs := []* jobsdb.JobT {}
216
+ var jobs []* jobsdb.JobT
216
217
for sc .Scan () {
217
218
lineByte := sc .Bytes ()
218
- uuid := uuid .MustParse ("69359037-9599-48e7-b8f2-48393c019135" )
219
219
job := & jobsdb.JobT {
220
- UUID : uuid ,
220
+ UUID : uuid . MustParse ( "69359037-9599-48e7-b8f2-48393c019135" ) ,
221
221
JobID : gjson .GetBytes (lineByte , "job_id" ).Int (),
222
222
UserID : gjson .GetBytes (lineByte , "user_id" ).String (),
223
223
CustomVal : gjson .GetBytes (lineByte , "custom_val" ).String (),
@@ -231,7 +231,7 @@ func readGzipJobFile(filename string) ([]*jobsdb.JobT, error) {
231
231
return jobs , nil
232
232
}
233
233
234
- func downloadFile (t * testing.T , fm filemanager.FileManager , fileToDownload string , cleanup * testhelper. Cleanup ) * os.File {
234
+ func downloadFile (t * testing.T , fm filemanager.FileManager , fileToDownload string ) * os.File {
235
235
file , err := os .CreateTemp ("" , "backedupfile" )
236
236
require .NoError (t , err , "expected no error while creating temporary file" )
237
237
@@ -240,12 +240,11 @@ func downloadFile(t *testing.T, fm filemanager.FileManager, fileToDownload strin
240
240
241
241
// reopening the file so to reset the pointer
242
242
// since file.Seek(0, io.SeekStart) doesn't work
243
- file .Close ()
243
+ _ = file .Close ()
244
244
file , err = os .Open (file .Name ())
245
245
require .NoError (t , err , "expected no error while reopening downloaded file" )
246
246
247
- require .NoError (t , err )
248
- cleanup .Cleanup (func () {
247
+ t .Cleanup (func () {
249
248
_ = file .Close ()
250
249
_ = os .Remove (file .Name ())
251
250
})
0 commit comments