|
1 |
| -from utils import session, server |
| 1 | +from utils import session, threaded_server |
2 | 2 |
|
3 | 3 |
|
4 | 4 | def test_text_message_of_length_1(session):
|
5 |
| - client, server = session |
| 5 | + conn, server = session |
6 | 6 | server.send_message_to_all('$')
|
7 |
| - assert client.recv() == '$' |
| 7 | + assert conn.recv() == '$' |
8 | 8 |
|
9 | 9 |
|
10 | 10 | def test_text_message_of_length_125B(session):
|
11 |
| - client, server = session |
| 11 | + conn, server = session |
12 | 12 | msg = 'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz'\
|
13 | 13 | 'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz'\
|
14 | 14 | 'abcdefghijklmnopqr125'
|
15 | 15 | server.send_message_to_all(msg)
|
16 |
| - assert client.recv() == msg |
| 16 | + assert conn.recv() == msg |
17 | 17 |
|
18 | 18 |
|
19 | 19 | def test_text_message_of_length_126B(session):
|
20 |
| - client, server = session |
| 20 | + conn, server = session |
21 | 21 | msg = 'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz'\
|
22 | 22 | 'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz'\
|
23 | 23 | 'abcdefghijklmnopqrs126'
|
24 | 24 | server.send_message_to_all(msg)
|
25 |
| - assert client.recv() == msg |
| 25 | + assert conn.recv() == msg |
26 | 26 |
|
27 | 27 |
|
28 | 28 | def test_text_message_of_length_127B(session):
|
29 |
| - client, server = session |
| 29 | + conn, server = session |
30 | 30 | msg = 'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz'\
|
31 | 31 | 'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz'\
|
32 | 32 | 'abcdefghijklmnopqrst127'
|
33 | 33 | server.send_message_to_all(msg)
|
34 |
| - assert client.recv() == msg |
| 34 | + assert conn.recv() == msg |
35 | 35 |
|
36 | 36 |
|
37 | 37 | def test_text_message_of_length_208B(session):
|
38 |
| - client, server = session |
| 38 | + conn, server = session |
39 | 39 | msg = 'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz'\
|
40 | 40 | 'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz'\
|
41 | 41 | 'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz'\
|
42 | 42 | 'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvw208'
|
43 | 43 | server.send_message_to_all(msg)
|
44 |
| - assert client.recv() == msg |
| 44 | + assert conn.recv() == msg |
45 | 45 |
|
46 | 46 |
|
47 | 47 | def test_text_message_of_length_1251B(session):
|
48 |
| - client, server = session |
| 48 | + conn, server = session |
49 | 49 | msg = ('abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz'\
|
50 | 50 | 'abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz'\
|
51 | 51 | 'abcdefghijklmnopqr125'*10)+'1'
|
52 | 52 | server.send_message_to_all(msg)
|
53 |
| - assert client.recv() == msg |
| 53 | + assert conn.recv() == msg |
54 | 54 |
|
55 | 55 |
|
56 | 56 | def test_text_message_of_length_68KB(session):
|
57 |
| - client, server = session |
| 57 | + conn, server = session |
58 | 58 | msg = '$'+('a'*67993)+'68000'+'^'
|
59 | 59 | assert len(msg) == 68000
|
60 | 60 | server.send_message_to_all(msg)
|
61 |
| - assert client.recv() == msg |
| 61 | + assert conn.recv() == msg |
62 | 62 |
|
63 | 63 |
|
64 | 64 | def test_text_message_of_length_1500KB(session):
|
65 | 65 | """ An enormous message (well beyond 65K) """
|
66 |
| - client, server = session |
| 66 | + conn, server = session |
67 | 67 | msg = '$'+('a'*1499991)+'1500000'+'^'
|
68 | 68 | assert len(msg) == 1500000
|
69 | 69 | server.send_message_to_all(msg)
|
70 |
| - assert client.recv() == msg |
| 70 | + assert conn.recv() == msg |
71 | 71 |
|
72 | 72 |
|
73 | 73 | def test_text_message_with_unicode_characters(session):
|
74 |
| - client, server = session |
| 74 | + conn, server = session |
75 | 75 | msg = '$äüö^'
|
76 | 76 | server.send_message_to_all(msg)
|
77 |
| - assert client.recv() == msg |
| 77 | + assert conn.recv() == msg |
78 | 78 |
|
79 | 79 |
|
80 | 80 | def test_text_message_stress_bursts(session):
|
81 |
| - """ Scenario: server sends multiple different message to the same client |
| 81 | + """ Scenario: server sends multiple different message to the same conn |
82 | 82 | at once """
|
83 | 83 | from threading import Thread
|
84 | 84 | NUM_THREADS = 100
|
85 | 85 | MESSAGE_LEN = 1000
|
86 |
| - client, server = session |
| 86 | + conn, server = session |
87 | 87 | messages_received = []
|
88 | 88 |
|
89 | 89 | # Threads receing
|
90 | 90 | threads_receiving = []
|
91 | 91 | for i in range(NUM_THREADS):
|
92 | 92 | th = Thread(
|
93 | 93 | target=lambda fn: messages_received.append(fn()),
|
94 |
| - args=(client.recv,) |
| 94 | + args=(conn.recv,) |
95 | 95 | )
|
96 | 96 | th.daemon = True
|
97 | 97 | threads_receiving.append(th)
|
|
0 commit comments