tests/unit/benchmark/contexts/test_node.py

   1 #!/usr/bin/env python
   2
   3 ##############################################################################
   4 # Copyright (c) 2015 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 # Unittest for yardstick.benchmark.contexts.node
  13
  14 from __future__ import absolute_import
  15 import os
  16 import unittest
  17
  18 from yardstick.benchmark.contexts import node
  19
  20
  21 class NodeContextTestCase(unittest.TestCase):
  22
  23     NODES_SAMPLE = "nodes_sample.yaml"
  24     NODES_DUPLICATE_SAMPLE = "nodes_duplicate_sample.yaml"
  25
  26     def setUp(self):
  27         self.test_context = node.NodeContext()
  28
  29     def test_construct(self):
  30
  31         self.assertIsNone(self.test_context.name)
  32         self.assertIsNone(self.test_context.file_path)
  33         self.assertEqual(self.test_context.nodes, [])
  34         self.assertEqual(self.test_context.controllers, [])
  35         self.assertEqual(self.test_context.computes, [])
  36         self.assertEqual(self.test_context.baremetals, [])
  37
  38     def test_unsuccessful_init(self):
  39
  40         attrs = {
  41             'name': 'foo',
  42             'file': self._get_file_abspath("error_file")
  43         }
  44
  45         self.assertRaises(IOError, self.test_context.init, attrs)
  46
  47     def test_successful_init(self):
  48
  49         attrs = {
  50             'name': 'foo',
  51             'file': self._get_file_abspath(self.NODES_SAMPLE)
  52         }
  53
  54         self.test_context.init(attrs)
  55
  56         self.assertEqual(self.test_context.name, "foo")
  57         self.assertEqual(len(self.test_context.nodes), 4)
  58         self.assertEqual(len(self.test_context.controllers), 2)
  59         self.assertEqual(len(self.test_context.computes), 1)
  60         self.assertEqual(self.test_context.computes[0]["name"], "node3")
  61         self.assertEqual(len(self.test_context.baremetals), 1)
  62         self.assertEqual(self.test_context.baremetals[0]["name"], "node4")
  63
  64     def test__get_server_with_dic_attr_name(self):
  65
  66         attrs = {
  67             'name': 'foo',
  68             'file': self._get_file_abspath(self.NODES_SAMPLE)
  69         }
  70
  71         self.test_context.init(attrs)
  72
  73         attr_name = {'name': 'foo.bar'}
  74         result = self.test_context._get_server(attr_name)
  75
  76         self.assertEqual(result, None)
  77
  78     def test__get_server_not_found(self):
  79
  80         attrs = {
  81             'name': 'foo',
  82             'file': self._get_file_abspath(self.NODES_SAMPLE)
  83         }
  84
  85         self.test_context.init(attrs)
  86
  87         attr_name = 'bar.foo'
  88         result = self.test_context._get_server(attr_name)
  89
  90         self.assertEqual(result, None)
  91
  92     def test__get_server_duplicate(self):
  93
  94         attrs = {
  95             'name': 'foo',
  96             'file': self._get_file_abspath(self.NODES_DUPLICATE_SAMPLE)
  97         }
  98
  99         self.test_context.init(attrs)
 100
 101         attr_name = 'node1.foo'
 102
 103         self.assertRaises(ValueError, self.test_context._get_server, attr_name)
 104
 105     def test__get_server_found(self):
 106
 107         attrs = {
 108             'name': 'foo',
 109             'file': self._get_file_abspath(self.NODES_SAMPLE)
 110         }
 111
 112         self.test_context.init(attrs)
 113
 114         attr_name = 'node1.foo'
 115         result = self.test_context._get_server(attr_name)
 116
 117         self.assertEqual(result['ip'], '10.229.47.137')
 118         self.assertEqual(result['name'], 'node1.foo')
 119         self.assertEqual(result['user'], 'root')
 120         self.assertEqual(result['key_filename'], '/root/.yardstick_key')
 121
 122     def _get_file_abspath(self, filename):
 123         curr_path = os.path.dirname(os.path.abspath(__file__))
 124         file_path = os.path.join(curr_path, filename)
 125         return file_path