From a5f24f0b077e581b1adcedf153e9f6617fbbdeef Mon Sep 17 00:00:00 2001
From: AAlon <avishaya@amazon.com>
Date: Tue, 3 Sep 2019 15:23:53 -0700
Subject: [PATCH] Add coverage report for ros2_build and remove python3-boto3

---
 ros2_build.sh | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

diff --git a/ros2_build.sh b/ros2_build.sh
index fdbd440..a712939 100755
--- a/ros2_build.sh
+++ b/ros2_build.sh
@@ -5,9 +5,7 @@ set -xe
 apt-key adv --keyserver hkp://keyserver.ubuntu.com:80 --recv-keys 421C365BD9FF1F717815A3895523BAEEB01FA116
 echo "deb http://packages.ros.org/ros/ubuntu `lsb_release -sc` main" > /etc/apt/sources.list.d/ros-latest.list
 apt update && apt install -y python3 python3-pip lcov cmake && rosdep update
-apt update && apt install -y python3-rosinstall python3-colcon-common-extensions && pip3 install -U setuptools coverage
-# Manually install python3-boto3 (temporary measure until this dependency is available via rosdep).
-apt install -y python3-boto3
+apt update && apt install -y python3-rosinstall python3-colcon-common-extensions && pip3 install -U setuptools coverage pytest
 apt list --upgradable 2>/dev/null | awk {'print $1'} | sed 's/\/.*//g' | grep $ROS_DISTRO | xargs apt install -y
 
 REPO_NAME=$(basename -- ${TRAVIS_BUILD_DIR})
@@ -52,10 +50,8 @@ if [ -z "${NO_TEST}" ]; then
             ;;
         "python")
             # this doesn't actually support multiple packages
-            cd "/${ROS_DISTRO}_ws/build/${PACKAGE_NAMES}"
-            # No Python coverage support through ament/colcon yet.
-            # coverage xml
-            # cp coverage.xml /shared/coverage.info
+            pytest --cov=src/${REPO_NAME}/${PACKAGE_NAMES} --cov-report=xml
+            cp coverage.xml /shared/coverage.info
             ;;
     esac
 fi