def test_join_filled_group(self):
"""Joining a filled group should fail"""
groupid = lruc(KVS.create_group(self.user.userid, self.game.gameid))
for _ in range(3):
lruc(KVS.join_group(groupid, uuid4()))
coro = KVS.join_group(groupid, uuid4())
self.assertRaises(GroupIsFull, lruc, coro)
def test_join_queue_not_all_ready(self):
"""Joining the queue while all the players are not ready should fail"""
player_2 = uuid4()
groupid = lruc(KVS.create_group(self.user.userid, self.game.gameid))
lruc(KVS.mark_as_ready(self.user.userid))
lruc(KVS.join_group(groupid, player_2))
coro = KVS.join_queue(groupid)
self.assertRaises(GroupNotReady, lruc, coro)
def test_join_queued_group(self):
"""Joining a queued group should fail"""
player_2 = uuid4()
groupid = lruc(KVS.create_group(self.user.userid, self.game.gameid))
lruc(KVS.mark_as_ready(self.user.userid))
lruc(KVS.join_queue(groupid))
coro = KVS.join_group(groupid, player_2)
self.assertRaises(WrongGroupState, lruc, coro)
def test_leave_group_not_beeing_in(self):
"""Leaving a group the user is not in must fail"""
player_2 = uuid4()
groupid = lruc(KVS.create_group(self.user.userid, self.game.gameid))
lruc(KVS.join_group(groupid, player_2))
lruc(KVS.leave_group(self.user.userid))
coro = KVS.leave_group(self.user.userid)
self.assertRaises(PlayerNotInGroup, lruc, coro)
def test_clear_ready_while_queued(self):
"""Mark a player as not ready while in queue should leave the queue"""
groupid = lruc(KVS.create_group(self.user.userid, self.game.gameid))
lruc(KVS.mark_as_ready(self.user.userid))
lruc(KVS.join_queue(groupid))
lruc(KVS.mark_as_not_ready(self.user.userid))
group = lruc(KVS.get_group(groupid))
self.assertEqual(group.state, State.GROUP_CHECK)
def test_mark_unmark_ready(self):
"""Test mark_as_ready/mark_as_not_ready on a couple of game state"""
coro = KVS.mark_as_ready(self.user.userid)
self.assertRaises(PlayerNotInGroup, lruc, coro)
groupid = lruc(KVS.create_group(self.user.userid, self.game.gameid))
lruc(KVS.mark_as_ready(self.user.userid))
lruc(KVS.join_queue(groupid))
coro = KVS.mark_as_ready(self.user.userid)
self.assertRaises(WrongGroupState, lruc, coro)
lruc(KVS.mark_as_not_ready(self.user.userid))
def setUp(self):
try:
if config.webapi.PRODUCTION:
lruc(
asyncio.gather(connect_to_postgres(None, loop),
connect_to_redis(None, loop)))
else:
RDB(SQLite())
KVS(InMemory())
userid = uuid4()
lruc(
RDB.create_user(userid, "toto", "*****@*****.**",
b"suchpasssword"))
self.user = lruc(RDB.get_user_by_id(userid))
gameid = lruc(
RDB.create_game("Shifumi", self.user.userid, 4,
"shifumi-server", [22451]))
self.game = lruc(RDB.get_game_by_id(gameid))
except:
logger.exception("Error in setUp KVS")
def tearDown(self):
try:
if config.webapi.PRODUCTION:
conn = lruc(RDB.pool.acquire())
lruc(conn.execute("TRUNCATE tbusers CASCADE"))
lruc(conn.execute("TRUNCATE tbgames CASCADE"))
lruc(RDB.pool.release(conn))
lruc(KVS.redis.flushdb())
lruc(
asyncio.gather(disconnect_from_postgres(None, loop),
disconnect_from_redis(None, loop)))
else:
RDB.conn.close()
self.user = None
self.game = None
except:
logger.exception("Error in teamDown RDB")
def test_join_leave_queue(self):
"""Join the queue and leave it a couple of time should pass"""
groupid = lruc(KVS.create_group(self.user.userid, self.game.gameid))
lruc(KVS.mark_as_ready(self.user.userid))
group = lruc(KVS.get_group(groupid))
self.assertEqual(group.state, State.GROUP_CHECK)
# Join
lruc(KVS.join_queue(groupid))
group = lruc(KVS.get_group(groupid))
self.assertEqual(group.state, State.IN_QUEUE)
# Leave
lruc(KVS.leave_queue(groupid))
group = lruc(KVS.get_group(groupid))
self.assertEqual(group.state, State.GROUP_CHECK)
# Join
lruc(KVS.join_queue(groupid))
group = lruc(KVS.get_group(groupid))
self.assertEqual(group.state, State.IN_QUEUE)
# Leave
lruc(KVS.leave_queue(groupid))
group = lruc(KVS.get_group(groupid))
self.assertEqual(group.state, State.GROUP_CHECK)
def test_join_group_while_in_group(self):
"""Joining a group beeing in a group must pass"""
groupid = lruc(KVS.create_group(self.user.userid, self.game.gameid))
coro = KVS.join_group(groupid, self.user.userid)
self.assertRaises(PlayerInGroupAlready, lruc, coro)
def test_create_group_while_in_group(self):
"""Creating a group beeing in a group must fail"""
lruc(KVS.create_group(self.user.userid, self.game.gameid))
coro = KVS.create_group(self.user.userid, self.game.gameid)
self.assertRaises(PlayerInGroupAlready, lruc, coro)
def test_create_group_while_alone(self):
"""Creating a group beeing alone must pass"""
coro = KVS.create_group(self.user.userid, self.game.gameid)
self.assertTrue(isinstance(lruc(coro), UUID))
def test_start_game(self):
"""Make a match out of several groups of players"""
KVS.start_game = CoroutineMock()
player_1 = uuid4()
player_2 = uuid4()
player_3 = uuid4()
player_4 = uuid4()
player_5 = uuid4()
player_6 = uuid4()
group_1 = lruc(KVS.create_group(player_1, self.game.gameid))
lruc(KVS.mark_as_ready(player_1))
lruc(KVS.join_group(group_1, player_2))
lruc(KVS.mark_as_ready(player_2))
lruc(KVS.join_group(group_1, player_3))
lruc(KVS.mark_as_ready(player_3))
group_2 = lruc(KVS.create_group(player_4, self.game.gameid))
lruc(KVS.mark_as_ready(player_4))
lruc(KVS.join_group(group_2, player_5))
lruc(KVS.mark_as_ready(player_5))
group_3 = lruc(KVS.create_group(player_6, self.game.gameid))
lruc(KVS.mark_as_ready(player_6))
# 3 players join, need 4 to start
queue_filled = lruc(KVS.join_queue(group_1))
# 2 more players join, need 4 (3 & 2 don't fit 4)
queue_filled = lruc(KVS.join_queue(group_2))
# 1 more player join (3 & 1 fit)
queue_filled = lruc(KVS.join_queue(group_3))
group = lruc(KVS.get_group(group_3))
KVS.start_game.assert_called_once_with(self.game.gameid, group.slotid)