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
101
102
103
104
105
106
107
108
109
110
|
# conftest -- Common test fixtures
# Copyright (C) 2021 Nicolas Schodet
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
from unittest.mock import Mock, patch
import pytest
import nxt.brick
def pytest_addoption(parser):
parser.addoption(
"--run-nxt",
action="append",
choices=("usb", "bluetooth", "devfile", "socket"),
default=[],
help="run tests needing a real NXT connected over given interface",
)
def pytest_configure(config):
config.addinivalue_line("markers", "nxt: mark test as needing a real NXT")
def pytest_collection_modifyitems(config, items):
run_nxt = config.getoption("--run-nxt")
for item in items:
for marker in item.iter_markers("nxt"):
if marker.args[0] not in run_nxt:
item.add_marker(
pytest.mark.skip(
reason="need --run-nxt=%s option to run" % marker.args[0]
)
)
@pytest.fixture
def mtime():
"""Mock time.time() and time.sleep()."""
current_time = 0
def timef():
nonlocal current_time
return current_time
def sleepf(delay):
nonlocal current_time
current_time += delay
mtime = Mock(spec_set=("time", "sleep"))
mtime.time.side_effect = timef
mtime.sleep.side_effect = sleepf
with (
patch("nxt.brick.time", new=mtime),
patch("nxt.motcont.time", new=mtime),
patch("nxt.motor.time", new=mtime),
patch("nxt.sensor.analog.time", new=mtime),
patch("nxt.sensor.digital.time", new=mtime),
):
yield mtime
def make_brick_mock():
b = Mock()
def find_files(pattern):
return nxt.brick.Brick.find_files(b, pattern)
def find_modules(pattern):
return nxt.brick.Brick.find_modules(b, pattern)
def open_file(*args, **kwargs):
return nxt.brick.Brick.open_file(b, *args, **kwargs)
def get_sensor(*args, **kwargs):
return nxt.brick.Brick.get_sensor(b, *args, **kwargs)
def get_motor(*args, **kwargs):
return nxt.brick.Brick.get_motor(b, *args, **kwargs)
b._sock.bsize = 60
b._sock.type = "usb"
b.find_files = find_files
b.find_modules = find_modules
b.open_file = open_file
b.get_sensor = get_sensor
b.get_motor = get_motor
return b
@pytest.fixture
def mbrick(mtime):
"""A brick with mocked low level functions."""
return make_brick_mock()
@pytest.fixture
def mbrick2(mtime):
"""A second brick with mocked low level functions."""
return make_brick_mock()
|
