import random import time from typing import Any, Dict, List, Optional from bemani.client.base import BaseClient from bemani.protocol import Node class ReflecBeatVolzza2(BaseClient): NAME = 'TEST' def verify_pcb_rb5_pcb_boot(self, loc: str) -> None: call = self.call_node() pcb = Node.void('pcb') pcb.set_attribute('method', 'rb5_pcb_boot') pcb.add_child(Node.string('lid', loc)) pcb.add_child(Node.string('rno', 'MBR-JA-C01')) call.add_child(pcb) # Swap with server resp = self.exchange('', call) # Verify that response is correct self.assert_path(resp, "response/pcb/sinfo/nm") self.assert_path(resp, "response/pcb/sinfo/cl_enbl") self.assert_path(resp, "response/pcb/sinfo/cl_h") self.assert_path(resp, "response/pcb/sinfo/cl_m") self.assert_path(resp, "response/pcb/sinfo/shop_flag") def verify_pcb_rb5_pcb_error(self, loc: str) -> None: call = self.call_node() pcb = Node.void('pcb') call.add_child(pcb) pcb.set_attribute('method', 'rb5_pcb_error') pcb.add_child(Node.string('lid', loc)) pcb.add_child(Node.string('code', 'exception')) pcb.add_child(Node.string('msg', 'exceptionstring')) # Swap with server resp = self.exchange('', call) # Verify that response is correct self.assert_path(resp, "response/pcb/@status") def verify_info_rb5_info_read(self, loc: str) -> None: call = self.call_node() info = Node.void('info') call.add_child(info) info.set_attribute('method', 'rb5_info_read') info.add_child(Node.string('lid', loc)) # Swap with server resp = self.exchange('', call) # Verify that response is correct self.assert_path(resp, "response/info/event_ctrl") self.assert_path(resp, "response/info/item_lock_ctrl") self.assert_path(resp, "response/info/mycourse_ctrl") def verify_info_rb5_info_read_shop_ranking(self, loc: str) -> None: call = self.call_node() info = Node.void('info') call.add_child(info) info.set_attribute('method', 'rb5_info_read_shop_ranking') # Arbitrarily chosen based on the song IDs we send in the # score section below. info.add_child(Node.s16('min', 1)) info.add_child(Node.s16('max', 10)) info.add_child(Node.string('lid', loc)) # Swap with server resp = self.exchange('', call) # Verify that response is correct self.assert_path(resp, "response/info/shop_score/time") self.assert_path(resp, "response/info/shop_score/data/rank") self.assert_path(resp, "response/info/shop_score/data/music_id") self.assert_path(resp, "response/info/shop_score/data/note_grade") self.assert_path(resp, "response/info/shop_score/data/clear_type") self.assert_path(resp, "response/info/shop_score/data/user_id") self.assert_path(resp, "response/info/shop_score/data/icon_id") self.assert_path(resp, "response/info/shop_score/data/score") self.assert_path(resp, "response/info/shop_score/data/time") self.assert_path(resp, "response/info/shop_score/data/name") def verify_info_rb5_info_read_hit_chart(self) -> None: call = self.call_node() info = Node.void('info') info.set_attribute('method', 'rb5_info_read_hit_chart') info.add_child(Node.s32('ver', 0)) call.add_child(info) # Swap with server resp = self.exchange('', call) # Verify that response is correct self.assert_path(resp, "response/info/ver") self.assert_path(resp, "response/info/ranking/weekly/bt") self.assert_path(resp, "response/info/ranking/weekly/et") self.assert_path(resp, "response/info/ranking/weekly/new/d/mid") self.assert_path(resp, "response/info/ranking/weekly/new/d/cnt") self.assert_path(resp, "response/info/ranking/monthly/bt") self.assert_path(resp, "response/info/ranking/monthly/et") self.assert_path(resp, "response/info/ranking/monthly/new/d/mid") self.assert_path(resp, "response/info/ranking/monthly/new/d/cnt") self.assert_path(resp, "response/info/ranking/total/bt") self.assert_path(resp, "response/info/ranking/total/et") self.assert_path(resp, "response/info/ranking/total/new/d/mid") self.assert_path(resp, "response/info/ranking/total/new/d/cnt") def verify_player_rb5_player_start(self, refid: str) -> None: call = self.call_node() player = Node.void('player') player.set_attribute('method', 'rb5_player_start') player.add_child(Node.string('rid', refid)) player.add_child(Node.u8_array('ga', [127, 0, 0, 1])) player.add_child(Node.u16('gp', 10573)) player.add_child(Node.u8_array('la', [16, 0, 0, 0])) player.add_child(Node.u8_array('pnid', [39, 16, 0, 0, 0, 23, 62, 60, 39, 127, 0, 0, 1, 23, 62, 60])) call.add_child(player) # Swap with server resp = self.exchange('', call) # Verify that response is correct self.assert_path(resp, "response/player/plyid") self.assert_path(resp, "response/player/start_time") self.assert_path(resp, "response/player/event_ctrl") self.assert_path(resp, "response/player/item_lock_ctrl") self.assert_path(resp, "response/player/mycourse_ctrl") def verify_player_rb5_player_end(self, refid: str) -> None: call = self.call_node() player = Node.void('player') player.set_attribute('method', 'rb5_player_end') 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") def verify_player_rb5_player_read_rank_5(self) -> None: call = self.call_node() player = Node.void('player') player.set_attribute('method', 'rb5_player_read_rank_5') player.add_child(Node.s32('uid', 0)) player.add_child(Node.s32_array('sc', [897, 897, 0, 0, 0])) player.add_child(Node.s8('mg_id', 0)) player.add_child(Node.s32('mg_sc', 220)) call.add_child(player) # Swap with server resp = self.exchange('', call) # Verify that response is correct self.assert_path(resp, "response/player/tbs/new_rank") self.assert_path(resp, "response/player/tbs/old_rank") self.assert_path(resp, "response/player/mng/new_rank") self.assert_path(resp, "response/player/mng/old_rank") def verify_player_rb5_player_read_rival_score_5(self, extid: int) -> None: call = self.call_node() player = Node.void('player') player.set_attribute('method', 'rb5_player_read_rival_score_5') player.add_child(Node.s32('uid', extid)) player.add_child(Node.s32('music_id', 6)) player.add_child(Node.s8('note_grade', 0)) call.add_child(player) # Swap with server resp = self.exchange('', call) # Verify that response is correct self.assert_path(resp, "response/player/@status") # Verify that we got a score if the extid is nonzero if extid != 0: self.assert_path(resp, "response/player/player_select_score/user_id") self.assert_path(resp, "response/player/player_select_score/name") self.assert_path(resp, "response/player/player_select_score/m_score") self.assert_path(resp, "response/player/player_select_score/m_scoreTime") self.assert_path(resp, "response/player/player_select_score/m_iconID") if resp.child_value('player/player_select_score/name') != self.NAME: raise Exception( f'Invalid name {resp.child_value("player/player_select_score/name")} returned on score read!' ) if resp.child_value('player/player_select_score/user_id') != extid: raise Exception( f'Invalid name {resp.child_value("player/player_select_score/user_id")} returned on score read!' ) def verify_player_rb5_player_read_rival_ranking_data_5(self, extid: int) -> None: call = self.call_node() player = Node.void('player') player.set_attribute('method', 'rb5_player_read_rival_ranking_data_5') player.add_child(Node.s32('uid', extid)) player.add_child(Node.s8('mgid', -1)) call.add_child(player) # Swap with server resp = self.exchange('', call) # Verify that response is correct self.assert_path(resp, "response/player/rival_data/rl/uid") self.assert_path(resp, "response/player/rival_data/rl/nm") self.assert_path(resp, "response/player/rival_data/rl/ic") self.assert_path(resp, "response/player/rival_data/rl/sl/mid") self.assert_path(resp, "response/player/rival_data/rl/sl/m") self.assert_path(resp, "response/player/rival_data/rl/sl/t") if resp.child_value('player/rival_data/rl/nm') != self.NAME: raise Exception( f'Invalid name {resp.child_value("player/rival_data/rl/nm")} returned on rival ranking read!' ) def verify_player_rb5_player_succeed(self, refid: str) -> None: call = self.call_node() player = Node.void('player') player.set_attribute('method', 'rb5_player_succeed') 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/name") self.assert_path(resp, "response/player/grd") self.assert_path(resp, "response/player/ap") self.assert_path(resp, "response/player/uattr") def verify_player_rb5_player_read(self, refid: str, cardid: str, location: str) -> Dict[str, Any]: call = self.call_node() player = Node.void('player') player.set_attribute('method', 'rb5_player_read') player.add_child(Node.string('rid', refid)) player.add_child(Node.string('lid', location)) player.add_child(Node.s16('ver', 0)) player.add_child(Node.string('card_id', cardid)) player.add_child(Node.s16('card_type', 1)) call.add_child(player) # Swap with server resp = self.exchange('', call) # Verify that response is correct self.assert_path(resp, "response/player/pdata/account/usrid") self.assert_path(resp, "response/player/pdata/account/tpc") self.assert_path(resp, "response/player/pdata/account/dpc") self.assert_path(resp, "response/player/pdata/account/crd") self.assert_path(resp, "response/player/pdata/account/brd") self.assert_path(resp, "response/player/pdata/account/tdc") self.assert_path(resp, "response/player/pdata/account/intrvld") self.assert_path(resp, "response/player/pdata/account/ver") self.assert_path(resp, "response/player/pdata/account/succeed") self.assert_path(resp, "response/player/pdata/account/pst") self.assert_path(resp, "response/player/pdata/account/st") self.assert_path(resp, "response/player/pdata/account/opc") self.assert_path(resp, "response/player/pdata/account/lpc") self.assert_path(resp, "response/player/pdata/account/cpc") self.assert_path(resp, "response/player/pdata/account/mpc") self.assert_path(resp, "response/player/pdata/base/name") 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/cmnt") self.assert_path(resp, "response/player/pdata/base/uattr") self.assert_path(resp, "response/player/pdata/base/money") self.assert_path(resp, "response/player/pdata/base/tbs_5") self.assert_path(resp, "response/player/pdata/base/tbgs_5") self.assert_path(resp, "response/player/pdata/base/mlog") self.assert_path(resp, "response/player/pdata/base/class") self.assert_path(resp, "response/player/pdata/base/class_ar") self.assert_path(resp, "response/player/pdata/base/skill_point") self.assert_path(resp, "response/player/pdata/base/meteor_flg") self.assert_path(resp, "response/player/pdata/rival") self.assert_path(resp, "response/player/pdata/config") self.assert_path(resp, "response/player/pdata/custom") self.assert_path(resp, "response/player/pdata/released") self.assert_path(resp, "response/player/pdata/announce") self.assert_path(resp, "response/player/pdata/dojo") self.assert_path(resp, "response/player/pdata/player_param") self.assert_path(resp, "response/player/pdata/shop_score") self.assert_path(resp, "response/player/pdata/minigame") self.assert_path(resp, "response/player/pdata/derby/is_open") self.assert_path(resp, "response/player/pdata/mylist/list/idx") self.assert_path(resp, "response/player/pdata/mylist/list/mlst") self.assert_path(resp, "response/player/pdata/mycourse/mycourse_id") self.assert_path(resp, "response/player/pdata/mycourse_f") 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!') mycourse = [ { 'music_id': resp.child_value('player/pdata/mycourse/music_id_1'), 'note_grade': resp.child_value('player/pdata/mycourse/note_grade_1'), 'score': resp.child_value('player/pdata/mycourse/score_1'), }, { 'music_id': resp.child_value('player/pdata/mycourse/music_id_2'), 'note_grade': resp.child_value('player/pdata/mycourse/note_grade_2'), 'score': resp.child_value('player/pdata/mycourse/score_2'), }, { 'music_id': resp.child_value('player/pdata/mycourse/music_id_3'), 'note_grade': resp.child_value('player/pdata/mycourse/note_grade_3'), 'score': resp.child_value('player/pdata/mycourse/score_3'), }, { 'music_id': resp.child_value('player/pdata/mycourse/music_id_4'), 'note_grade': resp.child_value('player/pdata/mycourse/note_grade_4'), 'score': resp.child_value('player/pdata/mycourse/score_4'), }, ] return { 'mycourse': mycourse, } def verify_player_rb5_player_read_score_5(self, refid: str, location: str) -> List[Dict[str, int]]: call = self.call_node() player = Node.void('player') call.add_child(player) player.set_attribute('method', 'rb5_player_read_score_5') player.add_child(Node.string('rid', refid)) player.add_child(Node.s16('ver', 1)) # Swap with server resp = self.exchange('', call) scores = [] for child in resp.child('player/pdata/record').children: if child.name != 'rec': continue self.assert_path(child, 'rec/mid') self.assert_path(child, 'rec/ntgrd') self.assert_path(child, 'rec/pc') self.assert_path(child, 'rec/ct') self.assert_path(child, 'rec/ar') self.assert_path(child, 'rec/scr') self.assert_path(child, 'rec/ms') self.assert_path(child, 'rec/param') self.assert_path(child, 'rec/bscrt') self.assert_path(child, 'rec/bart') self.assert_path(child, 'rec/bctt') self.assert_path(child, 'rec/bmst') self.assert_path(child, 'rec/time') self.assert_path(child, 'rec/k_flag') score = { 'id': child.child_value('mid'), 'chart': child.child_value('ntgrd'), 'clear_type': child.child_value('ct'), 'combo_type': child.child_value('param'), 'achievement_rate': child.child_value('ar'), 'score': child.child_value('scr'), 'miss_count': child.child_value('ms'), } scores.append(score) return scores def verify_player_rb5_player_read_score_old_5(self, refid: str, location: str) -> List[Dict[str, int]]: call = self.call_node() player = Node.void('player') call.add_child(player) player.set_attribute('method', 'rb5_player_read_score_old_5') player.add_child(Node.string('rid', refid)) player.add_child(Node.s16('ver', 1)) # Swap with server resp = self.exchange('', call) scores = [] for child in resp.child('player/pdata/record_old').children: if child.name != 'rec': continue self.assert_path(child, 'rec/mid') self.assert_path(child, 'rec/ntgrd') self.assert_path(child, 'rec/pc') self.assert_path(child, 'rec/ct') self.assert_path(child, 'rec/ar') self.assert_path(child, 'rec/scr') self.assert_path(child, 'rec/ms') self.assert_path(child, 'rec/param') self.assert_path(child, 'rec/bscrt') self.assert_path(child, 'rec/bart') self.assert_path(child, 'rec/bctt') self.assert_path(child, 'rec/bmst') self.assert_path(child, 'rec/time') self.assert_path(child, 'rec/k_flag') score = { 'id': child.child_value('mid'), 'chart': child.child_value('ntgrd'), 'clear_type': child.child_value('ct'), 'combo_type': child.child_value('param'), 'achievement_rate': child.child_value('ar'), 'score': child.child_value('scr'), 'miss_count': child.child_value('ms'), } scores.append(score) return scores def verify_player_rb5_player_write_5( self, refid: str, loc: str, scores: List[Dict[str, int]]=[], rivals: List[int]=[], mycourse: Optional[List[Dict[str, int]]]=None, ) -> int: call = self.call_node() player = Node.void('player') call.add_child(player) player.set_attribute('method', 'rb5_player_write_5') pdata = Node.void('pdata') player.add_child(pdata) account = Node.void('account') pdata.add_child(account) account.add_child(Node.s32('usrid', 0)) account.add_child(Node.s32('plyid', 0)) account.add_child(Node.s32('tpc', 1)) account.add_child(Node.s32('dpc', 1)) account.add_child(Node.s32('crd', 1)) account.add_child(Node.s32('brd', 1)) account.add_child(Node.s32('tdc', 1)) account.add_child(Node.string('rid', refid)) account.add_child(Node.string('lid', loc)) account.add_child(Node.u8('wmode', 0)) account.add_child(Node.u8('gmode', 0)) account.add_child(Node.s16('ver', 0)) account.add_child(Node.bool('pp', False)) account.add_child(Node.bool('ps', False)) account.add_child(Node.bool('continue', False)) account.add_child(Node.bool('firstfree', False)) account.add_child(Node.s16('pay', 0)) account.add_child(Node.s16('pay_pc', 0)) account.add_child(Node.u64('st', int(time.time() * 1000))) base = Node.void('base') pdata.add_child(base) base.add_child(Node.string('name', self.NAME)) base.add_child(Node.s32('mg', 0)) base.add_child(Node.s32('ap', 0)) base.add_child(Node.s32('uattr', 0)) base.add_child(Node.s32('money', 0)) base.add_child(Node.bool('is_tut', False)) base.add_child(Node.s32('class', -1)) base.add_child(Node.s32('class_ar', 0)) base.add_child(Node.s32('skill_point', 0)) stglog = Node.void('stglog') pdata.add_child(stglog) index = 0 for score in scores: log = Node.void('log') stglog.add_child(log) log.add_child(Node.s8('stg', index)) log.add_child(Node.s16('mid', score['id'])) log.add_child(Node.s8('ng', score['chart'])) log.add_child(Node.s8('col', 1)) log.add_child(Node.s8('mt', 0)) log.add_child(Node.s8('rt', 0)) log.add_child(Node.s8('ct', score['clear_type'])) log.add_child(Node.s16('param', score['combo_type'])) log.add_child(Node.s16('grd', 0)) log.add_child(Node.s16('ar', score['achievement_rate'])) log.add_child(Node.s16('sc', score['score'])) log.add_child(Node.s16('jt_jst', 0)) log.add_child(Node.s16('jt_grt', 0)) log.add_child(Node.s16('jt_gd', 0)) log.add_child(Node.s16('jt_ms', score['miss_count'])) log.add_child(Node.s16('jt_jr', 0)) log.add_child(Node.s32('r_uid', 0)) log.add_child(Node.s32('r_plyid', 0)) log.add_child(Node.s8('r_stg', 0)) log.add_child(Node.s8('r_ct', -1)) log.add_child(Node.s16('r_sc', 0)) log.add_child(Node.s16('r_grd', 0)) log.add_child(Node.s16('r_ar', 0)) log.add_child(Node.s8('r_cpuid', -1)) log.add_child(Node.s32('time', int(time.time()))) log.add_child(Node.s8('decide', 0)) log.add_child(Node.s16('g_gauge', 5000)) log.add_child(Node.s32('k_flag', 0)) index = index + 1 rivalnode = Node.void('rival') pdata.add_child(rivalnode) for rival in rivals: r = Node.void('r') rivalnode.add_child(r) r.add_child(Node.s32('id', rival)) r.add_child(Node.s8('regist_type', 3)) if mycourse is not None: mycoursenode = Node.void('mycourse') pdata.add_child(mycoursenode) mycoursenode.add_child(Node.s16('mycourse_id', 1)) mycoursenode.add_child(Node.s32('music_id_1', mycourse[0]['music_id'])) mycoursenode.add_child(Node.s16('note_grade_1', mycourse[0]['note_grade'])) mycoursenode.add_child(Node.s32('score_1', mycourse[0]['score'])) mycoursenode.add_child(Node.s32('music_id_2', mycourse[1]['music_id'])) mycoursenode.add_child(Node.s16('note_grade_2', mycourse[1]['note_grade'])) mycoursenode.add_child(Node.s32('score_2', mycourse[1]['score'])) mycoursenode.add_child(Node.s32('music_id_3', mycourse[2]['music_id'])) mycoursenode.add_child(Node.s16('note_grade_3', mycourse[2]['note_grade'])) mycoursenode.add_child(Node.s32('score_3', mycourse[2]['score'])) mycoursenode.add_child(Node.s32('music_id_4', mycourse[3]['music_id'])) mycoursenode.add_child(Node.s16('note_grade_4', mycourse[3]['note_grade'])) mycoursenode.add_child(Node.s32('score_4', mycourse[3]['score'])) mycoursenode.add_child(Node.s32('insert_time', -1)) mycoursenode.add_child(Node.s32('def_music_id_1', -1)) mycoursenode.add_child(Node.s16('def_note_grade_1', -1)) mycoursenode.add_child(Node.s32('def_music_id_2', -1)) mycoursenode.add_child(Node.s16('def_note_grade_2', -1)) mycoursenode.add_child(Node.s32('def_music_id_3', -1)) mycoursenode.add_child(Node.s16('def_note_grade_3', -1)) mycoursenode.add_child(Node.s32('def_music_id_4', -1)) mycoursenode.add_child(Node.s16('def_note_grade_4', -1)) # Swap with server resp = self.exchange('', call) # Verify that response is correct self.assert_path(resp, "response/player/uid") return resp.child_value('player/uid') def verify_lobby_rb5_lobby_read(self, location: str, extid: int) -> None: call = self.call_node() lobby = Node.void('lobby') lobby.set_attribute('method', 'rb5_lobby_read') lobby.add_child(Node.s32('uid', extid)) lobby.add_child(Node.s32('plyid', 0)) 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', [])) lobby.add_child(Node.u8('var', 4)) 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_rb5_lobby_entry(self, location: str, extid: int) -> int: call = self.call_node() lobby = Node.void('lobby') lobby.set_attribute('method', 'rb5_lobby_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.s32('plyid', 0)) e.add_child(Node.s16('mg', 255)) e.add_child(Node.s32('mopt', 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', 4)) 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])) e.add_child(Node.u8('ver', 2)) 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") self.assert_path(resp, "response/lobby/eid") 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/uattr") self.assert_path(resp, "response/lobby/e/pn") self.assert_path(resp, "response/lobby/e/plyid") self.assert_path(resp, "response/lobby/e/mg") self.assert_path(resp, "response/lobby/e/mopt") 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") self.assert_path(resp, "response/lobby/e/ver") return resp.child_value('lobby/eid') def verify_lobby_rb5_lobby_delete_entry(self, eid: int) -> None: call = self.call_node() lobby = Node.void('lobby') lobby.set_attribute('method', 'rb5_lobby_delete_entry') 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/@status") 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', 'lobby2', 'ntp', 'keepalive' ] ) paseli_enabled = self.verify_pcbtracker_alive() self.verify_message_get() self.verify_package_list() self.verify_dlstatus_progress() location = self.verify_facility_get() self.verify_pcbevent_put() self.verify_info_rb5_info_read(location) self.verify_pcb_rb5_pcb_boot(location) self.verify_pcb_rb5_pcb_error(location) # 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_rb5_player_start(ref_id) self.verify_player_rb5_player_succeed(ref_id) extid = self.verify_player_rb5_player_write_5( ref_id, 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_rb5_lobby_read(location, extid) eid = self.verify_lobby_rb5_lobby_entry(location, extid) self.verify_lobby_rb5_lobby_delete_entry(eid) # Verify we start with empty scores scores = self.verify_player_rb5_player_read_score_5(ref_id, location) oldscores = self.verify_player_rb5_player_read_score_old_5(ref_id, location) if len(scores) > 0 or len(oldscores) > 0: raise Exception('Existing scores returned on new card?') profile = self.verify_player_rb5_player_read(ref_id, card, location) for val in profile['mycourse']: if val['score'] != -1: raise Exception('Existing score returned for new card on course?') if val['music_id'] != -1: raise Exception('Existing score returned for new card on course?') if val['note_grade'] != -1: raise Exception('Existing score returned for new card on course?') 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': 9, 'chart': 1, 'clear_type': 9, 'combo_type': 0, 'achievement_rate': 7543, 'score': 432, 'miss_count': 5, }, # A good score on an easier chart of the same song { 'id': 9, 'chart': 0, 'clear_type': 9, 'combo_type': 1, 'achievement_rate': 9876, 'score': 543, 'miss_count': 0, }, # A bad score on a hard chart { 'id': 6, 'chart': 2, 'clear_type': 9, 'combo_type': 0, 'achievement_rate': 1234, 'score': 123, 'miss_count': 54, }, # A terrible score on an easy chart { 'id': 6, 'chart': 0, 'clear_type': 9, 'combo_type': 0, 'achievement_rate': 1024, 'score': 50, 'miss_count': 90, }, ] if phase == 2: dummyscores = [ # A better score on the same chart { 'id': 9, 'chart': 1, 'clear_type': 9, 'combo_type': 0, 'achievement_rate': 8765, 'score': 469, 'miss_count': 1, }, # A worse score on another same chart { 'id': 9, 'chart': 0, 'clear_type': 9, 'combo_type': 0, 'achievement_rate': 8765, 'score': 432, 'miss_count': 15, 'expected_score': 543, 'expected_clear_type': 9, 'expected_combo_type': 1, 'expected_achievement_rate': 9876, 'expected_miss_count': 0, }, ] self.verify_player_rb5_player_write_5(ref_id, location, scores=dummyscores) self.verify_player_rb5_player_read(ref_id, card, location) scores = self.verify_player_rb5_player_read_score_5(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_type' in expected: expected_combo_type = expected['expected_combo_type'] else: expected_combo_type = expected['combo_type'] 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_type'] != expected_combo_type: raise Exception(f'Expected a combo_type of \'{expected_combo_type}\' for song \'{expected["id"]}\' chart \'{expected["chart"]}\' but got combo_type \'{actual["combo_type"]}\'') 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 mycourse saving and updating firstcourse = [ { 'music_id': 51, 'note_grade': 1, 'score': 1086, }, { 'music_id': 37, 'note_grade': 1, 'score': 1041, }, { 'music_id': 102, 'note_grade': 1, 'score': 935, }, { 'music_id': 58, 'note_grade': 1, 'score': 973, }, ] self.verify_player_rb5_player_write_5(ref_id, location, mycourse=firstcourse) profile = self.verify_player_rb5_player_read(ref_id, card, location) for i in range(len(profile['mycourse'])): if profile['mycourse'][i]['music_id'] != firstcourse[i]['music_id']: raise Exception(f'invalid music ID for mycourse entry {i}!') if profile['mycourse'][i]['note_grade'] != firstcourse[i]['note_grade']: raise Exception(f'invalid chart for mycourse entry {i}!') if profile['mycourse'][i]['score'] != firstcourse[i]['score']: raise Exception(f'invalid score for mycourse entry {i}!') # Do a worse job on a different course secondcourse = [ { 'music_id': 2, 'note_grade': 1, 'score': 987, }, { 'music_id': 6, 'note_grade': 1, 'score': 234, }, { 'music_id': 102, 'note_grade': 1, 'score': 876, }, { 'music_id': 58, 'note_grade': 1, 'score': 573, }, ] self.verify_player_rb5_player_write_5(ref_id, location, mycourse=secondcourse) profile = self.verify_player_rb5_player_read(ref_id, card, location) for i in range(len(profile['mycourse'])): if profile['mycourse'][i]['music_id'] != firstcourse[i]['music_id']: raise Exception(f'invalid music ID for mycourse entry {i}!') if profile['mycourse'][i]['note_grade'] != firstcourse[i]['note_grade']: raise Exception(f'invalid chart for mycourse entry {i}!') if profile['mycourse'][i]['score'] != firstcourse[i]['score']: raise Exception(f'invalid score for mycourse entry {i}!') # Now, do better on our course, and verify it updates thirdcourse = [ { 'music_id': 51, 'note_grade': 1, 'score': 1186, }, { 'music_id': 37, 'note_grade': 1, 'score': 1141, }, { 'music_id': 102, 'note_grade': 1, 'score': 1035, }, { 'music_id': 58, 'note_grade': 1, 'score': 1073, }, ] self.verify_player_rb5_player_write_5(ref_id, location, mycourse=thirdcourse) profile = self.verify_player_rb5_player_read(ref_id, card, location) for i in range(len(profile['mycourse'])): if profile['mycourse'][i]['music_id'] != thirdcourse[i]['music_id']: raise Exception(f'invalid music ID for mycourse entry {i}!') if profile['mycourse'][i]['note_grade'] != thirdcourse[i]['note_grade']: raise Exception(f'invalid chart for mycourse entry {i}!') if profile['mycourse'][i]['score'] != thirdcourse[i]['score']: raise Exception(f'invalid score for mycourse entry {i}!') # Verify ending game self.verify_player_rb5_player_end(ref_id) # Verify empty and non-empty select score self.verify_player_rb5_player_read_rival_score_5(0) self.verify_player_rb5_player_read_rival_score_5(extid) # Verify rival score loading after rivaling ourselves self.verify_player_rb5_player_write_5(ref_id, location, rivals=[extid]) self.verify_player_rb5_player_read_rival_ranking_data_5(extid) # Verify high score tables and shop rank self.verify_info_rb5_info_read_hit_chart() self.verify_info_rb5_info_read_shop_ranking(location) self.verify_player_rb5_player_read_rank_5() # 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)