AI_UIAutomation/tests/water_detector/water_detector_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,
  navigateToFirmwarePage,
  checkFirmwareVersion,
  navigateToDeviceInfo,
  getDeviceInfo,
  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('waterDetector', 'WATER_DETECTOR_DEVICE');

describe('Water Detector Settings - 设备设置页', () => {
  let driver: DeviceDriver;
  let reporter: TestReporter;
  const automationName = `WDAutoTest_${Date.now().toString().slice(-4)}`;

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

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

      // 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} → WDTest → ${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 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;
    }
  });

  it('创建自动化', async () => {
    const start = Date.now();
    try {
      const entered = await enterDeviceSettings(driver, deviceName);
      expect(entered).toBe(true);

      // Find automation/scene option in settings
      const tapped = await scrollToAndTap(driver, 'Automation') ||
        await scrollToAndTap(driver, 'Scene') ||
        await scrollToAndTap(driver, 'Create Automation');
      expect(tapped).toBe(true);
      await sleep(3000);

      let source = await driver.getSource();
      const hasAutomationPage = source.includes('Automation') || source.includes('Scene') ||
        source.includes('Create') || source.includes('Add');

      // Tap Add/Create button if available
      const addBtn = await driver.findElementRaw('name', 'Add') ||
        await driver.findElementRaw('name', 'Create') ||
        await driver.findElementRaw('name', '+');
      if (addBtn) {
        await driver.tapElement(addBtn);
        await sleep(2000);
      }

      // Look for water leak trigger option
      const triggerOption = await driver.findElementRaw('name', 'Water Leak') ||
        await driver.findElementRaw('name', 'Detects water') ||
        await driver.findElementRaw('name', 'Water detected');
      if (triggerOption) {
        await driver.tapElement(triggerOption);
        await sleep(2000);
      }

      // Save/confirm the automation
      const saveBtn = await driver.findElementRaw('name', 'Save') ||
        await driver.findElementRaw('name', 'Done') ||
        await driver.findElementRaw('name', 'Confirm');
      if (saveBtn) {
        await driver.tapElement(saveBtn);
        await sleep(2000);
      }

      source = await driver.getSource();
      const detail = `自动化页面=${hasAutomationPage}`;
      console.log(detail);
      expect(hasAutomationPage).toBe(true);

      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);

      // Navigate to automation list
      const tapped = await scrollToAndTap(driver, 'Automation') ||
        await scrollToAndTap(driver, 'Scene') ||
        await scrollToAndTap(driver, 'Manage Automation');
      expect(tapped).toBe(true);
      await sleep(3000);

      const source = await driver.getSource();
      const hasAutomation = source.includes('Automation') || source.includes('Scene') ||
        source.includes('Water');

      // Find and delete the automation
      const automationEl = await driver.findElementRaw('name', 'Water Leak') ||
        await driver.findElementRaw('name', 'Water detected') ||
        await driver.findElementRaw('name', automationName);
      if (automationEl) {
        await driver.tapElement(automationEl);
        await sleep(2000);

        // Find delete button
        const deleteBtn = await driver.findElementRaw('name', 'Delete') ||
          await driver.findElementRaw('name', 'Remove');
        if (deleteBtn) {
          await driver.tapElement(deleteBtn);
          await sleep(1000);

          // Confirm deletion
          const confirmBtn = await driver.findElementRaw('name', 'Confirm') ||
            await driver.findElementRaw('name', 'Delete') ||
            await driver.findElementRaw('name', 'OK');
          if (confirmBtn) {
            await driver.tapElement(confirmBtn);
            await sleep(2000);
          }
        }
      }

      const detail = `自动化列表页面=${hasAutomation}, 删除操作完成`;
      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;
    }
  });
});