yardstick/tests/unit/benchmark/contexts/test_dummy.py

   1 ##############################################################################
   2 # Copyright (c) 2015 Huawei Technologies Co.,Ltd and others.
   3 #
   4 # All rights reserved. This program and the accompanying materials
   5 # are made available under the terms of the Apache License, Version 2.0
   6 # which accompanies this distribution, and is available at
   7 # http://www.apache.org/licenses/LICENSE-2.0
   8 ##############################################################################
   9
  10 import unittest
  11
  12 from yardstick.benchmark.contexts import dummy
  13
  14
  15 class DummyContextTestCase(unittest.TestCase):
  16
  17     def setUp(self):
  18         self.attrs = {
  19             'name': 'foo',
  20             'task_id': '1234567890',
  21         }
  22         self.test_context = dummy.DummyContext()
  23         self.addCleanup(self.test_context._delete_context)
  24
  25     def test___init__(self):
  26         self.assertFalse(self.test_context._flags.no_setup)
  27         self.assertFalse(self.test_context._flags.no_teardown)
  28         self.assertIsNone(self.test_context._name)
  29         self.assertIsNone(self.test_context._task_id)
  30
  31     def test_init(self):
  32         self.test_context.init(self.attrs)
  33         self.assertEqual(self.test_context._name, 'foo')
  34         self.assertEqual(self.test_context._task_id, '1234567890')
  35         self.assertFalse(self.test_context._flags.no_setup)
  36         self.assertFalse(self.test_context._flags.no_teardown)
  37
  38         self.assertEqual(self.test_context.name, 'foo-12345678')
  39         self.assertEqual(self.test_context.assigned_name, 'foo')
  40
  41     def test_init_flags_no_setup(self):
  42         self.attrs['flags'] = {'no_setup': True, 'no_teardown': False}
  43
  44         self.test_context.init(self.attrs)
  45
  46         self.assertEqual(self.test_context._name, 'foo')
  47         self.assertEqual(self.test_context._task_id, '1234567890')
  48         self.assertTrue(self.test_context._flags.no_setup)
  49         self.assertFalse(self.test_context._flags.no_teardown)
  50
  51         self.assertEqual(self.test_context.name, 'foo')
  52         self.assertEqual(self.test_context.assigned_name, 'foo')
  53
  54     def test_init_flags_no_teardown(self):
  55         self.attrs['flags'] = {'no_setup': False, 'no_teardown': True}
  56
  57         self.test_context.init(self.attrs)
  58
  59         self.assertFalse(self.test_context._flags.no_setup)
  60         self.assertTrue(self.test_context._flags.no_teardown)
  61
  62         self.assertEqual(self.test_context.name, 'foo')
  63         self.assertEqual(self.test_context.assigned_name, 'foo')
  64
  65     def test__get_server(self):
  66         self.test_context.init(self.attrs)
  67         self.test_context.deploy()
  68
  69         result = self.test_context._get_server(None)
  70         self.assertEqual(result, None)
  71
  72         self.test_context.undeploy()