119 lines
8.1 KiB
Python
119 lines
8.1 KiB
Python
"""主库使用 sw8/sw9，平台写入使用独立本地 SDK 夹具；核对验密、重提、解绑与新增渠道。"""
|
|
|
|
import argparse
|
|
import json
|
|
import time
|
|
from pathlib import Path
|
|
|
|
import pymysql
|
|
|
|
from sw_api_compare import compare_data
|
|
from sw_write_verify import Rehearsal
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
parser.add_argument("--runtime", type=Path, required=True)
|
|
parser.add_argument("--config", type=Path, default=Path("configs/config.yaml"))
|
|
parser.add_argument("--old-url", default="http://127.0.0.1:18081")
|
|
parser.add_argument("--new-url", default="http://127.0.0.1:19002")
|
|
parser.add_argument("--output", type=Path, required=True)
|
|
args = parser.parse_args()
|
|
run = Rehearsal(args)
|
|
fixture = json.loads((args.runtime / "sdk-fixture/fixture.json").read_text())
|
|
owner = fixture["owner_id"]
|
|
local = pymysql.connect(host="127.0.0.1", port=13309, user="root", autocommit=True, cursorclass=pymysql.cursors.DictCursor)
|
|
channel = "migchannel" + str(int(time.time()))
|
|
|
|
def query(sql, params=()):
|
|
with local.cursor() as cursor:
|
|
cursor.execute(sql, params)
|
|
return cursor.fetchall()
|
|
|
|
def api_path(side, path):
|
|
return ("/system/user" if side == "legacy" else "/sw/business") + path
|
|
|
|
def bind(side, password=None, success=True, identity="testlevel3"):
|
|
return run.call(side, identity, "POST", api_path(side, "/bindPlatform"),
|
|
{"sourceDb": "lx", "account": fixture["account"], "password": password or fixture["password"]}, success=success)
|
|
|
|
def snapshot(label, status):
|
|
values = []
|
|
for side, database in (("legacy", "sw8"), ("new", "sw9")):
|
|
rows = run.query(f"SELECT user_id AS userId,source_db AS sourceDb,platform_busier_id AS platformBusierId,platform_busier_account AS platformBusierAccount,status,reject_reason AS rejectReason,review_account AS reviewAccount FROM {database}.sys_user_bind_platform WHERE source_db='lx' AND platform_busier_id=%s", (owner,))
|
|
run.check(label + " 行数和状态 " + side, len(rows) == 1 and rows[0]["status"] == status)
|
|
values.append(rows)
|
|
compared = compare_data(*values, {}, run.mapping)
|
|
run.check(label + " 字段一致", compared["semantic_equal"], differences=compared["differences"])
|
|
|
|
try:
|
|
for database in ("sw_test_sdk_old", "sw_test_sdk_new"):
|
|
for table in ("tab_promote", "tab_promote_settlement_time"):
|
|
run.check("本地平台写入夹具为空 " + database + "." + table, query(f"SELECT COUNT(*) n FROM {database}.{table}")[0]["n"] == 0)
|
|
query(f"ALTER TABLE {database}.{table} AUTO_INCREMENT=1")
|
|
for side, database in (("legacy", "sw8"), ("new", "sw9")):
|
|
run.check("夹具绑定键未占用 " + side, not run.query(f"SELECT user_id FROM {database}.sys_user_bind_platform WHERE source_db='lx' AND platform_busier_id=%s", (owner,)))
|
|
denied = bind(side, password="invalid-password", success=False)
|
|
run.check("错误密码不允许绑定 " + side, denied["code"] != (200 if side == "legacy" else 0))
|
|
denied = bind(side, success=False, identity="testlevel1")
|
|
run.check("仅三级允许绑定 " + side, denied["code"] != (200 if side == "legacy" else 0))
|
|
started = time.monotonic()
|
|
for side in ("legacy", "new"):
|
|
bind(side)
|
|
snapshot("首次绑定待审核", "3")
|
|
for side in ("legacy", "new"):
|
|
denied = bind(side, success=False)
|
|
run.check("待审核重复提交被拒绝 " + side, denied["code"] != (200 if side == "legacy" else 0))
|
|
# 仅在主库夹具模拟 OA 的审核结果，不调用外部 OA 写接口。
|
|
for database in ("sw8", "sw9"):
|
|
run.query(f"UPDATE {database}.sys_user_bind_platform SET status='2',reject_reason='测试驳回',review_account='migration-test',review_time=NOW() WHERE source_db='lx' AND platform_busier_id=%s", (owner,))
|
|
time.sleep(max(0, 5.2 - (time.monotonic() - started)))
|
|
for side in ("legacy", "new"):
|
|
bind(side)
|
|
snapshot("驳回后重提并清空旧审核信息", "3")
|
|
for database in ("sw8", "sw9"):
|
|
run.query(f"UPDATE {database}.sys_user_bind_platform SET status='0',review_account='migration-test',review_time=NOW() WHERE source_db='lx' AND platform_busier_id=%s", (owner,))
|
|
snapshot("审核通过绑定", "0")
|
|
payload = {"sourceDb": "lx", "platformBusierId": owner, "account": channel, "password": "Channel168", "registerType": 0}
|
|
for side in ("legacy", "new"):
|
|
denied = run.call(side, "testlevel1", "POST", run.prefix(side) + "/promote", payload, success=False)
|
|
run.check("保留原一级角色的新增渠道权限限制 " + side, denied["code"] != (200 if side == "legacy" else 0))
|
|
run.call(side, "admin", "POST", run.prefix(side) + "/promote", payload)
|
|
saved, settlements = [], []
|
|
for side, database in (("legacy", "sw_test_sdk_old"), ("new", "sw_test_sdk_new")):
|
|
rows = query(f"SELECT id,account,password,status,busier_id,register_type,promote_level,parent_id,top_promote_id,game_ids2 FROM {database}.tab_promote WHERE account=%s", (channel,))
|
|
run.check("源平台渠道只创建一条 " + side, len(rows) == 1)
|
|
saved.append(rows[0])
|
|
settlement = query(f"SELECT * FROM {database}.tab_promote_settlement_time WHERE promote_id=%s", (rows[0]["id"],))
|
|
run.check("七日结算及时间关系 " + side, len(settlement) == 1 and settlement[0]["day_period"] == 7 and settlement[0]["time_type"] == 0 and settlement[0]["next_count_time"] - settlement[0]["create_time"] == 7 * 86400)
|
|
relation = query(f"SELECT promote_ids FROM {database}.tab_promote_business WHERE id=%s", (owner,))[0]["promote_ids"]
|
|
run.check("平台商务渠道关系已追加 " + side, str(rows[0]["id"]) in (relation or "").split(","))
|
|
run.check("新渠道属性及兼容密码完全一致", saved[0] == saved[1] and saved[0]["status"] == -3 and saved[0]["promote_level"] == 1)
|
|
for side, database in (("legacy", "sw8"), ("new", "sw9")):
|
|
denied = run.call(side, "admin", "POST", run.prefix(side) + "/promote", payload, success=False)
|
|
run.check("重复渠道账号被拒绝 " + side, denied["code"] != (200 if side == "legacy" else 0))
|
|
run.call(side, "testlevel3", "DELETE", api_path(side, "/bindPlatform"), {"sourceDb": "lx", "platformBusierId": owner})
|
|
run.check("解绑只删除当前绑定 " + side, not run.query(f"SELECT user_id FROM {database}.sys_user_bind_platform WHERE user_id=%s AND source_db='lx' AND platform_busier_id=%s", (run.owner(side, 3), owner)))
|
|
except Exception as error:
|
|
if all(check["passed"] for check in run.checks):
|
|
run.check("SDK 验证执行异常", False, error_type=type(error).__name__)
|
|
raise
|
|
finally:
|
|
for side, database in (("legacy", "sw8"), ("new", "sw9")):
|
|
run.query(f"DELETE FROM {database}.sys_user_bind_platform WHERE user_id=%s AND source_db='lx' AND platform_busier_id=%s AND platform_busier_account=%s", (run.owner(side, 3), owner, fixture["account"]))
|
|
run.query("DELETE FROM sw9.sys_business_binding_locks WHERE source_db='lx' AND platform_busier_id=%s", (owner,))
|
|
for database in ("sw_test_sdk_old", "sw_test_sdk_new"):
|
|
rows = query(f"SELECT id FROM {database}.tab_promote WHERE account=%s", (channel,))
|
|
for row in rows:
|
|
query(f"DELETE FROM {database}.tab_promote_settlement_time WHERE promote_id=%s", (row["id"],))
|
|
query(f"DELETE FROM {database}.tab_promote WHERE account=%s", (channel,))
|
|
query(f"UPDATE {database}.tab_promote_business SET promote_ids='' WHERE id=%s AND account=%s", (owner, fixture["account"]))
|
|
remaining = sum(run.query(f"SELECT COUNT(*) n FROM {db}.sys_user_bind_platform WHERE source_db='lx' AND platform_busier_id=%s", (owner,))[0]["n"] for db in ("sw8", "sw9"))
|
|
run.check("主库 SDK 绑定夹具已清理", remaining == 0)
|
|
local.close()
|
|
run.db.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|