|
1
|
|
|
import asyncio |
|
|
|
|
|
|
2
|
|
|
from unittest import mock |
|
3
|
|
|
import pytest |
|
4
|
|
|
|
|
5
|
|
|
from server import LobbyConnection |
|
6
|
|
|
|
|
7
|
|
|
|
|
8
|
|
|
@pytest.fixture() |
|
9
|
|
|
def lobbythread(): |
|
10
|
|
|
return mock.Mock( |
|
11
|
|
|
sendJSON=lambda obj: None |
|
12
|
|
|
) |
|
13
|
|
|
|
|
14
|
|
|
|
|
15
|
|
View Code Duplication |
@pytest.fixture |
|
|
|
|
|
|
16
|
|
|
def game_connection(request, game, loop, player_service, players, game_service, transport): |
|
17
|
|
|
from server import GameConnection, LobbyConnection |
|
|
|
|
|
|
18
|
|
|
conn = GameConnection(loop=loop, |
|
19
|
|
|
lobby_connection=mock.create_autospec(LobbyConnection(loop)), |
|
20
|
|
|
player_service=player_service, |
|
21
|
|
|
games=game_service) |
|
22
|
|
|
conn._transport = transport |
|
|
|
|
|
|
23
|
|
|
conn.player = players.hosting |
|
24
|
|
|
conn.game = game |
|
25
|
|
|
conn.lobby = mock.Mock(spec=LobbyConnection) |
|
26
|
|
|
|
|
27
|
|
|
def fin(): |
|
28
|
|
|
conn.abort() |
|
29
|
|
|
|
|
30
|
|
|
request.addfinalizer(fin) |
|
31
|
|
|
return conn |
|
32
|
|
|
|
|
33
|
|
|
|
|
34
|
|
View Code Duplication |
@pytest.fixture |
|
|
|
|
|
|
35
|
|
|
def connections(loop, player_service, game_service, transport, game): |
|
36
|
|
|
from server import GameConnection |
|
37
|
|
|
|
|
38
|
|
|
def make_connection(player, connectivity): |
|
39
|
|
|
lc = LobbyConnection(loop) |
|
40
|
|
|
lc.protocol = mock.Mock() |
|
41
|
|
|
conn = GameConnection(loop=loop, |
|
42
|
|
|
lobby_connection=lc, |
|
43
|
|
|
player_service=player_service, |
|
44
|
|
|
games=game_service) |
|
45
|
|
|
conn.player = player |
|
46
|
|
|
conn.game = game |
|
47
|
|
|
conn._transport = transport |
|
|
|
|
|
|
48
|
|
|
conn._connectivity_state.set_result(connectivity) |
|
|
|
|
|
|
49
|
|
|
return conn |
|
50
|
|
|
|
|
51
|
|
|
return mock.Mock( |
|
52
|
|
|
make_connection=make_connection |
|
53
|
|
|
) |
|
54
|
|
|
|