mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-20 20:35:00 +00:00
34 lines
895 B
Python
34 lines
895 B
Python
import os
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from aider.main import main
|
|
|
|
|
|
class TestBrowser(unittest.TestCase):
|
|
@patch("aider.main.launch_gui")
|
|
def test_browser_flag_imports_streamlit(self, mock_launch_gui):
|
|
os.environ["AIDER_ANALYTICS"] = "false"
|
|
|
|
# Run main with --browser and --yes flags
|
|
main(["--browser", "--yes"])
|
|
|
|
# Check that launch_gui was called
|
|
mock_launch_gui.assert_called_once()
|
|
|
|
# Try to import streamlit
|
|
try:
|
|
import streamlit # noqa: F401
|
|
|
|
streamlit_imported = True
|
|
except ImportError:
|
|
streamlit_imported = False
|
|
|
|
# Assert that streamlit was successfully imported
|
|
self.assertTrue(
|
|
streamlit_imported, "Streamlit should be importable after running with --browser flag"
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|