Cool_Tools/bemaniutils
1
0
Fork 0
mirror of synced 2024-11-28 07:50:51 +01:00
Code Releases Activity
4790385022
bemaniutils/bemani/client/ddr/ddrace.py

909 lines
37 KiB
Python
Raw Normal View History

Initial commit of BEMANI Utilities to GitHub.
2019-12-08 22:43:49 +01:00
import base64
import random
import time
from typing import Optional, Dict, List, Tuple, Any
from bemani.client.base import BaseClient
from bemani.common import ID, Time
from bemani.protocol import Node
def b64str(string: str) -> str:
return base64.b64encode(string.encode()).decode('ascii')
class DDRAceClient(BaseClient):
NAME = 'TEST'
def verify_eventlog_write(self, location: str) -> None:
call = self.call_node()
# Construct node
eventlog = Node.void('eventlog')
call.add_child(eventlog)
eventlog.set_attribute('method', 'write')
eventlog.add_child(Node.u32('retrycnt', 0))
data = Node.void('data')
eventlog.add_child(data)
data.add_child(Node.string('eventid', 'S_PWRON'))
data.add_child(Node.s32('eventorder', 0))
data.add_child(Node.u64('pcbtime', int(time.time() * 1000)))
data.add_child(Node.s64('gamesession', -1))
data.add_child(Node.string('strdata1', b64str('2.4.0')))
data.add_child(Node.string('strdata2', ''))
data.add_child(Node.s64('numdata1', 1))
data.add_child(Node.s64('numdata2', 0))
data.add_child(Node.string('locationid', location))
# Swap with server
resp = self.exchange('', call)
# Verify that response is correct
self.assert_path(resp, "response/eventlog/gamesession")
self.assert_path(resp, "response/eventlog/logsendflg")
self.assert_path(resp, "response/eventlog/logerrlevel")
self.assert_path(resp, "response/eventlog/evtidnosendflg")
def verify_system_convcardnumber(self, cardno: str) -> None:
call = self.call_node()
# Construct node
system = Node.void('system')
call.add_child(system)
system.set_attribute('method', 'convcardnumber')
info = Node.void('info')
system.add_child(info)
info.add_child(Node.s32('version', 1))
data = Node.void('data')
system.add_child(data)
data.add_child(Node.string('card_id', cardno))
data.add_child(Node.s32('card_type', 1))
# Swap with server
resp = self.exchange('', call)
# Verify that response is correct
self.assert_path(resp, "response/system/data/card_number")
self.assert_path(resp, "response/system/result")
def verify_playerdata_usergamedata_advanced_usernew(self, refid: str) -> int:
call = self.call_node()
# Construct node
playerdata = Node.void('playerdata')
call.add_child(playerdata)
playerdata.set_attribute('method', 'usergamedata_advanced')
playerdata.add_child(Node.u32('retrycnt', 0))
info = Node.void('info')
playerdata.add_child(info)
info.add_child(Node.s32('version', 1))
data = Node.void('data')
playerdata.add_child(data)
data.add_child(Node.string('mode', 'usernew'))
data.add_child(Node.string('shoparea', '.'))
data.add_child(Node.s64('gamesession', 123456))
data.add_child(Node.string('refid', refid))
data.add_child(Node.string('dataid', refid))
data.add_child(Node.string('gamekind', 'MDX'))
data.add_child(Node.string('pcbid', self.pcbid))
data.add_child(Node.void('record'))
# Swap with server
resp = self.exchange('', call)
# Verify that response is correct
self.assert_path(resp, "response/playerdata/seq")
self.assert_path(resp, "response/playerdata/code")
self.assert_path(resp, "response/playerdata/shoparea")
self.assert_path(resp, "response/playerdata/result")
return resp.child_value('playerdata/code')
def verify_playerdata_usergamedata_advanced_ghostload(self, refid: str, ghostid: int) -> Dict[str, Any]:
call = self.call_node()
# Construct node
playerdata = Node.void('playerdata')
call.add_child(playerdata)
playerdata.set_attribute('method', 'usergamedata_advanced')
playerdata.add_child(Node.u32('retrycnt', 0))
info = Node.void('info')
playerdata.add_child(info)
info.add_child(Node.s32('version', 1))
data = Node.void('data')
playerdata.add_child(data)
data.add_child(Node.string('mode', 'ghostload'))
data.add_child(Node.s32('ghostid', ghostid))
data.add_child(Node.s64('gamesession', 123456))
data.add_child(Node.string('refid', refid))
data.add_child(Node.string('dataid', refid))
data.add_child(Node.string('gamekind', 'MDX'))
data.add_child(Node.string('pcbid', self.pcbid))
data.add_child(Node.void('record'))
# Swap with server
resp = self.exchange('', call)
# Verify that response is correct
self.assert_path(resp, "response/playerdata/ghostdata/code")
self.assert_path(resp, "response/playerdata/ghostdata/mcode")
self.assert_path(resp, "response/playerdata/ghostdata/notetype")
self.assert_path(resp, "response/playerdata/ghostdata/ghostsize")
self.assert_path(resp, "response/playerdata/ghostdata/ghost")
return {
'extid': resp.child_value('playerdata/ghostdata/code'),
'id': resp.child_value('playerdata/ghostdata/mcode'),
'chart': resp.child_value('playerdata/ghostdata/notetype'),
'ghost': resp.child_value('playerdata/ghostdata/ghost'),
}
def verify_playerdata_usergamedata_advanced_rivalload(self, refid: str, loadflag: int) -> None:
call = self.call_node()
# Construct node
playerdata = Node.void('playerdata')
call.add_child(playerdata)
playerdata.set_attribute('method', 'usergamedata_advanced')
playerdata.add_child(Node.u32('retrycnt', 0))
info = Node.void('info')
playerdata.add_child(info)
info.add_child(Node.s32('version', 1))
data = Node.void('data')
playerdata.add_child(data)
data.add_child(Node.string('mode', 'rivalload'))
data.add_child(Node.u64('targettime', Time.now() * 1000))
data.add_child(Node.string('shoparea', '.'))
data.add_child(Node.bool('isdouble', False))
data.add_child(Node.s32('loadflag', loadflag))
data.add_child(Node.s32('ddrcode', 0))
data.add_child(Node.s64('gamesession', 123456))
data.add_child(Node.string('refid', refid))
data.add_child(Node.string('dataid', refid))
data.add_child(Node.string('gamekind', 'MDX'))
data.add_child(Node.string('pcbid', self.pcbid))
data.add_child(Node.void('record'))
# Swap with server
resp = self.exchange('', call)
# Verify that response is correct
self.assert_path(resp, "response/playerdata/data/recordtype")
if loadflag != 2:
# As implemented, its possible for a machine not in an arcade to have scores.
# So, if the test PCBID we're using isn't in an arcade, we won't fetch scores
# for area records (flag 2), so don't check for these in that case.
self.assert_path(resp, "response/playerdata/data/record/mcode")
self.assert_path(resp, "response/playerdata/data/record/notetype")
self.assert_path(resp, "response/playerdata/data/record/rank")
self.assert_path(resp, "response/playerdata/data/record/clearkind")
self.assert_path(resp, "response/playerdata/data/record/flagdata")
self.assert_path(resp, "response/playerdata/data/record/name")
self.assert_path(resp, "response/playerdata/data/record/area")
self.assert_path(resp, "response/playerdata/data/record/code")
self.assert_path(resp, "response/playerdata/data/record/score")
self.assert_path(resp, "response/playerdata/data/record/ghostid")
if resp.child_value('playerdata/data/recordtype') != loadflag:
raise Exception('Invalid record type returned!')
def verify_playerdata_usergamedata_advanced_userload(self, refid: str) -> Tuple[bool, List[Dict[str, Any]]]:
call = self.call_node()
# Construct node
playerdata = Node.void('playerdata')
call.add_child(playerdata)
playerdata.set_attribute('method', 'usergamedata_advanced')
playerdata.add_child(Node.u32('retrycnt', 0))
info = Node.void('info')
playerdata.add_child(info)
info.add_child(Node.s32('version', 1))
data = Node.void('data')
playerdata.add_child(data)
data.add_child(Node.string('mode', 'userload'))
data.add_child(Node.s64('gamesession', 123456))
data.add_child(Node.string('refid', refid))
data.add_child(Node.string('dataid', refid))
data.add_child(Node.string('gamekind', 'MDX'))
data.add_child(Node.string('pcbid', self.pcbid))
data.add_child(Node.void('record'))
# Swap with server
resp = self.exchange('', call)
# Verify that response is correct
self.assert_path(resp, "response/playerdata/result")
self.assert_path(resp, "response/playerdata/is_new")
music = []
for child in resp.child('playerdata').children:
if child.name != 'music':
continue
songid = child.child_value('mcode')
chart = 0
for note in child.children:
if note.name != 'note':
continue
if note.child_value('count') != 0:
# Actual song
music.append({
'id': songid,
'chart': chart,
'rank': note.child_value('rank'),
'halo': note.child_value('clearkind'),
'score': note.child_value('score'),
'ghostid': note.child_value('ghostid'),
})
chart = chart + 1
return (
resp.child_value('playerdata/is_new'),
music,
)
def verify_playerdata_usergamedata_advanced_inheritance(self, refid: str, locid: str) -> None:
call = self.call_node()
# Construct node
playerdata = Node.void('playerdata')
call.add_child(playerdata)
playerdata.set_attribute('method', 'usergamedata_advanced')
playerdata.add_child(Node.u32('retrycnt', 0))
info = Node.void('info')
playerdata.add_child(info)
info.add_child(Node.s32('version', 1))
data = Node.void('data')
playerdata.add_child(data)
data.add_child(Node.string('mode', 'inheritance'))
data.add_child(Node.string('locid', locid))
data.add_child(Node.s64('gamesession', 123456))
data.add_child(Node.string('refid', refid))
data.add_child(Node.string('dataid', refid))
data.add_child(Node.string('gamekind', 'MDX'))
data.add_child(Node.string('pcbid', self.pcbid))
data.add_child(Node.void('record'))
# Swap with server
resp = self.exchange('', call)
# Verify that response is correct
self.assert_path(resp, "response/playerdata/InheritanceStatus")
self.assert_path(resp, "response/playerdata/result")
def verify_playerdata_usergamedata_advanced_usersave(self, refid: str, extid: int, locid: str, score: Dict[str, Any], scorepos: int=0) -> None:
call = self.call_node()
# Construct node
playerdata = Node.void('playerdata')
call.add_child(playerdata)
playerdata.set_attribute('method', 'usergamedata_advanced')
playerdata.add_child(Node.u32('retrycnt', 0))
info = Node.void('info')
playerdata.add_child(info)
info.add_child(Node.s32('version', 1))
data = Node.void('data')
playerdata.add_child(data)
data.add_child(Node.string('mode', 'usersave'))
data.add_child(Node.string('name', self.NAME))
data.add_child(Node.s32('ddrcode', extid))
data.add_child(Node.s32('playside', 1))
data.add_child(Node.s32('playstyle', 0))
data.add_child(Node.s32('area', 58))
data.add_child(Node.s32('weight100', 0))
data.add_child(Node.string('shopname', 'gmw='))
data.add_child(Node.bool('ispremium', False))
data.add_child(Node.bool('iseapass', True))
data.add_child(Node.bool('istakeover', False))
data.add_child(Node.bool('isrepeater', False))
data.add_child(Node.bool('isgameover', scorepos < 0))
data.add_child(Node.string('locid', locid))
data.add_child(Node.string('shoparea', '.'))
data.add_child(Node.s64('gamesession', 123456))
data.add_child(Node.string('refid', refid))
data.add_child(Node.string('dataid', refid))
data.add_child(Node.string('gamekind', 'MDX'))
data.add_child(Node.string('pcbid', self.pcbid))
data.add_child(Node.void('record'))
for i in range(5):
if i == scorepos:
# Fill in score here
note = Node.void('note')
data.add_child(note)
note.add_child(Node.u8('stagenum', i + 1))
note.add_child(Node.u32('mcode', score['id']))
note.add_child(Node.u8('notetype', score['chart']))
note.add_child(Node.u8('rank', score['rank']))
note.add_child(Node.u8('clearkind', score['halo']))
note.add_child(Node.s32('score', score['score']))
note.add_child(Node.s32('exscore', 0))
note.add_child(Node.s32('maxcombo', 0))
note.add_child(Node.s32('life', 0))
note.add_child(Node.s32('fastcount', 0))
note.add_child(Node.s32('slowcount', 0))
note.add_child(Node.s32('judge_marvelous', 0))
note.add_child(Node.s32('judge_perfect', 0))
note.add_child(Node.s32('judge_great', 0))
note.add_child(Node.s32('judge_good', 0))
note.add_child(Node.s32('judge_boo', 0))
note.add_child(Node.s32('judge_miss', 0))
note.add_child(Node.s32('judge_ok', 0))
note.add_child(Node.s32('judge_ng', 0))
note.add_child(Node.s32('calorie', 0))
note.add_child(Node.s32('ghostsize', len(score['ghost'])))
note.add_child(Node.string('ghost', score['ghost']))
note.add_child(Node.u8('opt_speed', 0))
note.add_child(Node.u8('opt_boost', 0))
note.add_child(Node.u8('opt_appearance', 0))
note.add_child(Node.u8('opt_turn', 0))
note.add_child(Node.u8('opt_dark', 0))
note.add_child(Node.u8('opt_scroll', 0))
note.add_child(Node.u8('opt_arrowcolor', 0))
note.add_child(Node.u8('opt_cut', 0))
note.add_child(Node.u8('opt_freeze', 0))
note.add_child(Node.u8('opt_jump', 0))
note.add_child(Node.u8('opt_arrowshape', 0))
note.add_child(Node.u8('opt_filter', 0))
note.add_child(Node.u8('opt_guideline', 0))
note.add_child(Node.u8('opt_gauge', 0))
note.add_child(Node.u8('opt_judgepriority', 0))
note.add_child(Node.u8('opt_timing', 0))
note.add_child(Node.string('basename', ''))
note.add_child(Node.string('title_b64', ''))
note.add_child(Node.string('artist_b64', ''))
note.add_child(Node.u16('bpmMax', 0))
note.add_child(Node.u16('bpmMin', 0))
note.add_child(Node.u8('level', 0))
note.add_child(Node.u8('series', 0))
note.add_child(Node.u32('bemaniFlag', 0))
note.add_child(Node.u32('genreFlag', 0))
note.add_child(Node.u8('limited', 0))
note.add_child(Node.u8('region', 0))
note.add_child(Node.s32('gr_voltage', 0))
note.add_child(Node.s32('gr_stream', 0))
note.add_child(Node.s32('gr_chaos', 0))
note.add_child(Node.s32('gr_freeze', 0))
note.add_child(Node.s32('gr_air', 0))
note.add_child(Node.bool('share', False))
note.add_child(Node.u64('endtime', 0))
note.add_child(Node.s32('folder', 0))
else:
note = Node.void('note')
data.add_child(note)
note.add_child(Node.u8('stagenum', 0))
note.add_child(Node.u32('mcode', 0))
note.add_child(Node.u8('notetype', 0))
note.add_child(Node.u8('rank', 0))
note.add_child(Node.u8('clearkind', 0))
note.add_child(Node.s32('score', 0))
note.add_child(Node.s32('exscore', 0))
note.add_child(Node.s32('maxcombo', 0))
note.add_child(Node.s32('life', 0))
note.add_child(Node.s32('fastcount', 0))
note.add_child(Node.s32('slowcount', 0))
note.add_child(Node.s32('judge_marvelous', 0))
note.add_child(Node.s32('judge_perfect', 0))
note.add_child(Node.s32('judge_great', 0))
note.add_child(Node.s32('judge_good', 0))
note.add_child(Node.s32('judge_boo', 0))
note.add_child(Node.s32('judge_miss', 0))
note.add_child(Node.s32('judge_ok', 0))
note.add_child(Node.s32('judge_ng', 0))
note.add_child(Node.s32('calorie', 0))
note.add_child(Node.s32('ghostsize', 0))
note.add_child(Node.string('ghost', ''))
note.add_child(Node.u8('opt_speed', 0))
note.add_child(Node.u8('opt_boost', 0))
note.add_child(Node.u8('opt_appearance', 0))
note.add_child(Node.u8('opt_turn', 0))
note.add_child(Node.u8('opt_dark', 0))
note.add_child(Node.u8('opt_scroll', 0))
note.add_child(Node.u8('opt_arrowcolor', 0))
note.add_child(Node.u8('opt_cut', 0))
note.add_child(Node.u8('opt_freeze', 0))
note.add_child(Node.u8('opt_jump', 0))
note.add_child(Node.u8('opt_arrowshape', 0))
note.add_child(Node.u8('opt_filter', 0))
note.add_child(Node.u8('opt_guideline', 0))
note.add_child(Node.u8('opt_gauge', 0))
note.add_child(Node.u8('opt_judgepriority', 0))
note.add_child(Node.u8('opt_timing', 0))
note.add_child(Node.string('basename', ''))
note.add_child(Node.string('title_b64', ''))
note.add_child(Node.string('artist_b64', ''))
note.add_child(Node.u16('bpmMax', 0))
note.add_child(Node.u16('bpmMin', 0))
note.add_child(Node.u8('level', 0))
note.add_child(Node.u8('series', 0))
note.add_child(Node.u32('bemaniFlag', 0))
note.add_child(Node.u32('genreFlag', 0))
note.add_child(Node.u8('limited', 0))
note.add_child(Node.u8('region', 0))
note.add_child(Node.s32('gr_voltage', 0))
note.add_child(Node.s32('gr_stream', 0))
note.add_child(Node.s32('gr_chaos', 0))
note.add_child(Node.s32('gr_freeze', 0))
note.add_child(Node.s32('gr_air', 0))
note.add_child(Node.bool('share', False))
note.add_child(Node.u64('endtime', 0))
note.add_child(Node.s32('folder', 0))
# Swap with server
resp = self.exchange('', call)
# Verify that response is correct
self.assert_path(resp, "response/playerdata/result")
def verify_usergamedata_send(self, ref_id: str, ext_id: int, msg_type: str, send_only_common: bool=False) -> None:
call = self.call_node()
# Set up profile write
profiledata = {
'COMMON': [
b'1',
b'0', # shoparea spot, filled in below
b'3c880f8',
b'1',
b'0',
b'0',
b'0',
b'0',
b'0',
b'ffffffffffffffff',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0.000000',
b'0.000000',
b'0.000000',
b'0.000000',
b'0.000000',
b'0.000000',
b'0.000000',
b'0.000000',
b'', # Name spot, filled in below
ID.format_extid(ext_id).encode('ascii'),
b'',
b'',
b'',
b'',
b'',
b'',
],
'OPTION': [
b'0',
b'3',
b'0',
b'0',
b'0',
b'0',
b'0',
b'3',
b'0',
b'0',
b'0',
b'0',
b'1',
b'2',
b'0',
b'0',
b'0',
b'10.000000',
b'10.000000',
b'10.000000',
b'10.000000',
b'0.000000',
b'0.000000',
b'0.000000',
b'0.000000',
b'',
b'',
b'',
b'',
b'',
b'',
b'',
b'',
],
'LAST': [
b'1',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0.000000',
b'0.000000',
b'0.000000',
b'0.000000',
b'0.000000',
b'0.000000',
b'0.000000',
b'0.000000',
b'',
b'',
b'',
b'',
b'',
b'',
b'',
b'',
],
'RIVAL': [
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0',
b'0.000000',
b'0.000000',
b'0.000000',
b'0.000000',
b'0.000000',
b'0.000000',
b'0.000000',
b'0.000000',
b'',
b'',
b'',
b'',
b'',
b'',
b'',
b'',
]
}
if msg_type == 'new':
# New profile gets blank name, because we save over it at the end of the round.
profiledata['COMMON'][1] = b'0'
profiledata['COMMON'][25] = b''
elif msg_type == 'existing':
# Exiting profile gets our hardcoded name saved.
profiledata['COMMON'][1] = b'3a'
profiledata['COMMON'][25] = self.NAME.encode('shift-jis')
else:
Convert most of the format() string calls to f-strings using libcst. Exact commands run were: python3 -m libcst.tool codemod convert_format_to_fstring.ConvertFormatStringCommand . --no-format python3 setup.py build_ext --inplace
2020-01-07 22:29:07 +01:00
raise Exception(f'Unknown message type {msg_type}!')
Initial commit of BEMANI Utilities to GitHub.
2019-12-08 22:43:49 +01:00
if send_only_common:
profiledata = {'COMMON': profiledata['COMMON']}
# Construct node
playerdata = Node.void('playerdata')
call.add_child(playerdata)
playerdata.set_attribute('method', 'usergamedata_send')
playerdata.add_child(Node.u32('retrycnt', 0))
info = Node.void('info')
playerdata.add_child(info)
info.add_child(Node.s32('version', 1))
data = Node.void('data')
playerdata.add_child(data)
data.add_child(Node.string('refid', ref_id))
data.add_child(Node.string('dataid', ref_id))
data.add_child(Node.string('gamekind', 'MDX'))
data.add_child(Node.u32('datanum', len(profiledata.keys())))
record = Node.void('record')
data.add_child(record)
for ptype in profiledata:
profile = [b'ffffffff', ptype.encode('ascii')] + profiledata[ptype]
d = Node.string('d', base64.b64encode(b','.join(profile)).decode('ascii'))
record.add_child(d)
d.add_child(Node.string('bin1', ''))
# Swap with server
resp = self.exchange('', call)
self.assert_path(resp, "response/playerdata/result")
def verify_usergamedata_recv(self, ref_id: str) -> str:
call = self.call_node()
# Construct node
playerdata = Node.void('playerdata')
call.add_child(playerdata)
playerdata.set_attribute('method', 'usergamedata_recv')
info = Node.void('info')
playerdata.add_child(info)
info.add_child(Node.s32('version', 1))
data = Node.void('data')
playerdata.add_child(data)
data.add_child(Node.string('refid', ref_id))
data.add_child(Node.string('dataid', ref_id))
data.add_child(Node.string('gamekind', 'MDX'))
data.add_child(Node.u32('recv_num', 4))
data.add_child(Node.string('recv_csv', 'COMMON,3fffffffff,OPTION,3fffffffff,LAST,3fffffffff,RIVAL,3fffffffff'))
# Swap with server
resp = self.exchange('', call)
self.assert_path(resp, "response/playerdata/result")
self.assert_path(resp, "response/playerdata/player/record/d/bin1")
self.assert_path(resp, "response/playerdata/player/record_num")
profiles = 0
name = ''
for child in resp.child('playerdata/player/record').children:
if child.name != 'd':
continue
if profiles == 0:
bindata = child.value
profiledata = base64.b64decode(bindata).split(b',')
name = profiledata[25].decode('ascii')
profiles = profiles + 1
if profiles != 4:
raise Exception('Didn\'t receive all four profiles in the right order!')
return name
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_eventlog_write(location)
# Verify the game-wide packets Ace insists on sending before profile load
is_new, music = self.verify_playerdata_usergamedata_advanced_userload('X0000000000000000000000000123456')
if not is_new:
raise Exception('Fake profiles should be new!')
if len(music) > 0:
raise Exception('Fake profiles should have no scores associated!')
# Verify card registration and profile lookup
if cardid is not None:
card = cardid
else:
card = self.random_card()
Convert most of the format() string calls to f-strings using libcst. Exact commands run were: python3 -m libcst.tool codemod convert_format_to_fstring.ConvertFormatStringCommand . --no-format python3 setup.py build_ext --inplace
2020-01-07 22:29:07 +01:00
print(f"Generated random card ID {card} for use.")
Initial commit of BEMANI Utilities to GitHub.
2019-12-08 22:43:49 +01:00
if cardid is None:
self.verify_cardmng_inquire(card, msg_type='unregistered', paseli_enabled=paseli_enabled)
self.verify_system_convcardnumber(card)
ref_id = self.verify_cardmng_getrefid(card)
if len(ref_id) != 16:
Convert most of the format() string calls to f-strings using libcst. Exact commands run were: python3 -m libcst.tool codemod convert_format_to_fstring.ConvertFormatStringCommand . --no-format python3 setup.py build_ext --inplace
2020-01-07 22:29:07 +01:00
raise Exception(f'Invalid refid \'{ref_id}\' returned when registering card')
Initial commit of BEMANI Utilities to GitHub.
2019-12-08 22:43:49 +01:00
if ref_id != self.verify_cardmng_inquire(card, msg_type='new', paseli_enabled=paseli_enabled):
Convert most of the format() string calls to f-strings using libcst. Exact commands run were: python3 -m libcst.tool codemod convert_format_to_fstring.ConvertFormatStringCommand . --no-format python3 setup.py build_ext --inplace
2020-01-07 22:29:07 +01:00
raise Exception(f'Invalid refid \'{ref_id}\' returned when querying card')
Initial commit of BEMANI Utilities to GitHub.
2019-12-08 22:43:49 +01:00
extid = self.verify_playerdata_usergamedata_advanced_usernew(ref_id)
self.verify_usergamedata_send(ref_id, extid, 'new')
self.verify_playerdata_usergamedata_advanced_inheritance(ref_id, location)
name = self.verify_usergamedata_recv(ref_id)
if name != '':
raise Exception('Name stored on profile we just created!')
self.verify_usergamedata_send(ref_id, extid, 'existing', send_only_common=True)
name = self.verify_usergamedata_recv(ref_id)
if name != self.NAME:
raise Exception('Name stored on profile is incorrect!')
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):
Convert most of the format() string calls to f-strings using libcst. Exact commands run were: python3 -m libcst.tool codemod convert_format_to_fstring.ConvertFormatStringCommand . --no-format python3 setup.py build_ext --inplace
2020-01-07 22:29:07 +01:00
raise Exception(f'Invalid refid \'{ref_id}\' returned when querying card')
Initial commit of BEMANI Utilities to GitHub.
2019-12-08 22:43:49 +01:00
if cardid is None:
is_new, music = self.verify_playerdata_usergamedata_advanced_userload(ref_id)
if is_new:
raise Exception('Profile should not be new!')
if len(music) > 0:
raise Exception('Created profile should have no scores associated!')
# Verify score saving and updating
for phase in [1, 2]:
if phase == 1:
dummyscores = [
# An okay score on a chart
{
'id': 10,
'chart': 3,
'rank': 5,
'halo': 6,
'score': 765432,
'ghost': '765432',
},
# A good score on an easier chart of the same song
{
'id': 10,
'chart': 2,
'rank': 2,
'halo': 8,
'score': 876543,
'ghost': '876543',
},
# A bad score on a hard chart
{
'id': 479,
'chart': 2,
'rank': 11,
'halo': 6,
'score': 654321,
'ghost': '654321',
},
# A terrible score on an easy chart
{
'id': 479,
'chart': 1,
'rank': 15,
'halo': 6,
'score': 123456,
'ghost': '123456',
},
]
if phase == 2:
dummyscores = [
# A better score on the same chart
{
'id': 10,
'chart': 3,
'rank': 4,
'halo': 7,
'score': 888888,
'ghost': '888888',
},
# A worse score on another same chart
{
'id': 10,
'chart': 2,
'rank': 3,
'halo': 7,
'score': 654321,
'ghost': '654321',
'expected_score': 876543,
'expected_halo': 8,
'expected_rank': 2,
'expected_ghost': '876543',
},
]
pos = 0
for dummyscore in dummyscores:
self.verify_playerdata_usergamedata_advanced_usersave(
ref_id,
extid,
location,
dummyscore,
pos,
)
pos = pos + 1
is_new, scores = self.verify_playerdata_usergamedata_advanced_userload(ref_id)
if is_new:
raise Exception('Profile should not be new!')
if len(scores) == 0:
raise Exception('Expected some scores after saving!')
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:
Convert most of the format() string calls to f-strings using libcst. Exact commands run were: python3 -m libcst.tool codemod convert_format_to_fstring.ConvertFormatStringCommand . --no-format python3 setup.py build_ext --inplace
2020-01-07 22:29:07 +01:00
raise Exception(f"Didn't find song {expected['id']} chart {expected['chart']} in response!")
Initial commit of BEMANI Utilities to GitHub.
2019-12-08 22:43:49 +01:00
if 'expected_score' in expected:
expected_score = expected['expected_score']
else:
expected_score = expected['score']
if 'expected_rank' in expected:
expected_rank = expected['expected_rank']
else:
expected_rank = expected['rank']
if 'expected_halo' in expected:
expected_halo = expected['expected_halo']
else:
expected_halo = expected['halo']
if actual['score'] != expected_score:
Convert most of the format() string calls to f-strings using libcst. Exact commands run were: python3 -m libcst.tool codemod convert_format_to_fstring.ConvertFormatStringCommand . --no-format python3 setup.py build_ext --inplace
2020-01-07 22:29:07 +01:00
raise Exception(f'Expected a score of \'{expected_score}\' for song \'{expected["id"]}\' chart \'{expected["chart"]}\' but got score \'{actual["score"]}\'')
Initial commit of BEMANI Utilities to GitHub.
2019-12-08 22:43:49 +01:00
if actual['rank'] != expected_rank:
Convert most of the format() string calls to f-strings using libcst. Exact commands run were: python3 -m libcst.tool codemod convert_format_to_fstring.ConvertFormatStringCommand . --no-format python3 setup.py build_ext --inplace
2020-01-07 22:29:07 +01:00
raise Exception(f'Expected a rank of \'{expected_rank}\' for song \'{expected["id"]}\' chart \'{expected["chart"]}\' but got rank \'{actual["rank"]}\'')
Initial commit of BEMANI Utilities to GitHub.
2019-12-08 22:43:49 +01:00
if actual['halo'] != expected_halo:
Convert most of the format() string calls to f-strings using libcst. Exact commands run were: python3 -m libcst.tool codemod convert_format_to_fstring.ConvertFormatStringCommand . --no-format python3 setup.py build_ext --inplace
2020-01-07 22:29:07 +01:00
raise Exception(f'Expected a halo of \'{expected_halo}\' for song \'{expected["id"]}\' chart \'{expected["chart"]}\' but got halo \'{actual["halo"]}\'')
Initial commit of BEMANI Utilities to GitHub.
2019-12-08 22:43:49 +01:00
# Now verify that the ghost for this score is what we saved
ghost = self.verify_playerdata_usergamedata_advanced_ghostload(ref_id, received['ghostid'])
if 'expected_ghost' in expected:
expected_ghost = expected['expected_ghost']
else:
expected_ghost = expected['ghost']
if ghost['id'] != received['id']:
Convert most of the format() string calls to f-strings using libcst. Exact commands run were: python3 -m libcst.tool codemod convert_format_to_fstring.ConvertFormatStringCommand . --no-format python3 setup.py build_ext --inplace
2020-01-07 22:29:07 +01:00
raise Exception(f'Wrong song ID \'{ghost["id"]}\' returned for ghost, expected ID \'{received["id"]}\'')
Initial commit of BEMANI Utilities to GitHub.
2019-12-08 22:43:49 +01:00
if ghost['chart'] != received['chart']:
Convert most of the format() string calls to f-strings using libcst. Exact commands run were: python3 -m libcst.tool codemod convert_format_to_fstring.ConvertFormatStringCommand . --no-format python3 setup.py build_ext --inplace
2020-01-07 22:29:07 +01:00
raise Exception(f'Wrong song chart \'{ghost["chart"]}\' returned for ghost, expected chart \'{received["chart"]}\'')
Initial commit of BEMANI Utilities to GitHub.
2019-12-08 22:43:49 +01:00
if ghost['ghost'] != expected_ghost:
Convert most of the format() string calls to f-strings using libcst. Exact commands run were: python3 -m libcst.tool codemod convert_format_to_fstring.ConvertFormatStringCommand . --no-format python3 setup.py build_ext --inplace
2020-01-07 22:29:07 +01:00
raise Exception(f'Wrong ghost data \'{ghost["ghost"]}\' returned for ghost, expected \'{expected_ghost}\'')
Initial commit of BEMANI Utilities to GitHub.
2019-12-08 22:43:49 +01:00
if ghost['extid'] != extid:
Convert most of the format() string calls to f-strings using libcst. Exact commands run were: python3 -m libcst.tool codemod convert_format_to_fstring.ConvertFormatStringCommand . --no-format python3 setup.py build_ext --inplace
2020-01-07 22:29:07 +01:00
raise Exception(f'Wrong extid \'{ghost["extid"]}\' returned for ghost, expected \'{extid}\'')
Initial commit of BEMANI Utilities to GitHub.
2019-12-08 22:43:49 +01:00
# Sleep so we don't end up putting in score history on the same second
time.sleep(1)
# Simulate game over conditions
self.verify_playerdata_usergamedata_advanced_usersave(
ref_id,
extid,
location,
{},
-1,
)
else:
print("Skipping score checks for existing card")
# Verify global scores now that we've inserted some
self.verify_playerdata_usergamedata_advanced_rivalload('X0000000000000000000000000123456', 1)
self.verify_playerdata_usergamedata_advanced_rivalload('X0000000000000000000000000123456', 2)
self.verify_playerdata_usergamedata_advanced_rivalload('X0000000000000000000000000123456', 4)
# 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)
Copy Permalink