prowler/providers/aws/services/directoryservice/directoryservice_service_test.py

from datetime import datetime
from unittest.mock import patch

import botocore
from moto import mock_ds
from moto.core import DEFAULT_ACCOUNT_ID

from providers.aws.lib.audit_info.audit_info import current_audit_info
from providers.aws.services.directoryservice.directoryservice_service import (
    AuthenticationProtocol,
    CertificateState,
    CertificateType,
    DirectoryService,
    EventTopicStatus,
    RadiusStatus,
)

# Mock Test Region
AWS_REGION = "eu-west-1"

# Mocking Access Analyzer Calls
make_api_call = botocore.client.BaseClient._make_api_call


def mock_make_api_call(self, operation_name, kwarg):
    """We have to mock every AWS API call using Boto3"""
    if operation_name == "DescribeDirectories":
        return {
            "DirectoryDescriptions": [
                {
                    "DirectoryId": "test-directory",
                    "Name": "test-directory",
                    "ShortName": "test-directory",
                    "RadiusSettings": {
                        "RadiusServers": [
                            "test-server",
                        ],
                        "RadiusPort": 9999,
                        "RadiusTimeout": 100,
                        "RadiusRetries": 100,
                        "SharedSecret": "test-shared-secret",
                        "AuthenticationProtocol": "MS-CHAPv2",
                        "DisplayLabel": "test-directory",
                        "UseSameUsername": True | False,
                    },
                    "RadiusStatus": "Creating",
                },
            ],
        }
    if operation_name == "ListLogSubscriptions":
        return {
            "LogSubscriptions": [
                {
                    "DirectoryId": "test-directory",
                    "LogGroupName": "test-log-group",
                    "SubscriptionCreatedDateTime": datetime(2022, 1, 1),
                },
            ],
        }
    if operation_name == "DescribeEventTopics":
        return {
            "EventTopics": [
                {
                    "DirectoryId": "test-directory",
                    "TopicName": "test-topic",
                    "TopicArn": f"arn:aws:sns:{AWS_REGION}:{DEFAULT_ACCOUNT_ID}:test-topic",
                    "CreatedDateTime": datetime(2022, 1, 1),
                    "Status": "Registered",
                },
            ]
        }

    if operation_name == "ListCertificates":
        return {
            "CertificatesInfo": [
                {
                    "CertificateId": "test-certificate",
                    "CommonName": "test-certificate",
                    "State": "Registered",
                    "ExpiryDateTime": datetime(2023, 1, 1),
                    "Type": "ClientLDAPS",
                },
            ]
        }
    if operation_name == "GetSnapshotLimits":
        return {
            "SnapshotLimits": {
                "ManualSnapshotsLimit": 123,
                "ManualSnapshotsCurrentCount": 123,
                "ManualSnapshotsLimitReached": True,
            }
        }

    return make_api_call(self, operation_name, kwarg)


# Mock generate_regional_clients()
def mock_generate_regional_clients(service, audit_info):
    regional_client = audit_info.audit_session.client(service, region_name=AWS_REGION)
    regional_client.region = AWS_REGION
    return {AWS_REGION: regional_client}


# Patch every AWS call using Boto3 and generate_regional_clients to have 1 client
@patch("botocore.client.BaseClient._make_api_call", new=mock_make_api_call)
@patch(
    "providers.aws.services.directoryservice.directoryservice_service.generate_regional_clients",
    new=mock_generate_regional_clients,
)
class Test_DirectoryService_Service:
    # Test DirectoryService Client
    @mock_ds
    def test__get_client__(self):
        directoryservice = DirectoryService(current_audit_info)
        assert (
            directoryservice.regional_clients[AWS_REGION].__class__.__name__
            == "DirectoryService"
        )

    # Test DirectoryService Session
    @mock_ds
    def test__get_session__(self):
        directoryservice = DirectoryService(current_audit_info)
        assert directoryservice.session.__class__.__name__ == "Session"

    # Test DirectoryService Service
    @mock_ds
    def test__get_service__(self):
        directoryservice = DirectoryService(current_audit_info)
        assert directoryservice.service == "ds"

    def test__describe_directories__(self):
        # Set partition for the service
        current_audit_info.audited_partition = "aws"
        directoryservice = DirectoryService(current_audit_info)

        # __describe_directories__
        assert directoryservice.directories["test-directory"]
        assert directoryservice.directories["test-directory"].name == "test-directory"
        assert directoryservice.directories["test-directory"].region == AWS_REGION
        assert (
            directoryservice.directories[
                "test-directory"
            ].radius_settings.authentication_protocol
            == AuthenticationProtocol.MS_CHAPv2
        )
        assert (
            directoryservice.directories["test-directory"].radius_settings.status
            == RadiusStatus.Creating
        )

        # __list_log_subscriptions__
        assert (
            len(directoryservice.directories["test-directory"].log_subscriptions) == 1
        )
        assert (
            directoryservice.directories["test-directory"]
            .log_subscriptions[0]
            .log_group_name
            == "test-log-group"
        )
        assert directoryservice.directories["test-directory"].log_subscriptions[
            0
        ].created_date_time == datetime(2022, 1, 1)

        # __describe_event_topics__
        assert len(directoryservice.directories["test-directory"].event_topics) == 1
        assert (
            directoryservice.directories["test-directory"].event_topics[0].topic_name
            == "test-topic"
        )
        assert (
            directoryservice.directories["test-directory"].event_topics[0].topic_arn
            == f"arn:aws:sns:{AWS_REGION}:{DEFAULT_ACCOUNT_ID}:test-topic"
        )
        assert (
            directoryservice.directories["test-directory"].event_topics[0].status
            == EventTopicStatus.Registered
        )
        assert directoryservice.directories["test-directory"].event_topics[
            0
        ].created_date_time == datetime(2022, 1, 1)

        # __list_certificates__
        assert len(directoryservice.directories["test-directory"].certificates) == 1
        assert (
            directoryservice.directories["test-directory"].certificates[0].id
            == "test-certificate"
        )
        assert (
            directoryservice.directories["test-directory"].certificates[0].common_name
            == "test-certificate"
        )
        assert (
            directoryservice.directories["test-directory"].certificates[0].state
            == CertificateState.Registered
        )
        assert directoryservice.directories["test-directory"].certificates[
            0
        ].expiry_date_time == datetime(2023, 1, 1)
        assert (
            directoryservice.directories["test-directory"].certificates[0].type
            == CertificateType.ClientLDAPS
        )

        # __get_snapshot_limits__
        assert directoryservice.directories["test-directory"].snapshots_limits
        assert (
            directoryservice.directories[
                "test-directory"
            ].snapshots_limits.manual_snapshots_limit
            == 123
        )
        assert (
            directoryservice.directories[
                "test-directory"
            ].snapshots_limits.manual_snapshots_current_count
            == 123
        )
        assert (
            directoryservice.directories[
                "test-directory"
            ].snapshots_limits.manual_snapshots_limit_reached
            is True
        )