#!/usr/bin/env python
""" VAPI test """

import unittest
import os
import signal
from framework import VppTestCase, running_extended_tests, \
    VppTestRunner, Worker


@unittest.skipUnless(running_extended_tests(), "part of extended tests")
class VOMTestCase(VppTestCase):
    """ VPP Object Model Test """

    def test_vom_cpp(self):
        """ run C++ VOM tests """
        var = "TEST_DIR"
        built_root = os.getenv(var, None)
        self.assertIsNotNone(built_root,
                             "Environment variable `%s' not set" % var)
        executable = "%s/build/vom_test/vom_test" % built_root
        worker = Worker(
            [executable, "vpp object model", self.shm_prefix], self.logger)
        worker.start()
        timeout = 120
        worker.join(timeout)
        self.logger.info("Worker result is `%s'" % worker.result)
        error = False
        if worker.result is None:
            try:
                error = True
                self.logger.error(
                    "Timeout! Worker did not finish in %ss" % timeout)
                os.killpg(os.getpgid(worker.process.pid), signal.SIGTERM)
                worker.join()
            except:
                raise Exception("Couldn't kill worker-spawned process")
        if error:
            raise Exception(
                "Timeout! Worker did not finish in %ss" % timeout)
        self.assert_equal(worker.result, 0, "Binary test return code")


if __name__ == '__main__':
    unittest.main(testRunner=VppTestRunner)

[((11, 21, 11, 45), 'framework.running_extended_tests', 'running_extended_tests', ({}, {}), '()', False, 'from framework import VppTestCase, running_extended_tests, VppTestRunner, Worker\n'), ((45, 4, 45, 43), 'unittest.main', 'unittest.main', (), '', False, 'import unittest\n'), ((18, 21, 18, 41), 'os.getenv', 'os.getenv', ({(18, 31, 18, 34): 'var', (18, 36, 18, 40): 'None'}, {}), '(var, None)', False, 'import os\n'), ((22, 17, 23, 75), 'framework.Worker', 'Worker', ({(23, 12, 23, 61): "[executable, 'vpp object model', self.shm_prefix]", (23, 63, 23, 74): 'self.logger'}, {}), "([executable, 'vpp object model', self.shm_prefix], self.logger)", False, 'from framework import VppTestCase, running_extended_tests, VppTestRunner, Worker\n'), ((34, 26, 34, 56), 'os.getpgid', 'os.getpgid', ({(34, 37, 34, 55): 'worker.process.pid'}, {}), '(worker.process.pid)', False, 'import os\n')]