#NW.js 桌面应用自动化测试：当 Chromium 套壳遇上 Playwright

# utils/nw_launcher.py
import subprocess
import time
import socket


class NwLauncher:
    """NW.js 进程管理器 —— 管杀管埋"""

    def __init__(self, nw_path: str, app_dir: str, cdp_port: int = 9222):
        self.nw_path = nw_path
        self.app_dir = app_dir
        self.cdp_host = "127.0.0.1"
        self.cdp_port = cdp_port
        self._proc = None

    def start(self, timeout: int = 30):
        """启动 NW.js 并等待 CDP 就绪"""
        self._proc = subprocess.Popen(
            [self.nw_path, self.app_dir, f"--remote-debugging-port={self.cdp_port}"],
            stdout=subprocess.DEVNULL,
            stderr=subprocess.DEVNULL,
        )
        self._wait_for_cdp(timeout)

    def stop(self):
        """优雅退出，必要时强杀"""
        if self._proc is None:
            return
        self._proc.terminate()
        try:
            self._proc.wait(timeout=10)
        except subprocess.TimeoutExpired:
            self._proc.kill()

    @property
    def cdp_url(self) -> str:
        return f"http://{self.cdp_host}:{self.cdp_port}"

    def _wait_for_cdp(self, timeout: int):
        deadline = time.time() + timeout
        while time.time() < deadline:
            if self._is_port_open():
                return
            time.sleep(0.5)
        self._proc.kill()
        raise RuntimeError(
            f"NW.js CDP 端口 {self.cdp_port} 在 {timeout}s 内未就绪"
        )

    def _is_port_open(self) -> bool:
        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
            s.settimeout(1)
            return s.connect_ex((self.cdp_host, self.cdp_port)) == 0

# conftest.py
import pytest
from playwright.sync_api import sync_playwright, Browser, Page
from utils.nw_launcher import NwLauncher

NW_EXECUTABLE = r"C:\path\to\nw.exe"
APP_DIR = r"C:\path\to\your\app"


@pytest.fixture(scope="session")
def nw_launcher():
    """Session 级别：启动一次，全局唯一"""
    launcher = NwLauncher(NW_EXECUTABLE, APP_DIR)
    launcher.start()
    yield launcher
    launcher.stop()


@pytest.fixture(scope="session")
def nw_browser(nw_launcher: NwLauncher) -> Browser:
    """Session 级别：通过 CDP 连接到 NW.js"""
    with sync_playwright() as p:
        browser = p.chromium.connect_over_cdp(nw_launcher.cdp_url)
        yield browser


@pytest.fixture(scope="function")
def page(nw_browser: Browser) -> Page:
    """每个测试用例独立的 page，互不污染"""
    if nw_browser.contexts:
        ctx = nw_browser.contexts[0]
        if ctx.pages:
            page = ctx.pages[0]
            page.bring_to_front()
            return page
    ctx = nw_browser.new_context()
    return ctx.new_page()

# 别这么写
def test_login(page):
    page.locator("#username").fill("admin")
    page.locator("#password").fill("123456")
    page.locator("#login-btn").click()
    assert page.locator(".welcome-text").inner_text() == "欢迎"

# pages/base_page.py
from playwright.sync_api import Page


class BasePage:
    """所有页面对象的基类 —— DRY 原则的第一道防线"""

    def __init__(self, page: Page):
        self.page = page

    def wait_visible(self, locator, timeout: int = 10000):
        """等待元素可见（所有页面都要用的操作，写一遍就够了）"""
        self.page.locator(locator).wait_for(state="visible", timeout=timeout)

    def click_by_testid(self, testid: str):
        self.page.get_by_test_id(testid).click()

    def fill_by_testid(self, testid: str, value: str):
        self.page.get_by_test_id(testid).fill(value)

    def get_text_by_testid(self, testid: str) -> str:
        return self.page.get_by_test_id(testid).inner_text()

    def get_title(self) -> str:
        return self.page.title()

    def screenshot(self, name: str):
        self.page.screenshot(path=f"screenshots/{name}.png", full_page=True)

# pages/login_page.py
from pages.base_page import BasePage


class LoginPage(BasePage):
    """登录页 —— 元素定位只出现在这里，别处不认"""

    # ── Locators（集中管理，改起来只改这一处）──
    USERNAME_INPUT = "username-input"
    PASSWORD_INPUT = "password-input"
    LOGIN_BUTTON = "login-btn"
    ERROR_MESSAGE = "login-error"

    # ── Actions ──
    def navigate(self):
        """导航到登录页"""
        self.page.goto("file:///C:/path/to/login.html")

    def login(self, username: str, password: str):
        """执行登录操作 —— 用例只调这个方法，不需要知道内部怎么点的"""
        self.fill_by_testid(self.USERNAME_INPUT, username)
        self.fill_by_testid(self.PASSWORD_INPUT, password)
        self.click_by_testid(self.LOGIN_BUTTON)

    # ── Assertions（页面状态的断言方法，供用例直接调）──
    def get_error_message(self) -> str:
        return self.get_text_by_testid(self.ERROR_MESSAGE)

    def is_login_button_visible(self) -> bool:
        return self.page.get_by_test_id(self.LOGIN_BUTTON).is_visible()

# pages/main_page.py
from pages.base_page import BasePage


class MainPage(BasePage):
    WELCOME_TEXT = "welcome-text"
    SETTINGS_LINK = "settings-link"
    PROJECT_LIST = "project-list"
    NEW_PROJECT_BTN = "new-project-btn"

    def get_welcome_message(self) -> str:
        return self.get_text_by_testid(self.WELCOME_TEXT)

    def go_to_settings(self):
        self.click_by_testid(self.SETTINGS_LINK)

    def create_new_project(self, project_name: str):
        self.click_by_testid(self.NEW_PROJECT_BTN)
        self.page.get_by_placeholder("项目名称").fill(project_name)
        self.page.get_by_role("button", name="确认").click()

    def get_project_count(self) -> int:
        items = self.page.get_by_test_id(self.PROJECT_LIST).locator("li")
        return items.count()

# tests/test_login.py
import pytest
from pages.login_page import LoginPage
from pages.main_page import MainPage


class TestLogin:
    """登录模块 —— 用例只描述业务场景"""

    def test_login_success(self, page):
        """正常登录：输入正确账号密码后进入主页"""
        login_page = LoginPage(page)
        login_page.navigate()
        login_page.login("admin", "password123")

        main_page = MainPage(page)
        main_page.wait_visible(main_page.WELCOME_TEXT)
        assert "欢迎" in main_page.get_welcome_message()

    def test_login_wrong_password(self, page):
        """错误密码：页面上显示错误提示"""
        login_page = LoginPage(page)
        login_page.navigate()
        login_page.login("admin", "wrong_password")

        assert "密码错误" in login_page.get_error_message()

    def test_login_empty_username(self, page):
        """空用户名：登录按钮不可点击"""
        login_page = LoginPage(page)
        login_page.navigate()
        login_page.login("", "password123")

        assert login_page.is_login_button_visible()

# tests/test_main.py
from pages.login_page import LoginPage
from pages.main_page import MainPage


class TestMain:
    """主页面模块"""

    def test_create_project(self, page):
        """创建新项目后，项目列表数量增加"""
        login_page = LoginPage(page)
        login_page.navigate()
        login_page.login("admin", "password123")

        main_page = MainPage(page)
        before = main_page.get_project_count()
        main_page.create_new_project("自动化测试项目")
        main_page.wait_visible(main_page.PROJECT_LIST)
        after = main_page.get_project_count()

        assert after == before + 1

# pages/node_utils.py
"""封装通过 page.evaluate 调用的 Node 能力"""


class NodeUtils:
    def __init__(self, page):
        self.page = page

    def read_file(self, filepath: str) -> str:
        """直接读取本地文件 —— 测试数据准备的神器"""
        return self.page.evaluate("""
            (path) => {
                const fs = require('fs');
                return fs.readFileSync(path, 'utf-8');
            }
        """, filepath)

    def write_file(self, filepath: str, content: str):
        """直接写本地文件 —— 环境清理和 Mock 数据注入"""
        self.page.evaluate("""
            ([path, data]) => {
                const fs = require('fs');
                fs.writeFileSync(path, data);
            }
        """, [filepath, content])

    def get_platform(self) -> str:
        """获取操作系统平台"""
        return self.page.evaluate("() => require('os').platform()")

    def file_exists(self, filepath: str) -> bool:
        return self.page.evaluate("""
            (path) => {
                const fs = require('fs');
                return fs.existsSync(path);
            }
        """, filepath)

def test_config_import(page):
    """验证导入配置文件的功能"""
    from pages.node_utils import NodeUtils

    node = NodeUtils(page)
    # 用 Node API 写入一份测试配置
    node.write_file("/tmp/test_config.json", '{"theme":"dark"}')

    # 触发应用的导入功能
    page.get_by_test_id("import-config-btn").click()
    page.get_by_test_id("config-path-input").fill("/tmp/test_config.json")
    page.get_by_text("加载").click()

    # 断言
    page.get_by_text("主题: dark").wait_for()

# codegen 输出（原始）
page.locator("#username").fill("admin")
page.locator("#password").fill("123456")
page.locator("#login-btn").click()

# 进 PO 的方法
def login(self, username: str, password: str):
    self.fill_by_testid(self.USERNAME_INPUT, username)
    self.fill_by_testid(self.PASSWORD_INPUT, password)
    self.click_by_testid(self.LOGIN_BUTTON)