guke
3dca126411
微信优惠券增加美团/京东中间页，领券前获取微信头像授权（主功能） --------- Co-authored-by: guke <guke@autohome.com.cn> Reviewed-on: #75 Co-authored-by: guke <guke@wonderable.ai> Co-committed-by: guke <guke@wonderable.ai>
88 lines
3.6 KiB
Python
88 lines
3.6 KiB
Python
"""美团/京东 CPS 落地页「领券前授权头像」插页(/c/{code})。
|
|
|
|
验证:微信内未授权头像 → 出插页(snsapi_userinfo 按钮 + 「直接领券」出口);
|
|
已授权(wx_uinfo cookie) / 非微信 → 直接 302 跳 target。
|
|
插页只渲染 HTML、不真调微信,故无需 mock 微信网络。
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from app.core.config import settings
|
|
from app.db.session import SessionLocal
|
|
from app.repositories import cps_link as cps_link_repo
|
|
|
|
_WECHAT_UA = "Mozilla/5.0 (iPhone; CPU iPhone OS 16_0 like Mac OS X) MicroMessenger/8.0.40"
|
|
_OTHER_UA = "Mozilla/5.0 (iPhone; CPU iPhone OS 16_0 like Mac OS X) Safari/605.1"
|
|
|
|
|
|
@pytest.fixture()
|
|
def _wx_oauth_on(monkeypatch):
|
|
"""打开微信网页授权(插页只在 settings.wx_oauth_active 时出)。"""
|
|
monkeypatch.setattr(settings, "WX_MP_APPID", "wxtestappid")
|
|
monkeypatch.setattr(settings, "WX_MP_SECRET", "wxtestsecret")
|
|
monkeypatch.setattr(settings, "WX_MP_OAUTH_ENABLED", True)
|
|
monkeypatch.setattr(settings, "CPS_REDIRECT_BASE", "https://coupon.example.com")
|
|
|
|
|
|
def _make_link(platform: str, target: str) -> tuple[str, str]:
|
|
with SessionLocal() as db:
|
|
link = cps_link_repo.create_link(
|
|
db, group_id=1, activity_id=1,
|
|
sid="g1" if platform == "meituan" else None,
|
|
target_url=target, platform=platform,
|
|
)
|
|
return link.code, link.target_url
|
|
|
|
|
|
@pytest.mark.parametrize("platform,target", [
|
|
("meituan", "https://dpurl.cn/mt-abc"),
|
|
("jd", "https://u.jd.com/jd-xyz"),
|
|
])
|
|
def test_wechat_unauthorized_shows_interstitial(client, _wx_oauth_on, platform, target):
|
|
"""微信内 + 有 openid + 无 wx_uinfo → 200 插页:userinfo 授权按钮 + 直接领券出口。"""
|
|
code, tgt = _make_link(platform, target)
|
|
r = client.get(
|
|
f"/c/{code}",
|
|
headers={"user-agent": _WECHAT_UA, "cookie": "wx_openid=o_test"},
|
|
follow_redirects=False,
|
|
)
|
|
assert r.status_code == 200, r.text
|
|
body = r.text
|
|
assert "snsapi_userinfo" in body # 唯一按钮走 userinfo 授权(交互手势触发)
|
|
assert code in body # 授权 state=uinfo:{code}
|
|
assert tgt in body # 取消授权时前端转跳的目标券页(绝不阻断领券)
|
|
assert "暂不授权" not in body # 已去掉单独的"暂不授权"出口,改为取消即自动转跳
|
|
|
|
|
|
def test_authorized_redirects_to_target(client, _wx_oauth_on):
|
|
"""已授权头像(wx_uinfo=1)→ 不再出插页,直接 302 跳 target。"""
|
|
code, tgt = _make_link("meituan", "https://dpurl.cn/mt-ok")
|
|
r = client.get(
|
|
f"/c/{code}",
|
|
headers={"user-agent": _WECHAT_UA, "cookie": "wx_openid=o_test; wx_uinfo=1"},
|
|
follow_redirects=False,
|
|
)
|
|
assert r.status_code == 302
|
|
assert r.headers["location"] == tgt
|
|
|
|
|
|
def test_non_wechat_redirects_to_target(client, _wx_oauth_on):
|
|
"""非微信浏览器(拿不了授权)→ 直接 302 跳 target,不出插页。"""
|
|
code, tgt = _make_link("meituan", "https://dpurl.cn/mt-other")
|
|
r = client.get(
|
|
f"/c/{code}",
|
|
headers={"user-agent": _OTHER_UA, "cookie": "wx_openid=o_test"},
|
|
follow_redirects=False,
|
|
)
|
|
assert r.status_code == 302
|
|
assert r.headers["location"] == tgt
|
|
|
|
|
|
def test_oauth_cb_cancel_no_code_redirects_back(client):
|
|
"""授权弹窗点「取消」、微信带空 code 回调 → 兜底 302 回落地页(不报 422)。
|
|
落地页再由前端转跳目标券页,保证拒绝也能拿到券。"""
|
|
r = client.get("/wx/oauth/cb?state=uinfo:ABC123", follow_redirects=False)
|
|
assert r.status_code == 302
|
|
assert r.headers["location"] == "/c/ABC123"
|