Kea2-python 0.0.1a0__py3-none-any.whl

kea2/assets/fastbot_configs/max.schema.strings

@@ -0,0 +1 @@
+snssdk1128://webcast_feed?gd_label=88888

kea2/assets/fastbot_configs/max.strings

@@ -0,0 +1,3 @@
+1   搜索
+2   打开
+3   检查

kea2/assets/fastbot_configs/max.tree.pruning

@@ -0,0 +1,27 @@
+[
+{
+    "activity":"com.ss.android.xxx.NewActivity",
+    "xpath": "//*[@resource-id='com.xxx.go:id/aaa']",
+    "resourceid": "",
+    "contentdesc":"",
+    "text":"",
+    "classname":""
+},
+{
+    "activity":"com.ss.android.xxx.MainActivity",
+    "xpath": "//*[@resource-id='com.xxx.go:id/bbb' and @text='other']",
+    "resourceid": "",
+    "contentdesc":"",
+    "text":"",
+    "classname":""
+},
+{
+    "activity":"com.ss.android.xxx.SplashActivity",
+    "xpath": "//*[@resource-id='com.com.xxx.go:id/ccc' and @text='other']",
+    "resourceid": "",
+    "contentdesc":"",
+    "text":"",
+    "classname":"",
+    "clickable":"false"
+}
+]

kea2/assets/fastbot_configs/widget.block.py

@@ -0,0 +1,22 @@
+from kea2.utils import precondition
+from uiautomator2 import Device
+
+
+@precondition(lambda d:
+    d(text="Setting")
+)
+def sample_block_list(d: "Device"):
+    return d(text="Omni Notes Alpha").exists
+
+
+if __name__ == "__main__":
+    from kea2.utils import BLOCK_WIDGET
+    func = getattr(sample_block_list, BLOCK_WIDGET)
+    import uiautomator2 as u2
+    d = u2.connect()
+    blocked_widgets = func(d)
+    if isinstance(blocked_widgets, u2.UiObject):
+        blocked_widgets = [blocked_widgets]
+    if not all([isinstance(w, u2.UiObject) for w in blocked_widgets]):
+        raise TypeError(f"Invalid widgets block list in {sample_block_list}")
+    

kea2/assets/quickstart.py

@@ -0,0 +1,90 @@
+import unittest
+import uiautomator2 as u2
+
+from time import sleep
+from kea2 import precondition, prob, KeaTestRunner, Options
+from kea2.u2Driver import U2Driver
+
+
+class Omni_Notes_Sample(unittest.TestCase):
+
+    def setUp(self):
+        self.d = u2.connect()
+
+    @prob(0.7)  # The probability of executing the function when precondition is satisfied.
+    @precondition(
+        lambda self: self.d(text="Omni Notes Alpha").exists
+        and self.d(text="Settings").exists
+    )
+    def test_goToPrivacy(self):
+        """
+        The ability to jump out of the UI tarpits
+
+        precond:
+            The drawer was opened
+        action:
+            go to settings -> privacy
+        """
+        print("trying to click Settings")
+        self.d(text="Settings").click()
+        sleep(0.5)
+        print("trying to click Privacy")
+        self.d(text="Privacy").click()
+
+    @precondition(
+        lambda self: self.d(resourceId="it.feio.android.omninotes.alpha:id/search_src_text").exists
+    )
+    def test_rotation(self):
+        """
+        The ability to make assertion to find functional bug
+
+        precond:
+            The search input box is opened
+        action:
+            rotate the device (set it to landscape, then back to natural)
+        assertion:
+            The search input box is still being opened
+        """
+        self.d.set_orientation("l")
+        sleep(2)
+        self.d.set_orientation("n")
+        sleep(2)
+        assert self.d(resourceId="it.feio.android.omninotes.alpha:id/search_src_text").exists()
+
+
+URL = "https://raw.githubusercontent.com/ecnusse/Kea2/refs/heads/dev_test_hidden_algorithm/omninotes.apk"
+PACKAGE_NAME = "it.feio.android.omninotes.alpha"
+FILE_NAME = "omninotes.apk"
+
+
+def check_installation():
+    import os
+    from pathlib import Path
+    if not os.path.exists(Path(".") / FILE_NAME):
+        print(f"[INFO] omninote.apk not exists. Downloading from {URL}", flush=True)
+        import urllib.request
+        urllib.request.urlretrieve(URL, FILE_NAME)
+
+    d = u2.connect()
+    # automatically install omni-notes
+    if PACKAGE_NAME not in d.app_list():
+        print("[INFO] Installing omninotes.", flush=True)
+        d.app_install(FILE_NAME)
+    d.stop_uiautomator()
+
+
+if __name__ == "__main__":
+    check_installation()
+    KeaTestRunner.setOptions(
+        Options(
+            driverName="d",
+            Driver=U2Driver,
+            packageNames=[PACKAGE_NAME],
+            # serial="emulator-5554",   # specify the serial
+            maxStep=5000,
+            # running_mins=10,  # specify the maximal running time in minutes, default value is 10m
+            # throttle=200,   # specify the throttle in milliseconds, default value is 200ms
+            # agent='native'  # 'native' for running the vanilla Fastbot, 'u2' for running Kea2
+        )
+    )
+    unittest.main(testRunner=KeaTestRunner)

kea2/cli.py

@@ -0,0 +1,176 @@
+# coding: utf-8
+# cli.py
+
+from __future__ import absolute_import, print_function
+from kea2.utils import getProjectRoot, getLogger
+from .kea_launcher import run
+import argparse
+
+import os
+from pathlib import Path
+
+
+logger = getLogger(__name__)
+
+def cmd_init(args):
+    cwd = Path(os.getcwd())
+    configs_dir = cwd / "configs"
+    if os.path.isdir(configs_dir):
+        logger.warning("Kea2 project already initialized")
+        return
+
+    import shutil
+    def copy_configs():
+        src = Path(__file__).parent / "assets" / "fastbot_configs"
+        dst = configs_dir
+        shutil.copytree(src, dst)
+
+    def copy_samples():
+        src = Path(__file__).parent / "assets" / "quickstart.py"
+        dst = cwd / "quickstart.py"
+        shutil.copyfile(src, dst)
+
+    copy_configs()
+    copy_samples()
+    logger.info("Kea2 project initialized.")
+
+
+def cmd_load_configs(args):
+    pass
+
+def cmd_run(args):
+    base_dir = getProjectRoot()
+    if base_dir is None:
+        logger.error("kea2 project not initialized. Use `kea2 init`.")
+        return
+    argv = [__file__]
+    argv.extend(args.args)
+    run(argv)
+
+
+def cmd_install(args):
+    pass
+
+
+def cmd_uninstall(args):
+    pass
+
+
+def cmd_start(args):
+    pass
+
+
+def cmd_stop(args):
+    pass
+
+
+def cmd_current(args):
+    pass
+
+
+def cmd_doctor(args):
+    pass
+
+
+# def cmd_version(args):
+#     print(__version__)
+
+
+_commands = [
+    # dict(action=cmd_version, command="version", help="show version"),
+    dict(
+        action=cmd_init,
+        command="init",
+        help="init the Kea2 project in current directory",
+    ),
+    dict(
+        action=cmd_run,
+        command="run",
+        help="run kea2",
+        flags=[
+            dict(args=["args"], nargs=argparse.REMAINDER, help="args for kea2 run"),
+        ],
+    ),
+    # dict(
+    #     action=cmd_install,
+    #     command="",
+    #     help="install packages",
+    #     flags=[
+    #         dict(args=["url"], help="package url"),
+    #     ],
+    # ),
+    # dict(
+    #     action=cmd_uninstall,
+    #     command="uninstall",
+    #     help="uninstall packages",
+    #     flags=[
+    #         dict(args=["--all"], action="store_true", help="uninstall all packages"),
+    #         dict(args=["package_name"], nargs="*", help="package name"),
+    #     ],
+    # ),
+    # dict(
+    #     action=cmd_start,
+    #     command="start",
+    #     help="start application",
+    #     flags=[dict(args=["package_name"], type=str, nargs=None, help="package name")],
+    # ),
+    # dict(
+    #     action=cmd_stop,
+    #     command="stop",
+    #     help="stop application",
+    #     flags=[
+    #         dict(args=["--all"], action="store_true", help="stop all"),
+    #         dict(args=["package_name"], nargs="*", help="package name"),
+    #     ],
+    # ),
+    # dict(action=cmd_current, command="current", help="show current application"),
+    # dict(action=cmd_doctor, command="doctor", help="detect connect problem"),
+]
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter
+    )
+    parser.add_argument("-d", "--debug", action="store_true",
+                        help="show log")
+    parser.add_argument('-s', '--serial', type=str,
+                        help='device serial number')
+
+    subparser = parser.add_subparsers(dest='subparser')
+
+    actions = {}
+    for c in _commands:
+        cmd_name = c['command']
+        actions[cmd_name] = c['action']
+        sp = subparser.add_parser(
+            cmd_name,
+            help=c.get('help'),
+            formatter_class=argparse.ArgumentDefaultsHelpFormatter
+        )
+        for f in c.get('flags', []):
+            args = f.get('args')
+            if not args:
+                args = ['-'*min(2, len(n)) + n for n in f['name']]
+            kwargs = f.copy()
+            kwargs.pop('name', None)
+            kwargs.pop('args', None)
+            sp.add_argument(*args, **kwargs)
+
+    args = parser.parse_args()
+    
+
+    if args.debug:
+        import logging
+        logging.basicConfig(level=logging.DEBUG)
+        logger.debug("args: %s", args)
+
+    if args.subparser:
+        actions[args.subparser](args)
+        return
+
+    parser.print_help()
+
+
+if __name__ == "__main__":
+    main()