-
Notifications
You must be signed in to change notification settings - Fork 14
/
test_clientcomms.py
100 lines (89 loc) · 3.72 KB
/
test_clientcomms.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
import unittest
import sys
import os
from collections import OrderedDict
sys.path.append(os.path.join(os.path.dirname(__file__), "..", ".."))
from pkg_resources import require
require("mock")
from mock import Mock
from malcolm.core.clientcomms import ClientComms, CLIENT_STOP
from malcolm.core.syncfactory import SyncFactory
class TestClientComms(unittest.TestCase):
def test_init(self):
process = Mock()
client = ClientComms("c", process)
process.create_queue.assert_called_once_with()
self.assertEqual(client.q, process.create_queue.return_value)
def test_not_implemented_error(self):
client = ClientComms("c", Mock())
self.assertRaises(NotImplementedError, client.send_to_server, Mock())
self.assertRaises(NotImplementedError, client.start_recv_loop)
self.assertRaises(NotImplementedError, client.stop_recv_loop)
def test_send_logs_error(self):
client = ClientComms("c", Mock())
client.send_to_server = Mock(side_effect=Exception)
request = Mock()
request.to_dict = Mock(return_value = "<to_dict>")
client.q.get = Mock(side_effect = [request, CLIENT_STOP])
client.log_exception = Mock()
client.send_loop()
client.log_exception.assert_called_once_with(
"Exception sending request %s", "<to_dict>")
def test_requests_are_stored(self):
client = ClientComms("c", Mock())
client._current_id = 1234
request = Mock()
client.send_to_server = Mock()
client.q.get = Mock(side_effect = [request, CLIENT_STOP])
client.send_loop()
expected = OrderedDict({1234 : request})
self.assertEquals(expected, client.requests)
def test_loop_starts(self):
process = Mock(spawn = lambda x: x())
client = ClientComms("c", process)
client.send_loop = Mock()
client.start_recv_loop = Mock()
client.log_exception = Mock()
client.start()
client.send_loop.assert_called_once_with()
client.start_recv_loop.assert_called_once_with()
client.log_exception.assert_not_called()
def test_sends_to_server(self):
client = ClientComms("c", Mock())
client.send_to_server = Mock()
request = Mock()
client.q.get = Mock(side_effect = [request, CLIENT_STOP])
client.log_exception = Mock()
client.send_loop()
client.send_to_server.assert_called_once_with(request)
client.log_exception.assert_not_called()
def test_start_stop(self):
sync_factory = SyncFactory("s")
process = Mock()
process.spawn = sync_factory.spawn
process.create_queue = sync_factory.create_queue
client = ClientComms("c", process)
client.send_loop = Mock(side_effect = client.send_loop)
client.start_recv_loop = Mock()
client.stop_recv_loop = Mock()
client.log_exception = Mock()
client.start()
self.assertFalse(client._send_spawned.ready())
client.start_recv_loop.assert_called_once_with()
client.stop(0.1)
self.assertTrue(client._send_spawned.ready())
client.send_loop.assert_called_once_with()
client.stop_recv_loop.assert_called_once_with()
client.log_exception.assert_not_called()
def test_request_id_provided(self):
client = ClientComms("c", Mock())
client._current_id = 1234
client.send_to_server = Mock()
request_1 = Mock(id_ = None)
request_2 = Mock(id_ = None)
client.q.get = Mock(side_effect = [request_1, request_2, CLIENT_STOP])
client.send_loop()
self.assertEqual(1234, request_1.id_)
self.assertEqual(1235, request_2.id_)
if __name__ == "__main__":
unittest.main(verbosity=2)