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)
