tests(cli): fixed some test warnings

Author: Zhe Yu

tests/test_common.py

@@ -629,6 +629,7 @@ async def _try_server(url):
         return "127.0.0.1" in url or "localhost" in url
 
     mock_process = AsyncMock()
+    mock_process.terminate = MagicMock()
     with (
         patch("vectorcode.common.start_server", return_value=mock_process),
         patch("vectorcode.common.try_server", side_effect=_try_server),
@@ -639,3 +640,4 @@ async def _try_server(url):
         assert len(manager.get_processes()) == 1
         await manager.kill_servers()
         mock_process.terminate.assert_called_once()
+        mock_process.wait.assert_awaited()

tests/test_main.py

@@ -4,7 +4,7 @@
 
 from vectorcode import __version__
 from vectorcode.cli_utils import CliAction
-from vectorcode.main import async_main, main
+from vectorcode.main import async_main
 
 
 @pytest.mark.asyncio
@@ -317,12 +317,3 @@ async def test_async_main_exception_handling(monkeypatch):
     with patch("vectorcode.main.logger") as mock_logger:
         assert await async_main() == 1
         mock_logger.error.assert_called_once()
-
-
-def test_main(monkeypatch):
-    mock_async_main = AsyncMock(return_value=0)
-    monkeypatch.setattr("vectorcode.main.async_main", mock_async_main)
-    monkeypatch.setattr("asyncio.run", MagicMock(return_value=0))
-
-    result = main()
-    assert result == 0