add ttl

layercake/tests/test_dynamodb.py

@@ -4,6 +4,7 @@ from ipaddress import IPv4Address
 import pytest
 from botocore.exceptions import ClientError
 
+from layercake.dateutils import ttl
 from layercake.dynamodb import (
     DynamoDBCollection,
     DynamoDBPersistenceLayer,
@@ -58,29 +59,30 @@ def test_transact_write_items(
         dynamodb_persistence_layer.transact_write_items(transact)
 
 
-def test_collection(
+def test_collection_get_item(
     dynamodb_seeds,
     dynamodb_persistence_layer: DynamoDBPersistenceLayer,
 ):
     collect = DynamoDBCollection(dynamodb_persistence_layer)
-    tenant_item = collect.get_item(
+    data_notfound = collect.get_item(
         key=KeyPair(
             pk='5OxmMjL-ujoR5IMGegQz',
-            sk=Key('tenant'),
+            sk='tenant',
         ),
         raise_if_missing=False,
         default={},
     )
-    assert tenant_item == {}
+    assert data_notfound == {}
 
-    email_item = collect.get_item(
+    # This item was added from seeds
+    data = collect.get_item(
         key=KeyPair(
             pk='5OxmMjL-ujoR5IMGegQz',
             sk=Key('sergio@somosbeta.com.br', prefix='emails'),
         ),
         default={},
     )
-    assert email_item == {
+    assert data == {
         'email_verified': True,
         'mx_record_exists': True,
         'sk': 'emails#sergio@somosbeta.com.br',
@@ -92,3 +94,30 @@ def test_collection(
 
     with pytest.raises(DynamoDBCollection.MissingError):
         collect.get_item(key=KeyPair('5OxmMjL-ujoR5IMGegQz', 'notfound'))
+
+
+def test_collection_put_item(
+    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
+):
+    collect = DynamoDBCollection(dynamodb_persistence_layer)
+
+    assert collect.put_item(
+        key=KeyPair(
+            '5OxmMjL-ujoR5IMGegQz',
+            Key('6d1044d5-18c5-437c-9219-fc2ace7e5ebc', prefix='orgs'),
+        ),
+        name='Beta Educação',
+        ttl=ttl(days=3),
+    )
+
+    data = collect.get_item(
+        key=KeyPair(
+            pk='5OxmMjL-ujoR5IMGegQz',
+            sk=Key('6d1044d5-18c5-437c-9219-fc2ace7e5ebc', prefix='orgs'),
+        ),
+    )
+
+    assert data['sk'] == 'orgs#6d1044d5-18c5-437c-9219-fc2ace7e5ebc'
+    assert 'name' in data
+    assert 'ttl' in data
+    assert 'ttl_date' in data