def test_create_robot_with_docking_station(self, create):
robot = Robot.create(None, self.docking_station.id)
db_robot = Robot.get(robot.id)
create.assert_not_called()
self.assertIsNotNone(db_robot)
self.assertIsNotNone(db_robot.created_ts)
self.assertIsNone(db_robot.modified_ts)
self.assertIsNone(db_robot.route_id)
self.assertEqual(db_robot.docking_station_id, self.docking_station.id)
def get_all(cls, query_string=None):
from app.models.robot import Robot
query_string = json.loads(query_string) if query_string else {}
if not query_string.get('robot_id'):
return cls.query.all()
robot = Robot.get(query_string.get('robot_id'))
if not robot:
raise NotFound({
'resource_name': 'Robot',
'id': query_string.get('robot_id')
})
target_dt, start_dt, end_dt = validate_and_extract_datetimes(
query_string)
if target_dt:
robot.find_route_by_ts(target_dt)
return robot.find_routes_by_ts_range(start_ts=start_dt, end_ts=end_dt)
def test_delete_robot(self):
robot = Robot.create(None, self.docking_station.id)
robot.delete()
robot_db = Robot.get(robot.id)
self.assertIsNone(robot_db)
def test_get_nonexistent_robot(self):
robot = Robot.get(100)
self.assertIsNone(robot)
def test_get_robot(self):
robot = Robot.create(None, self.docking_station.id)
db_robot = Robot.get(robot.id)
self.assertEqual(db_robot.id, robot.id)