mirror of
https://github.com/prowler-cloud/prowler.git
synced 2025-12-19 05:17:47 +00:00
3d59c34ec9
Signed-off-by: dependabot[bot] <support@github.com> Co-authored-by: Pepe Fagoaga <pepe@prowler.com> Co-authored-by: Nacho Rivera <nachor1992@gmail.com> Co-authored-by: Rubén De la Torre Vico <rubendltv22@gmail.com> Co-authored-by: Pedro Martín <pedromarting3@gmail.com> Co-authored-by: Hugo966 <148140670+Hugo966@users.noreply.github.com> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> Co-authored-by: Hugo Gálvez Ureña <hugogalvezu96@gmail.com> Co-authored-by: github-actions <noreply@github.com>
58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
from unittest.mock import patch
|
|
|
|
from prowler.providers.azure.models import AzureIdentityInfo
|
|
from prowler.providers.azure.services.entra.entra_service import (
|
|
AuthorizationPolicy,
|
|
Entra,
|
|
User,
|
|
)
|
|
from tests.providers.azure.azure_fixtures import DOMAIN, set_mocked_azure_provider
|
|
|
|
|
|
async def mock_entra_get_users(_):
|
|
return {
|
|
"user-1@tenant1.es": User(id="id-1", name="User 1"),
|
|
}
|
|
|
|
|
|
async def mock_entra_get_authorization_policy(_):
|
|
return AuthorizationPolicy(
|
|
id="id-1",
|
|
name="Name 1",
|
|
description="Description 1",
|
|
default_user_role_permissions=None,
|
|
)
|
|
|
|
|
|
@patch(
|
|
"prowler.providers.azure.services.entra.entra_service.Entra.__get_users__",
|
|
new=mock_entra_get_users,
|
|
)
|
|
@patch(
|
|
"prowler.providers.azure.services.entra.entra_service.Entra.__get_authorization_policy__",
|
|
new=mock_entra_get_authorization_policy,
|
|
)
|
|
class Test_Entra_Service:
|
|
def test__get_client__(self):
|
|
entra_client = Entra(
|
|
set_mocked_azure_provider(identity=AzureIdentityInfo(tenant_domain=DOMAIN))
|
|
)
|
|
assert entra_client.clients[DOMAIN].__class__.__name__ == "GraphServiceClient"
|
|
|
|
def test__get_subscriptions__(self):
|
|
entra_client = Entra(set_mocked_azure_provider())
|
|
assert entra_client.subscriptions.__class__.__name__ == "dict"
|
|
|
|
def test__get_users__(self):
|
|
entra_client = Entra(set_mocked_azure_provider())
|
|
assert len(entra_client.users) == 1
|
|
assert entra_client.users["user-1@tenant1.es"].id == "id-1"
|
|
assert entra_client.users["user-1@tenant1.es"].name == "User 1"
|
|
|
|
def test__get_authorization_policy__(self):
|
|
entra_client = Entra(set_mocked_azure_provider())
|
|
assert entra_client.authorization_policy.id == "id-1"
|
|
assert entra_client.authorization_policy.name == "Name 1"
|
|
assert entra_client.authorization_policy.description == "Description 1"
|
|
assert not entra_client.authorization_policy.default_user_role_permissions
|