25 lines
752 B
Python
25 lines
752 B
Python
import app.events.stopgap.add_tenant as app
|
|
from aws_lambda_powertools.utilities.typing import LambdaContext
|
|
from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair
|
|
|
|
|
|
def test_add_tenant(
|
|
dynamodb_client,
|
|
dynamodb_persistence_layer: DynamoDBPersistenceLayer,
|
|
lambda_context: LambdaContext,
|
|
):
|
|
event = {
|
|
'detail': {
|
|
'new_image': {
|
|
'id': 'orgmembers#cJtK9SsnJhKPyxESe7g3DG',
|
|
'sk': '5OxmMjL-ujoR5IMGegQz',
|
|
},
|
|
}
|
|
}
|
|
assert app.lambda_handler(event, lambda_context) # type: ignore
|
|
result = dynamodb_persistence_layer.collection.get_item(
|
|
KeyPair('5OxmMjL-ujoR5IMGegQz', '0')
|
|
)
|
|
|
|
assert 'cJtK9SsnJhKPyxESe7g3DG' in result['tenant_id']
|