chore: fix archiver test - inconsistent number of files

rudderlabs · Oct 12, 2023 · fefd591 · fefd591
1 parent f977194
commit fefd591
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/archiver/archiver_test.go b/archiver/archiver_test.go
@@ -34,8 +34,8 @@ import (
 
 func TestJobsArchival(t *testing.T) {
  var (
- prefix = trand.String(10)
- minioResource []*destination.MINIOResource
+ prefixByWorkspace = map[int]string{0: trand.String(10), 1: trand.String(10), 2: trand.String(10)}
+ minioResource  []*destination.MINIOResource
 
  // test data - contains jobs from 3 workspaces(1 - 1 source, 2 & 3 - 2 sources each)
  seedJobsFileName = "testdata/MultiWorkspaceBackupJobs.json.gz"
@@ -78,7 +78,7 @@ func TestJobsArchival(t *testing.T) {
  Type: "MINIO",
  Config: map[string]interface{}{
  "bucketName": minioResource[0].BucketName,
- "prefix": prefix,
+ "prefix": prefixByWorkspace[0],
  "endPoint": minioResource[0].Endpoint,
  "accessKeyID": minioResource[0].AccessKey,
  "secretAccessKey": minioResource[0].SecretKey,
@@ -96,7 +96,7 @@ func TestJobsArchival(t *testing.T) {
  Type: "MINIO",
  Config: map[string]interface{}{
  "bucketName": minioResource[1].BucketName,
- "prefix": prefix,
+ "prefix": prefixByWorkspace[1],
  "endPoint": minioResource[1].Endpoint,
  "accessKeyID": minioResource[1].AccessKey,
  "secretAccessKey": minioResource[1].SecretKey,
@@ -114,7 +114,7 @@ func TestJobsArchival(t *testing.T) {
  Type: "MINIO",
  Config: map[string]interface{}{
  "bucketName": minioResource[2].BucketName,
- "prefix": prefix,
+ "prefix": prefixByWorkspace[2],
  "endPoint": minioResource[2].Endpoint,
  "accessKeyID": minioResource[2].AccessKey,
  "secretAccessKey": minioResource[2].SecretKey,
@@ -170,7 +170,7 @@ func TestJobsArchival(t *testing.T) {
  workspace := "defaultWorkspaceID-" + strconv.Itoa(i+1)
  fm, err := fileUploaderProvider.GetFileManager(workspace)
  require.NoError(t, err)
- fileIter := fm.ListFilesWithPrefix(context.Background(), "", prefix, 20)
+ fileIter := fm.ListFilesWithPrefix(context.Background(), "", prefixByWorkspace[i], 20)
  files, err := getAllFileNames(fileIter)
  require.NoError(t, err)
  require.Equal(t, sourcesPerWorkspace[i], len(files),