Merge pull request #39 from LambdaTest/stage

Stage

Author: sushobhit-lt

selenium/lambdatest_selenium_driver/smartui.py

@@ -1,5 +1,6 @@
 import json
-from lambdatest_sdk_utils import is_smartui_enabled,fetch_dom_serializer,post_snapshot
+import uuid
+from lambdatest_sdk_utils import is_smartui_enabled,fetch_dom_serializer,post_snapshot,get_snapshot_status
 from lambdatest_sdk_utils import setup_logger,get_logger
 
 
@@ -34,13 +35,30 @@ def smartui_snapshot(driver, name,options={}):
 
         # Post the dom to smartui endpoint
         dom['name'] = name
-        res = post_snapshot(dom,'lambdatest-selenium-driver',options=options)
+        if 'sync' in options and options['sync'] is True:
+            options['contextId'] = str(uuid.uuid4())
+            res = post_snapshot(dom,'lambdatest-selenium-driver',options=options)
 
-        if res and res.get('data') and res['data'].get('warnings') and len(res['data']['warnings']) != 0:
-            for warning in res['data']['warnings']:
-                logger.warn(warning)
+            if res and res.get('data') and res['data'].get('warnings') and len(res['data']['warnings']) != 0:
+                for warning in res['data']['warnings']:
+                    logger.warn(warning)
 
-        logger.info(f'Snapshot captured {name}')
+            logger.info(f'Snapshot captured {name}')
+            timeout= 600
+            if 'timeout' in options and options['timeout'] is not None:
+                timeout = options['timeout']
+            if timeout > 900 or timeout < 30:
+                timeout=600
+                logger.warn("Timeout value is not valid, allowed range is 30-900 seconds. Defaulting to 600 seconds.")
+            return get_snapshot_status(name,options['contextId'],timeout)
+        else:
+            res = post_snapshot(dom,'lambdatest-selenium-driver',options=options)
+
+            if res and res.get('data') and res['data'].get('warnings') and len(res['data']['warnings']) != 0:
+                for warning in res['data']['warnings']:
+                    logger.warn(warning)
+
+            logger.info(f'Snapshot captured {name}')
     except Exception as e:
         logger.error(f'SmartUI snapshot failed  "${name}"')
         logger.error(e)

selenium/lambdatest_selenium_driver/version.py

@@ -1 +1 @@
-__version__ = '1.0.4'
+__version__ = '1.0.6'

selenium/setup.py

@@ -7,7 +7,7 @@
 
 setup(
     name="lambdatest-selenium-driver",
-    version="1.0.5",
+    version="1.0.6",
     author="LambdaTest <[email protected]>",
     description="Python Selenium SDK for testing with Smart UI",
     long_description=long_description,

utils/lambdatest_sdk_utils/__init__.py

@@ -1,4 +1,4 @@
 from lambdatest_sdk_utils.constants import get_smart_ui_server_address
 from lambdatest_sdk_utils.logger import log_level,setup_logger,get_logger
-from lambdatest_sdk_utils.rest import post_snapshot,is_smartui_enabled,fetch_dom_serializer
+from lambdatest_sdk_utils.rest import post_snapshot,is_smartui_enabled,fetch_dom_serializer,get_snapshot_status
 from lambdatest_sdk_utils.version import __version__

utils/lambdatest_sdk_utils/rest.py

@@ -50,4 +50,12 @@ def post_snapshot(snapshot,pkg,**kwargs):
         logger.debug(f'post snapshot failed : {msg}')
         raise Exception(f'post snapshot failed')
 
-
+def get_snapshot_status(snapshot_name,context_id,timeout):
+    try:
+        response = requests.get(f'{get_smart_ui_server_address()}/snapshot/status?contextId={context_id}&snapshotName={snapshot_name}&pollTimeout={timeout}')
+        response.raise_for_status()
+        return response.json()
+    except Exception as e:
+        logger.debug(f'get snapshot status failed : {e}')
+        raise Exception(f'get snapshot status failed')
+        

utils/lambdatest_sdk_utils/version.py

@@ -1 +1 @@
-__version__ = '1.0.3'
+__version__ = '1.0.4'

utils/setup.py

@@ -1,9 +1,8 @@
 from setuptools import setup, find_packages
-import lambdatest_sdk_utils
 
 setup(
     name="lambdatest-sdk-utils",
-    version="1.0.3",
+    version="1.0.4",
     author="LambdaTest <[email protected]>",
     description="SDK utils",
     long_description="SDK utils for lambdatest",