diff --git a/http-api/middlewares/audit_log_middleware.py b/http-api/middlewares/audit_log_middleware.py index 24bc7c4..6ec2573 100644 --- a/http-api/middlewares/audit_log_middleware.py +++ b/http-api/middlewares/audit_log_middleware.py @@ -89,6 +89,7 @@ class AuditLogMiddleware(BaseMiddlewareHandler): action=self.action, data=data, ip=ip_addr, + author='himself', ttl=retention_days, ) diff --git a/http-api/middlewares/tenant_middelware.py b/http-api/middlewares/tenant_middelware.py index 4b8bbc1..17fa2a0 100644 --- a/http-api/middlewares/tenant_middelware.py +++ b/http-api/middlewares/tenant_middelware.py @@ -4,7 +4,11 @@ from aws_lambda_powertools.event_handler.api_gateway import ( APIGatewayHttpResolver, Response, ) -from aws_lambda_powertools.event_handler.exceptions import BadRequestError, ServiceError +from aws_lambda_powertools.event_handler.exceptions import ( + BadRequestError, + NotFoundError, + ServiceError, +) from aws_lambda_powertools.event_handler.middlewares import ( BaseMiddlewareHandler, NextMiddleware, @@ -102,6 +106,8 @@ def _tenant( If tenant_id is not provided. ForbiddenError If the user lacks the necessary ACL permissions for the specified tenant_id. + NotFoundError + If tenant not found. """ if not tenant_id: raise BadRequestError('Missing tenant') @@ -116,5 +122,5 @@ def _tenant( if tenant_id == '*': return Tenant(id=tenant_id, name='default') - obj = collect.get_item(KeyPair(tenant_id, '0')) + obj = collect.get_item(KeyPair(tenant_id, '0'), exception_cls=NotFoundError) return Tenant.parse_obj(obj) diff --git a/http-api/tests/routes/test_courses.py b/http-api/tests/routes/test_courses.py index 48e8b24..09e14db 100644 --- a/http-api/tests/routes/test_courses.py +++ b/http-api/tests/routes/test_courses.py @@ -31,13 +31,14 @@ def test_post_course( lambda_context, ) - # print(r) - assert 'id' in json.loads(r['body']) assert r['statusCode'] == HTTPStatus.CREATED collect = DynamoDBCollection(dynamodb_persistence_layer) logs = collect.get_items( - PartitionKey(ComposeKey('5OxmMjL-ujoR5IMGegQz', prefix='log', delimiter=':')) + PartitionKey( + ComposeKey('5OxmMjL-ujoR5IMGegQz', prefix='log', delimiter=':'), + ) ) + assert len(logs['items']) == 3 print(logs)