AI_UIAutomation/tests/curtain/curtain_control.test.ts

import { describe, it, beforeAll, afterAll, beforeEach, expect } from 'vitest';
import { DeviceDriver } from '../../drivers/types';
import { createDriver } from '../../drivers/factory';
import { TestReporter } from '../../utils/test-reporter';
import {
  sleep,
  navigateToTimerPage,
  addTimer,
  deleteAllTimers,
  enterDeviceSettings,
  scrollToAndTap,
  waitForSource,
} from '../../utils/common';
import { getDeviceName } from '../../config/device.config';
import * as dotenv from 'dotenv';
import * as path from 'path';

dotenv.config({ path: path.resolve(__dirname, '../../.env') });

const deviceName = getDeviceName('curtain', 'CURTAIN_DEVICE'); // 真实卡名(注册表,如 "Curtain 4D")

// 设备维度动态控制锚点:锚点按型号解析(设备名≠型号)。base Curtain=Curtain 1A(step3)
import { onesCtrl } from '../../utils/common';
import { setupResilientHooks } from "../../utils/common";
const CURTAIN_MODEL = process.env.CURTAIN_MODEL || 'Curtain 1A';
const CTRL_CURTAIN = onesCtrl('curtain', CURTAIN_MODEL);

describe('Curtain Control - 窗帘控制功能', () => {
  let driver: DeviceDriver;
  let reporter: TestReporter;
  setupResilientHooks(() => driver);

  beforeAll(async () => {
    driver = createDriver();
    await driver.createSession();
    reporter = new TestReporter('Curtain_Control', driver.platform.toUpperCase());
  });

  beforeEach(async () => {
    await driver.dismissPopupIfPresent();
    await driver.goBackToHomepage();
    await sleep(500);
    await driver.dismissPopupIfPresent();
  });

  afterAll(async () => {
    reporter.generate();
    await driver.destroySession();
  });

  // 点首页卡片 → 快捷弹窗(Open/Pause/Close + 状态)。
  // RN 首页 scrollToAndTap(scrollIntoView)对窗帘卡失效 → 手动滚动 + 从 getSource 解析卡名 bounds,点卡片开弹窗。
  async function openCurtainPopup(): Promise<void> {
    await driver.goBackToHomepage(); await sleep(800); await driver.dismissPopupIfPresent();
    let rect: { x: number; y: number; w: number } | null = null;
    for (let i = 0; i < 8 && !rect; i++) {
      const src = await driver.getSource();
      for (const n of src.split('<')) {
        if (!/resource-id="[^"]*\/nameText"/.test(n)) continue;
        const t = (n.match(/text="([^"]*)"/) || [])[1];
        if (!t || !t.includes(deviceName)) continue;
        const b = n.match(/bounds="\[(\d+),(\d+)\]\[(\d+),(\d+)\]"/);
        if (b) { rect = { x: +b[1], y: +b[2], w: +b[3] - +b[1] }; break; }
      }
      if (!rect) { await driver.scrollDown(500); await sleep(700); }
    }
    if (!rect) throw new Error(`找不到${deviceName}卡片`);
    await driver.tap(rect.x + Math.min(rect.w - 10, 60), rect.y); // 点卡名处(卡片可点区)开快捷弹窗
    await sleep(2500);
  }
  const enterCurtainControl = openCurtainPopup; // 延迟开关/光线感应 等辅助步骤复用(经"More"进功能页找控件)

  // 读弹窗里的窗帘状态:Fully open / Fully closed / "X% Opened..." 等
  function readCurtainState(src: string): string {
    const m = src.match(/text="(Fully open|Fully closed|[0-9]{1,3}% [^"]*)"/);
    return m ? m[1] : 'unknown';
  }

  // 点按钮后轮询等状态满足 pred(最多 ~24s,窗帘物理移动慢)
  async function waitCurtainState(pred: (s: string) => boolean): Promise<string> {
    let st = readCurtainState(await driver.getSource());
    for (let i = 0; i < 12 && !pred(st); i++) {
      await sleep(2000);
      st = readCurtainState(await driver.getSource());
    }
    return st;
  }

  it('打开窗帘', { timeout: 90000 }, async () => {
    const start = Date.now();
    try {
      await openCurtainPopup();
      const openBtn = await driver.findElementRaw('-android uiautomator', 'new UiSelector().text("Open")');
      expect(openBtn).not.toBeNull();
      await driver.tapElement(openBtn!);
      const st = await waitCurtainState(s => /open/i.test(s)); // 等到 "Fully open"
      console.log('打开后状态:', st);
      expect(/open/i.test(st)).toBe(true); // 硬断言:已打开
      expect(st).not.toBe('Fully closed');
      reporter.record(`${CTRL_CURTAIN} 打开窗帘`, 'PASS', Date.now() - start, `状态=${st}`);
    } catch (e: any) {
      const ss = await driver.screenshot().catch(() => '');
      reporter.record(`${CTRL_CURTAIN} 打开窗帘`, 'FAIL', Date.now() - start, e.message, ss);
      throw e;
    }
  });

  it('关闭窗帘', { timeout: 90000 }, async () => {
    const start = Date.now();
    try {
      await openCurtainPopup();
      const closeBtn = await driver.findElementRaw('-android uiautomator', 'new UiSelector().text("Close")');
      expect(closeBtn).not.toBeNull();
      await driver.tapElement(closeBtn!);
      const st = await waitCurtainState(s => s === 'Fully closed'); // 等到 "Fully closed"
      console.log('关闭后状态:', st);
      expect(st).toBe('Fully closed'); // 硬断言:已关闭
      reporter.record(`${CTRL_CURTAIN} 关闭窗帘`, 'PASS', Date.now() - start, `状态=${st}`);
    } catch (e: any) {
      const ss = await driver.screenshot().catch(() => '');
      reporter.record(`${CTRL_CURTAIN} 关闭窗帘`, 'FAIL', Date.now() - start, e.message, ss);
      throw e;
    }
  });

  it('暂停窗帘', { timeout: 90000 }, async () => {
    const start = Date.now();
    try {
      // Pause 非 ONES P0 step(step3 仅开/关);此处验证 Pause 控制可达:移动中点 Pause。
      await openCurtainPopup();
      const pauseBtn0 = await driver.findElementRaw('-android uiautomator', 'new UiSelector().text("Pause")');
      expect(pauseBtn0).not.toBeNull(); // 硬断言:快捷弹窗有 Pause 控制
      // 启动移动后尽快点 Pause(裸窗帘行程快,缩短间隔)
      const openBtn = await driver.findElementRaw('-android uiautomator', 'new UiSelector().text("Open")');
      if (openBtn) { await driver.tapElement(openBtn); await sleep(800); }
      const pauseBtn = await driver.findElementRaw('-android uiautomator', 'new UiSelector().text("Pause")');
      if (pauseBtn) { await driver.tapElement(pauseBtn); await sleep(2500); }
      const st = readCurtainState(await driver.getSource());
      console.log('暂停后状态:', st);
      reporter.record('暂停窗帘', 'PASS', Date.now() - start, `Pause 控制可达并执行, 状态=${st}`);
    } catch (e: any) {
      const ss = await driver.screenshot().catch(() => '');
      reporter.record('暂停窗帘', 'FAIL', Date.now() - start, e.message, ss);
      throw e;
    }
  });

});