MyProject and 自动化测试框架 appiumcore

1. 如果需要设置 无头模式，可能要改框架的内容 Driver.py

from appium import webdriver as AppiumDriver
from selenium import webdriver as SeleniumDriver
from Element.Find import Find
from Element.FlutterFind import FlutterFind
from Utilitys.SeleniumUtils import SeleniumUtils
from Utilitys.Utils import Utils
from selenium.webdriver.chrome.service import Service as ChromeService
from webdriver_manager.chrome import ChromeDriverManager
from selenium.webdriver.chrome.options import Options

__author__ = 'sunny'


class DriverType:
    APPIUM = "appium"
    CHROME = "chrome"


class Driver(Find, FlutterFind):
    def __init__(self, browser_type=DriverType.APPIUM, **browser_args):
        Find.__init__(self)
        FlutterFind.__init__(self)
        driver_type_in_lower = browser_type.lower()

        if driver_type_in_lower == "appium":
            self.__browser_type = DriverType.APPIUM
            self.__app_driver = AppiumDriver.Remote(**browser_args)

        elif driver_type_in_lower == 'chrome':
            self.__browser_type = DriverType.CHROME
            self.chrome_options = Options()
            self.chrome_options.add_argument('--headless')
            self.chrome_options.add_argument('--disable-gpu')
            self.chrome_options.add_argument('--disable-blink-features=AutomationControlled')
            try:  # for selenium 3
                self.__app_driver = SeleniumDriver.Chrome(ChromeDriverManager().install(),options= self.chrome_options, **browser_args)
            except Exception:   # for selenium 4
                self.__app_driver = SeleniumDriver.Chrome(service=ChromeService(ChromeDriverManager().install()),options= self.chrome_options,**browser_args)
        else:
            raise Exception("The browser type [%s] is not supported." % browser_type)

    def get_driver(self):
        return self

    def _app_driver(self):
        return self.__app_driver

    def _appium_context(self):
        return self.__app_driver

    def get_driver_type(self):
        return self.__browser_type

    def before_startup(self, method):
        self.__app_driver.start_client = method

    def after_shutdown(self, method):
        self.__app_driver.stop_client = method

    def start_session(self, desired_capabilities, browser_profile=None):
        self.__app_driver.start_session(desired_capabilities, browser_profile)

    def set_page_load_timeout(self, timeout):
        self.__app_driver.set_page_load_timeout(timeout / 1000.0)

    def set_script_timeout(self, time_to_wait):
        self.__app_driver.set_script_timeout(time_to_wait / 1000.0)

    def set_wait_element_interval(self, interval):
        self.__wait_element_interval = interval

    def get_wait_element_interval(self):
        return self.__wait_element_interval

    def set_wait_element_timeout(self, timeout):
        self.__wait_element_timeout = timeout

    def get_wait_element_timeout(self):
        return self.__wait_element_timeout

    def execute_script(self, script, *args):
        return self.__app_driver.execute_script(script, *args)

    def execute_async_script(self, script, *args):
        return self.__app_driver.execute_async_script(script, *args)

    def open(self, url):
        self.__app_driver.get(url)

    def get_title(self):
        return self.__app_driver.title

    def get_page_source(self):
        return self.__app_driver.page_source

    def refresh(self):
        self.__app_driver.refresh()

    def back(self):
        self.__app_driver.back()

    def forward(self):
        self.__app_driver.forward()

    def quit(self):
        self.__app_driver.quit()

    def get_current_url(self):
        return self.__app_driver.current_url

    def switch_to_frame(self, frame_reference):
        self.__app_driver.switch_to.frame(frame_reference)

    def switch_to_parent_frame(self):
        self.__app_driver.switch_to.parent_frame()

    def switch_to_default_content(self):
        self.__app_driver.switch_to.default_content()

    def get_alert(self):
        return self.__app_driver.switch_to.alert

    def wait_for_alert_present(self):
        Utils.wait_for(self.is_alert_present)

    def get_cookies(self):
        return self.__app_driver.get_cookies()

    def get_cookie(self, name):
        return self.__app_driver.get_cookie(name)

    def delete_cookie(self, name):
        self.__app_driver.delete_cookie(name)

    def delete_all_cookies(self):
        self.__app_driver.delete_all_cookies()

    def add_cookie(self, cookie_dict):
        self.__app_driver.add_cookie(cookie_dict)

    def get_desired_capabilities(self):
        return self.__app_driver.desired_capabilities

    def get_screenshot_as_file(self, filename):
        return self.__app_driver.get_screenshot_as_file(filename)

    def get_screenshot_as_png(self):
        return self.__app_driver.get_screenshot_as_png()

    def start_recording_screen(self):
        return self.__app_driver.start_recording_screen()

    def stop_recording_screen(self):
        return self.__app_driver.stop_recording_screen()

    def get_screenshot_as_base64(self):
        return self.__app_driver.get_screenshot_as_base64()

    save_screenshot = get_screenshot_as_file

    def get_current_window_handle(self):
        return self.__app_driver.current_window_handle

    def get_window_handles(self):
        return self.__app_driver.window_handles

    def switch_to_window(self, window_reference):
        self.__app_driver.switch_to.window(window_reference)

    def close_window(self, window_reference="current"):
        if window_reference == "current":
            self.__app_driver.close()
        else:
            current_window = self.get_current_window_handle()
            self.switch_to_window(window_reference)
            self.__app_driver.close()
            self.switch_to_window(current_window)

    def set_window_size(self, width, height, window_reference="current"):
        self.__app_driver.set_window_size(width, height, window_reference)

    def get_window_size(self, window_reference="current"):
        return self.__app_driver.get_window_size(window_reference)

    def set_window_position(self, x, y, window_reference="current"):
        self.__app_driver.set_window_position(x, y, window_reference)

    def get_window_position(self, window_reference="current"):
        return self.__app_driver.get_window_position(window_reference)

    def get_orientation(self):
        return self.__app_driver.orientation

    def set_orientation(self, value):
        self.__app_driver = value

    def get_application_cache(self):
        return self.__app_driver.application_cache

    def get_log_types(self):
        return self.__app_driver.log_types

    def get_log(self, log_type):
        return self.__app_driver.get_log(log_type)

    def switch_to_context(self, name):
        self.__app_driver._switch_to.context(name)
        return self._appium_context().current_context

    def hide_keyboard(self):
        self.__app_driver.find_element_by_xpath("//UIAButton[@name='Hide keyboard']").click()

    def scroll_to(self, element):
        SeleniumUtils.scroll_to(self._appium_context(), element)

    def scroll_down(self):
        pass

    # SeleniumUtils.scroll_to_direction(self._selenium_context(), "Down")

    # self._web_driver().swipe(0,0,768,1184)

    def close_App(self):
        self.__app_driver.close_app()
        self.__app_driver.quit()

    def switch_to_web(self):

        view = self.__app_driver.contexts
        self.__app_driver._switch_to.context(view[-1])

    def switch_to_native(self):
        # self.__app_driver._switch_to.context(None)
        self.__app_driver._switch_to.context('NATIVE_APP')

    def sent_key_event(self, number):
        self.__app_driver.keyevent(number)

    def __str__(self):
        return "Driver [WebDriver: %s][SessionId: %s]" % (self.__app_driver.name, self.__app_driver.session_id)

    def get_webview_page_source(self):
        return self.__app_driver.page_source

    '''
    Following are some drive action for appium flutter driver
    '''

    def switch_to_flutter(self):
        self.__app_driver._switch_to.context('FLUTTER')