def test_toggle_ready(self): from lobbypy.models import Player, Lobby player = self.session.query(Player).first() lobby = Lobby('Lobby', player, '', '', '') lobby.set_team(player, 0) self.session.add(lobby) transaction.commit() from lobbypy.controllers import toggle_ready lobby = self.session.merge(lobby) player = self.session.merge(player) toggle_ready(self.session, lobby, player) transaction.commit() lobby = self.session.merge(lobby) player = self.session.merge(player) self.assertTrue(lobby.teams[0].players[0].ready)
def test_set_class(self): from lobbypy.models import Player, Lobby player = self.session.query(Player).first() lobby = Lobby('Lobby', player, '', '', '') lobby.set_team(player, 0) self.session.add(lobby) transaction.commit() from lobbypy.controllers import set_class lobby = self.session.merge(lobby) player = self.session.merge(player) set_class(self.session, lobby, player, 1) transaction.commit() lobby = self.session.merge(lobby) player = self.session.merge(player) self.assertEquals(lobby.teams[0].players[0].cls, 1)