Fix async in tests

Author: linuxdaemon

cloudbot/util/mapping.py

@@ -1,14 +1,6 @@
 import weakref
 from collections import defaultdict
-from typing import (
-    TYPE_CHECKING,
-    Generic,
-    Optional,
-    Protocol,
-    TypeVar,
-    Union,
-    cast,
-)
+from typing import TYPE_CHECKING, Generic, Optional, TypeVar, Union, cast
 
 __all__ = (
     "KeyFoldDict",
@@ -24,6 +16,7 @@
 
 
 if TYPE_CHECKING:
+    from typing import Protocol
 
     class MapBase(Protocol[K_contra, V]):
         def __getitem__(self, item: K_contra) -> V:

tests/core_tests/test_main.py

@@ -1,13 +1,19 @@
 import logging
 from unittest.mock import patch
 
-from cloudbot.__main__ import main
+import pytest
 
+from cloudbot.__main__ import async_main
+
+
+@pytest.mark.asyncio
+async def test_main():
+    async def run():
+        return False
 
-def test_main():
     with patch("cloudbot.__main__.CloudBot") as mocked:
-        mocked().run.return_value = False
-        main()
+        mocked().run = run
+        await async_main()
         assert logging._srcfile is None
         assert not logging.logThreads
         assert not logging.logProcesses