@@ -23,140 +23,134 @@ def mock_serial(mocker):
23
23
return serial_patch
24
24
25
25
26
+ @pytest .fixture
27
+ def mock_handler (mocker , mock_serial ):
28
+ mocker .patch ("PSL.packet_handler.Handler._check_udev" )
29
+ return Handler ()
30
+
31
+
26
32
@pytest .fixture
27
33
def mock_list_ports (mocker ):
28
34
return mocker .patch ("PSL.packet_handler.list_ports" )
29
35
30
36
31
- def test_connect_scan_port (mock_serial , mock_list_ports ):
37
+ def test_connect_scan_port (mocker , mock_serial , mock_list_ports ):
32
38
mock_serial ().is_open = False
33
39
mock_list_ports .grep .return_value = mock_ListPortInfo ()
40
+ mocker .patch ("PSL.packet_handler.Handler._check_udev" )
34
41
Handler ()
35
42
mock_serial ().open .assert_called ()
36
43
37
44
38
- def test_connect_scan_failure (mock_serial , mock_list_ports ):
45
+ def test_connect_scan_failure (mocker , mock_serial , mock_list_ports ):
39
46
mock_serial ().is_open = False
40
47
mock_list_ports .grep .return_value = mock_ListPortInfo (found = False )
48
+ mocker .patch ("PSL.packet_handler.Handler._check_udev" )
41
49
with pytest .raises (SerialException ):
42
50
Handler ()
43
51
44
52
45
- def test_disconnect (mock_serial ):
46
- H = Handler ()
47
- H .disconnect ()
53
+ def test_disconnect (mock_serial , mock_handler ):
54
+ mock_handler .disconnect ()
48
55
mock_serial ().close .assert_called ()
49
56
50
57
51
- def test_reconnect (mock_serial ):
52
- H = Handler ()
53
- H .reconnect ()
58
+ def test_reconnect (mock_serial , mock_handler ):
59
+ mock_handler .reconnect ()
54
60
mock_serial ().close .assert_called ()
55
61
56
62
57
- def test_get_version (mock_serial ):
58
- H = Handler ()
63
+ def test_get_version (mock_serial , mock_handler ):
64
+ mock_handler . get_version ()
59
65
mock_serial ().write .assert_called_with (CP .GET_VERSION )
60
- assert H .version == VERSION
66
+ assert mock_handler .version == VERSION
61
67
62
68
63
- def test_get_ack_success (mock_serial ):
69
+ def test_get_ack_success (mock_serial , mock_handler ):
64
70
H = Handler ()
65
71
success = 1
66
72
mock_serial ().read .return_value = CP .Byte .pack (success )
67
73
assert H .get_ack () == success
68
74
69
75
70
- def test_get_ack_burst_mode (mock_serial ):
71
- H = Handler ()
76
+ def test_get_ack_burst_mode (mock_serial , mock_handler ):
72
77
success = 1
73
- H .load_burst = True
74
- queue_size = H .input_queue_size
78
+ mock_handler .load_burst = True
79
+ queue_size = mock_handler .input_queue_size
75
80
mock_serial ().read .return_value = b""
76
81
77
- assert H .get_ack () == success
78
- assert H .input_queue_size == queue_size + 1
82
+ assert mock_handler .get_ack () == success
83
+ assert mock_handler .input_queue_size == queue_size + 1
79
84
80
85
81
- def test_get_ack_failure (mock_serial ):
82
- H = Handler ()
86
+ def test_get_ack_failure (mock_serial , mock_handler ):
83
87
error = 3
84
88
mock_serial ().read .return_value = b""
85
- assert H .get_ack () == error
89
+ assert mock_handler .get_ack () == error
86
90
87
91
88
- def test_send_bytes (mock_serial ):
89
- H = Handler ()
90
- H .send_byte (CP .Byte .pack (0xFF ))
92
+ def test_send_bytes (mock_serial , mock_handler ):
93
+ mock_handler .send_byte (CP .Byte .pack (0xFF ))
91
94
mock_serial ().write .assert_called_with (CP .Byte .pack (0xFF ))
92
95
93
96
94
- def test_send_byte (mock_serial ):
95
- H = Handler ()
96
- H .send_byte (0xFF )
97
+ def test_send_byte (mock_serial , mock_handler ):
98
+ mock_handler .send_byte (0xFF )
97
99
mock_serial ().write .assert_called_with (CP .Byte .pack (0xFF ))
98
100
99
101
100
- def test_send_byte_burst_mode (mock_serial ):
101
- H = Handler ()
102
- H .load_burst = True
103
- H .send_byte (0xFF )
104
- assert H .burst_buffer == CP .Byte .pack (0xFF )
102
+ def test_send_byte_burst_mode (mock_serial , mock_handler ):
103
+ mock_handler .load_burst = True
104
+ mock_handler .send_byte (0xFF )
105
+ assert mock_handler .burst_buffer == CP .Byte .pack (0xFF )
105
106
106
107
107
- def test_receive (mock_serial ):
108
- H = Handler ()
108
+ def test_receive (mock_serial , mock_handler ):
109
109
mock_serial ().read .return_value = CP .Byte .pack (0xFF )
110
- r = H .get_byte ()
110
+ r = mock_handler .get_byte ()
111
111
mock_serial ().read .assert_called_with (1 )
112
112
assert r == 0xFF
113
113
114
114
115
- def test_receive_failure (mock_serial ):
116
- H = Handler ()
115
+ def test_receive_failure (mock_serial , mock_handler ):
117
116
mock_serial ().read .return_value = b""
118
- r = H .get_byte ()
117
+ r = mock_handler .get_byte ()
119
118
mock_serial ().read .assert_called_with (1 )
120
119
assert r == - 1
121
120
122
121
123
- def test_wait_for_data (mock_serial ):
124
- H = Handler ()
122
+ def test_wait_for_data (mock_serial , mock_handler ):
125
123
mock_serial ().in_waiting = True
126
- assert H .wait_for_data ()
124
+ assert mock_handler .wait_for_data ()
127
125
128
126
129
- def test_wait_for_data_timeout (mock_serial ):
130
- H = Handler ()
127
+ def test_wait_for_data_timeout (mock_serial , mock_handler ):
131
128
mock_serial ().in_waiting = False
132
- assert not H .wait_for_data ()
129
+ assert not mock_handler .wait_for_data ()
133
130
134
131
135
- def test_send_burst (mock_serial ):
136
- H = Handler ()
137
- H .load_burst = True
132
+ def test_send_burst (mock_serial , mock_handler ):
133
+ mock_handler .load_burst = True
138
134
139
135
for b in b"abc" :
140
- H .send_byte (b )
141
- H .get_ack ()
136
+ mock_handler .send_byte (b )
137
+ mock_handler .get_ack ()
142
138
143
139
mock_serial ().read .return_value = b"\x01 \x01 \x01 "
144
- acks = H .send_burst ()
140
+ acks = mock_handler .send_burst ()
145
141
146
142
mock_serial ().write .assert_called_with (b"abc" )
147
143
mock_serial ().read .assert_called_with (3 )
148
- assert not H .load_burst
149
- assert H .burst_buffer == b""
150
- assert H .input_queue_size == 0
144
+ assert not mock_handler .load_burst
145
+ assert mock_handler .burst_buffer == b""
146
+ assert mock_handler .input_queue_size == 0
151
147
assert acks == [1 , 1 , 1 ]
152
148
153
149
154
- def test_get_integer_unsupported_size (mock_serial ):
155
- H = Handler ()
150
+ def test_get_integer_unsupported_size (mock_serial , mock_handler ):
156
151
with pytest .raises (ValueError ):
157
- H ._get_integer_type (size = 3 )
152
+ mock_handler ._get_integer_type (size = 3 )
158
153
159
154
160
- def test_list_ports (mock_serial ):
161
- H = Handler ()
162
- assert isinstance (H ._list_ports (), list )
155
+ def test_list_ports (mock_serial , mock_handler ):
156
+ assert isinstance (mock_handler ._list_ports (), list )
0 commit comments