Fix tests

internal/commands/root_test.go

@@ -204,6 +204,7 @@ func assertError(t *testing.T, err error, expectedMessage string) {
 func clearFlags() {
 	mock.Flags = wrappers.FeatureFlagsResponseModel{}
 	mock.Flag = wrappers.FeatureFlagResponseModel{}
+	mock.FFErr = nil
 	wrappers.ClearCache()
 }
 

internal/commands/scan_test.go

@@ -1,5 +1,3 @@
-//go:build !integration
-
 package commands
 
 import (
@@ -1947,20 +1945,20 @@ func TestValidateScanTypes(t *testing.T) {
 }
 
 func TestIsContainersEngineEnabled_FlagEnabled(t *testing.T) {
-	mockWrapper := mock.FeatureFlagsMockWrapper{
-		FlagResponse: wrappers.FeatureFlagResponseModel{Status: true},
-		Err:          nil,
-	}
+	clearFlags()
+	mock.Flag = wrappers.FeatureFlagResponseModel{Name: wrappers.ContainerEngineCLIEnabled, Status: true}
+	mock.FFErr = nil
 
-	result := isContainersEngineEnabled(mockWrapper)
-	assert.True(t, result, "Expected isContainersEngineEnabled to return true when the flag is enabled")
+	result := isContainersEngineEnabled(mock.FeatureFlagsMockWrapper{})
+	assert.Assert(t, result, "expected result to be true")
 }
 
 func TestIsContainersEngineEnabled_FlagRetrievalFails(t *testing.T) {
-	mockWrapper := mock.FeatureFlagsMockWrapper{
-		Err: errors.New("network error"),
-	}
+	clearFlags()
+	mock.Flag = wrappers.FeatureFlagResponseModel{Name: wrappers.ContainerEngineCLIEnabled, Status: false}
+	mock.FFErr = errors.New("something went wrong while fetching ff")
+
+	result := isContainersEngineEnabled(mock.FeatureFlagsMockWrapper{})
 
-	result := isContainersEngineEnabled(mockWrapper)
-	assert.False(t, result, "Expected isContainersEngineEnabled to return false when flag retrieval fails")
+	assert.Assert(t, !result, "expected result to be false")
 }

internal/wrappers/mock/feature-flags-mock.go

@@ -8,6 +8,7 @@ import (
 
 var Flags wrappers.FeatureFlagsResponseModel
 var Flag wrappers.FeatureFlagResponseModel
+var FFErr error
 
 type FeatureFlagsMockWrapper struct {
 }
@@ -22,5 +23,8 @@ func (f FeatureFlagsMockWrapper) GetAll() (*wrappers.FeatureFlagsResponseModel,
 
 func (f FeatureFlagsMockWrapper) GetSpecificFlag(specificFlag string) (*wrappers.FeatureFlagResponseModel, error) {
 	fmt.Println("Called GetSpecificFlag in FeatureFlagsMockWrapper with flag:", specificFlag)
+	if FFErr != nil {
+		return nil, FFErr
+	}
 	return &Flag, nil
 }