From 979c8630338c79038fc16bfa0fdc6147dc5cd7ee Mon Sep 17 00:00:00 2001
From: Oleg Kalachev <okalachev@gmail.com>
Date: Tue, 2 Jul 2019 01:21:49 +0300
Subject: [PATCH] Add some test for clever package

---
 clever/CMakeLists.txt  |  5 +++++
 clever/test/basic.py   | 29 +++++++++++++++++++++++++++++
 clever/test/basic.test | 37 +++++++++++++++++++++++++++++++++++++
 3 files changed, 71 insertions(+)
 create mode 100755 clever/test/basic.py
 create mode 100755 clever/test/basic.test

diff --git a/clever/CMakeLists.txt b/clever/CMakeLists.txt
index 4f9e3e31..04e6744c 100644
--- a/clever/CMakeLists.txt
+++ b/clever/CMakeLists.txt
@@ -239,3 +239,8 @@ target_link_libraries(clever
 
 ## Add folders to be run by python nosetests
 # catkin_add_nosetests(test)
+
+if (CATKIN_ENABLE_TESTING)
+  find_package(rostest REQUIRED)
+  add_rostest(test/basic.test)
+endif()
diff --git a/clever/test/basic.py b/clever/test/basic.py
new file mode 100755
index 00000000..67de1fbe
--- /dev/null
+++ b/clever/test/basic.py
@@ -0,0 +1,29 @@
+#!/usr/bin/env python
+import rospy
+import pytest
+from mavros_msgs.msg import State
+
+@pytest.fixture()
+def node():
+    return rospy.init_node('clever_test', anonymous=True)
+
+def test_state(node):
+    state = rospy.wait_for_message('mavros/state', State, timeout=10)
+    assert state.connected == False
+    assert state.armed == False
+    assert state.guided == False
+    assert state.mode == ''
+
+def test_simple_offboard_services_available():
+    rospy.wait_for_service('get_telemetry', timeout=5)
+    rospy.wait_for_service('navigate', timeout=5)
+    rospy.wait_for_service('navigate_global', timeout=5)
+    rospy.wait_for_service('set_position', timeout=5)
+    rospy.wait_for_service('set_velocity', timeout=5)
+    rospy.wait_for_service('set_attitude', timeout=5)
+    rospy.wait_for_service('set_rates', timeout=5)
+    rospy.wait_for_service('land', timeout=5)
+
+def test_web_video_server(node):
+    import urllib2
+    urllib2.urlopen("http://localhost:8080").read()
diff --git a/clever/test/basic.test b/clever/test/basic.test
new file mode 100755
index 00000000..cbb52beb
--- /dev/null
+++ b/clever/test/basic.test
@@ -0,0 +1,37 @@
+<launch>
+    <!-- Verify all the required nodes basically work -->
+
+    <node pkg="mavros" type="mavros_node" name="mavros" required="true" output="screen">
+        <param name="fcu_url" value="udp://@127.0.1:14557"/>
+        <rosparam command="load" file="$(find clever)/launch/mavros_config.yaml"/>
+    </node>
+
+    <node name="visualization" pkg="mavros_extras" type="visualization" required="true">
+        <remap to="mavros/local_position/pose" from="local_position"/>
+        <remap to="mavros/setpoint_position/local" from="local_setpoint"/>
+        <param name="fixed_frame_id" value="map"/>
+        <param name="child_frame_id" value="base_link"/>
+        <param name="marker_scale" value="1"/>
+        <param name="max_track_size" value="20"/>
+        <param name="num_rotors" value="4"/>
+    </node>
+
+    <node name="web_video_server" pkg="web_video_server" type="web_video_server" required="true" output="screen">
+        <param name="default_stream_type" value="ros_compressed"/>
+        <param name="publish_rate" value="1.0"/>
+    </node>
+
+    <node pkg="tf2_ros" type="static_transform_publisher" name="map_flipped_frame" args="0 0 0 3.1415926 3.1415926 0 map map_flipped" required="true"/>
+
+    <node name="simple_offboard" pkg="clever" type="simple_offboard" required="true" output="screen">
+        <param name="reference_frames/body" value="map"/>
+        <param name="reference_frames/base_link" value="map"/>
+    </node>
+
+    <node name="tf2_web_republisher" pkg="tf2_web_republisher" type="tf2_web_republisher" required="true"/>
+
+    <node name="rc" pkg="clever" type="rc" required="true" output="screen"/>
+
+    <param name="test_module" value="$(find clever)/test/basic.py"/>
+    <test test-name="basic_test" pkg="ros_pytest" type="ros_pytest_runner"/>
+</launch>