import random import time from typing import Dict, List, Optional from bemani.common import Time from bemani.client.base import BaseClient from bemani.protocol import Node class ReflecBeatLimelight(BaseClient): NAME = "ＴＥＳＴ" def verify_log_pcb_status(self, loc: str) -> None: call = self.call_node() pcb = Node.void("log") pcb.set_attribute("method", "pcb_status") pcb.add_child(Node.string("lid", loc)) pcb.add_child(Node.s32("cnt", 0)) call.add_child(pcb) # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/log/@status") def verify_pcbinfo_get(self, loc: str) -> None: call = self.call_node() pcb = Node.void("pcbinfo") pcb.set_attribute("method", "get") pcb.add_child(Node.string("lid", loc)) call.add_child(pcb) # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/pcbinfo/info/name") self.assert_path(resp, "response/pcbinfo/info/pref") self.assert_path(resp, "response/pcbinfo/info/close") self.assert_path(resp, "response/pcbinfo/info/hour") self.assert_path(resp, "response/pcbinfo/info/min") def verify_sysinfo_get(self) -> None: call = self.call_node() info = Node.void("sysinfo") info.set_attribute("method", "get") call.add_child(info) # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/sysinfo/trd") def verify_ranking_read(self) -> None: call = self.call_node() ranking = Node.void("ranking") ranking.set_attribute("method", "read") call.add_child(ranking) # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/ranking/lic_10/time") self.assert_path(resp, "response/ranking/org_10/time") def verify_player_start(self, refid: str) -> None: call = self.call_node() player = Node.void("player") player.set_attribute("method", "start") player.add_child(Node.string("rid", refid)) player.add_child(Node.s32("ver", 0)) call.add_child(player) # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/player/is_suc") self.assert_path(resp, "response/player/unlock_music") self.assert_path(resp, "response/player/unlock_item") self.assert_path(resp, "response/player/item_lock_ctrl") self.assert_path(resp, "response/player/lincle_link_4/qpro") self.assert_path(resp, "response/player/lincle_link_4/glass") self.assert_path(resp, "response/player/lincle_link_4/treasure") self.assert_path(resp, "response/player/lincle_link_4/for_iidx_0_0") self.assert_path(resp, "response/player/lincle_link_4/for_iidx_0_1") self.assert_path(resp, "response/player/lincle_link_4/for_iidx_0_2") self.assert_path(resp, "response/player/lincle_link_4/for_iidx_0_3") self.assert_path(resp, "response/player/lincle_link_4/for_iidx_0_4") self.assert_path(resp, "response/player/lincle_link_4/for_iidx_0_5") self.assert_path(resp, "response/player/lincle_link_4/for_iidx_0_6") self.assert_path(resp, "response/player/lincle_link_4/for_iidx_0") self.assert_path(resp, "response/player/lincle_link_4/for_iidx_1") self.assert_path(resp, "response/player/lincle_link_4/for_iidx_2") self.assert_path(resp, "response/player/lincle_link_4/for_iidx_3") self.assert_path(resp, "response/player/lincle_link_4/for_iidx_4") self.assert_path(resp, "response/player/lincle_link_4/for_rb_0_0") self.assert_path(resp, "response/player/lincle_link_4/for_rb_0_1") self.assert_path(resp, "response/player/lincle_link_4/for_rb_0_2") self.assert_path(resp, "response/player/lincle_link_4/for_rb_0_3") self.assert_path(resp, "response/player/lincle_link_4/for_rb_0_4") self.assert_path(resp, "response/player/lincle_link_4/for_rb_0_5") self.assert_path(resp, "response/player/lincle_link_4/for_rb_0_6") self.assert_path(resp, "response/player/lincle_link_4/for_rb_0") self.assert_path(resp, "response/player/lincle_link_4/for_rb_1") self.assert_path(resp, "response/player/lincle_link_4/for_rb_2") self.assert_path(resp, "response/player/lincle_link_4/for_rb_3") self.assert_path(resp, "response/player/lincle_link_4/for_rb_4") self.assert_path(resp, "response/player/lincle_link_4/qproflg") self.assert_path(resp, "response/player/lincle_link_4/glassflg") self.assert_path(resp, "response/player/lincle_link_4/complete") def verify_player_delete(self, refid: str) -> None: call = self.call_node() player = Node.void("player") player.set_attribute("method", "delete") player.add_child(Node.string("rid", refid)) call.add_child(player) # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/player/@status") def verify_player_end(self, refid: str) -> None: call = self.call_node() player = Node.void("player") player.set_attribute("method", "end") player.add_child(Node.string("rid", refid)) player.add_child(Node.s32("status", 4)) call.add_child(player) # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/player") def verify_player_read(self, refid: str, location: str) -> List[Dict[str, int]]: call = self.call_node() player = Node.void("player") player.set_attribute("method", "read") player.add_child(Node.string("rid", refid)) player.add_child(Node.string("lid", location)) player.add_child(Node.s32("ver", 2)) call.add_child(player) # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/player/pdata/base/uid") self.assert_path(resp, "response/player/pdata/base/name") self.assert_path(resp, "response/player/pdata/base/icon_id") self.assert_path(resp, "response/player/pdata/base/lv") self.assert_path(resp, "response/player/pdata/base/exp") self.assert_path(resp, "response/player/pdata/base/mg") self.assert_path(resp, "response/player/pdata/base/ap") self.assert_path(resp, "response/player/pdata/base/pc") self.assert_path(resp, "response/player/pdata/base/uattr") self.assert_path(resp, "response/player/pdata/con/day") self.assert_path(resp, "response/player/pdata/con/cnt") self.assert_path(resp, "response/player/pdata/con/total_cnt") self.assert_path(resp, "response/player/pdata/con/last") self.assert_path(resp, "response/player/pdata/con/now") self.assert_path(resp, "response/player/pdata/team/id") self.assert_path(resp, "response/player/pdata/team/name") self.assert_path(resp, "response/player/pdata/custom/s_gls") self.assert_path(resp, "response/player/pdata/custom/bgm_m") self.assert_path(resp, "response/player/pdata/custom/st_f") self.assert_path(resp, "response/player/pdata/custom/st_bg") self.assert_path(resp, "response/player/pdata/custom/st_bg_b") self.assert_path(resp, "response/player/pdata/custom/eff_e") self.assert_path(resp, "response/player/pdata/custom/se_s") self.assert_path(resp, "response/player/pdata/custom/se_s_v") self.assert_path(resp, "response/player/pdata/custom/last_music_id") self.assert_path(resp, "response/player/pdata/custom/last_note_grade") self.assert_path(resp, "response/player/pdata/custom/sort_type") self.assert_path(resp, "response/player/pdata/custom/narrowdown_type") self.assert_path(resp, "response/player/pdata/custom/is_begginer") self.assert_path(resp, "response/player/pdata/custom/is_tut") self.assert_path(resp, "response/player/pdata/custom/symbol_chat_0") self.assert_path(resp, "response/player/pdata/custom/symbol_chat_1") self.assert_path(resp, "response/player/pdata/custom/gauge_style") self.assert_path(resp, "response/player/pdata/custom/obj_shade") self.assert_path(resp, "response/player/pdata/custom/obj_size") self.assert_path(resp, "response/player/pdata/custom/byword") self.assert_path(resp, "response/player/pdata/custom/is_auto_byword") self.assert_path(resp, "response/player/pdata/custom/is_tweet") self.assert_path(resp, "response/player/pdata/custom/is_link_twitter") self.assert_path(resp, "response/player/pdata/custom/mrec_type") self.assert_path(resp, "response/player/pdata/custom/card_disp_type") self.assert_path(resp, "response/player/pdata/custom/tab_sel") self.assert_path(resp, "response/player/pdata/custom/hidden_param") self.assert_path(resp, "response/player/pdata/released") self.assert_path(resp, "response/player/pdata/record") self.assert_path(resp, "response/player/pdata/cmnt") self.assert_path(resp, "response/player/pdata/rival") self.assert_path(resp, "response/player/pdata/glass") self.assert_path(resp, "response/player/pdata/fav_music_slot") self.assert_path(resp, "response/player/pdata/narrow_down/adv_param") if resp.child_value("player/pdata/base/name") != self.NAME: raise Exception(f'Invalid name {resp.child_value("player/pdata/base/name")} returned on profile read!') scores = [] for child in resp.child("player/pdata/record").children: if child.name != "rec": continue score = { "id": child.child_value("mid"), "chart": child.child_value("ng"), "clear_type": child.child_value("mrec_0/ct"), "achievement_rate": child.child_value("mrec_0/ar"), "score": child.child_value("mrec_0/bs"), "combo": child.child_value("mrec_0/mc"), "miss_count": child.child_value("mrec_0/bmc"), } scores.append(score) return scores def verify_player_write( self, refid: str, extid: int, loc: str, records: List[Dict[str, int]], scores: List[Dict[str, int]], ) -> int: call = self.call_node() player = Node.void("player") call.add_child(player) player.set_attribute("method", "write") player.add_child(Node.string("rid", refid)) player.add_child(Node.string("lid", loc)) player.add_child(Node.u64("begin_time", Time.now() * 1000)) player.add_child(Node.u64("end_time", Time.now() * 1000)) pdata = Node.void("pdata") player.add_child(pdata) base = Node.void("base") pdata.add_child(base) base.add_child(Node.s32("uid", extid)) base.add_child(Node.string("name", self.NAME)) base.add_child(Node.s16("icon_id", 0)) base.add_child(Node.s16("lv", 1)) base.add_child(Node.s32("exp", 0)) base.add_child(Node.s16("mg", 0)) base.add_child(Node.s16("ap", 0)) base.add_child(Node.s32("pc", 0)) base.add_child(Node.s32("uattr", 0)) con = Node.void("con") pdata.add_child(con) con.add_child(Node.s32("day", 0)) con.add_child(Node.s32("cnt", 0)) con.add_child(Node.s32("total_cnt", 0)) con.add_child(Node.s32("last", 0)) con.add_child(Node.s32("now", 0)) custom = Node.void("custom") pdata.add_child(custom) custom.add_child(Node.u8("s_gls", 0)) custom.add_child(Node.u8("bgm_m", 0)) custom.add_child(Node.u8("st_f", 0)) custom.add_child(Node.u8("st_bg", 0)) custom.add_child(Node.u8("st_bg_b", 100)) custom.add_child(Node.u8("eff_e", 0)) custom.add_child(Node.u8("se_s", 0)) custom.add_child(Node.u8("se_s_v", 100)) custom.add_child(Node.s16("last_music_id", 85)) custom.add_child(Node.u8("last_note_grade", 0)) custom.add_child(Node.u8("sort_type", 0)) custom.add_child(Node.u8("narrowdown_type", 0)) custom.add_child(Node.bool("is_begginer", False)) custom.add_child(Node.bool("is_tut", False)) custom.add_child(Node.s16_array("symbol_chat_0", [0, 1, 2, 3, 4, 5])) custom.add_child(Node.s16_array("symbol_chat_1", [0, 1, 2, 3, 4, 5])) custom.add_child(Node.u8("gauge_style", 0)) custom.add_child(Node.u8("obj_shade", 0)) custom.add_child(Node.u8("obj_size", 0)) custom.add_child(Node.s16_array("byword", [0, 0])) custom.add_child(Node.bool_array("is_auto_byword", [True, True])) custom.add_child(Node.bool("is_tweet", False)) custom.add_child(Node.bool("is_link_twitter", False)) custom.add_child(Node.s16("mrec_type", 0)) custom.add_child(Node.s16("card_disp_type", 0)) custom.add_child(Node.s16("tab_sel", 0)) custom.add_child( Node.s32_array( "hidden_param", [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 1, 0, 0, 0, 0], ) ) pdata.add_child(Node.void("released")) pdata.add_child(Node.void("rival")) pdata.add_child(Node.void("glass")) pdata.add_child(Node.void("fav_music_slot")) lincle_link_4 = Node.void("lincle_link_4") pdata.add_child(lincle_link_4) lincle_link_4.add_child(Node.u32("qpro_add", 0)) lincle_link_4.add_child(Node.u32("glass_add", 0)) lincle_link_4.add_child(Node.bool("for_iidx_0_0", False)) lincle_link_4.add_child(Node.bool("for_iidx_0_1", False)) lincle_link_4.add_child(Node.bool("for_iidx_0_2", False)) lincle_link_4.add_child(Node.bool("for_iidx_0_3", False)) lincle_link_4.add_child(Node.bool("for_iidx_0_4", False)) lincle_link_4.add_child(Node.bool("for_iidx_0_5", False)) lincle_link_4.add_child(Node.bool("for_iidx_0_6", False)) lincle_link_4.add_child(Node.bool("for_iidx_0", False)) lincle_link_4.add_child(Node.bool("for_iidx_1", False)) lincle_link_4.add_child(Node.bool("for_iidx_2", False)) lincle_link_4.add_child(Node.bool("for_iidx_3", False)) lincle_link_4.add_child(Node.bool("for_iidx_4", False)) lincle_link_4.add_child(Node.bool("for_rb_0_0", False)) lincle_link_4.add_child(Node.bool("for_rb_0_1", False)) lincle_link_4.add_child(Node.bool("for_rb_0_2", False)) lincle_link_4.add_child(Node.bool("for_rb_0_3", False)) lincle_link_4.add_child(Node.bool("for_rb_0_4", False)) lincle_link_4.add_child(Node.bool("for_rb_0_5", False)) lincle_link_4.add_child(Node.bool("for_rb_0_6", False)) lincle_link_4.add_child(Node.bool("for_rb_0", False)) lincle_link_4.add_child(Node.bool("for_rb_1", False)) lincle_link_4.add_child(Node.bool("for_rb_2", False)) lincle_link_4.add_child(Node.bool("for_rb_3", False)) lincle_link_4.add_child(Node.bool("for_rb_4", False)) # First, filter down to only records that are also in the battle log def key(thing: Dict[str, int]) -> str: return f'{thing["id"]}-{thing["chart"]}' updates = [key(score) for score in scores] sortedrecords = {key(record): record for record in records if key(record) in updates} # Now, see what records need updating and update them for score in scores: if key(score) in sortedrecords: # Had a record, need to merge record = sortedrecords[key(score)] else: # First time playing record = { "clear_type": 0, "achievement_rate": 0, "score": 0, "combo": 0, "miss_count": 999999999, } sortedrecords[key(score)] = { "id": score["id"], "chart": score["chart"], "clear_type": max(record["clear_type"], score["clear_type"]), "achievement_rate": max(record["achievement_rate"], score["achievement_rate"]), "score": max(record["score"], score["score"]), "combo": max(record["combo"], score["combo"]), "miss_count": min(record["miss_count"], score["miss_count"]), } # Finally, send the records and battle logs recordnode = Node.void("record") pdata.add_child(recordnode) blog = Node.void("blog") pdata.add_child(blog) for _, record in sortedrecords.items(): rec = Node.void("rec") recordnode.add_child(rec) rec.add_child(Node.u16("mid", record["id"])) rec.add_child(Node.u8("ng", record["chart"])) rec.add_child(Node.s32("point", 2)) rec.add_child(Node.s32("played_time", Time.now())) mrec_0 = Node.void("mrec_0") rec.add_child(mrec_0) mrec_0.add_child(Node.s32("win", 1)) mrec_0.add_child(Node.s32("lose", 0)) mrec_0.add_child(Node.s32("draw", 0)) mrec_0.add_child(Node.u8("ct", record["clear_type"])) mrec_0.add_child(Node.s16("ar", record["achievement_rate"])) mrec_0.add_child(Node.s32("bs", record["score"])) mrec_0.add_child(Node.s16("mc", record["combo"])) mrec_0.add_child(Node.s16("bmc", record["miss_count"])) mrec_1 = Node.void("mrec_1") rec.add_child(mrec_1) mrec_1.add_child(Node.s32("win", 0)) mrec_1.add_child(Node.s32("lose", 0)) mrec_1.add_child(Node.s32("draw", 0)) mrec_1.add_child(Node.u8("ct", 0)) mrec_1.add_child(Node.s16("ar", 0)) mrec_1.add_child(Node.s32("bs", 0)) mrec_1.add_child(Node.s16("mc", 0)) mrec_1.add_child(Node.s16("bmc", -1)) scoreid = 0 for score in scores: log = Node.void("log") blog.add_child(log) log.add_child(Node.u8("id", scoreid)) log.add_child(Node.u16("mid", score["id"])) log.add_child(Node.u8("ng", score["chart"])) log.add_child(Node.u8("mt", 0)) log.add_child(Node.u8("rt", 0)) log.add_child(Node.s32("ruid", 0)) myself = Node.void("myself") log.add_child(myself) myself.add_child(Node.s16("mg", 0)) myself.add_child(Node.s16("ap", 0)) myself.add_child(Node.u8("ct", score["clear_type"])) myself.add_child(Node.s32("s", score["score"])) myself.add_child(Node.s16("ar", score["achievement_rate"])) rival = Node.void("rival") log.add_child(rival) rival.add_child(Node.s16("mg", 0)) rival.add_child(Node.s16("ap", 0)) rival.add_child(Node.u8("ct", 2)) rival.add_child(Node.s32("s", 177)) rival.add_child(Node.s16("ar", 500)) log.add_child(Node.s32("time", Time.now())) scoreid = scoreid + 1 # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/player/uid") self.assert_path(resp, "response/player/time") return resp.child_value("player/uid") def verify_log_play(self, extid: int, loc: str, scores: List[Dict[str, int]]) -> None: call = self.call_node() log = Node.void("log") call.add_child(log) log.set_attribute("method", "play") log.add_child(Node.s32("uid", extid)) log.add_child(Node.string("lid", loc)) play = Node.void("play") log.add_child(play) play.add_child(Node.s16("stage", len(scores))) play.add_child(Node.s32("sec", 700)) scoreid = 0 for score in scores: rec = Node.void("rec") log.add_child(rec) rec.add_child(Node.s16("idx", scoreid)) rec.add_child(Node.s16("mid", score["id"])) rec.add_child(Node.s16("grade", score["chart"])) rec.add_child(Node.s16("color", 0)) rec.add_child(Node.s16("match", 0)) rec.add_child(Node.s16("res", 0)) rec.add_child(Node.s32("score", score["score"])) rec.add_child(Node.s16("mc", score["combo"])) rec.add_child(Node.s16("jt_jr", 0)) rec.add_child(Node.s16("jt_ju", 0)) rec.add_child(Node.s16("jt_gr", 0)) rec.add_child(Node.s16("jt_gd", 0)) rec.add_child(Node.s16("jt_ms", score["miss_count"])) rec.add_child(Node.s32("sec", 200)) scoreid = scoreid + 1 # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/log/@status") def verify_lobby_read(self, location: str, extid: int) -> None: call = self.call_node() lobby = Node.void("lobby") lobby.set_attribute("method", "read") lobby.add_child(Node.s32("uid", extid)) lobby.add_child(Node.u8("m_grade", 255)) lobby.add_child(Node.string("lid", location)) lobby.add_child(Node.s32("max", 128)) lobby.add_child(Node.s32_array("friend", [])) call.add_child(lobby) # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/lobby/interval") self.assert_path(resp, "response/lobby/interval_p") def verify_lobby_entry(self, location: str, extid: int) -> int: call = self.call_node() lobby = Node.void("lobby") lobby.set_attribute("method", "entry") e = Node.void("e") lobby.add_child(e) e.add_child(Node.s32("eid", 0)) e.add_child(Node.u16("mid", 79)) e.add_child(Node.u8("ng", 0)) e.add_child(Node.s32("uid", extid)) e.add_child(Node.s32("uattr", 0)) e.add_child(Node.string("pn", self.NAME)) e.add_child(Node.s16("mg", 0)) e.add_child(Node.s32("mopt", 0)) e.add_child(Node.s32("tid", 0)) e.add_child(Node.string("tn", "")) e.add_child(Node.s32("topt", 0)) e.add_child(Node.string("lid", location)) e.add_child(Node.string("sn", "")) e.add_child(Node.u8("pref", 51)) e.add_child(Node.s8("stg", 0)) e.add_child(Node.s8("pside", 0)) e.add_child(Node.s16("eatime", 30)) e.add_child(Node.u8_array("ga", [127, 0, 0, 1])) e.add_child(Node.u16("gp", 10007)) e.add_child(Node.u8_array("la", [16, 0, 0, 0])) lobby.add_child(Node.s32_array("friend", [])) call.add_child(lobby) # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/lobby/eid") self.assert_path(resp, "response/lobby/interval") self.assert_path(resp, "response/lobby/interval_p") self.assert_path(resp, "response/lobby/e/eid") self.assert_path(resp, "response/lobby/e/mid") self.assert_path(resp, "response/lobby/e/ng") self.assert_path(resp, "response/lobby/e/uid") self.assert_path(resp, "response/lobby/e/pn") self.assert_path(resp, "response/lobby/e/uattr") self.assert_path(resp, "response/lobby/e/mopt") self.assert_path(resp, "response/lobby/e/mg") self.assert_path(resp, "response/lobby/e/tid") self.assert_path(resp, "response/lobby/e/tn") self.assert_path(resp, "response/lobby/e/topt") self.assert_path(resp, "response/lobby/e/lid") self.assert_path(resp, "response/lobby/e/sn") self.assert_path(resp, "response/lobby/e/pref") self.assert_path(resp, "response/lobby/e/stg") self.assert_path(resp, "response/lobby/e/pside") self.assert_path(resp, "response/lobby/e/eatime") self.assert_path(resp, "response/lobby/e/ga") self.assert_path(resp, "response/lobby/e/gp") self.assert_path(resp, "response/lobby/e/la") return resp.child_value("lobby/eid") def verify_lobby_delete(self, eid: int) -> None: call = self.call_node() lobby = Node.void("lobby") lobby.set_attribute("method", "delete") lobby.add_child(Node.s32("eid", eid)) call.add_child(lobby) # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/lobby") def verify_event_w_update_status(self, loc: str, extid: int) -> None: call = self.call_node() event_w = Node.void("event_w") call.add_child(event_w) event_w.set_attribute("method", "update_status") event_w.add_child(Node.s32("uid", extid)) event_w.add_child(Node.string("p_name", self.NAME)) event_w.add_child(Node.s32("exp", 0)) event_w.add_child(Node.s32("customize", 0)) event_w.add_child(Node.s32("tid", 0)) event_w.add_child(Node.string("t_name", "")) event_w.add_child(Node.string("lid", loc)) event_w.add_child(Node.string("s_name", "")) event_w.add_child(Node.s8("pref", 51)) event_w.add_child(Node.s32("time", Time.now())) event_w.add_child(Node.s8("status", 1)) event_w.add_child(Node.s8("stage", 0)) event_w.add_child(Node.s32("mid", -1)) event_w.add_child(Node.s8("ng", -1)) # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/event_w/@status") def verify_event_w_add_comment(self, loc: str, extid: int) -> None: call = self.call_node() event_w = Node.void("event_w") call.add_child(event_w) event_w.set_attribute("method", "add_comment") event_w.add_child(Node.s32("uid", extid)) event_w.add_child(Node.string("p_name", self.NAME)) event_w.add_child(Node.s32("exp", 0)) event_w.add_child(Node.s32("customize", 0)) event_w.add_child(Node.s32("tid", 0)) event_w.add_child(Node.string("t_name", "")) event_w.add_child(Node.string("lid", loc)) event_w.add_child(Node.string("s_name", "")) event_w.add_child(Node.s8("pref", 51)) event_w.add_child(Node.s32("time", Time.now())) event_w.add_child(Node.string("comment", "アメ〜〜！")) event_w.add_child(Node.bool("is_tweet", False)) # Swap with server resp = self.exchange("", call) # Verify that response is correct self.assert_path(resp, "response/event_w/@status") def verify_event_r_get_all(self, extid: int) -> None: call = self.call_node() event_r = Node.void("event_r") call.add_child(event_r) event_r.set_attribute("method", "get_all") event_r.add_child(Node.s32("uid", extid)) event_r.add_child(Node.s32("time", 0)) event_r.add_child(Node.s32("limit", 30)) # Swap with server resp = self.exchange("", call) # Verify that response is correct. We should have at least one # comment (the one we just wrote) and one status (because we just # called update_status). self.assert_path(resp, "response/event_r/time") self.assert_path(resp, "response/event_r/status/s/uid") self.assert_path(resp, "response/event_r/status/s/p_name") self.assert_path(resp, "response/event_r/status/s/exp") self.assert_path(resp, "response/event_r/status/s/customize") self.assert_path(resp, "response/event_r/status/s/tid") self.assert_path(resp, "response/event_r/status/s/t_name") self.assert_path(resp, "response/event_r/status/s/lid") self.assert_path(resp, "response/event_r/status/s/s_name") self.assert_path(resp, "response/event_r/status/s/pref") self.assert_path(resp, "response/event_r/status/s/time") self.assert_path(resp, "response/event_r/status/s/status") self.assert_path(resp, "response/event_r/status/s/stage") self.assert_path(resp, "response/event_r/status/s/mid") self.assert_path(resp, "response/event_r/status/s/ng") self.assert_path(resp, "response/event_r/comment/c/uid") self.assert_path(resp, "response/event_r/comment/c/p_name") self.assert_path(resp, "response/event_r/comment/c/exp") self.assert_path(resp, "response/event_r/comment/c/customize") self.assert_path(resp, "response/event_r/comment/c/tid") self.assert_path(resp, "response/event_r/comment/c/t_name") self.assert_path(resp, "response/event_r/comment/c/lid") self.assert_path(resp, "response/event_r/comment/c/s_name") self.assert_path(resp, "response/event_r/comment/c/pref") self.assert_path(resp, "response/event_r/comment/c/time") self.assert_path(resp, "response/event_r/comment/c/comment") self.assert_path(resp, "response/event_r/comment/c/is_tweet") # Verify we posted our comment earlier found = False for child in resp.child("event_r/comment").children: if child.name != "c": continue if child.child_value("uid") == extid: name = child.child_value("p_name") comment = child.child_value("comment") if name != self.NAME: raise Exception(f"Invalid name '{name}' returned for comment!") if comment != "アメ〜〜！": raise Exception(f"Invalid comment '{comment}' returned for comment!") found = True if not found: raise Exception("Comment we posted was not found!") # Verify our status came through found = False for child in resp.child("event_r/status").children: if child.name != "s": continue if child.child_value("uid") == extid: name = child.child_value("p_name") if name != self.NAME: raise Exception(f"Invalid name '{name}' returned for status!") found = True if not found: raise Exception("Status was not found!") def verify(self, cardid: Optional[str]) -> None: # Verify boot sequence is okay self.verify_services_get( expected_services=[ "pcbtracker", "pcbevent", "local", "message", "facility", "cardmng", "package", "posevent", "pkglist", "dlstatus", "eacoin", "lobby", "ntp", "keepalive", ] ) paseli_enabled = self.verify_pcbtracker_alive() self.verify_message_get() self.verify_package_list() location = self.verify_facility_get() self.verify_pcbevent_put() self.verify_log_pcb_status(location) self.verify_pcbinfo_get(location) self.verify_sysinfo_get() self.verify_ranking_read() # Verify card registration and profile lookup if cardid is not None: card = cardid else: card = self.random_card() print(f"Generated random card ID {card} for use.") if cardid is None: self.verify_cardmng_inquire(card, msg_type="unregistered", paseli_enabled=paseli_enabled) ref_id = self.verify_cardmng_getrefid(card) if len(ref_id) != 16: raise Exception(f"Invalid refid '{ref_id}' returned when registering card") if ref_id != self.verify_cardmng_inquire(card, msg_type="new", paseli_enabled=paseli_enabled): raise Exception(f"Invalid refid '{ref_id}' returned when querying card") # Always get a player start, regardless of new profile or not self.verify_player_start(ref_id) self.verify_player_delete(ref_id) extid = self.verify_player_write( ref_id, 0, location, [], [], ) else: print("Skipping new card checks for existing card") ref_id = self.verify_cardmng_inquire(card, msg_type="query", paseli_enabled=paseli_enabled) # Verify pin handling and return card handling self.verify_cardmng_authpass(ref_id, correct=True) self.verify_cardmng_authpass(ref_id, correct=False) if ref_id != self.verify_cardmng_inquire(card, msg_type="query", paseli_enabled=paseli_enabled): raise Exception(f"Invalid refid '{ref_id}' returned when querying card") # Verify lobby functionality self.verify_lobby_read(location, extid) eid = self.verify_lobby_entry(location, extid) self.verify_lobby_delete(eid) # Verify status updates and puzzle comments self.verify_event_w_update_status(location, extid) self.verify_event_w_add_comment(location, extid) self.verify_event_r_get_all(extid) # Limelight is weird and sends only the top record for each song you played, # and then a separate battle log. So, emulating that is kinda hard. scores: List[Dict[str, int]] = [] if cardid is None: # Verify score saving and updating for phase in [1, 2]: if phase == 1: dummyscores = [ # An okay score on a chart { "id": 1, "chart": 1, "clear_type": 2, "achievement_rate": 7543, "score": 432, "combo": 123, "miss_count": 5, }, # A good score on an easier chart of the same song { "id": 1, "chart": 0, "clear_type": 3, "achievement_rate": 9876, "score": 543, "combo": 543, "miss_count": 0, }, # A bad score on a hard chart { "id": 3, "chart": 2, "clear_type": 2, "achievement_rate": 1234, "score": 123, "combo": 42, "miss_count": 54, }, # A terrible score on an easy chart { "id": 3, "chart": 0, "clear_type": 2, "achievement_rate": 1024, "score": 50, "combo": 12, "miss_count": 90, }, ] if phase == 2: dummyscores = [ # A better score on the same chart { "id": 1, "chart": 1, "clear_type": 3, "achievement_rate": 8765, "score": 469, "combo": 468, "miss_count": 1, }, # A worse score on another same chart { "id": 1, "chart": 0, "clear_type": 2, "achievement_rate": 8765, "score": 432, "combo": 321, "miss_count": 15, "expected_score": 543, "expected_clear_type": 3, "expected_achievement_rate": 9876, "expected_combo": 543, "expected_miss_count": 0, }, ] self.verify_player_write(ref_id, extid, location, scores, dummyscores) self.verify_log_play(extid, location, dummyscores) scores = self.verify_player_read(ref_id, location) for expected in dummyscores: actual = None for received in scores: if received["id"] == expected["id"] and received["chart"] == expected["chart"]: actual = received break if actual is None: raise Exception(f"Didn't find song {expected['id']} chart {expected['chart']} in response!") if "expected_score" in expected: expected_score = expected["expected_score"] else: expected_score = expected["score"] if "expected_achievement_rate" in expected: expected_achievement_rate = expected["expected_achievement_rate"] else: expected_achievement_rate = expected["achievement_rate"] if "expected_clear_type" in expected: expected_clear_type = expected["expected_clear_type"] else: expected_clear_type = expected["clear_type"] if "expected_combo" in expected: expected_combo = expected["expected_combo"] else: expected_combo = expected["combo"] if "expected_miss_count" in expected: expected_miss_count = expected["expected_miss_count"] else: expected_miss_count = expected["miss_count"] if actual["score"] != expected_score: raise Exception( f'Expected a score of \'{expected_score}\' for song \'{expected["id"]}\' chart \'{expected["chart"]}\' but got score \'{actual["score"]}\'' ) if actual["achievement_rate"] != expected_achievement_rate: raise Exception( f'Expected an achievement rate of \'{expected_achievement_rate}\' for song \'{expected["id"]}\' chart \'{expected["chart"]}\' but got achievement rate \'{actual["achievement_rate"]}\'' ) if actual["clear_type"] != expected_clear_type: raise Exception( f'Expected a clear_type of \'{expected_clear_type}\' for song \'{expected["id"]}\' chart \'{expected["chart"]}\' but got clear_type \'{actual["clear_type"]}\'' ) if actual["combo"] != expected_combo: raise Exception( f'Expected a combo of \'{expected_combo}\' for song \'{expected["id"]}\' chart \'{expected["chart"]}\' but got combo \'{actual["combo"]}\'' ) if actual["miss_count"] != expected_miss_count: raise Exception( f'Expected a miss count of \'{expected_miss_count}\' for song \'{expected["id"]}\' chart \'{expected["chart"]}\' but got miss count \'{actual["miss_count"]}\'' ) # Sleep so we don't end up putting in score history on the same second time.sleep(1) else: print("Skipping score checks for existing card") # Verify ending game self.verify_player_end(ref_id) # Verify paseli handling if paseli_enabled: print("PASELI enabled for this PCBID, executing PASELI checks") else: print("PASELI disabled for this PCBID, skipping PASELI checks") return sessid, balance = self.verify_eacoin_checkin(card) if balance == 0: print("Skipping PASELI consume check because card has 0 balance") else: self.verify_eacoin_consume(sessid, balance, random.randint(0, balance)) self.verify_eacoin_checkout(sessid)