AI_UIAutomation/tests/plug/plug_setting.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,
  enterDeviceSettings,
  enterEditInfo,
  renameDevice,
  changeDeviceRoom,
  scrollToAndTap,
  waitForSource,
  navigateToFirmwarePage,
  checkFirmwareVersion,
  navigateToDeviceInfo,
  getDeviceInfo,
} 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('plug', 'PLUG_DEVICE');

describe('Plug Settings - 设备设置页', () => {
  let driver: DeviceDriver;
  let reporter: TestReporter;

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

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

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

  it('修改设备名称-取消', async () => {
    const start = Date.now();
    try {
      const entered = await enterDeviceSettings(driver, deviceName);
      expect(entered).toBe(true);
      const editOpened = await enterEditInfo(driver);
      expect(editOpened).toBe(true);

      const renamed = await renameDevice(driver, 'RenameTest', 'cancel');
      expect(renamed).toBe(true);

      const source = await driver.getSource();
      expect(source).toContain(deviceName);

      reporter.record('修改设备名称-取消', 'PASS', Date.now() - start, `输入"RenameTest"后取消, 名称保持"${deviceName}"`);
    } catch (e: any) {
      const ss = await driver.screenshot().catch(() => '');
      reporter.record('修改设备名称-取消', 'FAIL', Date.now() - start, e.message, ss);
      throw e;
    }
  });

  it('修改设备名称-保存还原', async () => {
    const start = Date.now();
    try {
      const entered = await enterDeviceSettings(driver, deviceName);
      expect(entered).toBe(true);
      const editOpened = await enterEditInfo(driver);
      expect(editOpened).toBe(true);

      const renamed = await renameDevice(driver, 'TestPlug', 'save');
      expect(renamed).toBe(true);

      let source = await driver.getSource();
      expect(source).toContain('TestPlug');
      console.log('名称已修改为: TestPlug');

      // Restore
      const restored = await renameDevice(driver, deviceName, 'save');
      expect(restored).toBe(true);

      source = await driver.getSource();
      expect(source).toContain(deviceName);
      console.log('名称还原: true');

      reporter.record('修改设备名称-保存还原', 'PASS', Date.now() - start, `${deviceName} → TestPlug → ${deviceName} (还原=true)`);
    } catch (e: any) {
      const ss = await driver.screenshot().catch(() => '');
      reporter.record('修改设备名称-保存还原', 'FAIL', Date.now() - start, e.message, ss);
      throw e;
    }
  });

  it('修改设备房间', async () => {
    const start = Date.now();
    try {
      const entered = await enterDeviceSettings(driver, deviceName);
      expect(entered).toBe(true);
      const editOpened = await enterEditInfo(driver);
      expect(editOpened).toBe(true);

      const result = await changeDeviceRoom(driver);
      expect(result.success).toBe(true);
      console.log('可选房间数:', result.roomCount);

      // Restore (tap first room again to toggle)
      await changeDeviceRoom(driver);

      reporter.record('修改设备房间', 'PASS', Date.now() - start, `Select Room页面, ${result.roomCount}个房间可选`);
    } catch (e: any) {
      const ss = await driver.screenshot().catch(() => '');
      reporter.record('修改设备房间', 'FAIL', Date.now() - start, e.message, ss);
      throw e;
    }
  });

  it('切换设备图标', async () => {
    const start = Date.now();
    try {
      const entered = await enterDeviceSettings(driver, deviceName);
      expect(entered).toBe(true);

      const tapped = await scrollToAndTap(driver, 'Customize Icon');
      expect(tapped).toBe(true);
      await sleep(2000);

      const source = await driver.getSource();
      const hasIconPage = source.includes('Icon') || source.includes('Customize') || source.includes('Choose');
      expect(hasIconPage).toBe(true);

      reporter.record('切换设备图标', 'PASS', Date.now() - start, 'Customize Icon页面打开成功');
    } catch (e: any) {
      const ss = await driver.screenshot().catch(() => '');
      reporter.record('切换设备图标', 'FAIL', Date.now() - start, e.message, ss);
      throw e;
    }
  });

  it('切换设备类型', async () => {
    const start = Date.now();
    try {
      const entered = await enterDeviceSettings(driver, deviceName);
      expect(entered).toBe(true);

      // Find device type selector
      const tapped = await scrollToAndTap(driver, 'Device Type') ||
        await scrollToAndTap(driver, 'Type');
      await sleep(2000);

      const source = await driver.getSource();
      const hasTypePage = source.includes('Type') || source.includes('Plug') || source.includes('Light');
      expect(hasTypePage).toBe(true);

      // Select a different type if available, then restore
      const altType = await driver.findElementRaw('name', 'Light') ||
        await driver.findElementRaw('name', 'Appliance');
      if (altType) {
        await driver.tapElement(altType);
        await sleep(2000);

        // Restore to Plug type
        const plugType = await driver.findElementRaw('name', 'Plug') ||
          await driver.findElementRaw('name', 'Outlet');
        if (plugType) {
          await driver.tapElement(plugType);
          await sleep(2000);
        }
      }

      reporter.record('切换设备类型', 'PASS', Date.now() - start, '设备类型页面打开成功');
    } catch (e: any) {
      const ss = await driver.screenshot().catch(() => '');
      reporter.record('切换设备类型', 'FAIL', Date.now() - start, e.message, ss);
      throw e;
    }
  });

  it('NFC页面', async () => {
    const start = Date.now();
    try {
      const entered = await enterDeviceSettings(driver, deviceName);
      expect(entered).toBe(true);

      const tapped = await scrollToAndTap(driver, 'NFC');
      expect(tapped).toBe(true);
      await sleep(2000);

      const source = await driver.getSource();
      const hasNFC = source.includes('NFC') || source.includes('Tag') || source.includes('Write');
      expect(hasNFC).toBe(true);

      reporter.record('NFC页面', 'PASS', Date.now() - start, 'NFC页面打开成功');
    } catch (e: any) {
      const ss = await driver.screenshot().catch(() => '');
      reporter.record('NFC页面', 'FAIL', Date.now() - start, e.message, ss);
      throw e;
    }
  });

  it('查看固件版本', async () => {
    const start = Date.now();
    try {
      const entered = await enterDeviceSettings(driver, deviceName);
      expect(entered).toBe(true);

      const navOk = await navigateToFirmwarePage(driver);
      expect(navOk).toBe(true);
      await sleep(3000);

      const version = await checkFirmwareVersion(driver);
      const detail = `固件版本=${version}`;
      console.log(detail);
      expect(version).not.toBe('unknown');

      reporter.record('查看固件版本', 'PASS', Date.now() - start, detail);
    } catch (e: any) {
      const ss = await driver.screenshot().catch(() => '');
      reporter.record('查看固件版本', 'FAIL', Date.now() - start, e.message, ss);
      throw e;
    }
  });

  it('查看设备信息', async () => {
    const start = Date.now();
    try {
      const entered = await enterDeviceSettings(driver, deviceName);
      expect(entered).toBe(true);

      const navOk = await navigateToDeviceInfo(driver);
      expect(navOk).toBe(true);
      await sleep(3000);

      const info = await getDeviceInfo(driver);
      const detail = `BLE MAC=${info.macAddress || 'not found'}`;
      console.log(detail);
      expect(info.macAddress).toBeDefined();

      reporter.record('查看设备信息', 'PASS', Date.now() - start, detail);
    } catch (e: any) {
      const ss = await driver.screenshot().catch(() => '');
      reporter.record('查看设备信息', 'FAIL', Date.now() - start, e.message, ss);
      throw e;
    }
  });
});