81 lines
3.0 KiB
Python
81 lines
3.0 KiB
Python
import importlib.util
|
|
from pathlib import Path
|
|
import tempfile
|
|
import unittest
|
|
|
|
ROOT = Path(__file__).resolve().parents[2]
|
|
spec = importlib.util.spec_from_file_location("room_server", ROOT / "tools/room_controller/server.py")
|
|
server = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(server)
|
|
|
|
|
|
class RoomTests(unittest.TestCase):
|
|
def setUp(self):
|
|
self.now = 100
|
|
self.room = server.Room(":memory:", lambda: self.now)
|
|
self.poll()
|
|
|
|
def tearDown(self): self.room.db.close()
|
|
|
|
def poll(self, reports=None, boot="boot1"):
|
|
return self.room.poll(dict(cabinet="cabinet_01", protocol=1, boot=boot, reports=reports or []))
|
|
|
|
def ready(self):
|
|
self.room.operator("prepare")
|
|
command = self.poll()["commands"][0]
|
|
self.assertEqual(self.room.state["phase"], "preparing")
|
|
self.poll([dict(kind="ack", id=command["id"], session=command["session"], ok=True)])
|
|
self.assertEqual(self.room.state["phase"], "ready")
|
|
|
|
def event(self, name, event_id):
|
|
return dict(kind="event", event=name, id=event_id, session=self.room.state["session"])
|
|
|
|
def test_ready_then_win_queues_one_reward(self):
|
|
self.ready()
|
|
self.poll([self.event("started", "1"), self.event("won", "2")])
|
|
reward = self.poll()["commands"]
|
|
self.assertEqual(len(reward), 1)
|
|
self.assertEqual(reward[0]["type"], "reward")
|
|
self.poll([self.event("won", "2"), self.event("won", "3")])
|
|
self.assertEqual(len(self.poll()["commands"]), 1)
|
|
|
|
def test_stale_session_cannot_reward_after_reset(self):
|
|
self.ready()
|
|
self.poll([self.event("started", "1")])
|
|
old_win = self.event("won", "2")
|
|
self.room.operator("reset")
|
|
self.poll([old_win])
|
|
self.assertEqual(self.room.state["reward"], "not requested")
|
|
|
|
def test_offline_and_expiry(self):
|
|
self.now += 5
|
|
self.assertFalse(self.room.snapshot()["online"])
|
|
with self.assertRaises(ValueError): self.room.operator("prepare")
|
|
self.poll()
|
|
self.room.operator("prepare")
|
|
self.now += 16
|
|
self.assertEqual(self.poll()["commands"], [])
|
|
self.assertEqual(self.room.state["phase"], "needs_prepare")
|
|
|
|
def test_new_agent_boot_requires_prepare(self):
|
|
self.ready()
|
|
self.poll(boot="boot2")
|
|
self.assertEqual(self.room.state["session"], "")
|
|
self.assertEqual(self.room.state["phase"], "needs_prepare")
|
|
|
|
def test_persistence(self):
|
|
with tempfile.TemporaryDirectory() as temp:
|
|
path = Path(temp) / "room.db"
|
|
room = server.Room(path, lambda: self.now)
|
|
room.poll(dict(cabinet="cabinet_01", protocol=1, boot="a"))
|
|
room.operator("prepare")
|
|
session = room.state["session"]
|
|
room.db.close()
|
|
restored = server.Room(path, lambda: self.now)
|
|
self.assertEqual(restored.state["session"], session)
|
|
self.assertFalse(restored.online())
|
|
restored.db.close()
|
|
|
|
|
|
if __name__ == "__main__": unittest.main()
|