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
|
import pytest
from labgrid.driver import SerialDriver
from labgrid.exceptions import NoSupplierFoundError
class TestSerialDriver:
def test_instanziation_fail_missing_port(self, target):
with pytest.raises(NoSupplierFoundError):
SerialDriver(target, "serial")
def test_instanziation(self, target, serial_port, mocker):
serial_mock = mocker.patch('serial.Serial')
s = SerialDriver(target, "serial")
assert (isinstance(s, SerialDriver))
assert (target.drivers[0] == s)
def test_write(self, target, serial_port, mocker):
serial_mock = mocker.patch('serial.Serial')
s = SerialDriver(target, "serial")
target.activate(s)
s.write(b"testdata")
serial_mock.return_value.open.assert_called_once_with()
serial_mock.return_value.write.assert_called_once_with(b"testdata")
@pytest.mark.parametrize("param", [[1, 1, None, 1], # old test case
[3, 2, None, 3], [3, 2, 5, 3], [3, 2, 1, 1], [1, 2, 1, 1]])
# param = [size, in_waiting, max_size, out]
def test_read(self, target, serial_port, mocker, param):
serial_mock = mocker.patch('serial.Serial')
serial_mock.return_value.in_waiting = param[1]
s = SerialDriver(target, "serial")
target.activate(s)
s.read(size=param[0], max_size=param[2])
serial_mock.return_value.open.assert_called_once_with()
# assert 'read' called once with correct return:
serial_mock.return_value.read.assert_called_once_with(param[3])
def test_close(self, target, serial_port, mocker):
serial_mock = mocker.patch('serial.Serial')
s = SerialDriver(target, "serial")
target.activate(s)
s.close()
serial_mock.return_value.open.assert_called_once_with()
serial_mock.return_value.close.assert_called_once_with()
def test_deactivate(self, target, serial_port, mocker):
serial_mock = mocker.patch('serial.Serial')
s = SerialDriver(target, "serial")
target.activate(s)
target.deactivate(s)
serial_mock.return_value.open.assert_called_once_with()
serial_mock.return_value.close.assert_called_once_with()
def test_rfc2711_instanziation(self, target, serial_rfc2711_port, mocker):
serial_mock = mocker.patch('serial.rfc2217.Serial')
s = SerialDriver(target, "serial")
assert (isinstance(s, SerialDriver))
assert (target.drivers[0] == s)
serial_mock.assert_called_once_with()
def test_raw_instanziation(self, target, serial_raw_port, mocker):
serial_mock = mocker.patch('serial.serial_for_url')
s = SerialDriver(target, "serial")
assert (isinstance(s, SerialDriver))
assert (target.drivers[0] == s)
serial_mock.assert_called_once_with('socket://', do_not_open=True)
|
