test/test_vapi.py

   1 #!/usr/bin/env python
   2 """ VAPI test """
   3
   4 import unittest
   5 import os
   6 import signal
   7 from framework import VppTestCase, running_extended_tests, \
   8     running_on_centos, VppTestRunner, Worker
   9
  10
  11 @unittest.skipUnless(running_extended_tests, "part of extended tests")
  12 class VAPITestCase(VppTestCase):
  13     """ VAPI test """
  14
  15     @classmethod
  16     def setUpClass(cls):
  17         super(VAPITestCase, cls).setUpClass()
  18
  19     @classmethod
  20     def tearDownClass(cls):
  21         super(VAPITestCase, cls).tearDownClass()
  22
  23     def test_vapi_c(self):
  24         """ run C VAPI tests """
  25         var = "TEST_DIR"
  26         built_root = os.getenv(var, None)
  27         self.assertIsNotNone(built_root,
  28                              "Environment variable `%s' not set" % var)
  29         executable = "%s/build/vapi_test/vapi_c_test" % built_root
  30         worker = Worker(
  31             [executable, "vapi client", self.shm_prefix], self.logger)
  32         worker.start()
  33         timeout = 60
  34         worker.join(timeout)
  35         self.logger.info("Worker result is `%s'" % worker.result)
  36         error = False
  37         if worker.result is None:
  38             try:
  39                 error = True
  40                 self.logger.error(
  41                     "Timeout! Worker did not finish in %ss" % timeout)
  42                 os.killpg(os.getpgid(worker.process.pid), signal.SIGTERM)
  43                 worker.join()
  44             except:
  45                 self.logger.debug("Couldn't kill worker-spawned process")
  46                 raise
  47         if error:
  48             raise Exception(
  49                 "Timeout! Worker did not finish in %ss" % timeout)
  50         self.assert_equal(worker.result, 0, "Binary test return code")
  51
  52     @unittest.skipIf(running_on_centos, "Centos's gcc can't compile our C++")
  53     def test_vapi_cpp(self):
  54         """ run C++ VAPI tests """
  55         var = "TEST_DIR"
  56         built_root = os.getenv(var, None)
  57         self.assertIsNotNone(built_root,
  58                              "Environment variable `%s' not set" % var)
  59         executable = "%s/build/vapi_test/vapi_cpp_test" % built_root
  60         worker = Worker(
  61             [executable, "vapi client", self.shm_prefix], self.logger)
  62         worker.start()
  63         timeout = 120
  64         worker.join(timeout)
  65         self.logger.info("Worker result is `%s'" % worker.result)
  66         error = False
  67         if worker.result is None:
  68             try:
  69                 error = True
  70                 self.logger.error(
  71                     "Timeout! Worker did not finish in %ss" % timeout)
  72                 os.killpg(os.getpgid(worker.process.pid), signal.SIGTERM)
  73                 worker.join()
  74             except:
  75                 raise Exception("Couldn't kill worker-spawned process")
  76         if error:
  77             raise Exception(
  78                 "Timeout! Worker did not finish in %ss" % timeout)
  79         self.assert_equal(worker.result, 0, "Binary test return code")
  80
  81
  82 if __name__ == '__main__':
  83     unittest.main(testRunner=VppTestRunner)