mirror of
https://github.com/prowler-cloud/prowler.git
synced 2026-07-18 01:51:50 +00:00
Mario Rodriguez Lopez
67 lines
2.3 KiB
Python
67 lines
2.3 KiB
Python
from boto3 import client
|
|
from mock import patch
|
|
from moto import mock_aws
|
|
|
|
from prowler.providers.aws.services.appsync.appsync_service import AppSync
|
|
from tests.providers.aws.utils import (
|
|
AWS_ACCOUNT_NUMBER,
|
|
AWS_REGION_US_EAST_1,
|
|
set_mocked_aws_provider,
|
|
)
|
|
|
|
|
|
def mock_generate_regional_clients(provider, service):
|
|
regional_client = provider._session.current_session.client(
|
|
service, region_name=AWS_REGION_US_EAST_1
|
|
)
|
|
regional_client.region = AWS_REGION_US_EAST_1
|
|
return {AWS_REGION_US_EAST_1: regional_client}
|
|
|
|
|
|
@patch(
|
|
"prowler.providers.aws.aws_provider.AwsProvider.generate_regional_clients",
|
|
new=mock_generate_regional_clients,
|
|
)
|
|
class Test_AppSync_Service:
|
|
# Test AppSync Service
|
|
def test_service(self):
|
|
aws_provider = set_mocked_aws_provider()
|
|
appsync = AppSync(aws_provider)
|
|
assert appsync.service == "appsync"
|
|
|
|
# Test AppSync Client
|
|
def test_client(self):
|
|
aws_provider = set_mocked_aws_provider()
|
|
appsync = AppSync(aws_provider)
|
|
assert appsync.client.__class__.__name__ == "AppSync"
|
|
|
|
# Test AppSync Session
|
|
def test__get_session__(self):
|
|
aws_provider = set_mocked_aws_provider()
|
|
appsync = AppSync(aws_provider)
|
|
assert appsync.session.__class__.__name__ == "Session"
|
|
|
|
# Test AppSync Session
|
|
def test_audited_account(self):
|
|
aws_provider = set_mocked_aws_provider()
|
|
appsync = AppSync(aws_provider)
|
|
assert appsync.audited_account == AWS_ACCOUNT_NUMBER
|
|
|
|
# Test AppSync Describe File Systems
|
|
@mock_aws
|
|
def test_list_graphql_apis(self):
|
|
aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1])
|
|
appsync = client("appsync", region_name=AWS_REGION_US_EAST_1)
|
|
api = appsync.create_graphql_api(
|
|
name="test-api",
|
|
authenticationType="API_KEY",
|
|
logConfig={"fieldLogLevel": "ALL", "cloudWatchLogsRoleArn": "test"},
|
|
)
|
|
api_arn = api["graphqlApi"]["arn"]
|
|
appsync_client = AppSync(aws_provider)
|
|
|
|
assert appsync_client.graphql_apis[api_arn].name == "test-api"
|
|
assert appsync_client.graphql_apis[api_arn].field_log_level == "ALL"
|
|
assert appsync_client.graphql_apis[api_arn].authentication_type == "API_KEY"
|
|
assert appsync_client.graphql_apis[api_arn].tags == [{}]
|