diff --git a/opax-mcp/test_phase2b_firestore_store.py b/opax-mcp/test_phase2b_firestore_store.py index 27abf3d..1734033 100644 --- a/opax-mcp/test_phase2b_firestore_store.py +++ b/opax-mcp/test_phase2b_firestore_store.py @@ -41,7 +41,7 @@ class TestPhase2BFirestoreStore(unittest.IsolatedAsyncioTestCase): # Mocks for record_audit_event self.mock_firestore_client.collection.return_value.document.return_value.create = AsyncMock() - self.store = FirestoreMemoryStore(project_id="test-project", client=self.mock_firestore_client) + self.store = FirestoreMemoryStore(project_id="test-project", client=self.mock_firestore_client) def tearDown(self): self.transaction_patcher.stop() @@ -56,9 +56,9 @@ class TestPhase2BFirestoreStore(unittest.IsolatedAsyncioTestCase): self.mock_firestore_client.collection.assert_called_with("emma_conversations") collection_mock = self.mock_firestore_client.collection.return_value - collection_mock.document.assert_called_with(convo.conversation_id) - document_mock = collection_mock.document.return_value - document_mock.set.assert_awaited_once() + collection_mock.document.assert_called_with(convo.conversation_id) + document_mock = collection_mock.document.return_value + document_mock.set.assert_awaited_once() async def test_append_message_uses_subcollection_and_updates_timestamp(self):