added helper to ensure browser is installed before running integratio…

…n tests that need it
owasp-sbot · Oct 4, 2024 · 9cab4a8 · 9cab4a8
1 parent bba2e72
commit 9cab4a8
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 3 deletions.
diff --git a/...routes/test__i__Playwright__Serverless.py → ...pi/routes/test__i__Playwright__Browser.py b/...routes/test__i__Playwright__Serverless.py → ...pi/routes/test__i__Playwright__Browser.py
diff --git a/...t/fast_api/routes/test_Routes__Browser.py → ...st_api/routes/test__u__Routes__Browser.py b/...t/fast_api/routes/test_Routes__Browser.py → ...st_api/routes/test__u__Routes__Browser.py
@@ -1,12 +1,14 @@
 from unittest import TestCase
 
+from integration.fast_api_objs_for_tests import ensure_browser_is_installed
 from osbot_serverless_flows.fast_api.routes.Routes__Browser import Routes__Browser
 
 
-class test_Routes_Browser(TestCase):
+class test__u__Routes_Browser(TestCase):
 
     @classmethod
     def setUpClass(cls):
+        ensure_browser_is_installed()
         cls.routes_browser = Routes__Browser()
 
     def test_url_html(self):

diff --git a/tests/integration/fast_api_objs_for_tests.py b/tests/integration/fast_api_objs_for_tests.py
@@ -7,3 +7,9 @@
     client__serverless_flows = fast_api__serverless_flows.client()
 
 assert duration.seconds < 1         # make sure the setup time is less than 1 second
+
+
+def ensure_browser_is_installed():
+    from osbot_serverless_flows.playwright.Playwright__Serverless import Playwright__Serverless
+    playwright_browser = Playwright__Serverless()
+    playwright_browser.browser__install()
diff --git a/...d/test_Flow__Playwright__Get_Page_Html.py → ...st__i__Flow__Playwright__Get_Page_Html.py b/...d/test_Flow__Playwright__Get_Page_Html.py → ...st__i__Flow__Playwright__Get_Page_Html.py
@@ -1,13 +1,17 @@
 from unittest                                                                   import TestCase
+
+from integration.fast_api_objs_for_tests import ensure_browser_is_installed
 from osbot_serverless_flows.flows.browser_based.Flow__Playwright__Get_Page_Html import Flow__Playwright__Get_Page_Html
 
 
-class test_Flow__Playwright__Get_Page_Html(TestCase):
+class test__i__Flow__Playwright__Get_Page_Html(TestCase):
 
     @classmethod
     def setUpClass(cls) -> None:
+        ensure_browser_is_installed()
         cls.flow__get_page_html = Flow__Playwright__Get_Page_Html()
 
+
     def test_run(self):
         flow_data = self.flow__get_page_html.run()
         page_content = flow_data.get('page_content')

diff --git a/..._Flow__Playwright__Get_Page_Screenshot.py → ..._Flow__Playwright__Get_Page_Screenshot.py b/..._Flow__Playwright__Get_Page_Screenshot.py → ..._Flow__Playwright__Get_Page_Screenshot.py
@@ -1,14 +1,16 @@
 from unittest                                                                         import TestCase
 
+from integration.fast_api_objs_for_tests import ensure_browser_is_installed
 from osbot_utils.utils.Misc import base64_to_bytes
 
 from osbot_serverless_flows.flows.browser_based.Flow__Playwright__Get_Page_Screenshot import Flow__Playwright__Get_Page_Screenshot
 
 
-class test_Flow__Playwright__Get_Page_Screenshot(TestCase):
+class test__i__Flow__Playwright__Get_Page_Screenshot(TestCase):
 
     @classmethod
     def setUpClass(cls) -> None:
+        ensure_browser_is_installed()
         cls.flow__get_page_screenshot = Flow__Playwright__Get_Page_Screenshot()
 
     def test_run(self):