Kea2-python 0.0.1b0__py3-none-any.whl → 0.0.1b1__py3-none-any.whl

kea2/__init__.py

@@ -1,4 +1,4 @@
-from .keaUtils import KeaTestRunner, precondition, prob, Options
+from .keaUtils import KeaTestRunner, precondition, prob, max_tries, Options
 
 import logging
 logging.basicConfig(level=logging.DEBUG)

kea2/absDriver.py

@@ -50,3 +50,7 @@ class AbstractDriver(abc.ABC):
     @abc.abstractmethod
     def getStaticChecker(self, hierarchy) -> AbstractStaticChecker:
         pass
+
+    @classmethod
+    @abc.abstractmethod
+    def tearDown(self): ...

kea2/adbUtils.py

@@ -1,5 +1,9 @@
 import subprocess
 from typing import List, Optional
+from .utils import getLogger
+
+logger = getLogger(__name__)
+
 
 def run_adb_command(cmd: List[str], timeout=10):
     """
@@ -13,6 +17,7 @@ def run_adb_command(cmd: List[str], timeout=10):
         str: The standard output from the command. If an error occurs, returns None.
     """
     full_cmd = ["adb"] + cmd
+    logger.debug(f"{' '.join(full_cmd)}")
     try:
         result = subprocess.run(full_cmd, capture_output=True, text=True, timeout=timeout)
         if result.returncode != 0:

kea2/assets/quicktest.py

@@ -57,7 +57,7 @@ PACKAGE_NAME = "it.feio.android.omninotes.alpha"
 FILE_NAME = "omninotes.apk"
 
 
-def check_installation():
+def check_installation(serial=None):
     import os
     from pathlib import Path
     if not os.path.exists(Path(".") / FILE_NAME):
@@ -65,7 +65,7 @@ def check_installation():
         import urllib.request
         urllib.request.urlretrieve(URL, FILE_NAME)
 
-    d = u2.connect()
+    d = u2.connect(serial)
     # automatically install omni-notes
     if PACKAGE_NAME not in d.app_list():
         print("[INFO] Installing omninotes.", flush=True)
@@ -74,7 +74,7 @@ def check_installation():
 
 
 if __name__ == "__main__":
-    check_installation()
+    check_installation(serial=None)
     KeaTestRunner.setOptions(
         Options(
             driverName="d",

kea2/cli.py

@@ -3,7 +3,7 @@
 
 from __future__ import absolute_import, print_function
 import sys
-from kea2.utils import getProjectRoot, getLogger
+from .utils import getProjectRoot, getLogger
 from .kea_launcher import run
 import argparse
 

kea2/keaUtils.py

@@ -19,9 +19,10 @@ from .utils import TimeStamp, getProjectRoot, getLogger
 from .u2Driver import StaticU2UiObject
 import uiautomator2 as u2
 import types
+
 PRECONDITIONS_MARKER = "preconds"
 PROP_MARKER = "prop"
-
+MAX_TRIES_MARKER = "max_tries"
 
 logger = getLogger(__name__)
 
@@ -30,14 +31,14 @@ logger = getLogger(__name__)
 PropName = NewType("PropName", str)
 PropertyStore = NewType("PropertyStore", Dict[PropName, TestCase])
 
-TIME_STAMP = TimeStamp().getTimeStamp()
-LOGFILE = f"fastbot_{TIME_STAMP}.log"
-RESFILE = f"result_{TIME_STAMP}.json"
+STAMP = TimeStamp().getTimeStamp()
+LOGFILE = f"fastbot_{STAMP}.log"
+RESFILE = f"result_{STAMP}.json"
 
 def precondition(precond: Callable[[Any], bool]) -> Callable:
     """the decorator @precondition
 
-    The precondition specifies when the property could be executed.
+    @precondition specifies when the property could be executed.
     A property could have multiple preconditions, each of which is specified by @precondition.
     """
     def accept(f):
@@ -56,7 +57,7 @@ def precondition(precond: Callable[[Any], bool]) -> Callable:
 def prob(p: float):
     """the decorator @prob
 
-    The prob specify the propbability of execution when a property is satisfied.
+    @prob specify the propbability of execution when a property is satisfied.
     """
     p = float(p)
     if not 0 < p <= 1.0:
@@ -73,6 +74,26 @@ def prob(p: float):
     return accept
 
 
+def max_tries(n: int):
+    """the decorator @max_tries
+
+    @max_tries specify the maximum tries of executing a property.
+    """
+    n = int(n)
+    if not n > 0:
+        raise ValueError("The maxium tries should be a positive integer.")
+    def accept(f):
+        @wraps(f)
+        def precondition_wrapper(*args, **kwargs):
+            return f(*args, **kwargs)
+
+        setattr(precondition_wrapper, MAX_TRIES_MARKER, n)
+
+        return precondition_wrapper
+
+    return accept
+
+
 @dataclass
 class Options:
     """
@@ -96,10 +117,16 @@ class Options:
     throttle: int = 200
     # the output_dir for saving logs and results
     output_dir: str = "output"
+    # the stamp for log file and result file, default: current time stamp
+    log_stamp: str = None
 
     def __post_init__(self):
         if self.serial and self.Driver:
             self.Driver.setDeviceSerial(self.serial)
+        if self.log_stamp:
+            global LOGFILE, RESFILE
+            LOGFILE = f"fastbot_{self.log_stamp}.log"
+            RESFILE = f"result_{self.log_stamp}.json"
 
 
 @dataclass
@@ -151,6 +178,9 @@ class JsonResult(TextTestResult):
         super().addError(test, err)
         self.res[getFullPropName(test)].error += 1
 
+    def getExcuted(self, test: TestCase):
+        return self.res[getFullPropName(test)].executed
+
 
 def activateFastbot(options: Options, port=None) -> threading.Thread:
     """
@@ -274,6 +304,8 @@ class KeaTestRunner(TextTestRunner):
         global LOGFILE, RESFILE
         LOGFILE = output_dir / Path(LOGFILE)
         RESFILE = output_dir / Path(RESFILE)
+        logger.debug(f"Log file: {LOGFILE}")
+        logger.debug(f"Result file: {RESFILE}")
 
     def run(self, test):
 
@@ -333,7 +365,7 @@ class KeaTestRunner(TextTestRunner):
                     , flush=True)
 
                     try:
-                        propsSatisfiedPrecond = self.getValidProperties()
+                        propsSatisfiedPrecond = self.getValidProperties(result)
                     except requests.ConnectionError:
                         print(
                             "[INFO] Exploration times up (--running-minutes)."
@@ -425,6 +457,8 @@ class KeaTestRunner(TextTestRunner):
         """
         block_widgets: List[str] = self._getBlockedWidgets()
         URL = f"http://localhost:{self.scriptDriver.lport}/stepMonkey"
+        logger.debug(f"Sending request: {URL}")
+        logger.debug(f"Blocking widgets: {block_widgets}")
         r = requests.post(
             url=URL,
             json={
@@ -440,12 +474,14 @@ class KeaTestRunner(TextTestRunner):
         """
         send a stop monkey request to the server and get the xml string.
         """
-        r = requests.get(f"http://localhost:{self.scriptDriver.lport}/stopMonkey")
+        URL = f"http://localhost:{self.scriptDriver.lport}/stopMonkey"
+        logger.debug(f"Sending request: {URL}")
+        r = requests.get(URL)
 
         res = r.content.decode(encoding="utf-8")
         print(f"[Server INFO] {res}", flush=True)
 
-    def getValidProperties(self) -> PropertyStore:
+    def getValidProperties(self, result: JsonResult) -> PropertyStore:
 
         xml_raw = self.stepMonkey()
         staticCheckerDriver = self.options.Driver.getStaticChecker(hierarchy=xml_raw)
@@ -470,6 +506,10 @@ class KeaTestRunner(TextTestRunner):
                     break
             # if all the precond passed. make it the candidate prop.
             if valid:
+                logger.debug(f"precond satisfied: {getFullPropName(test)}")
+                if result.getExcuted(test) >= getattr(prop, MAX_TRIES_MARKER, float("inf")):
+                    logger.debug(f"{getFullPropName(test)} has reached its max_tries. Skip.")
+                    continue
                 validProps[propName] = test
         return validProps
 
@@ -570,6 +610,8 @@ class KeaTestRunner(TextTestRunner):
 
         return blocked_widgets
 
-    def tearDown(self):
-        # TODO Add tearDown method (remove local port, etc.)
-        pass
+    def __del__(self):
+        """tearDown method. Cleanup the env.
+        """
+        if self.options.Driver:
+            self.options.Driver.tearDown()

kea2/kea_launcher.py

@@ -76,6 +76,14 @@ def _set_runner_parser(subparsers: "argparse._SubParsersAction[argparse.Argument
         help="The name of driver in script.",
     )
 
+    parser.add_argument(
+        "--log-stamp",
+        dest="log_stamp",
+        type=str,
+        required=False,
+        help="the stamp for log file and result file, default: current time stamp",
+    )
+
     parser.add_argument(
         "extra",
         nargs=argparse.REMAINDER,
@@ -138,7 +146,8 @@ def run(args=None):
         serial=args.serial,
         running_mins=args.running_minutes if args.running_minutes else 10,
         maxStep=args.max_step if args.max_step else 500,
-        throttle=args.throttle_ms if args.throttle_ms else 200
+        throttle=args.throttle_ms if args.throttle_ms else 200,
+        log_stamp=args.log_stamp
     )
 
     KeaTestRunner.setOptions(options)

kea2/u2Driver.py

@@ -8,7 +8,7 @@ from typing import Dict, List, Union
 from lxml import etree
 from .absDriver import AbstractScriptDriver, AbstractStaticChecker, AbstractDriver
 from .adbUtils import list_forwards, remove_forward, create_forward
-from .utils import TimeStamp
+from .utils import TimeStamp, getLogger
 
 TIME_STAMP = TimeStamp().getTimeStamp()
 
@@ -16,6 +16,8 @@ import logging
 logging.getLogger("urllib3").setLevel(logging.INFO)
 logging.getLogger("uiautomator2").setLevel(logging.INFO)
 
+logger = getLogger(__name__)
+
 """
 The definition of U2ScriptDriver
 """
@@ -57,16 +59,13 @@ class U2ScriptDriver(AbstractScriptDriver):
                 setattr(self.d._dev, "msg", "meta")
                 print(f"[U2] local port: {lport}", flush=True)
                 return lport
-            
+
             self._remove_remote_port(8090)
             self.d.lport = get_u2_forward_port()
             self._remove_remote_port(9008)
 
         return self.d
 
-    def tearDown(self):
-        self.d.stop_uiautomator()
-
     def _remove_remote_port(self, port:int):
         """remove the forward port
         """
@@ -76,6 +75,11 @@ class U2ScriptDriver(AbstractScriptDriver):
                 forward_local = forward["local"]
                 remove_forward(local_spec=forward_local, device=self.deviceSerial)
 
+    def tearDown(self):
+        logger.debug("U2Driver tearDown: stop_uiautomator")
+        self.d.stop_uiautomator()
+        logger.debug("U2Driver tearDown: remove forward")
+        self._remove_remote_port(8090)
 
 """
 The definition of U2StaticChecker
@@ -231,6 +235,7 @@ class U2StaticDevice(u2.Device):
     
     def __getattr__(self, attr):
         """Proxy other methods to script_driver"""
+        # logger.debug(f"{attr} not exists in static checker, proxy to script_driver.")
         return getattr(self._script_driver, attr)
 
 class _XPathEntry(u2.xpath.XPathEntry):
@@ -293,6 +298,10 @@ class U2Driver(AbstractDriver):
             self.staticChecker = U2StaticChecker()
         return self.staticChecker.getInstance(hierarchy)
 
+    @classmethod
+    def tearDown(self):
+        self.scriptDriver.tearDown()
+
 
 """
 Other Utils
@@ -311,6 +320,7 @@ def forward_port(self, remote: Union[int, str]) -> int:
                 return int(f.local[len("tcp:") :])
         local_port = get_free_port()
         self.forward("tcp:" + str(local_port), remote)
+        logger.debug(f"forwading port: tcp:{local_port} -> {remote}")
         return local_port
 
 

{kea2_python-0.0.1b0.dist-info → kea2_python-0.0.1b1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Kea2-python
-Version: 0.0.1b0
+Version: 0.0.1b1
 Summary: A python library for supporting and customizing automated UI testing for mobile apps
 Author-email: Xixian Liang <xixian@stu.ecnu.edu.cn>
 Requires-Python: >=3.8
@@ -16,6 +16,10 @@ Dynamic: license-file
 [![PyPI Downloads](https://static.pepy.tech/badge/kea2-python)](https://pepy.tech/projects/kea2-python)
 ![Python](https://img.shields.io/badge/python-3.8%2B-blue)
 
+<div>
+    <img src="https://github.com/user-attachments/assets/58f68b00-cc9c-4620-9e2e-66c43cf7caae" style="border-radius: 14px; width: 20%; height: 20%;"/> 
+</div>
+
 Kea2 is an easy-to-use Python library for supporting, customizing and improving automated UI testing for mobile apps. The library is currently built on top of [Fastbot](https://github.com/bytedance/Fastbot_Android) and [uiautomator2](https://github.com/openatx/uiautomator2), and targeting [Android](https://en.wikipedia.org/wiki/Android_(operating_system)) apps.
 
 ### Kea2 has three important features:

{kea2_python-0.0.1b0.dist-info → kea2_python-0.0.1b1.dist-info}/RECORD

@@ -1,16 +1,16 @@
-kea2/__init__.py,sha256=_IFDiyzb6OLyhjldBXrAol6OpOijTnJ2LYWMK4TqSno,121
-kea2/absDriver.py,sha256=M08ba0kpIer20ApMhX7yCmcLrfPb6a3udZxR_4FhriI,1224
-kea2/adbUtils.py,sha256=IV-5G_3mc89YP1NJZJIWUrgxb13QECGGXTTANkGH_Rk,8986
-kea2/cli.py,sha256=eeXWf5aFCEYDKqL1o4qMcxhcCp6yG8twsjaXv0ZaTy0,2756
-kea2/keaUtils.py,sha256=wu_FHlLAO8cj69XpP02sRb0xaJqGuW79uH9onNXD0Ag,20305
-kea2/kea_launcher.py,sha256=RzAoX9T0zoJzYIE-PeIcg7eNWPzwUSsBvA17nObTnpk,4133
+kea2/__init__.py,sha256=QW4OVa1NHcX-DxC7Nr-ozo1G1wxfARnRh63bB08a1Yw,132
+kea2/absDriver.py,sha256=X99d-OP77EIiDwTLvWVkgDUtqscoxmWeOb-EcaOPWGI,1294
+kea2/adbUtils.py,sha256=JhsQiJ_GZG3iPcFHsBDwkSfQJ9R5dMRKr5fVOgCAe_E,9088
+kea2/cli.py,sha256=E309FofSYGls6OP8BNE-d3Ft_hGQpm61gRKz9TgS-Bk,2752
+kea2/keaUtils.py,sha256=8apsnAlmVnNaMmFFZjTcPqlU-RpztFYvPPY2Gay-jhA,21808
+kea2/kea_launcher.py,sha256=90zppG2XkaU7YyGnqbb0FCfG5dDV6zCx08rMIsGLZbs,4374
 kea2/logWatcher.py,sha256=hd8banPiCa6aCQ6d_MznWKOdzK_A2X_dPbrx-usjxgE,1927
-kea2/u2Driver.py,sha256=8j21Rcb9ZuqBIU68Hcx1S_gl6zJ_ZngbIUMxXBBQpEU,10643
+kea2/u2Driver.py,sha256=IsQ0IvVuyUY16ZAVqeerzqr_3Cu5laeeGvoeCA1_ztU,11065
 kea2/utils.py,sha256=zjuoVwts2qVX9GnTaPoiqarfcSvyfW6cAD3ESf_dmrQ,1284
 kea2/assets/fastbot-thirdpart.jar,sha256=0SZ_OoZFWDGMnazgXKceHgKvXdUDoIa3Gb2bcifaikk,85664
 kea2/assets/framework.jar,sha256=rTluOJJKj2DFwh7ascXso1udYdWv00BxBwSQ3Vmv-fw,1149240
 kea2/assets/monkeyq.jar,sha256=TjszvflvE59ev9RhVhkfS_0L46VYZbL2wvpKRyA8ZMk,460366
-kea2/assets/quicktest.py,sha256=QP7P5l3LPmDDSyE2qZ1ZTnH-cFLVP8AYDqDjOBFdRfM,2859
+kea2/assets/quicktest.py,sha256=j-fIryvspXObaNm3kWzgM0LMFes4h7EXT5C9xUWGUVo,2887
 kea2/assets/u2.jar,sha256=G9fFf-AZ0CdaZrk27V1-pfJE2Y5eO6PRzPShTnIe-U8,3746525
 kea2/assets/fastbot_configs/ADBKeyBoard.apk,sha256=L3Kva4gCQALo10EzinCvHrIbffGE3p94qSNbnS8CxP0,163634
 kea2/assets/fastbot_configs/abl.strings,sha256=Rn8_YEbVGOJqndIY_-kWnR5NaoFI-cuB-ij10Ddhl90,75
@@ -25,9 +25,9 @@ kea2/assets/fastbot_libs/arm64-v8a/libfastbot_native.so,sha256=dA2Tf74-gDrCFdeCl
 kea2/assets/fastbot_libs/armeabi-v7a/libfastbot_native.so,sha256=GWcL8M8WfQAd9CfOM6pRYbOnxeycN-LiL7Mf59YIadE,1334420
 kea2/assets/fastbot_libs/x86/libfastbot_native.so,sha256=k-aw1gEXRWMKZRNHIggKNuZy0wC1y2BnveJGEIO6rbo,2036856
 kea2/assets/fastbot_libs/x86_64/libfastbot_native.so,sha256=tiofhlf4uMQcU5WAvrdLgTBME0lb83hVUoGtTwxmE8A,2121416
-kea2_python-0.0.1b0.dist-info/licenses/LICENSE,sha256=nM9PPjcsXVo5SzNsjRqWgA-gdJlwqZZcRDSC6Qf6bVE,2034
-kea2_python-0.0.1b0.dist-info/METADATA,sha256=NTKDhgUHAHCtc-jt8YRBpagrG8eZJncQFh9W-GWhV10,22456
-kea2_python-0.0.1b0.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-kea2_python-0.0.1b0.dist-info/entry_points.txt,sha256=mFX06TyxXiUAJQ6JZn8QHzfn8n5R8_KJ5W-pTm_fRCA,39
-kea2_python-0.0.1b0.dist-info/top_level.txt,sha256=TsgNH4PQoNOVhegpO7AcjutMVWp6Z4KDL1pBH9FnMmk,5
-kea2_python-0.0.1b0.dist-info/RECORD,,
+kea2_python-0.0.1b1.dist-info/licenses/LICENSE,sha256=nM9PPjcsXVo5SzNsjRqWgA-gdJlwqZZcRDSC6Qf6bVE,2034
+kea2_python-0.0.1b1.dist-info/METADATA,sha256=e307QrCqtPZ8y4Z8ucayAiydj___hwr3bxkhaQCcwow,22622
+kea2_python-0.0.1b1.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+kea2_python-0.0.1b1.dist-info/entry_points.txt,sha256=mFX06TyxXiUAJQ6JZn8QHzfn8n5R8_KJ5W-pTm_fRCA,39
+kea2_python-0.0.1b1.dist-info/top_level.txt,sha256=TsgNH4PQoNOVhegpO7AcjutMVWp6Z4KDL1pBH9FnMmk,5
+kea2_python-0.0.1b1.dist-info/RECORD,,