Skip to content

Commit 129be47

Browse files
committed
Cleanup test fixture syntax and imports
1 parent b49e9ec commit 129be47

5 files changed

Lines changed: 16 additions & 35 deletions

File tree

tests/conftest.py

Lines changed: 5 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -147,12 +147,6 @@ def state_machine(container: ApplicationContainer) -> StateMachine:
147147
return container.state_machine()
148148

149149

150-
@pytest.fixture()
151-
def robot() -> StubRobot:
152-
"""Fixture to provide a mock robot instance."""
153-
return StubRobot()
154-
155-
156150
@pytest.fixture()
157151
def uploader(container: ApplicationContainer) -> Uploader:
158152
"""Fixture to provide a mock robot instance."""
@@ -170,7 +164,7 @@ def scheduling_utilities(
170164
)
171165

172166

173-
@pytest.fixture
167+
@pytest.fixture()
174168
def state_machine_thread(
175169
container: ApplicationContainer,
176170
mocker: MockerFixture,
@@ -183,7 +177,7 @@ def state_machine_thread(
183177
state_machine_thread.join()
184178

185179

186-
@pytest.fixture
180+
@pytest.fixture()
187181
def state_machine_thread_with_db(
188182
setup_db_connection_string: str,
189183
container: ApplicationContainer,
@@ -202,7 +196,7 @@ def state_machine_thread_with_db(
202196
state_machine_thread.join()
203197

204198

205-
@pytest.fixture
199+
@pytest.fixture()
206200
def robot_service_thread(
207201
container: ApplicationContainer,
208202
) -> Generator[RobotServiceThreadMock]:
@@ -219,7 +213,7 @@ def robot_service_thread(
219213
robot_service_thread.join()
220214

221215

222-
@pytest.fixture
216+
@pytest.fixture()
223217
def robot_inspection_service_thread(
224218
container: ApplicationContainer,
225219
) -> Generator[Thread]:
@@ -237,7 +231,7 @@ def robot_inspection_service_thread(
237231
robot_inspection_service_thread.join()
238232

239233

240-
@pytest.fixture
234+
@pytest.fixture()
241235
def mocked_robot_service(
242236
container: ApplicationContainer, mocker: MockerFixture
243237
) -> RobotService:

tests/isar/services/utilities/test_scheduling_utilities.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88
from isar.apis.models.models import ControlMissionResponse
99
from isar.config.settings import settings
1010
from isar.models.events import APIEvent, Event, EventTimeoutError
11-
from isar.modules import ApplicationContainer
1211
from isar.services.utilities.scheduling_utilities import SchedulingUtilities
1312
from isar.state_machine.states.await_next_mission import AwaitNextMission
1413
from isar.state_machine.states.going_to_lockdown import GoingToLockdown
@@ -254,9 +253,8 @@ def test_return_home_twice_causes_conflict(
254253

255254

256255
def test_api_with_unsuccessful_return_home_stop(
257-
mocker: MockerFixture, container: ApplicationContainer
256+
mocker: MockerFixture, scheduling_utilities: SchedulingUtilities
258257
) -> None:
259-
scheduling_utilities: SchedulingUtilities = container.scheduling_utilities()
260258
stopped_mission_response: ControlMissionResponse = ControlMissionResponse(
261259
success=False, failure_reason="ISAR failed to stop mission"
262260
)

tests/isar/state_machine/states/test_await_next_mission_state.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44

55
from isar.config.settings import settings
66
from isar.models.events import Events
7-
from isar.modules import ApplicationContainer
87
from isar.services.utilities.scheduling_utilities import SchedulingUtilities
98
from isar.state_machine.state import EventHandlerMapping
109
from isar.state_machine.states.resuming import Resuming
@@ -24,7 +23,7 @@
2423

2524

2625
def test_state_machine_with_successful_mission_stop(
27-
container: ApplicationContainer,
26+
scheduling_utilities: SchedulingUtilities,
2827
robot_service_thread: RobotServiceThreadMock,
2928
state_machine_thread: StateMachineThreadMock,
3029
mocker: MockerFixture,
@@ -45,8 +44,6 @@ def test_state_machine_with_successful_mission_stop(
4544
tasks=[StubTask.take_image() for _ in range(20)],
4645
)
4746

48-
scheduling_utilities: SchedulingUtilities = container.scheduling_utilities()
49-
5047
state_machine_thread.start()
5148
robot_service_thread.start()
5249
wait_until(

tests/isar/state_machine/states/test_monitor_state.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88

99
from isar.config.settings import settings
1010
from isar.models.events import EmptyMessage, Events
11-
from isar.modules import ApplicationContainer
1211
from isar.services.utilities.scheduling_utilities import SchedulingUtilities
1312
from isar.state_machine.state import EventHandlerMapping
1413
from isar.state_machine.states.monitor import Monitor
@@ -121,7 +120,7 @@ def test_transition_from_resuming_to_monitor(events: Events) -> None:
121120

122121

123122
def test_state_machine_with_unsuccessful_mission_stop(
124-
container: ApplicationContainer,
123+
scheduling_utilities: SchedulingUtilities,
125124
mocker: MockerFixture,
126125
state_machine_thread: StateMachineThreadMock,
127126
robot_service_thread: RobotServiceThreadMock,
@@ -131,7 +130,6 @@ def test_state_machine_with_unsuccessful_mission_stop(
131130
id="id", name="Dummy misson", tasks=[StubTask.take_image()]
132131
)
133132

134-
scheduling_utilities: SchedulingUtilities = container.scheduling_utilities()
135133
mocker.patch.object(
136134
StubRobot, "mission_status", return_value=MissionStatus.InProgress
137135
)
@@ -168,7 +166,7 @@ def test_state_machine_with_unsuccessful_mission_stop(
168166

169167

170168
def test_state_machine_with_unsuccessful_mission_stop_with_mission_id(
171-
container: ApplicationContainer,
169+
scheduling_utilities: SchedulingUtilities,
172170
mocker: MockerFixture,
173171
state_machine_thread: StateMachineThreadMock,
174172
robot_service_thread: RobotServiceThreadMock,
@@ -180,7 +178,6 @@ def test_state_machine_with_unsuccessful_mission_stop_with_mission_id(
180178
id="id", name="Dummy misson", tasks=[StubTask.take_image()]
181179
)
182180

183-
scheduling_utilities: SchedulingUtilities = container.scheduling_utilities()
184181
mocker.patch.object(
185182
StubRobot, "mission_status", return_value=MissionStatus.InProgress
186183
)
@@ -217,7 +214,7 @@ def test_state_machine_with_unsuccessful_mission_stop_with_mission_id(
217214

218215

219216
def test_robot_mission_status_exception_handling(
220-
container: ApplicationContainer,
217+
scheduling_utilities: SchedulingUtilities,
221218
state_machine_thread: StateMachineThreadMock,
222219
robot_service_thread: RobotServiceThreadMock,
223220
) -> None:
@@ -226,7 +223,6 @@ def test_robot_mission_status_exception_handling(
226223
name="Dummy mission",
227224
tasks=[StubTask.take_image()],
228225
)
229-
scheduling_utilities: SchedulingUtilities = container.scheduling_utilities()
230226

231227
robot_service_thread.robot_service.robot = StubRobotMissionStatusRaisesException()
232228

tests/isar/state_machine/test_integration_test_for_states.py

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,9 @@
2727

2828

2929
def test_state_machine_transitions_when_running_full_mission(
30-
container: ApplicationContainer,
3130
state_machine_thread: StateMachineThreadMock,
3231
robot_service_thread: RobotServiceThreadMock,
32+
scheduling_utilities: SchedulingUtilities,
3333
mocker: MockerFixture,
3434
) -> None:
3535
mocker.patch.object(settings, "RETURN_HOME_DELAY", 5.0)
@@ -61,7 +61,6 @@ def test_state_machine_transitions_when_running_full_mission(
6161
)
6262
mission: Mission = Mission(id="id", name="Dummy mission", tasks=[task_1, task_2])
6363

64-
scheduling_utilities: SchedulingUtilities = container.scheduling_utilities()
6564
scheduling_utilities.start_mission(mission=mission)
6665

6766
expected_transitions = deque(
@@ -82,7 +81,7 @@ def test_state_machine_transitions_when_running_full_mission(
8281

8382

8483
def test_state_machine_failed_dependency(
85-
container: ApplicationContainer,
84+
scheduling_utilities: SchedulingUtilities,
8685
state_machine_thread: StateMachineThreadMock,
8786
robot_service_thread: RobotServiceThreadMock,
8887
mocker: MockerFixture,
@@ -109,7 +108,6 @@ def test_state_machine_failed_dependency(
109108
in state_machine_thread.state_machine.transitions_list,
110109
timeout=10,
111110
)
112-
scheduling_utilities: SchedulingUtilities = container.scheduling_utilities()
113111
scheduling_utilities.start_mission(mission=mission)
114112

115113
expected_transitions = deque(
@@ -135,6 +133,7 @@ def test_state_machine_failed_dependency(
135133

136134
def test_state_machine_with_successful_collection(
137135
container: ApplicationContainer,
136+
scheduling_utilities: SchedulingUtilities,
138137
state_machine_thread: StateMachineThreadMock,
139138
robot_service_thread: RobotServiceThreadMock,
140139
robot_inspection_service_thread: Thread,
@@ -153,7 +152,6 @@ def test_state_machine_with_successful_collection(
153152
mission: Mission = Mission(
154153
id="id", name="Dummy misson", tasks=[StubTask.take_image()]
155154
)
156-
scheduling_utilities: SchedulingUtilities = container.scheduling_utilities()
157155

158156
mocker.patch.object(settings, "RETURN_HOME_DELAY", 2.0)
159157
state_machine_thread.start()
@@ -184,6 +182,7 @@ def test_state_machine_with_successful_collection(
184182

185183
def test_state_machine_with_unsuccessful_collection(
186184
container: ApplicationContainer,
185+
scheduling_utilities: SchedulingUtilities,
187186
mocker: MockerFixture,
188187
state_machine_thread: StateMachineThreadMock,
189188
robot_service_thread: RobotServiceThreadMock,
@@ -208,7 +207,6 @@ def test_state_machine_with_unsuccessful_collection(
208207
mission: Mission = Mission(
209208
id="id", name="Dummy misson", tasks=[StubTask.take_image()]
210209
)
211-
scheduling_utilities: SchedulingUtilities = container.scheduling_utilities()
212210
scheduling_utilities.start_mission(mission=mission)
213211

214212
expected_transitions = deque(
@@ -233,7 +231,7 @@ def test_state_machine_with_unsuccessful_collection(
233231

234232

235233
def test_state_machine_with_mission_start_during_return_home_without_queueing_stop_response(
236-
container: ApplicationContainer,
234+
scheduling_utilities: SchedulingUtilities,
237235
mocker: MockerFixture,
238236
state_machine_thread: StateMachineThreadMock,
239237
robot_service_thread: RobotServiceThreadMock,
@@ -242,7 +240,6 @@ def test_state_machine_with_mission_start_during_return_home_without_queueing_st
242240
mission: Mission = Mission(
243241
id="id", name="Dummy misson", tasks=[StubTask.take_image()]
244242
)
245-
scheduling_utilities: SchedulingUtilities = container.scheduling_utilities()
246243
mocker.patch.object(
247244
StubRobot, "mission_status", return_value=MissionStatus.InProgress
248245
)
@@ -280,7 +277,7 @@ def test_state_machine_with_mission_start_during_return_home_without_queueing_st
280277

281278

282279
def test_state_machine_failed_to_initiate_mission_and_return_home(
283-
container: ApplicationContainer,
280+
scheduling_utilities: SchedulingUtilities,
284281
state_machine_thread: StateMachineThreadMock,
285282
robot_service_thread: RobotServiceThreadMock,
286283
mocker: MockerFixture,
@@ -309,7 +306,6 @@ def test_state_machine_failed_to_initiate_mission_and_return_home(
309306
timeout=10,
310307
)
311308

312-
scheduling_utilities: SchedulingUtilities = container.scheduling_utilities()
313309
scheduling_utilities.start_mission(mission=mission)
314310

315311
expected_transitions = deque(

0 commit comments

Comments
 (0)