shenleiandClaude Opus 5.5
a22c064587
Five per-slot toggles next to start/stop, shown only when the server sends the MTSB1 lock line; locked attributes are marked in the current attribute line. Server side: metin2-server docs/FEATURE-ATTR-LOCK.md. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
184 lines
12 KiB
Python
184 lines
12 KiB
Python
import re
|
|
import sys
|
|
import types
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
ROOT=Path(__file__).resolve().parents[2]
|
|
|
|
def script(name):return (ROOT/'src/platform/ScriptLib'/name).read_text().split('R"PY(',1)[1].rsplit(')PY";',1)[0]
|
|
|
|
|
|
class Widget(object):
|
|
def __init__(self,*a,**k):self.shown=False;self.text='';self.pos=(0,0);self.size=(0,0)
|
|
def __getattr__(self,name):return lambda *a,**k:None
|
|
def Show(self):self.shown=True
|
|
def Hide(self):self.shown=False
|
|
def IsShow(self):return self.shown
|
|
def SetText(self,text):self.text=text
|
|
def SetPosition(self,x,y):self.pos=(x,y)
|
|
def GetLocalPosition(self):return self.pos
|
|
def GetGlobalPosition(self):return self.pos
|
|
def SetSize(self,w,h):self.size=(w,h)
|
|
def GetWidth(self):return self.size[0]
|
|
def GetHeight(self):return self.size[1]
|
|
|
|
|
|
class Action(Widget):
|
|
def __init__(self,parent,x,y,width,text,event):
|
|
Widget.__init__(self);self.pos=(x,y);self.event=event;self.enabled=True;self.label=Widget();self.text=text;self.shown=True
|
|
def SetEnabled(self,enabled):self.enabled=enabled
|
|
|
|
|
|
class SwitchbotTests(unittest.TestCase):
|
|
def setUp(self):
|
|
self.time=1.;self.sent=[];self.chat=[];self.events=[]
|
|
self.inventory={10:(11209,1,[(3,2),(15,1)]),20:(71084,7,[]),21:(71084,3,[]),30:(27001,5,[])}
|
|
protos={11209:(2,0),71084:(3,18),27001:(3,0)}
|
|
mods={n:types.ModuleType(n) for n in ('app','chat','game','grp','item','net','player','ui','uiToolTip','wndMgr','mt_biolog','mt_wiki')}
|
|
mods['app'].GetTime=lambda:self.time
|
|
mods['chat'].CHAT_TYPE_INFO=1;mods['chat'].AppendChat=lambda t,m:self.chat.append(m)
|
|
mods['net'].SendChatPacket=self.sent.append
|
|
mods['game'].GameWindow=type('GameWindow',(),dict(Open=lambda s:None,Close=lambda s:None,BINARY_ServerCommand_Run=lambda s,line:7))
|
|
grp=mods['grp'];grp.SetColor=grp.RenderBar=grp.RenderBox=lambda *a:None
|
|
it=mods['item'];it.ITEM_TYPE_WEAPON=1;it.ITEM_TYPE_ARMOR=2;it.ITEM_TYPE_USE=3;self.selected=None
|
|
it.SelectItem=lambda v:setattr(self,'selected',v);it.GetItemType=lambda:protos[self.selected][0];it.GetItemSubType=lambda:protos[self.selected][1]
|
|
it.GetItemName=lambda:'N%d'%self.selected
|
|
pl=mods['player'];pl.INVENTORY=1;pl.INVENTORY_PAGE_SIZE=45;pl.INVENTORY_PAGE_COUNT=5
|
|
pl.GetItemIndex=lambda w,c:self.inventory.get(c,(0,))[0];pl.GetItemCount=lambda w,c:self.inventory[c][1]
|
|
pl.GetItemAttribute=lambda w,c,i:(self.inventory[c][2]+[(0,0)]*5)[i]
|
|
mods['ui'].Window=Widget
|
|
mods['uiToolTip'].ItemToolTip=type('ItemToolTip',(),dict(AFFECT_DICT={3:lambda v:'STR+%d'%v}))
|
|
mods['wndMgr'].GetScreenWidth=lambda:800;mods['wndMgr'].GetScreenHeight=lambda:600
|
|
mods['mt_biolog'].Action=Action;mods['mt_biolog'].Text=lambda p,x,y,t='',c=0:Widget();mods['mt_biolog'].L=lambda t:t if isinstance(t,str) else t.decode('utf-8')
|
|
mods['mt_wiki'].U=lambda t:t if isinstance(t,str) else t.decode('utf-8');mods['mt_wiki'].B=lambda t,limit=0:t
|
|
self.previous={n:sys.modules.get(n) for n in mods}
|
|
sys.modules.update(mods)
|
|
self.ns={'__name__':'mt_switchbot'};exec(script('SwitchbotScript.inc'),self.ns)
|
|
self.bot=self.ns['Bot'](self.sent.append,lambda:self.time,self.events.append)
|
|
|
|
def tearDown(self):
|
|
for name,value in self.previous.items():
|
|
if value is None:sys.modules.pop(name,None)
|
|
else:sys.modules[name]=value
|
|
|
|
def advance(self,seconds=.4):self.time+=seconds;self.bot.Tick()
|
|
|
|
def load(self,bot,cell=10):
|
|
for line in ('MTSB1 attrs %d 11209 3'%cell,'MTSB1 attr 0 3 12','MTSB1 attr 1 63 6','MTSB1 attr 2 71 30','MTSB1 attrs_end %d 3'%cell):bot.Receive(line)
|
|
|
|
def test_commands_are_paced_and_attrs_requests_collapse(self):
|
|
self.bot.RequestAttrs(10);self.bot.RequestAttrs(11);self.bot.RequestAttrs(12);self.bot.Status()
|
|
self.assertEqual(self.sent,['/mt_switchbot 1 attrs 10'])
|
|
self.load(self.bot);self.advance()
|
|
self.assertEqual(self.sent[-1],'/mt_switchbot 1 attrs 12')
|
|
self.bot.Receive('MTSB1 error attrs no_attr');self.advance()
|
|
self.assertEqual(self.sent[-1],'/mt_switchbot 1 status')
|
|
self.assertEqual(len(self.sent),3)
|
|
|
|
def test_atomic_attr_list_and_rejected_partials(self):
|
|
self.load(self.bot)
|
|
self.assertEqual(self.bot.allowed[10]['rows'],[(3,12),(63,6),(71,30)])
|
|
for bad in (['MTSB1 attrs 11 1 2','MTSB1 attr 0 3 12','MTSB1 attrs_end 11 2'],
|
|
['MTSB1 attrs 11 1 1','MTSB1 attr 1 3 12','MTSB1 attrs_end 11 1'],
|
|
['MTSB1 attrs 11 1 1','MTSB1 attr 0 3 -1','MTSB1 attrs_end 11 1'],
|
|
['MTSB1 attrs_end 11 0']):
|
|
for line in bad:self.assertTrue(self.bot.Receive(line))
|
|
self.assertNotIn(11,self.bot.allowed)
|
|
self.assertFalse(self.bot.Receive('MTF1 cap 1 1 127 0'))
|
|
|
|
def test_start_validates_against_server_list_and_builds_command(self):
|
|
self.load(self.bot)
|
|
self.assertEqual(self.bot.Start(10,71084,100,[[(3,13)],[]]),'target')
|
|
self.assertEqual(self.bot.Start(10,71084,100,[[(1,1)],[]]),'target')
|
|
self.assertEqual(self.bot.Start(10,71084,100,[[(3,1),(3,2)],[]]),'invalid')
|
|
self.assertEqual(self.bot.Start(10,71084,1001,[[(3,1)],[]]),'invalid')
|
|
self.assertEqual(self.bot.Start(10,71084,100,[[],[]]),'invalid')
|
|
self.assertIsNone(self.bot.Start(10,71084,100,[[(3,8),(63,6)],[(71,30)]]))
|
|
self.advance()
|
|
self.assertEqual(self.sent[-1],'/mt_switchbot 1 start 10 71084 100 2 3 8 63 6 1 71 30')
|
|
self.bot.Receive('MTSB1 error start running')
|
|
self.assertIsNone(self.bot.Start(10,71084,5,[[],[(71,1)]]));self.advance()
|
|
self.assertEqual(self.sent[-1],'/mt_switchbot 1 start 10 71084 5 1 71 1') # empty first alternative skipped
|
|
|
|
def test_state_errors_and_timeout(self):
|
|
self.bot.Receive('MTSB1 state 1 10 11209 3 100 71084 4 running')
|
|
self.assertTrue(self.bot.Running());self.assertEqual(self.bot.state['scrolls'],4)
|
|
self.bot.Receive('MTSB1 state 0 10 11209 4 100 71084 3 matched')
|
|
self.assertFalse(self.bot.Running());self.assertEqual(self.bot.state['reason'],'matched')
|
|
self.bot.Receive('MTSB1 state 0 10 11209 4 100 71084 3 exploded')
|
|
self.assertEqual(self.bot.state['reason'],'matched') # unknown reason ignored
|
|
self.bot.Receive('MTSB1 error start busy');self.assertEqual(self.bot.error,'busy')
|
|
self.bot.Status();self.advance(4.1);self.advance()
|
|
self.assertEqual(self.bot.error,'unsupported');self.assertTrue(self.bot.unsupported)
|
|
|
|
def test_server_constants_match(self):
|
|
server=ROOT.parent/'metin2-server/src/server/game/src/port_switchbot.h'
|
|
if not server.exists():self.skipTest('server checkout not installed')
|
|
text=server.read_text(encoding='latin1')
|
|
reasons=set(re.findall(r'return "([a-z_]+)"',text))|{'running','stopped','idle'}
|
|
errors=set(re.findall(r'error [a-z%]+ ([a-z_]+)"',text))|{'invalid','rate'}
|
|
lock_errors={'slot','max','db'}
|
|
self.assertTrue(reasons-{'invalid','running','item','equipped','no_attr','scroll','target','busy'}-lock_errors<=set(self.ns['REASONS']))
|
|
self.assertTrue(lock_errors<=set(self.ns['ERRORS']))
|
|
self.assertTrue(errors<=set(self.ns['ERRORS']))
|
|
self.assertIn('kPortSwitchLimit=1000',text);self.assertEqual(max(self.ns['LIMITS']),1000)
|
|
self.assertEqual(self.ns['MAX_TARGETS'],5)
|
|
|
|
def test_lock_protocol_and_buttons(self):
|
|
self.bot.Lock(10,0);self.assertEqual(self.sent,[]) # no lock line yet: old server, no request
|
|
self.load(self.bot);self.bot.Receive('MTSB1 lock 10 11209 2')
|
|
self.assertEqual(self.bot.LockMask(10,11209),2);self.assertIsNone(self.bot.LockMask(10,11210))
|
|
self.bot.Lock(10,5);self.bot.Lock(10,0);self.advance()
|
|
self.assertEqual(self.sent[-1],'/mt_switchbot 1 lock 10 0')
|
|
self.bot.Receive('MTSB1 lock 10 11209 3');self.assertIsNone(self.bot.waiting);self.assertEqual(self.bot.LockMask(10,11209),3)
|
|
self.bot.Lock(10,1);self.advance();self.bot.Receive('MTSB1 error lock max');self.assertEqual(self.bot.error,'max')
|
|
for bad in ('MTSB1 lock 10 11209','MTSB1 lock 10 11209 32'):self.bot.Receive(bad)
|
|
self.assertEqual(self.bot.LockMask(10,11209),3)
|
|
# Window: one button per filled slot, hidden while picking, locked attributes marked.
|
|
game=sys.modules['game'];gw=game.GameWindow();gw.interface=types.SimpleNamespace(wndMiniMap=Widget())
|
|
game.GameWindow.Open(gw);w=self.ns['g_window'];w.Open()
|
|
self.assertFalse(any(b.shown for b in w.lockButtons))
|
|
self.load(w.bot);w.bot.Receive('MTSB1 lock 10 11209 2')
|
|
self.assertEqual([b.shown for b in w.lockButtons],[True,True,False,False,False])
|
|
self.assertEqual([b.text for b in w.lockButtons[:2]],['锁1','【锁2】']);self.assertIn('[锁]',w.current.text.split(' / ')[1])
|
|
self.assertIn('保持不变',w.status.text)
|
|
self.time+=.5;w.OnUpdate();w.bot.Receive('MTSB1 state 0 0 0 0 0 0 0 idle') # answers the status request
|
|
sent=len(self.sent);w.lockButtons[0].event();self.time+=.5;w.OnUpdate()
|
|
self.assertEqual(self.sent[sent:],['/mt_switchbot 1 lock 10 0'])
|
|
w.TogglePick();self.assertFalse(any(b.shown for b in w.lockButtons));w.TogglePick()
|
|
w.bot.Receive('MTSB1 state 1 10 11209 0 100 71084 9 running');self.assertFalse(w.lockButtons[0].enabled)
|
|
sent=len(self.sent);w.lockButtons[0].event();self.time+=.5;w.OnUpdate()
|
|
self.assertNotIn('/mt_switchbot 1 lock 10 0',self.sent[sent:])
|
|
game.GameWindow.Close(gw)
|
|
|
|
def test_window_scan_pick_start_and_notify(self):
|
|
game=sys.modules['game'];window_type=game.GameWindow
|
|
gw=window_type();gw.interface=types.SimpleNamespace(wndMiniMap=Widget())
|
|
game.GameWindow.Open(gw);w=self.ns['g_window'];w.Open()
|
|
self.assertEqual(w.items,[(10,11209)]);self.assertEqual(w.scrolls,[(71084,10)])
|
|
self.assertEqual(self.sent,['/mt_switchbot 1 attrs 10']) # the status request waits for this reply
|
|
self.load(w.bot);self.time+=.5;w.OnUpdate()
|
|
self.assertEqual(self.sent[-1],'/mt_switchbot 1 status')
|
|
self.assertEqual(game.GameWindow.BINARY_ServerCommand_Run(gw,'MTSB1 state 0 0 0 0 0 0 0 idle'),1)
|
|
self.assertEqual(game.GameWindow.BINARY_ServerCommand_Run(gw,'other'),7)
|
|
w.TogglePick();w.Pick(0)
|
|
self.assertEqual(w.alternatives[0],[[3,12]])
|
|
w.Adjust(0,-1);self.assertEqual(w.alternatives[0][0][1],10);w.Adjust(0,1);w.Adjust(0,1);self.assertEqual(w.alternatives[0][0][1],12)
|
|
w.Start();self.time+=.5;w.OnUpdate()
|
|
self.assertEqual(self.sent[-1],'/mt_switchbot 1 start 10 71084 100 1 3 12')
|
|
w.bot.Receive('MTSB1 state 1 10 11209 1 100 71084 9 running');self.assertFalse(w.startButton.enabled)
|
|
self.inventory[40]=self.inventory.pop(10);sent=len(self.sent) # moved while running: follow the server's cell
|
|
w.bot.Receive('MTSB1 state 1 40 11209 2 100 71084 8 running');w.Scan()
|
|
self.assertEqual(w.Selected(),(40,11209));self.assertEqual(w.alternatives[0],[[3,12]])
|
|
self.assertEqual(w.Allowed(),[(3,12),(63,6),(71,30)]);self.time+=.5;w.OnUpdate();self.assertEqual(len(self.sent),sent)
|
|
w.bot.Receive('MTSB1 state 0 40 11209 2 100 71084 8 stopped')
|
|
self.inventory[5]=self.inventory.pop(40);w.Scan();self.assertEqual(w.Selected(),(5,11209)) # moved while idle: same vnum
|
|
self.inventory[10]=self.inventory.pop(5);w.Scan()
|
|
w.Hide();w.bot.Receive('MTSB1 state 0 10 11209 2 100 71084 8 matched')
|
|
self.assertTrue(w.IsShow());self.assertIn('已获得目标属性',self.chat[-1])
|
|
game.GameWindow.Close(gw);self.assertIsNone(self.ns['g_window'])
|
|
|
|
|
|
if __name__=='__main__':unittest.main()
|