✅ bot run coverage

.requirements/requirements.txt

@@ -21,7 +21,7 @@ click==8.1.6 ; python_version >= "3.10" and python_version < "4.0"
 colorama==0.4.6 ; python_version >= "3.10" and python_version < "4.0" and (sys_platform == "win32" or platform_system == "Windows")
 cryptography==41.0.2 ; python_version >= "3.10" and python_version < "4.0"
 cytoolz==0.12.2 ; python_version >= "3.10" and python_version < "4" and implementation_name == "cpython"
-dxsp==4.2.1 ; python_version >= "3.10" and python_version < "4.0"
+dxsp==4.2.2 ; python_version >= "3.10" and python_version < "4.0"
 dynaconf==3.2.0 ; python_version >= "3.10" and python_version < "4.0"
 emoji==2.7.0 ; python_version >= "3.10" and python_version < "4.0"
 eth-abi==4.1.0 ; python_version >= "3.10" and python_version < "4"
@@ -35,7 +35,7 @@ eth-typing==3.4.0 ; python_version >= "3.10" and python_version < "4"
 eth-utils==2.2.0 ; python_version >= "3.10" and python_version < "4"
 exceptiongroup==1.1.2 ; python_version >= "3.10" and python_version < "3.11"
 fastapi==0.100.0 ; python_version >= "3.10" and python_version < "4.0"
-findmyorder==1.7.0 ; python_version >= "3.10" and python_version < "4.0"
+findmyorder==1.7.1 ; python_version >= "3.10" and python_version < "4.0"
 frozendict==2.3.8 ; python_version >= "3.10" and python_version < "4.0"
 frozenlist==1.4.0 ; python_version >= "3.10" and python_version < "4.0"
 future==0.18.3 ; python_version >= "3.10" and python_version < "4.0"
@@ -47,15 +47,15 @@ html5lib==1.1 ; python_version >= "3.10" and python_version < "4.0"
 httpcore==0.17.3 ; python_version >= "3.10" and python_version < "4.0"
 httpx==0.24.1 ; python_version >= "3.10" and python_version < "4.0"
 hyperframe==6.0.1 ; python_version >= "3.10" and python_version < "4.0"
-iamlistening==3.0.1 ; python_version >= "3.10" and python_version < "4.0"
+iamlistening==3.0.2 ; python_version >= "3.10" and python_version < "4.0"
 idna==3.4 ; python_version >= "3.10" and python_version < "4.0"
 jsonschema-specifications==2023.7.1 ; python_version >= "3.10" and python_version < "4.0"
 jsonschema==4.18.4 ; python_version >= "3.10" and python_version < "4.0"
 logbook==1.5.3 ; python_version >= "3.10" and python_version < "4.0"
 loguru==0.7.0 ; python_version >= "3.10" and python_version < "4.0"
 lru-dict==1.2.0 ; python_version >= "3.10" and python_version < "4.0"
 lxml==4.9.3 ; python_version >= "3.10" and python_version < "4.0"
-markdown==3.4.3 ; python_version >= "3.10" and python_version < "4.0"
+markdown==3.4.4 ; python_version >= "3.10" and python_version < "4.0"
 matrix-nio==0.20.2 ; python_version >= "3.10" and python_version < "4.0"
 multidict==6.0.4 ; python_version >= "3.10" and python_version < "4.0"
 multitasking==0.0.11 ; python_version >= "3.10" and python_version < "4.0"
@@ -75,8 +75,8 @@ pycares==4.3.0 ; python_version >= "3.10" and python_version < "4.0"
 pycoingecko==3.1.0 ; python_version >= "3.10" and python_version < "4.0"
 pycparser==2.21 ; python_version >= "3.10" and python_version < "4.0"
 pycryptodome==3.18.0 ; python_version >= "3.10" and python_version < "4.0"
-pydantic-core==2.3.0 ; python_version >= "3.10" and python_version < "4.0"
-pydantic==2.0.3 ; python_version >= "3.10" and python_version < "4.0"
+pydantic-core==2.4.0 ; python_version >= "3.10" and python_version < "4.0"
+pydantic==2.1.1 ; python_version >= "3.10" and python_version < "4.0"
 pyparsing==3.1.0 ; python_version >= "3.10" and python_version < "4.0"
 python-cryptography-fernet-wrapper==1.0.4 ; python_version >= "3.10" and python_version < "4.0"
 python-dateutil==2.8.2 ; python_version >= "3.10" and python_version < "4.0"

pyproject.toml

@@ -50,7 +50,7 @@ ping3 = "^4.0.4"
 ccxt = "^4.0.0"
 dxsp = "^4.1.0"
 findmyorder = "^1.6.1"
-iamlistening = "^3.0.1"
+iamlistening = "^3.0.2"
 talkytrend = "^1.12.1"
 
 
@@ -73,6 +73,12 @@ exclude = [
 format = "github"
 fixable = ["ALL"]
 
+[tool.black]
+target-version = ['py39', 'py310', 'py311']
+line-length = 88
+# exclude = "((.eggs | .git | .pytype | .pytest_cache | build | dist))"
+
+
 [tool.pylint.exceptions]
 overgeneral-exceptions = [
     "builtins.BaseException",
@@ -103,6 +109,7 @@ pythonpath = ". tt"
 testpaths = "tests"
 python_classes = "Test*"
 log_format = "%(asctime)s - %(levelname)s - %(name)s - %(message)s"
+addopts = "-v"
 log_level = "DEBUG"
 filterwarnings = [
     " ignore:.*U.*mode is deprecated:DeprecationWarning",

tests/test_unit.py

@@ -2,11 +2,12 @@
  TT test
 """
 import asyncio
-from unittest.mock import AsyncMock
+from unittest.mock import AsyncMock, patch
 
 import pytest
 from fastapi.testclient import TestClient
-from iamlistening import Listener
+from iamlistening import ChatManager, Listener
+from iamlistening.platform import TelegramHandler
 
 from tt.bot import app
 from tt.config import settings
@@ -27,6 +28,12 @@
 def listener_test():
     return Listener()
 
+@pytest.fixture(name="ial_test")
+def ial_test():
+    listener = Listener()
+    listener.handler = TelegramHandler()
+    return listener
+
 @pytest.fixture(name="plugin_manager_obj")
 def pluginmngr_test():
     return PluginManager()
@@ -35,6 +42,30 @@
 def message():
     return "Test message"
 
+@pytest.fixture
+def mock_listener():
+   mock_listener = AsyncMock(spec=Listener)
+   mock_listener.handler = AsyncMock(spec=ChatManager)
+   return mock_listener
+
+@pytest.fixture
+def mock_plugin_manager():
+   return AsyncMock(spec=PluginManager)
+
+
+
+@pytest.mark.asyncio
+async def test_run_bot_task():
+    async with pytest.AsyncioFixture(None) as loop:
+
+        task = loop.create_task(run_bot())
+        await asyncio.sleep(0.1)
+        assert not task.done()
+        task.cancel()
+        await task
+        assert task.done()
+        assert task.cancelled()
+
 
 def test_app_endpoint_main():
     client = TestClient(app)
@@ -80,46 +111,29 @@
 
 
 @pytest.mark.asyncio
-async def test_start_bot(listener_obj, plugin_manager_obj):
-    start = AsyncMock()
-    task = asyncio.create_task(start_bot(listener_obj, plugin_manager_obj))
-    task.cancel()
-    with pytest.raises(asyncio.CancelledError):
-        start.assert_awaited
-        await task
+async def test_run_bot():
+    listener_instance = Listener()
+    start_bot = AsyncMock(side_effect=[listener_instance])
+    with patch('tt.utils.start_bot', start_bot):
+        task = asyncio.create_task(run_bot())
+        await asyncio.gather(task, asyncio.sleep(2))
+        start_bot.assert_awaited
+        listener_created = listener_instance
+        assert isinstance(listener_created, Listener) 
 
 
 @pytest.mark.asyncio
-async def test_run_bot(caplog):
-    start_bot = AsyncMock()
-    task = asyncio.create_task(run_bot())
-    start_bot.assert_awaited
-    task.cancel()
-    with pytest.raises(asyncio.CancelledError):
-        await task
-
-
-# @pytest.mark.asyncio
-# async def test_get_latest_message(message):
-#     listener = Listener()
-#     await listener.start()
-#     assert listener is not None
-#     assert settings.VALUE == "On Testing"
-#     await listener.handler.handle_message(message)
-#     assert await listener.handler.get_latest_message() == message
-
-
-# @pytest.mark.asyncio
-# async def test_listener_handler():
-#     listener_test = Listener()
-#     print(listener_test)
-#     assert listener_test is not None
-#     assert isinstance(listener_test, Listener)
-#     await listener_test.start()
-#     await listener_test.handler.handle_message("hello")
-#     msg = await listener_test.handler.get_latest_message()
-#     print(msg)
-#     assert msg == "hello"
+async def test_start_bot():
+
+    listener = AsyncMock(spec=Listener)
+    listener.handler = AsyncMock(spec=ChatManager)
+    plugin_manager = AsyncMock(spec=PluginManager)
+    await start_bot(
+        listener, 
+        plugin_manager,
+        max_iterations=1)
+    listener.start.assert_awaited_once()
+    listener.handler.get_latest_message.assert_awaited_once()
 
 
 @pytest.mark.asyncio

tt/ __init__.py

@@ -1 +0,0 @@
-__version__ = "2.2.2"

tt/config.py

@@ -71,7 +71,7 @@ def loguru_setup():
         }
     logging.basicConfig(handlers=[InterceptHandler()], level=0, force=True)
     loguru_logger.add(
-        sink=sys.stderr,
+        sink=sys.stdout,
         level=settings.loglevel,
         filter=log_filters,
     )

tt/utils.py

@@ -24,7 +24,7 @@ async def send_notification(msg):
         body_format=msg_format)
 
 
-async def run_bot(max_iterations=None):
+async def run_bot():
     """
     🤖 Run the chat bot & the plugins.
     """
@@ -43,15 +43,18 @@ async def start_plugins(plugin_manager):
         loop.create_task(plugin_manager.start_all_plugins())
 
 
-async def start_bot(listener, plugin_manager):
+async def start_bot(listener, plugin_manager, max_iterations=None):
     """
-    👂 Listen to the bot and dispatch to plugins
+    👂 Start the chat listener and dispatch to plugins
     """
     await listener.start()
     await start_plugins(plugin_manager)
-    while True:
+    iteration = 0
+    while iteration < max_iterations:
         msg = await listener.handler.get_latest_message()
         if msg and settings.plugin_enabled:
             await plugin_manager.process_message(msg)
+        iteration += 1
 
     await asyncio.sleep(1)
+