test_rospy_node.py

Go to the documentation of this file.

#!/usr/bin/env python

import rospy
import os, sys, unittest, rostest

try:
    from respeaker_ros import *
except RuntimeError as e:
    print("Catch runtime error ({}), check if this is expected".format(e.args[0]))
    assert(e.args[0] == 'Check the device is connected and recognized')

# https://stackoverflow.com/questions/10971033/backporting-python-3-openencoding-utf-8-to-python-2
if sys.version_info[0] > 2:
    # py3k
    pass
else:
    # py2
    import __builtin__
    def open(filename, encoding=None):
        return __builtin__.open(filename)

pkg_dir = os.path.abspath(os.path.join(os.path.realpath(__file__), os.pardir, os.pardir))
pkg_name = os.path.basename(pkg_dir)


class TestRospyNode(unittest.TestCase):

    def __init__(self, *args):
        unittest.TestCase.__init__(self, *args)

    def test_rosnode(self):
        __name__ = 'dummy'
        for scripts_dir in ['scripts', 'node_scripts']:
            full_scripts_dir = os.path.join(pkg_dir, scripts_dir)
            if not os.path.exists(full_scripts_dir):
                continue
            for filename in [f for f in map(lambda x: x, os.listdir(full_scripts_dir)) if os.path.isfile(f) and f.endswith('.py')]:
                print("Check if {} is loadable".format(filename))
                import subprocess
                try:
                    ret = subprocess.check_output(['rosrun', pkg_name, filename], stderr=subprocess.STDOUT)
                except subprocess.CalledProcessError as e:
                    print("Catch runtime error ({}), check if this is expect".format(e.output))
                    self.assertTrue('Check the device is connected and recognized' in e.output)


if __name__ == '__main__':
    rostest.rosrun('test_rospy_node', pkg_name, TestRospyNode, sys.argv)