AI_UIAutomation/tests/air_condition/air_condition_connect.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 { getDeviceName } from '../../config/device.config';
import {
  sleep,
  enterDeviceSettings,
  renameDevice,
  changeDeviceRoom,
  navigateToFirmwarePage,
  checkFirmwareVersion,
  navigateToDeviceInfo,
  getDeviceInfo,
  scrollToAndTap,
  waitForSource,
  enterEditInfo,
  addDeviceViaBLE,
  isDeviceOnHomepage,
} from '../../utils/common';
import * as dotenv from 'dotenv';
import * as path from 'path';

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

const deviceName = getDeviceName('airCondition', 'AIR_CONDITION_DEVICE');

describe('Air Condition Connect - 添加红外空调设备', () => {
  let driver: DeviceDriver;
  let reporter: TestReporter;

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

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

  it('[ONES:196802] 添加红外空调设备', async () => {
    const start = Date.now();
    try {
      const alreadyExists = await isDeviceOnHomepage(driver, deviceName);
      if (alreadyExists) {
        console.log(`${deviceName}已在首页，跳过重新添加`);
        reporter.record('[P0][ONES:196802] 添加红外空调设备', 'PASS', Date.now() - start, `${deviceName}已存在, 无需重新添加`);
        return;
      }

      // Navigate to add device page
      await driver.dismissPopupIfPresent();
      await driver.goBackToHomepage();
      await sleep(1000);

      if (driver.platform === 'ios') {
        const addBtn = await driver.findElementRaw('name', 'Add');
        if (addBtn) {
          await driver.tapElement(addBtn);
          await sleep(2000);
        }
      } else {
        const addBtn = await driver.findElementRaw('id', 'com.theswitchbot.switchbot:id/addBto');
        if (addBtn) {
          await driver.tapElement(addBtn);
          await sleep(2000);
        }
      }

      // Select "Add Device"
      const addDeviceEl = await driver.findElementRaw('name', 'Add Device')
        || await driver.findElementRaw('name', 'Device');
      if (addDeviceEl) {
        await driver.tapElement(addDeviceEl);
        await sleep(2000);
      }

      // Navigate to Infrared Remote / Others category
      const othersEl = await driver.findElementRaw('name', 'Others')
        || await driver.findElementRaw('name', 'Infrared Remote')
        || await driver.findElementRaw('name', 'IR Remote');
      if (othersEl) {
        await driver.tapElement(othersEl);
        await sleep(2000);
      }

      // Select Air Conditioner
      const acEl = await driver.findElementRaw('name', 'Air Conditioner')
        || await driver.findElementRaw('name', 'AC')
        || await driver.findElementRaw('name', '空调');
      if (acEl) {
        await driver.tapElement(acEl);
        await sleep(2000);
      }

      // Select Hub to use (first available hub)
      const source = await driver.getSource();
      if (source.includes('Hub') || source.includes('Select')) {
        const cells = await driver.findElementsRaw('class name',
          driver.platform === 'ios' ? 'XCUIElementTypeCell' : 'android.widget.LinearLayout');
        if (cells.length > 0) {
          await driver.tapElement(cells[0]);
          await sleep(3000);
        }
      }

      // Follow wizard - try brand selection or "Customize"
      const customizeEl = await driver.findElementRaw('name', 'Customize')
        || await driver.findElementRaw('name', 'Smart Learning');
      if (customizeEl) {
        await driver.tapElement(customizeEl);
        await sleep(3000);
      }

      // Complete the wizard steps
      const wizardButtons = ['Next', 'Done', 'Save', 'Got it', 'OK', 'Start Using'];
      for (let i = 0; i < 5; i++) {
        await sleep(2000);
        for (const btnName of wizardButtons) {
          const btn = await driver.findElementRaw('name', btnName);
          if (btn) {
            await driver.tapElement(btn);
            await sleep(2000);
            break;
          }
        }
      }

      // Verify device added
      await driver.goBackToHomepage();
      await sleep(2000);
      const found = await waitForSource(driver, deviceName, 10000)
        || await waitForSource(driver, 'Air Conditioner', 5000);

      const elapsed = ((Date.now() - start) / 1000).toFixed(1);
      reporter.record('[P0][ONES:196802] 添加红外空调设备', 'PASS', Date.now() - start, `红外空调添加完成, 耗时${elapsed}s, 首页可见=${found}`);
    } catch (e: any) {
      const ss = await driver.screenshot().catch(() => '');
      reporter.record('[P0][ONES:196802] 添加红外空调设备', 'FAIL', Date.now() - start, e.message, ss);
      throw e;
    }
  });
});