166 lines
4.9 KiB
Python
166 lines
4.9 KiB
Python
import base64
|
|
import json
|
|
import os
|
|
from dataclasses import dataclass
|
|
from http import HTTPMethod
|
|
|
|
import jsonlines
|
|
import pytest
|
|
|
|
PYTEST_TABLE_NAME = 'pytest'
|
|
PK = 'id'
|
|
SK = 'sk'
|
|
|
|
|
|
# https://docs.pytest.org/en/7.1.x/reference/reference.html#pytest.hookspec.pytest_configure
|
|
def pytest_configure():
|
|
os.environ['TZ'] = 'America/Sao_Paulo'
|
|
os.environ['DYNAMODB_PARTITION_KEY'] = PK
|
|
os.environ['DYNAMODB_SORT_KEY'] = SK
|
|
os.environ['DOCUSEAL_KEY'] = 'gUWhWtYBgTaP8fc1q5GZ6JuUHaZzMgZna6KFBHz3Gzk'
|
|
os.environ['USER_TABLE'] = PYTEST_TABLE_NAME
|
|
os.environ['COURSE_TABLE'] = PYTEST_TABLE_NAME
|
|
os.environ['ENROLLMENT_TABLE'] = PYTEST_TABLE_NAME
|
|
os.environ['ORDER_TABLE'] = PYTEST_TABLE_NAME
|
|
os.environ['ENROLLMENT_TABLE'] = PYTEST_TABLE_NAME
|
|
os.environ['BUCKET_NAME'] = 'saladeaula.digital'
|
|
os.environ['LOG_LEVEL'] = 'DEBUG'
|
|
|
|
|
|
@dataclass
|
|
class LambdaContext:
|
|
function_name: str = 'test'
|
|
memory_limit_in_mb: int = 128
|
|
invoked_function_arn: str = 'arn:aws:lambda:eu-west-1:809313241:function:test'
|
|
aws_request_id: str = '52fdfc07-2182-154f-163f-5f0f9a621d72'
|
|
|
|
|
|
@pytest.fixture
|
|
def lambda_context() -> LambdaContext:
|
|
return LambdaContext()
|
|
|
|
|
|
class HttpApiProxy:
|
|
def __call__(
|
|
self,
|
|
raw_path: str,
|
|
method: str = HTTPMethod.GET,
|
|
body: dict = {},
|
|
*,
|
|
headers: dict = {},
|
|
auth_flow_type: str = 'USER_AUTH',
|
|
queryStringParameters: dict = {},
|
|
**kwargs,
|
|
) -> dict:
|
|
return {
|
|
'version': '2.0',
|
|
'routeKey': '$default',
|
|
'rawPath': raw_path,
|
|
'rawQueryString': 'parameter1=value1¶meter1=value2¶meter2=value',
|
|
'cookies': ['cookie1', 'cookie2'],
|
|
'headers': headers,
|
|
'queryStringParameters': queryStringParameters,
|
|
'requestContext': {
|
|
'accountId': '123456789012',
|
|
'apiId': 'api-id',
|
|
'authorizer': {
|
|
'lambda': {
|
|
'user': {
|
|
'name': 'Sérgio R Siqueira',
|
|
'email': 'sergio@somosbeta.com.br',
|
|
'email_verified': 'true',
|
|
'custom:user_id': '5OxmMjL-ujoR5IMGegQz',
|
|
'sub': 'c4f30dbd-083e-4b84-aa50-c31afe9b9c01',
|
|
},
|
|
'auth_flow_type': auth_flow_type,
|
|
},
|
|
'jwt': {
|
|
'claims': {'claim1': 'value1', 'claim2': 'value2'},
|
|
'scopes': ['scope1', 'scope2'],
|
|
},
|
|
},
|
|
'domainName': 'id.execute-api.us-east-1.amazonaws.com',
|
|
'domainPrefix': 'id',
|
|
'http': {
|
|
'method': str(method),
|
|
'path': raw_path,
|
|
'protocol': 'HTTP/1.1',
|
|
'sourceIp': '192.168.0.1/32',
|
|
'userAgent': 'agent',
|
|
},
|
|
'requestId': 'id',
|
|
'routeKey': '$default',
|
|
'stage': '$default',
|
|
'time': '12/Mar/2020:19:03:58 +0000',
|
|
'timeEpoch': 1583348638390,
|
|
},
|
|
'body': _base64_dict(body),
|
|
'pathParameters': {'parameter1': 'value1'},
|
|
'isBase64Encoded': True,
|
|
'stageVariables': {'stageVariable1': 'value1', 'stageVariable2': 'value2'},
|
|
}
|
|
|
|
|
|
def _base64_dict(obj: dict = {}) -> str | None:
|
|
if not obj:
|
|
return None
|
|
return base64.b64encode(json.dumps(obj).encode()).decode()
|
|
|
|
|
|
@pytest.fixture
|
|
def http_api_proxy():
|
|
return HttpApiProxy()
|
|
|
|
|
|
@pytest.fixture
|
|
def dynamodb_client():
|
|
from boto3clients import dynamodb_client as client
|
|
|
|
client.create_table(
|
|
AttributeDefinitions=[
|
|
{'AttributeName': PK, 'AttributeType': 'S'},
|
|
{'AttributeName': SK, 'AttributeType': 'S'},
|
|
],
|
|
TableName=PYTEST_TABLE_NAME,
|
|
KeySchema=[
|
|
{'AttributeName': PK, 'KeyType': 'HASH'},
|
|
{'AttributeName': SK, 'KeyType': 'RANGE'},
|
|
],
|
|
ProvisionedThroughput={
|
|
'ReadCapacityUnits': 123,
|
|
'WriteCapacityUnits': 123,
|
|
},
|
|
)
|
|
|
|
yield client
|
|
|
|
client.delete_table(TableName=PYTEST_TABLE_NAME)
|
|
|
|
|
|
@pytest.fixture()
|
|
def dynamodb_persistence_layer(dynamodb_client):
|
|
from layercake.dynamodb import DynamoDBPersistenceLayer
|
|
|
|
return DynamoDBPersistenceLayer(PYTEST_TABLE_NAME, dynamodb_client)
|
|
|
|
|
|
@pytest.fixture()
|
|
def dynamodb_seeds(dynamodb_client):
|
|
from layercake.dynamodb import serialize
|
|
|
|
with open('tests/seeds.jsonl', 'rb') as fp:
|
|
reader = jsonlines.Reader(fp)
|
|
|
|
for line in reader.iter(type=dict, skip_invalid=True):
|
|
dynamodb_client.put_item(
|
|
TableName=PYTEST_TABLE_NAME,
|
|
Item=serialize(line),
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def app():
|
|
import app
|
|
|
|
return app
|