|
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