✅ 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"
@@ -101,7 +101,7 @@ six==1.16.0 ; python_version >= "3.10" and python_version < "4.0"
 sniffio==1.3.0 ; python_version >= "3.10" and python_version < "4.0"
 soupsieve==2.4.1 ; python_version >= "3.10" and python_version < "4.0"
 starlette==0.27.0 ; python_version >= "3.10" and python_version < "4.0"
-talkytrend==1.12.1 ; python_version >= "3.10" and python_version < "4.0"
+talkytrend==1.12.3 ; python_version >= "3.10" and python_version < "4.0"
 telethon==1.29.2 ; python_version >= "3.10" and python_version < "4.0"
 toml==0.10.2 ; python_version >= "3.10" and python_version < "4.0"
 toolz==0.12.0 ; python_version >= "3.10" and python_version < "4" and (implementation_name == "pypy" or implementation_name == "cpython")

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_cex_exchange_plugin.py

@@ -142,3 +142,43 @@ async def test_execute_order(plugin, order_parsed):
     print(result)
     assert result is not None
     assert "⚠️ order execution" in result
+
+
+# @pytest.mark.asyncio
+# async def test_cex_exchange():
+#     exchange_instance = CexExchange()
+#     ccxt_client_mock = AsyncMock()
+#     ccxt_client_mock.uid = '12345'
+#     ccxt_client_mock.fetchTicker.side_effect = lambda symbol: {'last': 5000}
+#     ccxt_client_mock.fetchBalance = AsyncMock(return_value={'BTC': {'free': 1}})
+#     ccxt_client_mock.create_order = AsyncMock(
+#         return_value={
+#             'id': '12345',
+#             'amount': 1,
+#             'price': 5000,
+#             'datetime': '2022-01-01 00:00:00'})
+#     exchange_instance.cex = ccxt_client_mock
+
+#     with patch('ccxt.binance', return_value=ccxt_client_mock):
+
+#         result = await exchange_instance.get_info()
+
+#         assert result is not None
+#         assert '💱' in result
+#         assert '🪪' in result
+
+#         order_params = {
+#             'action': 'BUY',
+#             'instrument': 'BTCUSDT',
+#             'quantity': 100
+#         }
+
+#         result = await ccxt_client_mock.execute_order(order_params)
+#         print(result)
+#         assert result is not None
+#         assert '⬆️ BTC/USD' in result
+
+#         ccxt_client_mock.fetchTicker.assert_awaited_once_with('BTCUSDT')
+#         ccxt_client_mock.fetchBalance.assert_awaited_once()
+#         ccxt_client_mock.create_order.assert_awaited_once_with(
+#             'BTCUSDT', settings.cex_ordertype, 'BUY', 0.02, price=None)

tests/test_example_plugin.py

@@ -2,6 +2,7 @@
 from unittest.mock import AsyncMock
 
 import pytest
+from asyncz.triggers import CronTrigger
 
 from tt.config import settings
 from tt.plugins.default_plugins.example_plugin import ExamplePlugin
@@ -30,11 +31,23 @@
     assert plugin_manager is not None
 
 
+@pytest.mark.asyncio
+async def test_baseplugin_class():
+    plugin = BasePlugin()
+    assert callable(plugin.start) 
+    assert callable(plugin.stop)
+    assert callable(plugin.send_notification)
+    assert callable(plugin.should_handle)
+    assert callable(plugin.handle_message)
+
+
 @pytest.mark.asyncio
 async def test_baseplugin():
     plugin = BasePlugin()
     await plugin.start()
     assert plugin is not None
+    await plugin.stop()
+    assert plugin is not None
 
 
 @pytest.mark.asyncio
@@ -48,6 +61,7 @@
     assert len(plugin_manager.plugins) >= 1
     assert isinstance(plugin_manager.plugins[0], ExamplePlugin)
 
+
 @pytest.mark.asyncio
 async def test_load_plugins(caplog):
     settings.talkytrend_enabled = False
@@ -60,6 +74,7 @@
     await plugin_manager.start_all_plugins()
     assert 'Loading plugins from' in caplog.text
 
+
 @pytest.mark.asyncio
 async def test_start_plugin(caplog):
     plugin_module = importlib.import_module(
@@ -74,6 +89,7 @@
     assert 'plugin started' in caplog.text
     assert 'plugin enabled' in caplog.text
 
+
 @pytest.mark.asyncio
 async def test_plugin(plugin, plugin_manager):
     handle_message = AsyncMock()
@@ -82,9 +98,56 @@
     assert plugin.should_handle("any message") is True
     handle_message.assert_awaited_once
 
+
 @pytest.mark.asyncio
 async def test_plugin_notification(plugin, plugin_manager):
     """Test notification """
     send_notification = AsyncMock()
     await plugin.handle_message(f"{settings.bot_prefix}{settings.bot_command_help}")
     send_notification.assert_awaited_once
+
+
+@pytest.mark.asyncio
+async def test_plugin_notify_cron_task():
+    plugin = BasePlugin()
+    plugin.scheduler = AsyncMock()
+    plugin.send_notification = AsyncMock()
+    function_result = 'Test Result'
+    function_mock = AsyncMock(return_value=function_result)
+
+    await plugin.plugin_notify_cron_task(
+        user_name='Test User',
+        user_day_of_week='mon-fri',
+        user_hours='6,12,18',
+        user_timezone='UTC',
+        function=function_mock
+    )
+
+    plugin.scheduler.add_task.assert_called_once_with(
+        name='Test User',
+        fn=plugin.send_notification,
+        args=[function_result],
+        trigger=CronTrigger(
+            day_of_week='mon-fri',
+            hour='6,12,18',
+            timezone='UTC'
+        ),
+        is_enabled=True
+    )
+
+
+@pytest.mark.asyncio
+async def test_plugin_notify_schedule_task():
+    plugin = BasePlugin()
+    plugin.scheduler = AsyncMock()
+    plugin.send_notification = AsyncMock()
+    function_result = 'Test Result'
+    function_mock = AsyncMock(return_value=function_result)
+    await plugin.plugin_notify_schedule_task(
+        user_name='Test User',
+        frequency=8,
+        function=function_mock
+    )
+
+    plugin.scheduler.add_task.assert_awaited_once
+

tests/test_unit.py

@@ -2,15 +2,17 @@
  TT test
 """
 import asyncio
-from unittest.mock import AsyncMock
+from unittest.mock import AsyncMock, patch
 
 import pytest
+import uvicorn
 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.bot import app, start_bot_task
 from tt.config import settings
-from tt.plugins.plugin_manager import BasePlugin, PluginManager
+from tt.plugins.plugin_manager import PluginManager
 from tt.utils import run_bot, send_notification, start_bot, start_plugins
 
 
@@ -27,6 +29,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 +43,23 @@
 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_start_bot_task():
+    run_bot = AsyncMock()
+    await start_bot_task()
+    assert run_bot.assert_awaited_once
+
 
 def test_app_endpoint_main():
     client = TestClient(app)
@@ -80,53 +105,32 @@
 
 
 @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
-async def test_baseplugins():
-    plugin = BasePlugin
-    assert callable(plugin.start) 
-    assert callable(plugin.stop)
-    assert callable(plugin.send_notification) 
-    assert callable(plugin.should_handle)
-    assert callable(plugin.handle_message)
+def test_main():
+    client = TestClient(app)
+    uvicorn.run = AsyncMock(client)
+    uvicorn.run.assert_called_once

tt/ __init__.py

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

tt/bot.py

@@ -17,7 +17,7 @@
 
 @app.on_event("startup")
 async def start_bot_task():
-    """⛓️🤖🙊BOT"""
+    """⛓️🤖BOT"""
     event_loop = asyncio.get_event_loop()
     event_loop.create_task(run_bot())
 
@@ -41,6 +41,6 @@ async def webhook(request: Request):
     return {"status": "OK"}
 
 
+
 if __name__ == "__main__":
     uvicorn.run(app, host=settings.host, port=int(settings.port))
-