Skip to content

Commit a09f39f

Browse files
authored
chore: improve error logs during processor stash test (#4317)
chore: improve error logs during processor stash test:
1 parent 0771999 commit a09f39f

File tree

1 file changed

+16
-17
lines changed

1 file changed

+16
-17
lines changed

processor/stash/stash_test.go

+16-17
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,8 @@ import (
99
"testing"
1010
"time"
1111

12+
"github.com/samber/lo"
13+
1214
"github.com/rudderlabs/rudder-go-kit/testhelper/docker/resource"
1315

1416
"github.com/google/uuid"
@@ -21,7 +23,6 @@ import (
2123
backendconfig "github.com/rudderlabs/rudder-server/backend-config"
2224
"github.com/rudderlabs/rudder-server/jobsdb"
2325
"github.com/rudderlabs/rudder-server/services/fileuploader"
24-
"github.com/rudderlabs/rudder-server/testhelper"
2526
)
2627

2728
var prefix = "proc_error_jobs_"
@@ -35,12 +36,10 @@ func TestStoreErrorsToObjectStorage(t *testing.T) {
3536
// running minio container on docker
3637
pool, err := dockertest.NewPool("")
3738
require.NoError(t, err, "Failed to create docker pool")
38-
cleanup := &testhelper.Cleanup{}
39-
defer cleanup.Run()
4039

4140
minioResource := make([]*resource.MinioResource, uniqueWorkspaces)
4241
for i := 0; i < uniqueWorkspaces; i++ {
43-
minioResource[i], err = resource.SetupMinio(pool, cleanup)
42+
minioResource[i], err = resource.SetupMinio(pool, t)
4443
require.NoError(t, err)
4544
}
4645

@@ -107,7 +106,7 @@ func TestStoreErrorsToObjectStorage(t *testing.T) {
107106
},
108107
}
109108

110-
fileuploaderProvider := fileuploader.NewStaticProvider(storageSettings)
109+
fileUploaderProvider := fileuploader.NewStaticProvider(storageSettings)
111110

112111
jobs := []*jobsdb.JobT{
113112
{
@@ -137,7 +136,7 @@ func TestStoreErrorsToObjectStorage(t *testing.T) {
137136
}
138137

139138
st := New()
140-
st.fileuploader = fileuploaderProvider
139+
st.fileuploader = fileUploaderProvider
141140
st.logger = logger.NOP
142141

143142
jobsCount := countJobsByWorkspace(jobs)
@@ -156,16 +155,18 @@ func TestStoreErrorsToObjectStorage(t *testing.T) {
156155
return true
157156
}
158157
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)
161162
require.NoError(t, err)
162163
return false
163164
}
164165
return true
165166
}, 20*time.Second, 1*time.Second, "no backup files found in backup store: ", err)
166167

167168
if storageSettings[workspace].Preferences.ProcErrors {
168-
f := downloadFile(t, fm, file[0].Key, cleanup)
169+
f := downloadFile(t, fm, file[0].Key)
169170
jobsFromFile, err := readGzipJobFile(f.Name())
170171
require.NoError(t, err)
171172
require.NotZero(t, jobsCount[workspace], "jobsCount for workspace: ", workspace, " is zero")
@@ -203,7 +204,7 @@ func readGzipJobFile(filename string) ([]*jobsdb.JobT, error) {
203204
if err != nil {
204205
return []*jobsdb.JobT{}, err
205206
}
206-
defer gz.Close()
207+
defer func() { _ = gz.Close() }()
207208

208209
sc := bufio.NewScanner(gz)
209210
// default scanner buffer maxCapacity is 64K
@@ -212,12 +213,11 @@ func readGzipJobFile(filename string) ([]*jobsdb.JobT, error) {
212213
buf := make([]byte, maxCapacity)
213214
sc.Buffer(buf, maxCapacity)
214215

215-
jobs := []*jobsdb.JobT{}
216+
var jobs []*jobsdb.JobT
216217
for sc.Scan() {
217218
lineByte := sc.Bytes()
218-
uuid := uuid.MustParse("69359037-9599-48e7-b8f2-48393c019135")
219219
job := &jobsdb.JobT{
220-
UUID: uuid,
220+
UUID: uuid.MustParse("69359037-9599-48e7-b8f2-48393c019135"),
221221
JobID: gjson.GetBytes(lineByte, "job_id").Int(),
222222
UserID: gjson.GetBytes(lineByte, "user_id").String(),
223223
CustomVal: gjson.GetBytes(lineByte, "custom_val").String(),
@@ -231,7 +231,7 @@ func readGzipJobFile(filename string) ([]*jobsdb.JobT, error) {
231231
return jobs, nil
232232
}
233233

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 {
235235
file, err := os.CreateTemp("", "backedupfile")
236236
require.NoError(t, err, "expected no error while creating temporary file")
237237

@@ -240,12 +240,11 @@ func downloadFile(t *testing.T, fm filemanager.FileManager, fileToDownload strin
240240

241241
// reopening the file so to reset the pointer
242242
// since file.Seek(0, io.SeekStart) doesn't work
243-
file.Close()
243+
_ = file.Close()
244244
file, err = os.Open(file.Name())
245245
require.NoError(t, err, "expected no error while reopening downloaded file")
246246

247-
require.NoError(t, err)
248-
cleanup.Cleanup(func() {
247+
t.Cleanup(func() {
249248
_ = file.Close()
250249
_ = os.Remove(file.Name())
251250
})

0 commit comments

Comments
 (0)