diff --git a/test/web_agent_test.py b/test/web_agent_test.py index 1a1fa82..7d2e14a 100644 --- a/test/web_agent_test.py +++ b/test/web_agent_test.py @@ -1,9 +1,10 @@ -import asyncio -import unittest from datetime import datetime -from unittest.mock import Mock, MagicMock, patch, call -import xml.etree.ElementTree as ET from typing import Iterator +from unittest.mock import Mock, MagicMock, patch, call +import asyncio +import time +import unittest +import xml.etree.ElementTree as ET from sia.web_io_buffer import WebIOBuffer from sia.working_memory import WorkingMemory @@ -199,6 +200,7 @@ class WebAgentTest(unittest.TestCase): # Approve response self.agent.approve_response() + time.sleep(1) # Verify entry added to working memory self.assertEqual(self.working_memory.get_entries_count(), 1)