Add message to assert

Author: strawgate

key-value/key-value-aio/src/key_value/aio/stores/dynamodb/store.py

@@ -200,11 +200,13 @@ async def _get_managed_entry(self, *, key: str, collection: str) -> ManagedEntry
         if not json_value:
             return None
 
-        managed_entry = self._serialization_adapter.load_json(json_str=json_value)
+        managed_entry: ManagedEntry = self._serialization_adapter.load_json(json_str=json_value)
 
-        ttl = item.get("ttl", {}).get("N")
-        if ttl:
-            managed_entry.expires_at = datetime.fromtimestamp(int(ttl), tz=timezone.utc)
+        expires_at_epoch = item.get("ttl", {}).get("N")
+
+        # Our managed entry may carry a TTL, but the TTL in DynamoDB takes precedence.
+        if expires_at_epoch:
+            managed_entry.expires_at = datetime.fromtimestamp(int(expires_at_epoch), tz=timezone.utc)
 
         return managed_entry
 

key-value/key-value-aio/tests/stores/base.py

@@ -174,7 +174,7 @@ async def test_put_ttl_get_ttl(self, store: BaseStore):
 
         assert value == {"test": "test"}
         assert ttl is not None
-        assert ttl == IsFloat(approx=100)
+        assert ttl == IsFloat(approx=100), f"TTL should be ~100, but is {ttl}"
 
     async def test_negative_ttl(self, store: BaseStore):
         """Tests that a negative ttl will return None when getting the key."""

key-value/key-value-sync/tests/code_gen/stores/base.py

@@ -171,7 +171,7 @@ def test_put_ttl_get_ttl(self, store: BaseStore):
 
         assert value == {"test": "test"}
         assert ttl is not None
-        assert ttl == IsFloat(approx=100)
+        assert ttl == IsFloat(approx=100), f"TTL should be ~100, but is {ttl}"
 
     def test_negative_ttl(self, store: BaseStore):
         """Tests that a negative ttl will return None when getting the key."""