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
|
from traffic_light.__main__ import main
from traffic_light.core import ID_PRODUCT_ORIGINAL
from unittest import mock
import pytest
@mock.patch("usb.core.find", return_value=mock.MagicMock(idProduct=ID_PRODUCT_ORIGINAL))
@pytest.mark.parametrize(
"test_input,expected",
[
(["--all", "on"], 0),
(["--all", "off"], 0),
(["-a", "off"], 0),
],
)
def test_set_all_leds(usb_find_mock, test_input, expected):
assert main(test_input) == expected
@pytest.mark.parametrize("test_input", [(["--all", "foo"]), (["--blue", "on"])])
def test_should_fail_for_wrong_argument_and_value(test_input):
with pytest.raises(SystemExit):
main(test_input)
@mock.patch("usb.core.find", return_value=mock.MagicMock(idProduct=ID_PRODUCT_ORIGINAL))
@pytest.mark.parametrize(
"test_input,expected",
[
(["--red", "on"], 0),
(["--red", "off"], 0),
(["-r", "off"], 0),
(["--yellow", "on"], 0),
(["--yellow", "off"], 0),
(["-y", "off"], 0),
(["--green", "on"], 0),
(["--green", "off"], 0),
(["-g", "off"], 0),
],
)
def test_set_diffrent_color_leds(usb_find_mock, test_input, expected):
assert main(test_input) == expected
@pytest.mark.parametrize(
"test_input,expected",
[
(["-adr", "2"], 1),
(["--address", "2"], 1),
],
)
def test_should_fail_for_no_given_color(test_input, expected):
assert main(test_input) == expected
@mock.patch("usb.core.find", return_value=mock.MagicMock(idProduct=ID_PRODUCT_ORIGINAL))
def test_should_fail_for_multiple_lights_whitout_given_address(usb_find_mock):
# given
def side_effect(*args, **kwargs):
if kwargs.get("find_all", False):
return [mock.MagicMock(), mock.MagicMock()]
else:
return mock.MagicMock()
# usb_find_mock.return_value = [mock.MagicMock(), mock.MagicMock()]
usb_find_mock.side_effect = side_effect
# then
assert main(["--red", "on"]) == 1
@mock.patch("usb.core.find", return_value=mock.MagicMock(idProduct=ID_PRODUCT_ORIGINAL))
def test_should_fail_for_no_connected_light(usb_find_mock):
# given
usb_find_mock.return_value = None
# then
assert main(["--red", "on"]) == 1
|
