Add test_get_nodemanagers()

The unittest for plugins/vanilla/test_utils.py isn't completed.
The function get_nodemanagers() hasn't been tested.
Change-Id: Ib7a9300785025d502ee7bb3d0bb706dbc21f81cb
This commit is contained in:
jiasen.lin 2017-02-10 19:11:48 +08:00
parent 1622e57074
commit 1a87461187
1 changed files with 12 additions and 0 deletions

View File

@ -61,6 +61,18 @@ class TestUtils(base.SaharaWithDbTestCase):
[self.ng_manager])
self.assertIsNone(u.get_namenode(cl))
def test_get_nodemanagers(self):
cl = tu.create_cluster('cl1', 't1', 'vanilla', '2.7.1',
[self.ng_manager, self.ng_nodemanager])
nodemanagers = u.get_nodemanagers(cl)
self.assertEqual(2, len(nodemanagers))
self.assertEqual(set(['tt1', 'tt2']),
set([nodemanagers[0].instance_id,
nodemanagers[1].instance_id]))
cl = tu.create_cluster('cl1', 't1', 'vanilla', '2.7.1',
[self.ng_namenode])
self.assertEqual([], u.get_nodemanagers(cl))
def test_get_oozie(self):
cl = tu.create_cluster('cl1', 't1', 'vanilla', '2.7.1',
[self.ng_manager, self.ng_oozie])