78 lines
2.2 KiB
Python
78 lines
2.2 KiB
Python
import pytest
|
|
from unittest.mock import Mock
|
|
|
|
from atst.domain.csp.cloud import AWSCloudProvider
|
|
|
|
|
|
AWS_CONFIG = {
|
|
"AWS_ACCESS_KEY_ID": "",
|
|
"AWS_SECRET_KEY": "",
|
|
"AWS_REGION_NAME": "us-fake-1",
|
|
}
|
|
AUTH_CREDENTIALS = {
|
|
"aws_access_key_id": AWS_CONFIG["AWS_ACCESS_KEY_ID"],
|
|
"aws_secret_access_key": AWS_CONFIG["AWS_SECRET_KEY"],
|
|
}
|
|
|
|
|
|
def mock_boto_organizations(_config=None, **kwargs):
|
|
describe_create_account_status = (
|
|
"SUCCEEDED"
|
|
if _config.get("organizations.describe_create_account.failure", False) == False
|
|
else "FAILED"
|
|
)
|
|
|
|
import boto3
|
|
|
|
mock = Mock(wraps=boto3.client("organizations", **kwargs))
|
|
|
|
# https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/organizations.html#Organizations.Client.create_account
|
|
mock.create_account = Mock(
|
|
return_value={
|
|
"CreateAccountStatus": {
|
|
"Id": "create-account-status-id",
|
|
"AccountName": "account-name",
|
|
"AccountId": "account-id",
|
|
"State": "SUCCEEDED",
|
|
}
|
|
}
|
|
)
|
|
|
|
# https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/organizations.html#Organizations.Client.describe_create_account_status
|
|
mock.describe_create_account_status = Mock(
|
|
return_value={
|
|
"CreateAccountStatus": {
|
|
"Id": "create-account-status-id",
|
|
"AccountName": "account-name",
|
|
"AccountId": "account-id",
|
|
"State": describe_create_account_status,
|
|
}
|
|
}
|
|
)
|
|
return mock
|
|
|
|
|
|
class MockBoto3:
|
|
def __init__(self, config=None):
|
|
self.config = config or {}
|
|
|
|
def client(self, client_name, **kwargs):
|
|
return {"organizations": mock_boto_organizations}[client_name](
|
|
**kwargs, _config=self.config
|
|
)
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def mock_boto3(request):
|
|
marks = request.node.get_closest_marker("mock_boto3")
|
|
if marks:
|
|
mock_config = marks.args[0] if len(marks.args) else {}
|
|
else:
|
|
mock_config = {}
|
|
return MockBoto3(mock_config)
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def mock_aws(mock_boto3):
|
|
return AWSCloudProvider(AWS_CONFIG, boto3=mock_boto3)
|