From 2a0fca79b62ed3f404d352fa67162bf29d769d5d Mon Sep 17 00:00:00 2001
From: Alex Schofield <git@ajschof.me>
Date: Fri, 7 Feb 2025 20:16:32 +0000
Subject: fix tests not working due to incorrect imports

---
 src/main.py            | 16 ++++++++++++++++
 tests/test_discover.py |  6 +++---
 2 files changed, 19 insertions(+), 3 deletions(-)
 create mode 100644 src/main.py

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"},
-- 
cgit v1.2.3