update address

http-api/tests/routes/test_users.py

@@ -2,7 +2,6 @@ import json
 from http import HTTPMethod, HTTPStatus
 
 from layercake.dynamodb import (
-    DynamoDBCollection,
     DynamoDBPersistenceLayer,
     KeyPair,
     SortKey,
@@ -32,13 +31,12 @@ def test_update_user_cpf(
     )
     assert r['statusCode'] == HTTPStatus.OK
 
-    collect = DynamoDBCollection(dynamodb_persistence_layer)
-    user = collect.get_items(
+    user = dynamodb_persistence_layer.collection.get_items(
         TransactKey('5OxmMjL-ujoR5IMGegQz')
         + SortKey('0')
-        + SortKey('last_profile_edit')
+        + SortKey('rate_limit#user_update')
     )
-    assert 'last_profile_edit' in user
+    assert 'rate_limit#user_update' in user
 
 
 def test_update_user_name(
@@ -216,8 +214,9 @@ def test_post_email(
 
     assert r['statusCode'] == HTTPStatus.CREATED
 
-    collect = DynamoDBCollection(dynamodb_persistence_layer)
-    user = collect.get_item(KeyPair('5OxmMjL-ujoR5IMGegQz', '0'))
+    user = dynamodb_persistence_layer.collection.get_item(
+        KeyPair('5OxmMjL-ujoR5IMGegQz', '0')
+    )
     assert user['emails'] == {
         'sergio@somosbeta.com.br',
         'osergiosiqueira@gmail.com',
@@ -274,8 +273,9 @@ def test_delete_email(
 
     assert r['statusCode'] == HTTPStatus.OK
 
-    collect = DynamoDBCollection(dynamodb_persistence_layer)
-    user = collect.get_item(KeyPair('5OxmMjL-ujoR5IMGegQz', '0'))
+    user = dynamodb_persistence_layer.collection.get_item(
+        KeyPair('5OxmMjL-ujoR5IMGegQz', '0')
+    )
     assert user['emails'] == {
         'sergio@somosbeta.com.br',
     }