wehub-resource-sync
87 lines
2.7 KiB
Python
87 lines
2.7 KiB
Python
"""
|
|
Focused WeChat task tests.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import copy
|
|
import json
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
from bench_env.task.wechat import tasks as _tasks_module
|
|
from bench_env.tests.conftest import make_judge_input
|
|
from bench_env.tests.weather.test_tasks import TEST_OS_STATE
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[3]
|
|
DEFAULT_ROUTE = {"app": "launcher", "path": "/"}
|
|
|
|
|
|
def _load_json(*parts: str) -> dict[str, Any]:
|
|
return json.loads(ROOT.joinpath(*parts).read_text(encoding="utf-8"))
|
|
|
|
|
|
WECHAT_BASE_STATE = _load_json("apps", "Wechat", "data", "defaults.json")
|
|
|
|
|
|
def _make_input(
|
|
init_wechat: dict[str, Any],
|
|
curr_wechat: dict[str, Any],
|
|
):
|
|
os_state = {"time": copy.deepcopy(TEST_OS_STATE["time"]), "providers": {}}
|
|
return make_judge_input(
|
|
{"apps": {"wechat": init_wechat}, "os": os_state},
|
|
{"apps": {"wechat": curr_wechat}, "os": os_state},
|
|
route=DEFAULT_ROUTE,
|
|
)
|
|
|
|
|
|
def _append_wechat_moment(state: dict[str, Any], content: str, *, images: list[str] | None = None) -> None:
|
|
state["moments"].insert(
|
|
0,
|
|
{
|
|
"id": f"mo_test_{len(state['moments']) + 1}",
|
|
"wxid": state["user"]["wxid"],
|
|
"userName": state["user"]["name"],
|
|
"userAvatar": state["user"]["avatar"],
|
|
"content": content,
|
|
"timestamp": TEST_OS_STATE["time"]["timestamp"],
|
|
"images": list(images or []),
|
|
},
|
|
)
|
|
|
|
|
|
def test_scenic_photo_to_moments_with_phrase_positive():
|
|
task = _tasks_module.ScenicPhotoToMomentsWithPhrase(
|
|
time_hint="上周",
|
|
place_name="颐和园万寿山",
|
|
required_phrase="春天真好",
|
|
)
|
|
init_wechat = copy.deepcopy(WECHAT_BASE_STATE)
|
|
curr_wechat = copy.deepcopy(WECHAT_BASE_STATE)
|
|
_append_wechat_moment(
|
|
curr_wechat,
|
|
"上周拍的颐和园万寿山，春天真好",
|
|
images=[task.p._photo_path],
|
|
)
|
|
checks = task.check_goals(_make_input(init_wechat, curr_wechat))
|
|
assert all(check["passed"] for check in checks), checks
|
|
|
|
|
|
def test_scenic_photo_to_moments_with_phrase_negative():
|
|
task = _tasks_module.ScenicPhotoToMomentsWithPhrase(
|
|
time_hint="上周",
|
|
place_name="颐和园万寿山",
|
|
required_phrase="春天真好",
|
|
)
|
|
init_wechat = copy.deepcopy(WECHAT_BASE_STATE)
|
|
curr_wechat = copy.deepcopy(WECHAT_BASE_STATE)
|
|
_append_wechat_moment(
|
|
curr_wechat,
|
|
"上周拍的颐和园万寿山，春天真好",
|
|
images=["/sdcard/DCIM/Camera/IMG_other.jpg"],
|
|
)
|
|
checks = task.check_goals(_make_input(init_wechat, curr_wechat))
|
|
assert any(check["passed"] is False for check in checks), checks
|