def _makeLobby(self, name='test', owner=None, server_address='test',
game_map='test', password='******', teams=lambda: []):
from lobbypy.models import Lobby
if owner is None:
from lobbypy.models import Player
owner = Player('')
return Lobby(name, owner, server_address, game_map, password, teams)
def _makeLobby(self,
owner=None,
name='Lobby',
server_address='server:9999',
game_map='cp_lololol',
password='******'):
from lobbypy.models import Lobby
if owner is None:
from lobbypy.models import Player
owner = Player('-1')
return Lobby(name, owner, server_address, game_map, password, teams=[])
def test_on_get_lobby_listing(self, magic_emit,
magic_make_lobby_dict, magic_Lobby):
from lobbypy.models import Lobby, Player
p = Player('0')
lobbies = [
Lobby('A', p, '', '', '', teams=[]),
Lobby('B', p, '', '', '', teams=[]),
]
magic_Lobby.query.all.return_value = lobbies
lobbies_dict = [
{'name': 'A'},
{'name': 'B'},
]
json_rvs = list(lobbies_dict)
def side_effect(*args, **kwargs):
return json_rvs.pop(0)
magic_make_lobby_dict.side_effect = side_effect
instance = self._makeOne()
rvs = instance.on_get_lobby_listing()
self.assertTrue(rvs[0])
self.assertEqual(rvs[1], lobbies_dict)
def post(self):
o_id = request.form['owner_id']
name = request.form['name']
server_address = request.form['server_address']
game_map = request.form['game_map']
password = request.form['password']
o = Player.query.get(o_id)
if not o:
abort(405)
l = Lobby(name, o, server_address, game_map, password)
db.session.add(l)
db.session.commit()
return jsonify(201, id=l.id)
def test_with_owned_lobbies(self):
from lobbypy.models import Player, Lobby
p = Player('0')
db.session.add(p)
l = Lobby('test', p, 'test', 'test', 'test', teams=[])
db.session.add(l)
db.session.commit()
rv = self._callFUT(p)
l_dels = []
for l_del in rv:
l_dels.append(l_del)
self.assertEqual(len(l_dels), 1)
self.assertEqual(l_dels[0], (l, True))
def test_with_speced_lobbies(self):
from lobbypy.models import Player, Lobby
o = Player('1')
p = Player('0')
db.session.add(o)
db.session.add(p)
l = Lobby('test', o, 'test', 'test', 'test', teams=[])
l.spectators.append(p)
db.session.add(l)
db.session.commit()
rv = self._callFUT(p)
l_dels = []
for l_del in rv:
l_dels.append(l_del)
self.assertEqual(len(l_dels), 1)
self.assertEqual(l_dels[0], (l, False))
def test_with_teamed_lobbies(self):
from lobbypy.models import Player, Lobby, Team
o = Player('1')
p = Player('0')
db.session.add(o)
db.session.add(p)
l = Lobby('test', o, 'test', 'test', 'test', teams=[])
t = Team('Red')
l.teams.append(t)
l.teams[0].join(p)
db.session.add(l)
db.session.commit()
rv = self._callFUT(p)
l_dels = []
for l_del in rv:
l_dels.append(l_del)
self.assertEqual(len(l_dels), 1)
self.assertEqual(l_dels[0], (l, False))
