Skip to content
Draft
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
60 changes: 60 additions & 0 deletions tests/interfaces/test_robotinterface.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,51 @@
from alitra import Frame, Position

from robot_interface.test_robot_interface import interface_test
from robot_interface.models.mission.mission import Mission
from robot_interface.models.mission.task import TakeImage

from isar_robot.robotinterface import Robot

from time import sleep

dummy_position = Position(x=0, y=0, z=0, frame=Frame("asset"))
dummy_inspection_task = TakeImage(id="dummy_task", target=dummy_position)
dummy_mission = Mission(id="dummy", tasks=[dummy_inspection_task], name="Dummy Mission")


def test_robotinterface():
interface_test(Robot())


def test_initiate_mission():
robot = Robot()
robot.initiate_mission(mission=dummy_mission)
assert robot.mission_simulation is not None


def test_task_status():
robot = Robot()
robot.initiate_mission(mission=dummy_mission)
status = robot.task_status(task_id="dummy_task")
assert status is not None


def test_stop():
robot = Robot()
robot.initiate_mission(mission=dummy_mission)
sleep(2) # Give some time for the mission to start
robot.stop()
sleep(2) # Give some time for the mission to stop
assert robot.mission_simulation.mission_done is True


def test_get_inspection():
robot = Robot()
robot.initiate_mission(mission=dummy_mission)
inspection = robot.get_inspection(task=dummy_inspection_task)
assert inspection is not None


def test_get_telemetry_publishers():
robot = Robot()
publishers = robot.get_telemetry_publishers(
Expand Down Expand Up @@ -45,3 +84,24 @@ def test_pressure_telemetry():
isar_id="test_id", robot_name="test_robot"
)
assert pressure_telemetry is not None


def test_robot_status():
robot = Robot()
status = robot.robot_status()
assert status is not None


def test_pause() -> None:
robot = Robot()
robot.initiate_mission(mission=dummy_mission)
robot.pause()
assert robot.mission_simulation.mission_done is False


def test_resume() -> None:
robot = Robot()
robot.initiate_mission(mission=dummy_mission)
robot.pause()
robot.resume()
assert robot.mission_simulation.mission_done is False
Loading