def sendMission(args, client):
interop_client = AsyncClient(args.interop_url,
args.interop_username,
args.interop_password)
mission = interop_client.get_mission(args.mission_id).result()
mission_json = json_format.MessageToJson(mission)
mission_dictionary = json.loads(mission_json)
print(mission_dictionary)
def test_timeout(self):
"""Test connection timeout"""
# We are assuming that there is no machine at this address.
addr = "http://10.255.255.254"
timeout = 0.0001
with self.assertRaises(requests.Timeout):
Client(addr, username, password, timeout=timeout)
with self.assertRaises(requests.Timeout):
AsyncClient(addr, username, password, timeout=timeout)
def main():
# Setup logging
logging.basicConfig(
level=logging.INFO,
stream=sys.stdout,
format='%(asctime)s: %(name)s: %(levelname)s: %(message)s')
# Parse command line args.
parser = argparse.ArgumentParser(description='AUVSI SUAS Interop CLI.')
parser.add_argument('--url',
required=True,
help='URL for interoperability.')
parser.add_argument('--username',
required=True,
help='Username for interoperability.')
parser.add_argument('--password', help='Password for interoperability.')
subparsers = parser.add_subparsers(help='Sub-command help.')
subparser = subparsers.add_parser('mission', help='Get mission details.')
subparser.set_defaults(func=mission)
subparser.add_argument('--mission_id',
type=int,
required=True,
help='ID of the mission to get.')
subparser = subparsers.add_parser('obstacles', help='Get obstaclesj.')
subparser.set_defaults(func=obstacles)
subparser = subparsers.add_parser(
'odlcs',
help='Upload odlcs.',
description='''Download or upload odlcs to/from the interoperability
server.
Without extra arguments, this prints all odlcs that have been uploaded to the
server.
With --odlc_dir, this uploads new odlcs to the server.
This tool searches for odlc JSON and images files within --odlc_dir
conforming to the 2017 Object File Format and uploads the odlc
characteristics and thumbnails to the interoperability server.
There is no deduplication logic. Odlcs will be uploaded multiple times, as
unique odlcs, if the tool is run multiple times.''',
formatter_class=argparse.RawDescriptionHelpFormatter)
subparser.set_defaults(func=odlcs)
subparser.add_argument(
'--odlc_dir',
help='Enables odlc upload. Directory containing odlc data.')
subparser = subparsers.add_parser('probe', help='Send dummy requests.')
subparser.set_defaults(func=probe)
subparser.add_argument('--interop_time',
type=float,
default=1.0,
help='Time between sent requests (sec).')
subparser = subparsers.add_parser(
'mavlink',
help='''Receive MAVLink GLOBAL_POSITION_INT packets and
forward as telemetry to interop server.''')
subparser.set_defaults(func=mavlink)
subparser.add_argument(
'--device',
type=str,
help='pymavlink device name to read from. E.g. tcp:localhost:8080.')
# Parse args, get password if not provided.
args = parser.parse_args()
if args.password:
password = args.password
else:
password = getpass.getpass('Interoperability Password: ')
# Create client and dispatch subcommand.
client = AsyncClient(args.url, args.username, password)
args.func(args, client)
async def load_mission(args, client):
interop_client = AsyncClient(args.interop_url,
args.interop_username,
args.interop_password)
await client.load_mission(args.mission_id, interop_client)
def setUp(self):
"""Create a logged in Client."""
# Test rest with non-admin clients.
self.client = Client(server, username, password)
self.async_client = AsyncClient(server, username, password)
class TestClient(unittest.TestCase):
"""Test the Client class.
The Client class is a very thin wrapper, so there is very little to test.
"""
def setUp(self):
"""Create a logged in Client."""
# Test rest with non-admin clients.
self.client = Client(server, username, password)
self.async_client = AsyncClient(server, username, password)
def test_get_mission(self):
"""Test getting a mission."""
mission = self.client.get_mission(1)
async_mission = self.async_client.get_mission(1).result()
# Check basic field info.
self.assertEqual(1, mission.id)
self.assertEqual(1, async_mission.id)
def test_post_telemetry(self):
"""Test sending some telemetry."""
t = interop_api_pb2.Telemetry()
t.latitude = 38
t.longitude = -76
t.altitude = 100
t.heading = 90
# Raises an exception on error.
self.client.post_telemetry(t)
self.async_client.post_telemetry(t).result()
def test_post_bad_telemetry(self):
"""Test sending some (incorrect) telemetry."""
t = interop_api_pb2.Telemetry()
t.latitude = 38
t.longitude = -76
t.altitude = 100
t.heading = 400 # Out of range.
with self.assertRaises(InteropError):
self.client.post_telemetry(t)
with self.assertRaises(InteropError):
self.async_client.post_telemetry(t).result()
def test_get_obstacles(self):
"""Test getting obstacles."""
obstacles = self.client.get_obstacles()
async_obstacles = self.async_client.get_obstacles().result()
# No exceptions is a good sign, let's see if the data matches the fixture.
self.assertEqual(2, len(obstacles.stationary_obstacles))
self.assertEqual(2, len(async_obstacles.stationary_obstacles))
radii = [o.radius for o in obstacles.stationary_obstacles]
async_radii = [o.radius for o in async_obstacles.stationary_obstacles]
self.assertIn(50, radii)
self.assertIn(50, async_radii)
self.assertIn(150, radii)
self.assertIn(150, async_radii)
heights = [o.height for o in obstacles.stationary_obstacles]
self.assertIn(300, heights)
self.assertIn(200, heights)
async_heights = [
o.height for o in async_obstacles.stationary_obstacles
]
self.assertIn(300, async_heights)
self.assertIn(200, async_heights)
def test_odlcs(self):
"""Test odlc workflow."""
# Post a odlc gets an updated odlc.
odlc = interop_api_pb2.Odlc()
odlc.type = interop_api_pb2.Odlc.STANDARD
post_odlc = self.client.post_odlc(odlc)
async_post_odlc = self.client.post_odlc(odlc)
self.assertIsNotNone(post_odlc.id)
self.assertIsNotNone(async_post_odlc.id)
self.assertEqual(interop_api_pb2.Odlc.STANDARD, post_odlc.type)
self.assertEqual(interop_api_pb2.Odlc.STANDARD, async_post_odlc.type)
self.assertNotEqual(post_odlc.id, async_post_odlc.id)
# Get odlcs.
get_odlc = self.client.get_odlc(post_odlc.id)
async_get_odlc = self.async_client.get_odlc(
async_post_odlc.id).result()
get_odlcs = self.client.get_odlcs()
async_get_odlcs = self.async_client.get_odlcs().result()
self.assertEquals(post_odlc, get_odlc)
self.assertEquals(async_post_odlc, async_get_odlc)
self.assertIn(post_odlc, get_odlcs)
self.assertIn(async_post_odlc, async_get_odlcs)
# Update odlc.
post_odlc.shape = interop_api_pb2.Odlc.CIRCLE
async_post_odlc.shape = interop_api_pb2.Odlc.CIRCLE
put_odlc = self.client.put_odlc(post_odlc.id, post_odlc)
async_put_odlc = self.async_client.put_odlc(async_post_odlc.id,
async_post_odlc).result()
self.assertEquals(post_odlc, put_odlc)
self.assertEquals(async_post_odlc, async_put_odlc)
# Upload odlc image.
test_image_filepath = os.path.join(os.path.dirname(__file__),
"testdata/A.jpg")
with open(test_image_filepath, 'rb') as f:
image_data = f.read()
self.client.put_odlc_image(post_odlc.id, image_data)
self.async_client.put_odlc_image(async_post_odlc.id,
image_data).result()
# Get the odlc image.
get_image = self.client.get_odlc_image(post_odlc.id)
async_get_image = self.async_client.get_odlc_image(
async_post_odlc.id).result()
self.assertEquals(image_data, get_image)
self.assertEquals(image_data, async_get_image)
# Delete the odlc image.
self.client.delete_odlc_image(post_odlc.id)
self.async_client.delete_odlc_image(async_post_odlc.id).result()
with self.assertRaises(InteropError):
self.client.get_odlc_image(post_odlc.id)
with self.assertRaises(InteropError):
self.async_client.get_odlc_image(async_post_odlc.id).result()
# Delete odlc.
self.client.delete_odlc(post_odlc.id)
self.async_client.delete_odlc(async_post_odlc.id).result()
self.assertNotIn(post_odlc, self.client.get_odlcs())
self.assertNotIn(async_post_odlc,
self.async_client.get_odlcs().result())
def test_bad_login(self):
"""Bad login raises exception"""
with self.assertRaises(InteropError):
Client(server, "foo", "bar")
with self.assertRaises(InteropError):
AsyncClient(server, "foo", "bar")
def test_login(self):
"""Simple login test."""
# Simply creating a Client causes a login.
# If it doesn't raise an exception, it worked!
Client(server, username, password)
AsyncClient(server, username, password)
class TestClient(unittest.TestCase):
"""Test the Client class.
The Client class is a very thin wrapper, so there is very little to test.
"""
def setUp(self):
"""Create a logged in Client."""
# Test rest with non-admin clients.
self.client = Client(server, username, password)
self.async_client = AsyncClient(server, username, password)
def test_get_teams(self):
"""Tests getting team status."""
teams = self.client.get_teams()
async_teams = self.async_client.get_teams().result()
self.assertEqual(1, len(teams))
self.assertEqual(1, len(async_teams))
self.assertEqual('testuser', teams[0].team.username)
self.assertEqual('testuser', async_teams[0].team.username)
def test_get_mission(self):
"""Test getting a mission."""
mission = self.client.get_mission(1)
async_mission = self.async_client.get_mission(1).result()
# Check basic field info.
self.assertEqual(1, mission.id)
self.assertEqual(1, async_mission.id)
def test_post_telemetry(self):
"""Test sending some telemetry."""
t = interop_api_pb2.Telemetry()
t.latitude = 38
t.longitude = -76
t.altitude = 100
t.heading = 90
# Raises an exception on error.
self.client.post_telemetry(t)
self.async_client.post_telemetry(t).result()
def test_post_bad_telemetry(self):
"""Test sending some (incorrect) telemetry."""
t = interop_api_pb2.Telemetry()
t.latitude = 38
t.longitude = -76
t.altitude = 100
t.heading = 400 # Out of range.
with self.assertRaises(InteropError):
self.client.post_telemetry(t)
with self.assertRaises(InteropError):
self.async_client.post_telemetry(t).result()
def test_odlcs(self):
"""Test odlc workflow."""
# Post a odlc gets an updated odlc.
odlc = interop_api_pb2.Odlc()
odlc.mission = 1
odlc.type = interop_api_pb2.Odlc.STANDARD
post_odlc = self.client.post_odlc(odlc)
async_post_odlc = self.async_client.post_odlc(odlc).result()
self.assertIsNotNone(post_odlc.id)
self.assertIsNotNone(async_post_odlc.id)
self.assertEqual(1, post_odlc.mission)
self.assertEqual(1, async_post_odlc.mission)
self.assertEqual(interop_api_pb2.Odlc.STANDARD, post_odlc.type)
self.assertEqual(interop_api_pb2.Odlc.STANDARD, async_post_odlc.type)
self.assertNotEqual(post_odlc.id, async_post_odlc.id)
# Get odlc.
get_odlc = self.client.get_odlc(post_odlc.id)
async_get_odlc = self.async_client.get_odlc(
async_post_odlc.id).result()
get_odlcs = self.client.get_odlcs()
async_get_odlcs = self.async_client.get_odlcs().result()
get_odlcs_mission = self.client.get_odlcs(mission=1)
async_get_odlcs_mission = self.async_client.get_odlcs(
mission=1).result()
get_odlcs_bad_mission = self.client.get_odlcs(mission=2)
async_get_odlcs_bad_mission = self.async_client.get_odlcs(
mission=2).result()
self.assertEquals(post_odlc, get_odlc)
self.assertEquals(async_post_odlc, async_get_odlc)
self.assertIn(post_odlc, get_odlcs)
self.assertIn(async_post_odlc, async_get_odlcs)
self.assertIn(post_odlc, get_odlcs_mission)
self.assertIn(async_post_odlc, async_get_odlcs_mission)
self.assertNotIn(post_odlc, get_odlcs_bad_mission)
self.assertNotIn(async_post_odlc, async_get_odlcs_bad_mission)
# Update odlc.
post_odlc.shape = interop_api_pb2.Odlc.CIRCLE
async_post_odlc.shape = interop_api_pb2.Odlc.CIRCLE
put_odlc = self.client.put_odlc(post_odlc.id, post_odlc)
async_put_odlc = self.async_client.put_odlc(async_post_odlc.id,
async_post_odlc).result()
self.assertEquals(post_odlc, put_odlc)
self.assertEquals(async_post_odlc, async_put_odlc)
# Upload odlc image.
test_image_filepath = os.path.join(os.path.dirname(__file__),
"testdata/A.jpg")
with open(test_image_filepath, 'rb') as f:
image_data = f.read()
self.client.put_odlc_image(post_odlc.id, image_data)
self.async_client.put_odlc_image(async_post_odlc.id,
image_data).result()
# Get the odlc image.
get_image = self.client.get_odlc_image(post_odlc.id)
async_get_image = self.async_client.get_odlc_image(
async_post_odlc.id).result()
self.assertEquals(image_data, get_image)
self.assertEquals(image_data, async_get_image)
# Delete the odlc image.
self.client.delete_odlc_image(post_odlc.id)
self.async_client.delete_odlc_image(async_post_odlc.id).result()
with self.assertRaises(InteropError):
self.client.get_odlc_image(post_odlc.id)
with self.assertRaises(InteropError):
self.async_client.get_odlc_image(async_post_odlc.id).result()
# Delete odlc.
self.client.delete_odlc(post_odlc.id)
self.async_client.delete_odlc(async_post_odlc.id).result()
self.assertNotIn(post_odlc, self.client.get_odlcs())
self.assertNotIn(async_post_odlc,
self.async_client.get_odlcs().result())
class TestClient(unittest.TestCase):
"""Test the Client class.
The Client class is a very thin wrapper, so there is very little to test.
"""
def setUp(self):
"""Create a logged in Client."""
# Test rest with non-admin clients.
self.client = Client(server, username, password)
self.async_client = AsyncClient(server, username, password)
def test_get_missions(self):
"""Test getting missions."""
missions = self.client.get_missions()
async_missions = self.async_client.get_missions().result()
# Check one mission returned.
self.assertEqual(1, len(missions))
self.assertEqual(1, len(async_missions))
# Check a few fields.
self.assertTrue(missions[0].active)
self.assertTrue(async_missions[0].active)
self.assertEqual(1, missions[0].id)
self.assertEqual(1, async_missions[0].id)
self.assertEqual(38.14792, missions[0].home_pos.latitude)
self.assertEqual(38.14792, async_missions[0].home_pos.latitude)
def test_post_telemetry(self):
"""Test sending some telemetry."""
t = Telemetry(latitude=38,
longitude=-76,
altitude_msl=100,
uas_heading=90)
# Raises an exception on error.
self.client.post_telemetry(t)
self.async_client.post_telemetry(t).result()
def test_post_bad_telemetry(self):
"""Test sending some (incorrect) telemetry."""
t0 = Telemetry(latitude=38,
longitude=-76,
altitude_msl=100,
uas_heading=90)
# The Telemetry constructor prevents us from passing invalid
# values, but we can still screw things up in an update
t0.latitude = 'baz'
with self.assertRaises(InteropError):
self.client.post_telemetry(t0)
with self.assertRaises(InteropError):
self.async_client.post_telemetry(t0).result()
# We only accept Telemetry objects (or objects that behave like
# Telemetry, not dicts.
t1 = {
'latitude': 38,
'longitude': -76,
'altitude_msl': 100,
'uas_heading': 90
}
with self.assertRaises(AttributeError):
self.client.post_telemetry(t1)
with self.assertRaises(AttributeError):
self.async_client.post_telemetry(t1).result()
def test_get_obstacles(self):
"""Test getting obstacles."""
stationary = self.client.get_obstacles()
async_future = self.async_client.get_obstacles()
async_stationary = async_future.result()
# No exceptions is a good sign, let's see if the data matches the fixture.
self.assertEqual(2, len(stationary))
self.assertEqual(2, len(async_stationary))
radii = [o.cylinder_radius for o in stationary]
async_radii = [o.cylinder_radius for o in async_stationary]
self.assertIn(50, radii)
self.assertIn(50, async_radii)
self.assertIn(150, radii)
self.assertIn(150, async_radii)
heights = [o.cylinder_height for o in stationary]
self.assertIn(300, heights)
self.assertIn(200, heights)
async_heights = [o.cylinder_height for o in async_stationary]
self.assertIn(300, async_heights)
self.assertIn(200, async_heights)
def test_odlcs(self):
"""Test odlc workflow."""
# Post a odlc gets an updated odlc.
odlc = Odlc(type='standard')
post_odlc = self.client.post_odlc(odlc)
async_post_odlc = self.client.post_odlc(odlc)
self.assertIsNotNone(post_odlc.id)
self.assertIsNotNone(async_post_odlc.id)
self.assertIsNotNone(post_odlc.user)
self.assertIsNotNone(async_post_odlc.user)
self.assertEqual('standard', post_odlc.type)
self.assertEqual('standard', async_post_odlc.type)
self.assertNotEqual(post_odlc.id, async_post_odlc.id)
# Get odlcs.
get_odlc = self.client.get_odlc(post_odlc.id)
async_get_odlc = self.async_client.get_odlc(
async_post_odlc.id).result()
get_odlcs = self.client.get_odlcs()
async_get_odlcs = self.async_client.get_odlcs().result()
self.assertEquals(post_odlc, get_odlc)
self.assertEquals(async_post_odlc, async_get_odlc)
self.assertIn(post_odlc, get_odlcs)
self.assertIn(async_post_odlc, async_get_odlcs)
# Update odlc.
post_odlc.shape = 'circle'
async_post_odlc.shape = 'circle'
put_odlc = self.client.put_odlc(post_odlc.id, post_odlc)
async_put_odlc = self.async_client.put_odlc(async_post_odlc.id,
async_post_odlc).result()
self.assertEquals(post_odlc, put_odlc)
self.assertEquals(async_post_odlc, async_put_odlc)
# Upload odlc image.
test_image_filepath = os.path.join(os.path.dirname(__file__),
"testdata/A.jpg")
with open(test_image_filepath, 'rb') as f:
image_data = f.read()
self.client.put_odlc_image(post_odlc.id, image_data)
self.async_client.put_odlc_image(async_post_odlc.id,
image_data).result()
# Get the odlc image.
get_image = self.client.get_odlc_image(post_odlc.id)
async_get_image = self.async_client.get_odlc_image(
async_post_odlc.id).result()
self.assertEquals(image_data, get_image)
self.assertEquals(image_data, async_get_image)
# Delete the odlc image.
self.client.delete_odlc_image(post_odlc.id)
self.async_client.delete_odlc_image(async_post_odlc.id).result()
with self.assertRaises(InteropError):
self.client.get_odlc_image(post_odlc.id)
with self.assertRaises(InteropError):
self.async_client.get_odlc_image(async_post_odlc.id).result()
# Delete odlc.
self.client.delete_odlc(post_odlc.id)
self.async_client.delete_odlc(async_post_odlc.id).result()
self.assertNotIn(post_odlc, self.client.get_odlcs())
self.assertNotIn(async_post_odlc,
self.async_client.get_odlcs().result())
