unit test case

Author: cx-sumit-morchhale

internal/commands/scan.go

@@ -2092,7 +2092,7 @@ func uploadZip(uploadsWrapper wrappers.UploadsWrapper, zipFilePath string, unzip
 	logger.PrintIfVerbose(fmt.Sprintf("Zip size before upload:  %.2fMB\n", float64(fileInfo.Size())/mbBytes))
 
 	// check for INCREASE_FILE_UPLOAD_LIMIT feature flag
-	flagResponse, _ := wrappers.GetSpecificFeatureFlag(featureFlagsWrapper, wrappers.IncreaseFileUploadLimit)
+	flagResponse, _ := featureFlagsWrapper.GetSpecificFlag(wrappers.IncreaseFileUploadLimit)
 
 	fmt.Println("flagResponse.Status: ", flagResponse.Status)
 	fmt.Println("fileInfo.Size(): ", fileInfo.Size())

internal/commands/scan_test.go

@@ -4286,8 +4286,8 @@ func TestUploadZip_AsMultipartUpload_when_FF_Enable_ZIP_Exceeds_5GB(t *testing.T
 	MaxSizeBytes = 1
 	defer func() { MaxSizeBytes = 5 * 1024 * 1024 * 1024 }() // Reset after test
 	uploadWrapper := mock.UploadsMockWrapper{}
-	featureFlagsWrapper := &mock.FeatureFlagsMockWrapper{}
 	mock.Flag = wrappers.FeatureFlagResponseModel{Name: wrappers.IncreaseFileUploadLimit, Status: true}
+	featureFlagsWrapper := &mock.FeatureFlagsMockWrapper{}
 	url, zipPath, err := uploadZip(&uploadWrapper, "data/sources.zip", false, true, featureFlagsWrapper)
 	assert.NilError(t, err)
 	assert.Equal(t, zipPath, "")
@@ -4304,8 +4304,8 @@ func TestUploadZip_AsMultipartUpload_when_FF_Disable_ZIP_Exceeds_5GB(t *testing.
 	defer func() { MaxSizeBytes = 5 * 1024 * 1024 * 1024 }() // Reset after test
 
 	uploadWrapper := mock.UploadsMockWrapper{}
-	featureFlagsWrapper := &mock.FeatureFlagsMockWrapper{}
 	mock.Flag = wrappers.FeatureFlagResponseModel{Name: wrappers.IncreaseFileUploadLimit, Status: false}
+	featureFlagsWrapper := &mock.FeatureFlagsMockWrapper{}
 	url, zipPath, err := uploadZip(&uploadWrapper, "data/sources.zip", false, true, featureFlagsWrapper)
 	assert.NilError(t, err)
 	assert.Equal(t, zipPath, "")
@@ -4314,8 +4314,8 @@ func TestUploadZip_AsMultipartUpload_when_FF_Disable_ZIP_Exceeds_5GB(t *testing.
 
 func TestUploadZip_AsMultipartUpload_when_FF_Enable_ZIP_LessThan_5GB(t *testing.T) {
 	uploadWrapper := mock.UploadsMockWrapper{}
-	featureFlagsWrapper := &mock.FeatureFlagsMockWrapper{}
 	mock.Flag = wrappers.FeatureFlagResponseModel{Name: wrappers.IncreaseFileUploadLimit, Status: true}
+	featureFlagsWrapper := &mock.FeatureFlagsMockWrapper{}
 	url, zipPath, err := uploadZip(&uploadWrapper, "data/sources.zip", false, true, featureFlagsWrapper)
 	assert.NilError(t, err)
 	assert.Equal(t, zipPath, "")
@@ -4324,8 +4324,8 @@ func TestUploadZip_AsMultipartUpload_when_FF_Enable_ZIP_LessThan_5GB(t *testing.
 
 func TestUploadZip_AsMultipartUpload_when_FF_Disable_ZIP_LessThan_5GB(t *testing.T) {
 	uploadWrapper := mock.UploadsMockWrapper{}
-	featureFlagsWrapper := &mock.FeatureFlagsMockWrapper{}
 	mock.Flag = wrappers.FeatureFlagResponseModel{Name: wrappers.IncreaseFileUploadLimit, Status: false}
+	featureFlagsWrapper := &mock.FeatureFlagsMockWrapper{}
 	url, zipPath, err := uploadZip(&uploadWrapper, "data/sources.zip", false, true, featureFlagsWrapper)
 	assert.NilError(t, err)
 	assert.Equal(t, zipPath, "")
@@ -4366,8 +4366,8 @@ func TestUploadZip_AsMultipartUpload_when_FF_Enable_ZIP_Exceeds_5GB_Error(t *tes
 	defer func() { MaxSizeBytes = 5 * 1024 * 1024 * 1024 }() //
 
 	uploadWrapper := mock.UploadsMockWrapper{}
-	featureFlagsWrapper := &mock.FeatureFlagsMockWrapper{}
 	mock.Flag = wrappers.FeatureFlagResponseModel{Name: wrappers.IncreaseFileUploadLimit, Status: true}
+	featureFlagsWrapper := &mock.FeatureFlagsMockWrapper{}
 	_, zipPath, err := uploadZip(&uploadWrapper, zipPathTemp, false, true, featureFlagsWrapper)
 	assert.Assert(t, err != nil)
 	assert.Assert(t, strings.Contains(err.Error(), "error from UploadFileInMultipart"), err.Error())