add postgres

konviva-events/tests/conftest.py

@@ -1,5 +1,8 @@
+import base64
+import json
 import os
 from dataclasses import dataclass
+from http import HTTPMethod
 
 import jsonlines
 import pytest
@@ -27,11 +30,83 @@ class LambdaContext:
     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 = {},
+        auth_flow_type: str = 'USER_AUTH',
+        queryStringParameters: dict = {},
+        **kwargs,
+    ) -> dict:
+        return {
+            'version': '2.0',
+            'routeKey': '$default',
+            'rawPath': raw_path,
+            'rawQueryString': 'parameter1=value1&parameter1=value2&parameter2=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 lambda_context() -> LambdaContext:
     return LambdaContext()
 
 
+@pytest.fixture
+def http_api_proxy():
+    return HttpApiProxy()
+
+
 @pytest.fixture
 def dynamodb_client():
     from boto3clients import dynamodb_client as client
@@ -57,8 +132,29 @@ def dynamodb_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):
-    with jsonlines.open('tests/seeds.jsonl') as lines:
-        for line in lines:
-            dynamodb_client.put_item(TableName=PYTEST_TABLE_NAME, Item=line)
+def 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