fix tests not working due to incorrect imports

2 files changed, 19 insertions, 3 deletions

diff --git a/src/main.py b/src/main.py
new file mode 100644
index 0000000..d705374
--- /dev/null
+++ b/src/main.py
@@ -0,0 +1,16 @@
+from discover import discover
+import os
+from logger import get_logger
+import asyncio
+
+logger = get_logger("Main")
+
+async def main(debug_flag=None):
+    check_debug = debug_flag if debug_flag is not None else os.getenv("DEBUG", "FALSE").upper() == "TRUE"
+    if check_debug:
+        logger.info("Debug mode is enabled")
+    devices = await discover(debug=check_debug)
+    print(devices)
+
+if __name__ == "__main__":
+    asyncio.run(main())
diff --git a/tests/test_discover.py b/tests/test_discover.py
index 9a2debd..28f62e0 100644
--- a/tests/test_discover.py
+++ b/tests/test_discover.py
@@ -1,6 +1,6 @@
 from unittest import IsolatedAsyncioTestCase
 from unittest.mock import AsyncMock, patch
-from discover import main
+from discover import discover
 from bleak.backends.device import BLEDevice
 
 class TestDiscover(IsolatedAsyncioTestCase):
@@ -10,7 +10,7 @@ class TestDiscover(IsolatedAsyncioTestCase):
 
         mock_discover.return_value = [device_1]
 
-        result = await main(debug_flag=False)
+        result = await discover()
 
         expected = [
             {"name": "TP350S", "address": "AA:BB:CC:DD:EE:FF"},
@@ -25,7 +25,7 @@ class TestDiscover(IsolatedAsyncioTestCase):
 
         mock_discover.return_value = [device_1, device_2]
 
-        result = await main(debug_flag=True)
+        result = await discover()
 
         expected = [
             {"name": "TP350S", "address": "AA:BB:CC:DD:EE:FF"},