tidy

Author: nhsdevws

ack_backend/tests/test_audit_table.py

@@ -4,7 +4,7 @@
 from errors import UnhandledAuditTableError
 
 class TestAuditTable(unittest.TestCase):
-    
+
     def setUp(self):
         self.logger_patcher = patch('audit_table.logger')
         self.mock_logger = self.logger_patcher.start()
@@ -24,14 +24,14 @@ def test_get_next_queued_file_details_returns_oldest(self):
         self.mock_dynamodb_resource.Table.return_value = mock_table
         mock_table.query.return_value = {
             "Items": [
-                {"timestamp": 2, "foo": "bar"},
-                {"timestamp": 1, "foo": "baz"},
+                {"timestamp": 2, "my-key": "value2"},
+                {"timestamp": 1, "my-key": "value1"},
             ]
         }
         # Act
         result = audit_table.get_next_queued_file_details("queue1")
         # Assert
-        self.assertEqual(result, {"timestamp": 1, "foo": "baz"})
+        self.assertEqual(result, {"timestamp": 1, "my-key": "value1"})
 
     def test_get_next_queued_file_details_returns_none_if_empty(self):
         mock_table = MagicMock()