96 lines
2.7 KiB
Python
96 lines
2.7 KiB
Python
import base64
|
|
import json
|
|
from dataclasses import dataclass
|
|
from http import HTTPMethod
|
|
|
|
import pytest
|
|
|
|
|
|
@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'
|
|
|
|
|
|
class HttpApiProxy:
|
|
def __call__(
|
|
self,
|
|
raw_path: str,
|
|
method: str = HTTPMethod.GET,
|
|
body: dict = {},
|
|
*,
|
|
headers: 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': {
|
|
'parameter1': 'value1,value2',
|
|
'parameter2': 'value',
|
|
},
|
|
'requestContext': {
|
|
'accountId': '123456789012',
|
|
'apiId': 'api-id',
|
|
'authorizer': {
|
|
'lambda': {
|
|
'user': {},
|
|
'tenant': '*',
|
|
},
|
|
'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 lambda_context() -> LambdaContext:
|
|
return LambdaContext()
|
|
|
|
|
|
@pytest.fixture
|
|
def http_api_proxy():
|
|
return HttpApiProxy()
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_app(monkeypatch):
|
|
monkeypatch.setattr('settings.ELASTIC_CONN', {'hosts': 'http://127.0.0.1:9200'})
|
|
|
|
import app
|
|
|
|
return app
|