ghndrx/prowler
1
0
Fork 0
mirror of https://github.com/ghndrx/prowler.git synced 2026-02-10 14:55:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(azure): Change class names from azure services and fix typing error (#3350)

Browse Source
This commit is contained in:
Pedro Martín
2024-02-05 15:43:04 +01:00
committed by GitHub
parent 8c51094df1
commit 4587a9f651
44 changed files with 581 additions and 584 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
tests/providers/azure/services/defender/defender_service_test.py
View File

@@ -2,22 +2,22 @@ from datetime import timedelta
from unittest.mock import patch
from prowler.providers.azure.services.defender.defender_service import (
Assesment,
AutoProvisioningSetting,
Defender,
Defender_Assessments,
Defender_Pricing,
Defender_Security_Contacts,
Pricing,
SecurityContacts,
)
from tests.providers.azure.azure_fixtures import (
AZURE_SUSCRIPTION,
AZURE_SUBSCRIPTION,
set_mocked_azure_audit_info,
)
def mock_defender_get_pricings(_):
return {
AZURE_SUSCRIPTION: {
"Standard": Defender_Pricing(
AZURE_SUBSCRIPTION: {
"Standard": Pricing(
resource_id="resource_id",
pricing_tier="pricing_tier",
free_trial_remaining_time=timedelta(days=1),
@@ -28,7 +28,7 @@ def mock_defender_get_pricings(_):
def mock_defender_get_auto_provisioning_settings(_):
return {
AZURE_SUSCRIPTION: {
AZURE_SUBSCRIPTION: {
"default": AutoProvisioningSetting(
resource_id="/subscriptions/resource_id",
resource_name="default",
@@ -41,8 +41,8 @@ def mock_defender_get_auto_provisioning_settings(_):
def mock_defender_get_assessments(_):
return {
AZURE_SUSCRIPTION: {
"default": Defender_Assessments(
AZURE_SUBSCRIPTION: {
"default": Assesment(
resource_id="/subscriptions/resource_id",
resource_name="default",
status="Healthy",
@@ -53,8 +53,8 @@ def mock_defender_get_assessments(_):
def mock_defender_get_security_contacts(_):
return {
AZURE_SUSCRIPTION: {
"default": Defender_Security_Contacts(
AZURE_SUBSCRIPTION: {
"default": SecurityContacts(
resource_id="/subscriptions/resource_id",
emails="user@user.com, test@test.es",
phone="666666666",
@@ -87,7 +87,7 @@ class Test_Defender_Service:
def test__get_client__(self):
defender = Defender(set_mocked_azure_audit_info())
assert (
defender.clients[AZURE_SUSCRIPTION].__class__.__name__ == "SecurityCenter"
defender.clients[AZURE_SUBSCRIPTION].__class__.__name__ == "SecurityCenter"
)
def test__get_subscriptions__(self):
@@ -99,14 +99,14 @@ class Test_Defender_Service:
defender = Defender(set_mocked_azure_audit_info())
assert len(defender.pricings) == 1
assert (
defender.pricings[AZURE_SUSCRIPTION]["Standard"].resource_id
defender.pricings[AZURE_SUBSCRIPTION]["Standard"].resource_id
== "resource_id"
)
assert (
defender.pricings[AZURE_SUSCRIPTION]["Standard"].pricing_tier
defender.pricings[AZURE_SUBSCRIPTION]["Standard"].pricing_tier
== "pricing_tier"
)
assert defender.pricings[AZURE_SUSCRIPTION][
assert defender.pricings[AZURE_SUBSCRIPTION][
"Standard"
].free_trial_remaining_time == timedelta(days=1)
@@ -114,25 +114,25 @@ class Test_Defender_Service:
defender = Defender(set_mocked_azure_audit_info())
assert len(defender.auto_provisioning_settings) == 1
assert (
defender.auto_provisioning_settings[AZURE_SUSCRIPTION][
defender.auto_provisioning_settings[AZURE_SUBSCRIPTION][
"default"
].resource_id
== "/subscriptions/resource_id"
)
assert (
defender.auto_provisioning_settings[AZURE_SUSCRIPTION][
defender.auto_provisioning_settings[AZURE_SUBSCRIPTION][
"default"
].resource_name
== "default"
)
assert (
defender.auto_provisioning_settings[AZURE_SUSCRIPTION][
defender.auto_provisioning_settings[AZURE_SUBSCRIPTION][
"default"
].resource_type
== "Microsoft.Security/autoProvisioningSettings"
)
assert (
defender.auto_provisioning_settings[AZURE_SUSCRIPTION][
defender.auto_provisioning_settings[AZURE_SUBSCRIPTION][
"default"
].auto_provision
== "On"
@@ -142,47 +142,47 @@ class Test_Defender_Service:
defender = Defender(set_mocked_azure_audit_info())
assert len(defender.assessments) == 1
assert (
defender.assessments[AZURE_SUSCRIPTION]["default"].resource_id
defender.assessments[AZURE_SUBSCRIPTION]["default"].resource_id
== "/subscriptions/resource_id"
)
assert (
defender.assessments[AZURE_SUSCRIPTION]["default"].resource_name
defender.assessments[AZURE_SUBSCRIPTION]["default"].resource_name
== "default"
)
assert defender.assessments[AZURE_SUSCRIPTION]["default"].status == "Healthy"
assert defender.assessments[AZURE_SUBSCRIPTION]["default"].status == "Healthy"
def test__get_security_contacts__(self):
defender = Defender(set_mocked_azure_audit_info())
assert len(defender.security_contacts) == 1
assert (
defender.security_contacts[AZURE_SUSCRIPTION]["default"].resource_id
defender.security_contacts[AZURE_SUBSCRIPTION]["default"].resource_id
== "/subscriptions/resource_id"
)
assert (
defender.security_contacts[AZURE_SUSCRIPTION]["default"].emails
defender.security_contacts[AZURE_SUBSCRIPTION]["default"].emails
== "user@user.com, test@test.es"
)
assert (
defender.security_contacts[AZURE_SUSCRIPTION]["default"].phone
defender.security_contacts[AZURE_SUBSCRIPTION]["default"].phone
== "666666666"
)
assert (
defender.security_contacts[AZURE_SUSCRIPTION][
defender.security_contacts[AZURE_SUBSCRIPTION][
"default"
].alert_notifications_minimal_severity
== "High"
)
assert (
defender.security_contacts[AZURE_SUSCRIPTION][
defender.security_contacts[AZURE_SUBSCRIPTION][
"default"
].alert_notifications_state
== "On"
)
assert defender.security_contacts[AZURE_SUSCRIPTION][
assert defender.security_contacts[AZURE_SUBSCRIPTION][
"default"
].notified_roles == ["Owner", "Contributor"]
assert (
defender.security_contacts[AZURE_SUSCRIPTION][
defender.security_contacts[AZURE_SUBSCRIPTION][
"default"
].notified_roles_state
== "On"