update conftest

layercake/layercake/dynamodb.py

@@ -11,10 +11,10 @@ from botocore.exceptions import ClientError
 from .dateutils import now, timestamp
 
 TZ = os.getenv('TZ', 'UTC')
-DELIMITER = os.getenv('DELIMITER', '#')
-LIMIT = int(os.getenv('LIMIT', 25))
-PARTITION_KEY = os.getenv('PARTITION_KEY', 'pk')
-SORT_KEY = os.getenv('SORT_KEY', 'sk')
+PK = os.getenv('DYNAMODB_PARTITION_KEY', 'pk')
+SK = os.getenv('DYNAMODB_SORT_KEY', 'sk')
+LIMIT = int(os.getenv('DYNAMODB_LIMIT', 25))
+DELIMITER = os.getenv('DYNAMODB_DELIMITER', '#')
 
 
 logger = Logger(__name__)
@@ -85,20 +85,20 @@ class PartitionKey(PrimaryKey):
     """Represents a partition key for DynamoDB queries"""
 
     def __init__(self, pk: str) -> None:
-        super().__init__(**{PARTITION_KEY: pk})
+        super().__init__(**{PK: pk})
 
     def expr_attr_name(self) -> dict:
-        return {'#pk': PARTITION_KEY}
+        return {'#pk': PK}
 
     def expr_attr_values(self) -> dict:
-        return {':pk': self[PARTITION_KEY]}
+        return {':pk': self[PK]}
 
 
 class KeyPair(PrimaryKey):
     """Represents a composite key (partition key and sort key) for DynamoDB queries"""
 
     def __init__(self, pk: str, sk: str) -> None:
-        super().__init__(**{PARTITION_KEY: pk, SORT_KEY: sk})
+        super().__init__(**{PK: pk, SK: sk})
 
     def __repr__(self) -> str:
         pk, sk = self.values()
@@ -106,14 +106,14 @@ class KeyPair(PrimaryKey):
 
     def expr_attr_name(self) -> dict:
         return {
-            '#pk': PARTITION_KEY,
-            '#sk': SORT_KEY,
+            '#pk': PK,
+            '#sk': SK,
         }
 
     def expr_attr_values(self) -> dict:
         return {
-            ':pk': self[PARTITION_KEY],
-            ':sk': self[SORT_KEY],
+            ':pk': self[PK],
+            ':sk': self[SK],
         }