add validation to data

konviva-events/tests/test_app.py

@@ -1,7 +1,13 @@
 from http import HTTPMethod, HTTPStatus
 
 from layercake.dateutils import now
-from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair, PartitionKey
+from layercake.dynamodb import (
+    DynamoDBPersistenceLayer,
+    KeyPair,
+    PartitionKey,
+    SortKey,
+    TransactKey,
+)
 
 from .conftest import HttpApiProxy, LambdaContext
 
@@ -22,7 +28,7 @@ def test_start_progress(
                 'ID_MATRICULA': '123',
                 'APROVEITAMENTO': '23.152173913043477',
                 'ANDAMENTO': '38.888888888888886',
-                'event_name': 'UPDATING',
+                'status': 'IN_PROGRESS',
             },
         ),
         lambda_context,
@@ -58,7 +64,7 @@ def test_update_progress(
                 'ID_MATRICULA': '456',
                 'APROVEITAMENTO': '23.152173913043477',
                 'ANDAMENTO': '12.888888888888886',
-                'event_name': 'UPDATING',
+                'status': 'IN_PROGRESS',
             },
         ),
         lambda_context,
@@ -99,7 +105,7 @@ def test_set_as_completed(
                 'ID_MATRICULA': '567',
                 'APROVEITAMENTO': '89.152173913043477',
                 'ANDAMENTO': '100',
-                'event_name': 'COMPLETED',
+                'status': 'COMPLETED',
             },
         ),
         lambda_context,
@@ -119,7 +125,6 @@ def test_set_as_completed(
         item.get('sk') == 'SCHEDULE#REMINDER_CERT_EXPIRATION_BEFORE_30_DAYS'
         for item in r['items']
     )
-    assert any(item.get('sk') == 'SCHEDULE#SET_CERT_EXPIRED' for item in r['items'])
 
     r = dynamodb_persistence_layer.collection.query(PartitionKey('LOCK'))
     assert len(r['items']) == 1
@@ -148,7 +153,7 @@ def test_set_as_failed(
                 'ID_MATRICULA': '567',
                 'APROVEITAMENTO': '12.152173913043477',
                 'ANDAMENTO': '100',
-                'event_name': 'COMPLETED',
+                'status': 'COMPLETED',
             },
         ),
         lambda_context,
@@ -156,7 +161,10 @@ def test_set_as_failed(
     assert r['statusCode'] == HTTPStatus.NO_CONTENT
 
     # Check `seeds.jsonl` for sample data related to this query
-    r = dynamodb_persistence_layer.collection.query(
-        PartitionKey('6c7e3d9b-f5d1-4da4-9e55-0825bb6ff2b8')
+    r = dynamodb_persistence_layer.collection.get_items(
+        TransactKey('6c7e3d9b-f5d1-4da4-9e55-0825bb6ff2b8')
+        + SortKey('0')
+        + SortKey('METADATA#DEDUPLICATION_WINDOW', rename_key='dedup_window')
     )
-    assert any(item.get('sk') == 'FAILED' for item in r['items'])
+    assert 'failed_at' in r
+    assert 'dedup_window' in r