AI_UIAutomation/tests/bot/bot_setting.test.ts

import { describe, it, beforeAll, afterAll, beforeEach, expect } from 'vitest';
import { DeviceDriver } from '../../drivers/types';
import { createDriver } from '../../drivers/factory';
import { BotHelper } from '../../utils/bot-helper';
import { TestReporter } from '../../utils/test-reporter';
import { getDeviceName } from '../../config/device.config';
import {
  sleep,
  waitForSource,
  enterDeviceSettings,
  enterEditInfo,
  scrollToAndTap,
  renameDevice,
  changeDeviceRoom,
  changeDeviceIcon,
  navigateToTimerPage,
  addTimer,
  deleteAllTimers,
  navigateToDeviceInfo,
  getDeviceInfo,
  navigateToFirmwarePage,
  checkFirmwareVersion,
} from '../../utils/common';
import * as dotenv from 'dotenv';
import * as path from 'path';

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

const deviceName = getDeviceName('bot', 'BOT_DEVICE');

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

  beforeAll(async () => {
    driver = createDriver();
    await driver.createSession();
    bot = new BotHelper(driver);
    reporter = new TestReporter('Bot_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, 'TestBot', 'save');
      expect(renamed).toBe(true);

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

      // 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} → TestBot → ${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 changed = await changeDeviceIcon(driver);
      expect(changed).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('定时器 - 添加Only once定时', async () => {
    const start = Date.now();
    try {
      const entered = await enterDeviceSettings(driver, deviceName);
      expect(entered).toBe(true);

      const navOk = await navigateToTimerPage(driver);
      expect(navOk).toBe(true);
      await sleep(2000);

      const added = await addTimer(driver);
      expect(added).toBe(true);

      const source = await driver.getSource();
      expect(source).toContain('Only once');
      console.log('定时器列表有Only once: true');

      reporter.record('添加Only once定时', 'PASS', Date.now() - start, '添加Only once定时成功');
    } catch (e: any) {
      const ss = await driver.screenshot().catch(() => '');
      reporter.record('添加Only once定时', '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 navigateToTimerPage(driver);
      expect(navOk).toBe(true);
      await sleep(2000);

      // Ensure at least one schedule exists
      let source = await driver.getSource();
      if (!source.includes('Only once') && !source.includes('Press Mode')) {
        await addTimer(driver);
      }

      const deleted = await deleteAllTimers(driver);
      expect(deleted).toBe(true);
      console.log('定时器已清空: true');

      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('查看固件版本信息', 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 source = await driver.getSource();
      const hasBattery = source.includes('Battery');

      const detail = `固件版本=${version}, Battery信息=${hasBattery}`;
      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('查看设备信息 - BLE MAC', 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;
    }
  });

  it('查看操作日志', async () => {
    const start = Date.now();
    try {
      const entered = await enterDeviceSettings(driver, deviceName);
      expect(entered).toBe(true);

      const tapped = await scrollToAndTap(driver, 'Logs');
      expect(tapped).toBe(true);
      await waitForSource(driver, 'Logs', 5000);

      const source = await driver.getSource();
      expect(source).toContain('Logs');
      const hasEntries = source.includes('Ran successfully') || source.includes('Turn on') ||
        source.includes('Turn off') || source.includes('Press') || source.includes('App Control') ||
        source.includes('May') || source.includes('No more data') || source.includes('No logs');

      const detail = `Logs页面加载成功, 有操作记录=${hasEntries}`;
      console.log(detail);
      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;
    }
  });
});