234 lines
8.3 KiB
Python
234 lines
8.3 KiB
Python
"""uiautomator2 采购演练 Adapter 测试；不连接真实手机。"""
|
|
|
|
import unittest
|
|
|
|
from src.pdd_device_service import PddDeviceService
|
|
from src.pdd_purchase_adapter import PddPurchaseError
|
|
from src.pdd_u2_purchase_adapter import U2PddPurchaseAdapter
|
|
from src.pdd_u2_purchase_adapter import U2PddLivePurchaseAdapter
|
|
|
|
|
|
GOODS_URL = "https://mobile.yangkeduo.com/goods.html?goods_id=753136429979"
|
|
|
|
|
|
def home_xml() -> str:
|
|
return """<hierarchy>
|
|
<node package="com.xunmeng.pinduoduo" bounds="[0,0][1080,2376]">
|
|
<node clickable="true" enabled="true" visible-to-user="true"
|
|
bounds="[500,2100][1080,2328]">
|
|
<node text="立即购买 ¥5.03" enabled="true" visible-to-user="true"
|
|
bounds="[600,2150][1000,2280]"/>
|
|
</node>
|
|
</node>
|
|
</hierarchy>"""
|
|
|
|
|
|
def panel_xml(quantity: int = 1) -> str:
|
|
return f"""<hierarchy>
|
|
<node package="com.xunmeng.pinduoduo" bounds="[0,0][1080,2376]">
|
|
<node text="已选： 黑色 3XL【140-165斤】" bounds="[300,500][1000,560]"/>
|
|
<node content-desc="黑色" selected="true" bounds="[30,900][330,1000]"/>
|
|
<node text="3XL【140-165斤】" selected="true"
|
|
bounds="[40,1200][500,1300]"/>
|
|
<node text="折后¥5.03" bounds="[300,400][600,470]"/>
|
|
<node class="android.widget.EditText" text="{quantity}"
|
|
bounds="[480,700][560,780]"/>
|
|
<node content-desc="减少数量" clickable="true" bounds="[390,700][470,780]"/>
|
|
<node content-desc="增加数量" clickable="true" bounds="[570,700][650,780]"/>
|
|
<node clickable="true" bounds="[0,2181][1080,2328]">
|
|
<node text="提交订单 ¥5.03" enabled="true" visible-to-user="true"
|
|
bounds="[380,2220][700,2290]"/>
|
|
</node>
|
|
</node>
|
|
</hierarchy>"""
|
|
|
|
|
|
class FakeEditor:
|
|
def __init__(self, device) -> None:
|
|
self.device = device
|
|
self.count = 1
|
|
|
|
def set_text(self, value: str) -> None:
|
|
self.device.quantity = int(value)
|
|
|
|
|
|
class FakeDevice:
|
|
def __init__(self, initial_xml: str = "") -> None:
|
|
self.mode = "special" if initial_xml else "home"
|
|
self.special_xml = initial_xml
|
|
self.quantity = 1
|
|
self.clicks = []
|
|
self.opened_urls = []
|
|
|
|
def app_current(self):
|
|
return {"package": "com.xunmeng.pinduoduo"}
|
|
|
|
def app_start(self, _package):
|
|
pass
|
|
|
|
def app_wait(self, _package, timeout=10):
|
|
return True
|
|
|
|
def open_url(self, url):
|
|
self.opened_urls.append(url)
|
|
|
|
def dump_hierarchy(self):
|
|
if self.mode == "special":
|
|
return self.special_xml
|
|
return home_xml() if self.mode == "home" else panel_xml(self.quantity)
|
|
|
|
def click(self, x, y):
|
|
self.clicks.append((x, y))
|
|
self.mode = "panel"
|
|
|
|
def __call__(self, **selector):
|
|
if selector == {"className": "android.widget.EditText"}:
|
|
return FakeEditor(self)
|
|
raise AssertionError(f"未预期的 selector: {selector}")
|
|
|
|
|
|
class U2PddPurchaseAdapterTest(unittest.TestCase):
|
|
def _adapter(self, device, calls):
|
|
def select_color_fn(_device, _xml, target, **_kwargs):
|
|
calls.append(("color", target))
|
|
return target == "黑色"
|
|
|
|
def select_size_fn(_device, _xml, target, **_kwargs):
|
|
calls.append(("size", target))
|
|
return target == "3XL【140-165斤】"
|
|
|
|
return U2PddPurchaseAdapter(
|
|
"USB-001",
|
|
device_service=PddDeviceService(connector=lambda _serial: device),
|
|
sleeper=lambda _seconds: None,
|
|
select_color_fn=select_color_fn,
|
|
select_size_fn=select_size_fn,
|
|
)
|
|
|
|
def _live_adapter(self, device, calls):
|
|
def select_color_fn(_device, _xml, target, **_kwargs):
|
|
calls.append(("color", target))
|
|
return target == "黑色"
|
|
|
|
def select_size_fn(_device, _xml, target, **_kwargs):
|
|
calls.append(("size", target))
|
|
return target == "3XL【140-165斤】"
|
|
|
|
return U2PddLivePurchaseAdapter(
|
|
"USB-001",
|
|
device_service=PddDeviceService(connector=lambda _serial: device),
|
|
sleeper=lambda _seconds: None,
|
|
select_color_fn=select_color_fn,
|
|
select_size_fn=select_size_fn,
|
|
)
|
|
|
|
def test_dry_run_reaches_confirmation_without_clicking_submit(self):
|
|
device = FakeDevice()
|
|
calls = []
|
|
adapter = self._adapter(device, calls)
|
|
|
|
adapter.open_goods(GOODS_URL)
|
|
goods = adapter.read_state()
|
|
adapter.select_options(
|
|
{"color": "黑色", "size": "3XL【140-165斤】"}
|
|
)
|
|
adapter.set_quantity(2)
|
|
confirmation = adapter.read_state()
|
|
adapter.enter_confirmation()
|
|
adapter.stop_before_submit()
|
|
adapter.close()
|
|
|
|
self.assertEqual(goods.page_kind, "goods")
|
|
self.assertEqual(goods.goods_id, "753136429979")
|
|
self.assertEqual(
|
|
confirmation.selected_options,
|
|
{"color": "黑色", "size": "3XL【140-165斤】"},
|
|
)
|
|
self.assertEqual(confirmation.quantity, 2)
|
|
self.assertEqual(confirmation.price_cent, 503)
|
|
self.assertEqual(confirmation.page_kind, "order_confirmation")
|
|
self.assertEqual(calls, [("color", "黑色"), ("size", "3XL【140-165斤】")])
|
|
# 只点击一次商品页采购入口；最终“提交订单”从不点击。
|
|
self.assertEqual(len(device.clicks), 1)
|
|
|
|
def test_unsupported_dynamic_option_stops_before_click(self):
|
|
device = FakeDevice()
|
|
adapter = self._adapter(device, [])
|
|
adapter.open_goods(GOODS_URL)
|
|
|
|
with self.assertRaises(PddPurchaseError) as raised:
|
|
adapter.select_options({"bundle": "标准版"})
|
|
|
|
self.assertEqual(raised.exception.code, "PURCHASE_OPTIONS_UNSUPPORTED")
|
|
self.assertEqual(device.clicks, [])
|
|
adapter.close()
|
|
|
|
def test_captcha_stops_without_waiting_or_clicking(self):
|
|
device = FakeDevice(
|
|
'<hierarchy><node package="com.xunmeng.pinduoduo" '
|
|
'text="请完成验证" bounds="[0,0][1080,2376]"/></hierarchy>'
|
|
)
|
|
adapter = self._adapter(device, [])
|
|
|
|
with self.assertRaises(PddPurchaseError) as raised:
|
|
adapter.open_goods(GOODS_URL)
|
|
|
|
self.assertEqual(raised.exception.code, "PDD_PAGE_CAPTCHA")
|
|
self.assertEqual(device.clicks, [])
|
|
adapter.close()
|
|
|
|
def test_invalid_non_pdd_url_is_rejected_before_connect(self):
|
|
device = FakeDevice()
|
|
adapter = self._adapter(device, [])
|
|
|
|
with self.assertRaises(PddPurchaseError) as raised:
|
|
adapter.open_goods("https://example.test/goods?goods_id=1")
|
|
|
|
self.assertEqual(raised.exception.code, "PURCHASE_GOODS_URL_INVALID")
|
|
self.assertEqual(device.opened_urls, [])
|
|
|
|
def test_live_adapter_clicks_unique_submit_target_only_once(self):
|
|
device = FakeDevice()
|
|
adapter = self._live_adapter(device, [])
|
|
adapter.open_goods(GOODS_URL)
|
|
adapter.select_options(
|
|
{"color": "黑色", "size": "3XL【140-165斤】"}
|
|
)
|
|
adapter.set_quantity(2)
|
|
|
|
state = adapter.read_state()
|
|
adapter.submit_order_once()
|
|
with self.assertRaises(PddPurchaseError) as raised:
|
|
adapter.submit_order_once()
|
|
|
|
self.assertEqual(state.submit_candidate_count, 1)
|
|
self.assertEqual(raised.exception.code, "PURCHASE_SUBMIT_ALREADY_ATTEMPTED")
|
|
# 第一次点击采购入口，第二次且仅一次点击最终提交。
|
|
self.assertEqual(len(device.clicks), 2)
|
|
adapter.close()
|
|
|
|
def test_live_adapter_rejects_multiple_submit_targets_without_click(self):
|
|
device = FakeDevice()
|
|
adapter = self._live_adapter(device, [])
|
|
adapter.open_goods(GOODS_URL)
|
|
adapter.select_options(
|
|
{"color": "黑色", "size": "3XL【140-165斤】"}
|
|
)
|
|
device.special_xml = panel_xml().replace(
|
|
"</hierarchy>",
|
|
'<node text="确认购买" enabled="true" visible-to-user="true" '
|
|
'bounds="[50,2200][300,2300]"/></hierarchy>',
|
|
)
|
|
device.mode = "special"
|
|
|
|
with self.assertRaises(PddPurchaseError) as raised:
|
|
adapter.submit_order_once()
|
|
|
|
self.assertEqual(raised.exception.code, "PURCHASE_SUBMIT_TARGET_AMBIGUOUS")
|
|
self.assertEqual(len(device.clicks), 1)
|
|
adapter.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|