Device UI 测试性能优化记录 — 7.3x 加速

1
# 优化前：Python 循环 + 逐单元格 CDP 调用
2
def get_table_row_count(self) -> int:
3
    tables = self._page.query_selector_all("table")
4
    for table in tables:
5
        rows = tbody.query_selector_all("tr")
6
        for row in rows:
7
            cells = row.query_selector_all("td")
8
            for cell in cells:
9
                cell.inner_text()  # 每次调用都是一次 CDP 往返！

1
has_data = device_page.wait_for_table_data(timeout=5000)  # 内部调用 get_table_row_count
2
row_count = device_page.get_table_row_count()              # 冗余调用
3
has_no_data_msg = device_page.has_no_data_message()        # 冗余调用

1
_DATA_ROW_CHECK = """Array.from(document.querySelectorAll('table tbody tr')).filter(row => {
2
    const text = row.textContent || '';
3
    if (/没有.*匹配|没有.*记录|暂无.*数据/.test(text)) return false;
4
    return Array.from(row.querySelectorAll('td')).some(cell => cell.textContent.trim());
5
}).length"""
6

7
def get_table_row_count(self) -> int:
8
    return self._page.evaluate(_DATA_ROW_CHECK)  # 一次 CDP 调用

1
def has_no_data_message(self) -> bool:
2
    return self._page.locator("text=没有找到匹配的记录").count() > 0

1
def wait_for_table_data(self, timeout: int = 5000) -> bool:
2
    try:
3
        self._page.wait_for_function(
4
            f"{_DATA_ROW_CHECK} > 0 || "
5
            "document.body.innerText.includes('没有找到匹配的记录')",
6
            timeout=timeout
7
        )
8
        return self._page.evaluate(f"{_DATA_ROW_CHECK} > 0")
9
    except Exception:
10
        return self._page.evaluate(f"{_DATA_ROW_CHECK} > 0")

1
with step(f"验证{btn_name}表格数据"):
2
    has_data = device_page.wait_for_table_data(timeout=5000)
3
    if allow_empty:
4
        if not has_data:
5
            step(f"{btn_name}表格数据为空（允许）")
6
    else:
7
        assert has_data, f"{btn_name}表格数据为空"