tests/unit/benchmark/runner/test_base.py

   1 #!/usr/bin/env python
   2
   3 ##############################################################################
   4 # Copyright (c) 2017 Huawei Technologies Co.,Ltd and others.
   5 #
   6 # All rights reserved. This program and the accompanying materials
   7 # are made available under the terms of the Apache License, Version 2.0
   8 # which accompanies this distribution, and is available at
   9 # http://www.apache.org/licenses/LICENSE-2.0
  10 ##############################################################################
  11
  12 from __future__ import print_function
  13 from __future__ import absolute_import
  14
  15 import unittest
  16 import time
  17
  18 from yardstick.benchmark.runners.iteration import IterationRunner
  19
  20
  21 class RunnerTestCase(unittest.TestCase):
  22
  23     def test_get_output(self):
  24         runner = IterationRunner({})
  25         runner.output_queue.put({'case': 'opnfv_yardstick_tc002'})
  26         runner.output_queue.put({'criteria': 'PASS'})
  27
  28         idle_result = {
  29             'case': 'opnfv_yardstick_tc002',
  30             'criteria': 'PASS'
  31         }
  32
  33         time.sleep(1)
  34         actual_result = runner.get_output()
  35         self.assertEqual(idle_result, actual_result)
  36
  37
  38 def main():
  39     unittest.main()
  40
  41
  42 if __name__ == '__main__':
  43     main()