add

order-events/tests/events/test_assign_tenant.py

@@ -0,0 +1,28 @@
+from aws_lambda_powertools.utilities.typing import LambdaContext
+from layercake.dynamodb import DynamoDBPersistenceLayer, PartitionKey
+
+import events.assign_tenant_cnpj as app
+
+
+def test_assign_tenant_cnpj(
+    dynamodb_seeds,
+    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
+    lambda_context: LambdaContext,
+):
+    event = {
+        'detail': {
+            'new_image': {
+                'id': '9omWNKymwU5U4aeun6mWzZ',
+                'cnpj': '15608435000190',
+                'email': 'sergio@somosbeta.com.br',
+            }
+        }
+    }
+
+    assert app.lambda_handler(event, lambda_context)  # type: ignore
+
+    result = dynamodb_persistence_layer.collection.query(
+        PartitionKey('9omWNKymwU5U4aeun6mWzZ')
+    )
+
+    assert 4 == len(result['items'])