Merge pull request #72 from kayqueGovetri/ENH/insert-context-browser

ENH: Implement __enter__ and __exit__ to open and stop browser

Author: hhslepicka

botcity/web/bot.py

@@ -77,6 +77,12 @@ def __init__(self, headless=False):
 
         self._download_folder_path = os.getcwd()
 
+    def __enter__(self):
+        pass
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        self.stop_browser()
+
     @property
     def driver(self):
         """

tests/test_browser.py

@@ -6,6 +6,13 @@
 from botcity.web import WebBot, By
 
 
+def test_context(web: WebBot):
+    with web:
+        web.browse(conftest.INDEX_PAGE)
+        assert web.driver
+    assert web.driver is None
+
+
 def test_create_tab(web: WebBot):
     web.browse(conftest.INDEX_PAGE)
     title = web.page_title()