Merge "Extended Context class with get_physical_nodes functionality"
[yardstick.git] / 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 base
13 from yardstick.benchmark.contexts import dummy
14
15
16 class DummyContextTestCase(unittest.TestCase):
17
18     def setUp(self):
19         self.attrs = {
20             'name': 'foo',
21             'task_id': '1234567890',
22         }
23         self.test_context = dummy.DummyContext()
24         self.addCleanup(self._delete_contexts)
25
26     @staticmethod
27     def _delete_contexts():
28         for context in base.Context.list:
29             context._delete_context()
30
31     def test___init__(self):
32         self.assertFalse(self.test_context._flags.no_setup)
33         self.assertFalse(self.test_context._flags.no_teardown)
34         self.assertIsNone(self.test_context._name)
35         self.assertIsNone(self.test_context._task_id)
36
37     def test_init(self):
38         self.test_context.init(self.attrs)
39         self.assertEqual(self.test_context._name, 'foo')
40         self.assertEqual(self.test_context._task_id, '1234567890')
41         self.assertFalse(self.test_context._flags.no_setup)
42         self.assertFalse(self.test_context._flags.no_teardown)
43
44         self.assertEqual(self.test_context.name, 'foo-12345678')
45         self.assertEqual(self.test_context.assigned_name, 'foo')
46
47     def test_init_flags_no_setup(self):
48         self.attrs['flags'] = {'no_setup': True, 'no_teardown': False}
49
50         self.test_context.init(self.attrs)
51
52         self.assertEqual(self.test_context._name, 'foo')
53         self.assertEqual(self.test_context._task_id, '1234567890')
54         self.assertTrue(self.test_context._flags.no_setup)
55         self.assertFalse(self.test_context._flags.no_teardown)
56
57         self.assertEqual(self.test_context.name, 'foo')
58         self.assertEqual(self.test_context.assigned_name, 'foo')
59
60     def test_init_flags_no_teardown(self):
61         self.attrs['flags'] = {'no_setup': False, 'no_teardown': True}
62
63         self.test_context.init(self.attrs)
64
65         self.assertFalse(self.test_context._flags.no_setup)
66         self.assertTrue(self.test_context._flags.no_teardown)
67
68         self.assertEqual(self.test_context.name, 'foo')
69         self.assertEqual(self.test_context.assigned_name, 'foo')
70
71     def test__get_server(self):
72         self.test_context.init(self.attrs)
73         self.test_context.deploy()
74
75         result = self.test_context._get_server(None)
76         self.assertEqual(result, None)
77
78         self.test_context.undeploy()
79
80     def test__get_physical_nodes(self):
81         result = self.test_context._get_physical_nodes()
82         self.assertIsNone(result)
83
84     def test__get_physical_node_for_server(self):
85         result = self.test_context._get_physical_node_for_server("fake")
86         self.assertIsNone(result)