chore(refactor): make Provider generation global (#4961)

Co-authored-by: pedrooot <pedromarting3@gmail.com>
This commit is contained in:
Sergio Garciaandpedrooot authored and GitHub committed 2024-09-12 16:56:58 +02:00
1 parent 8f37252676
commit 8ace8c01cf
8 files changed
+42 -19

No files matched your search

@@ -4,16 +4,23 @@ AWS_REGION = "eu-west-1"
class Test_accessanalyzer_enabled_fixer:
def test_accessanalyzer_enabled_fixer(self):
accessanalyzer_client = mock.MagicMock
accessanalyzer_client.analyzers = []
with mock.patch(
"prowler.providers.aws.services.accessanalyzer.accessanalyzer_service.AccessAnalyzer",
new=accessanalyzer_client,
):
# Test Check
from prowler.providers.aws.services.accessanalyzer.accessanalyzer_enabled.accessanalyzer_enabled_fixer import (
fixer,
)
@mock.patch(
"prowler.providers.aws.services.accessanalyzer.accessanalyzer_enabled.accessanalyzer_enabled_fixer.accessanalyzer_client"
)
def test_accessanalyzer_enabled_fixer(self, mock_accessanalyzer_client):
mock_client = mock.MagicMock()
mock_accessanalyzer_client.regional_clients = {AWS_REGION: mock_client}
mock_accessanalyzer_client.fixer_config = {
"accessanalyzer_enabled": {
"AnalyzerName": "DefaultAnalyzer",
"AnalyzerType": "ACCOUNT_UNUSED_ACCESS",
}
}
mock_client.create_analyzer.return_value = None
assert fixer(AWS_REGION)
from prowler.providers.aws.services.accessanalyzer.accessanalyzer_enabled.accessanalyzer_enabled_fixer import (
fixer,
)
result = fixer(AWS_REGION)
assert result
@@ -16,10 +16,14 @@ DETECTOR_ARN = f"arn:aws:guardduty:{AWS_REGION_EU_WEST_1}:{AWS_ACCOUNT_NUMBER}:d
class Test_guardduty_is_enabled_fixer:
@mock_aws
def test_guardduty_is_enabled_fixer(self):
regional_client = mock.MagicMock()
guardduty_client = mock.MagicMock
guardduty_client.region = AWS_REGION_EU_WEST_1
guardduty_client.detectors = []
guardduty_client.audited_account_arn = AWS_ACCOUNT_ARN
regional_client.create_detector.return_value = None
guardduty_client.regional_clients = {AWS_REGION_EU_WEST_1: regional_client}
with mock.patch(
"prowler.providers.aws.services.guardduty.guardduty_service.GuardDuty",
guardduty_client,