AI_UIAutomation/tests/hub/hub_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,
  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('hub', 'HUB_DEVICE');

describe('Hub Control - 功能页操作', () => {
  let driver: DeviceDriver;
  let reporter: TestReporter;

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

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

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

  async function enterHubControl(): Promise<void> {
    const el = await driver.findElementRaw('name', deviceName);
    if (!el) throw new Error(`找不到${deviceName}卡片`);
    await driver.tapElement(el);
    await sleep(3000);
  }

  it('切换温度单位为°C', async () => {
    const start = Date.now();
    try {
      await enterHubControl();

      const celsiusEl = await driver.findElementRaw('name', '°C') ||
        await driver.findElementRaw('name', 'C') ||
        await driver.findElementRaw('name', '℃');
      expect(celsiusEl).not.toBeNull();
      await driver.tapElement(celsiusEl!);
      await sleep(2000);

      const source = await driver.getSource();
      const hasCelsius = source.includes('°C') || source.includes('℃');
      expect(hasCelsius).toBe(true);

      reporter.record('切换温度单位为°C', 'PASS', Date.now() - start, '温度单位已切换为°C');
    } catch (e: any) {
      const ss = await driver.screenshot().catch(() => '');
      reporter.record('切换温度单位为°C', 'FAIL', Date.now() - start, e.message, ss);
      throw e;
    }
  });

  it('切换温度单位为°F', async () => {
    const start = Date.now();
    try {
      await enterHubControl();

      const fahrenheitEl = await driver.findElementRaw('name', '°F') ||
        await driver.findElementRaw('name', 'F') ||
        await driver.findElementRaw('name', '℉');
      expect(fahrenheitEl).not.toBeNull();
      await driver.tapElement(fahrenheitEl!);
      await sleep(2000);

      const source = await driver.getSource();
      const hasFahrenheit = source.includes('°F') || source.includes('℉');
      expect(hasFahrenheit).toBe(true);

      // Restore to °C
      const celsiusEl = await driver.findElementRaw('name', '°C') ||
        await driver.findElementRaw('name', 'C') ||
        await driver.findElementRaw('name', '℃');
      if (celsiusEl) {
        await driver.tapElement(celsiusEl);
        await sleep(1000);
      }

      reporter.record('切换温度单位为°F', 'PASS', Date.now() - start, '温度单位已切换为°F并还原');
    } catch (e: any) {
      const ss = await driver.screenshot().catch(() => '');
      reporter.record('切换温度单位为°F', 'FAIL', Date.now() - start, e.message, ss);
      throw e;
    }
  });

  it('ON按键配置', async () => {
    const start = Date.now();
    try {
      await enterHubControl();

      const onKeyEl = await driver.findElementRaw('name', 'ON Key') ||
        await driver.findElementRaw('name', 'ON');
      expect(onKeyEl).not.toBeNull();
      await driver.tapElement(onKeyEl!);
      await sleep(2000);

      const source = await driver.getSource();
      const hasSceneConfig = source.includes('Scene') || source.includes('Action') ||
        source.includes('Device') || source.includes('Automation');
      expect(hasSceneConfig).toBe(true);

      reporter.record('ON按键配置', 'PASS', Date.now() - start, `ON Key配置页面已打开, 含Scene配置=${hasSceneConfig}`);
    } catch (e: any) {
      const ss = await driver.screenshot().catch(() => '');
      reporter.record('ON按键配置', 'FAIL', Date.now() - start, e.message, ss);
      throw e;
    }
  });

  it('OFF按键配置', async () => {
    const start = Date.now();
    try {
      await enterHubControl();

      const offKeyEl = await driver.findElementRaw('name', 'OFF Key') ||
        await driver.findElementRaw('name', 'OFF');
      expect(offKeyEl).not.toBeNull();
      await driver.tapElement(offKeyEl!);
      await sleep(2000);

      const source = await driver.getSource();
      const hasSceneConfig = source.includes('Scene') || source.includes('Action') ||
        source.includes('Device') || source.includes('Automation');
      expect(hasSceneConfig).toBe(true);

      reporter.record('OFF按键配置', 'PASS', Date.now() - start, `OFF Key配置页面已打开, 含Scene配置=${hasSceneConfig}`);
    } catch (e: any) {
      const ss = await driver.screenshot().catch(() => '');
      reporter.record('OFF按键配置', 'FAIL', Date.now() - start, e.message, ss);
      throw e;
    }
  });

  it('温度数据详情', async () => {
    const start = Date.now();
    try {
      await enterHubControl();

      // Tap temperature display area to open data detail
      const tempEl = await driver.findElementRaw('name', 'Temperature') ||
        await driver.findElementRaw('name', 'Temp');
      if (tempEl) {
        await driver.tapElement(tempEl);
      } else {
        // Try tapping the temperature number area
        const source = await driver.getSource();
        const hasTemp = source.includes('°C') || source.includes('°F');
        expect(hasTemp).toBe(true);
        // Tap center area where temperature is typically displayed
        await driver.tap(200, 300);
      }
      await sleep(3000);

      const source = await driver.getSource();
      const hasDataDetail = source.includes('Data') || source.includes('History') ||
        source.includes('Chart') || source.includes('Detail') || source.includes('Temperature');
      expect(hasDataDetail).toBe(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 {
      await enterHubControl();

      const exportEl = await driver.findElementRaw('name', 'Export') ||
        await driver.findElementRaw('name', 'export');
      if (!exportEl) {
        // Try scrolling to find export
        await scrollToAndTap(driver, 'Export');
      } else {
        await driver.tapElement(exportEl);
      }
      await sleep(3000);

      const source = await driver.getSource();
      const hasExportOptions = source.includes('Export') || source.includes('CSV') ||
        source.includes('Share') || source.includes('Email') || source.includes('Loading');
      expect(hasExportOptions).toBe(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;
    }
  });
});