def setUp(self): self.nodes = NodeManager(mock.Mock()) self.node = mock.Mock(id='fakeid') self.lb = mock.Mock(id='lbfakeid')
class TestNodeManager(unittest2.TestCase): def setUp(self): self.nodes = NodeManager(mock.Mock()) self.node = mock.Mock(id='fakeid') self.lb = mock.Mock(id='lbfakeid') @mock.patch('balancerclient.common.base.Manager._get', autospec=True) def test_get(self, mock_get): self.nodes.get(self.lb, self.node) expected = mock.call(self.nodes, '/loadbalancers/lbfakeid/nodes/fakeid', 'node') self.assertTrue(mock_get.called) self.assertEqual(mock_get.mock_calls, [expected]) @mock.patch('balancerclient.common.base.Manager._create', autospec=True) def test_create(self, mock_create): self.nodes.create(self.lb, 'node1', 'HW', '10.0.0.1', 80, 10, 'ACTIVE') body = {'nodes': [{'name': 'node1', 'type': 'HW', 'address': '10.0.0.1', 'port': 80, 'weight': 10, 'condition': 'ACTIVE'}]} expected = mock.call(self.nodes, '/loadbalancers/lbfakeid/nodes', body, 'nodes', return_raw=True) self.assertTrue(mock_create.called) self.assertNotEqual(mock_create.mock_calls, []) self.assertEqual(mock_create.mock_calls[0], expected) @mock.patch('balancerclient.common.base.Manager._update', autospec=True) def test_update(self, mock_update): self.nodes.update(self.lb, self.node, name='node1', type='HW', address='10.0.0.1', port=80, weight=10, condition='ACTIVE') body = {'name': 'node1', 'type': 'HW', 'address': '10.0.0.1', 'port': 80, 'weight': 10, 'condition': 'ACTIVE'} expected = mock.call(self.nodes, '/loadbalancers/lbfakeid/nodes/fakeid', body, 'node') self.assertTrue(mock_update.called) self.assertEqual(mock_update.mock_calls, [expected]) @mock.patch('balancerclient.common.base.Manager._update', autospec=True) def test_update_condition(self, mock_update): self.nodes.update_condition(self.lb, self.node, 'FAKECONDITION') expected = mock.call(self.nodes, '/loadbalancers/lbfakeid/nodes/fakeid/FAKECONDITION', 'loadbalancers') self.assertTrue(mock_update.called) self.assertEqual(mock_update.mock_calls, [expected]) @mock.patch('balancerclient.common.base.Manager._delete', autospec=True) def test_delete(self, mock_delete): self.nodes.delete(self.lb, self.node) expected = mock.call(self.nodes, '/loadbalancers/lbfakeid/nodes/fakeid') self.assertTrue(mock_delete.called) self.assertEqual(mock_delete.mock_calls, [expected]) @mock.patch('balancerclient.common.base.Manager._list', autospec=True) def test_nodes_list(self, mock_list): self.nodes.list(self.lb) expected = mock.call(self.nodes, '/loadbalancers/lbfakeid/nodes', 'nodes') self.assertTrue(mock_list.called) self.assertEqual(mock_list.mock_calls, [expected])
class TestNodeManager(unittest2.TestCase): def setUp(self): self.nodes = NodeManager(mock.Mock()) self.node = mock.Mock(id='fakeid') self.lb = mock.Mock(id='lbfakeid') @mock.patch('balancerclient.common.base.Manager._get', autospec=True) def test_get(self, mock_get): self.nodes.get(self.lb, self.node) expected = mock.call(self.nodes, '/loadbalancers/lbfakeid/nodes/fakeid', 'node') self.assertTrue(mock_get.called) self.assertEqual(mock_get.mock_calls, [expected]) @mock.patch('balancerclient.common.base.Manager._create', autospec=True) def test_create(self, mock_create): self.nodes.create(self.lb, 'node1', 'HW', '10.0.0.1', 80, 10, 'ACTIVE') body = { 'nodes': [{ 'name': 'node1', 'type': 'HW', 'address': '10.0.0.1', 'port': 80, 'weight': 10, 'condition': 'ACTIVE' }] } expected = mock.call(self.nodes, '/loadbalancers/lbfakeid/nodes', body, 'nodes', return_raw=True) self.assertTrue(mock_create.called) self.assertNotEqual(mock_create.mock_calls, []) self.assertEqual(mock_create.mock_calls[0], expected) @mock.patch('balancerclient.common.base.Manager._update', autospec=True) def test_update(self, mock_update): self.nodes.update(self.lb, self.node, name='node1', type='HW', address='10.0.0.1', port=80, weight=10, condition='ACTIVE') body = { 'name': 'node1', 'type': 'HW', 'address': '10.0.0.1', 'port': 80, 'weight': 10, 'condition': 'ACTIVE' } expected = mock.call(self.nodes, '/loadbalancers/lbfakeid/nodes/fakeid', body, 'node') self.assertTrue(mock_update.called) self.assertEqual(mock_update.mock_calls, [expected]) @mock.patch('balancerclient.common.base.Manager._update', autospec=True) def test_update_condition(self, mock_update): self.nodes.update_condition(self.lb, self.node, 'FAKECONDITION') expected = mock.call( self.nodes, '/loadbalancers/lbfakeid/nodes/fakeid/FAKECONDITION', 'loadbalancers') self.assertTrue(mock_update.called) self.assertEqual(mock_update.mock_calls, [expected]) @mock.patch('balancerclient.common.base.Manager._delete', autospec=True) def test_delete(self, mock_delete): self.nodes.delete(self.lb, self.node) expected = mock.call(self.nodes, '/loadbalancers/lbfakeid/nodes/fakeid') self.assertTrue(mock_delete.called) self.assertEqual(mock_delete.mock_calls, [expected]) @mock.patch('balancerclient.common.base.Manager._list', autospec=True) def test_nodes_list(self, mock_list): self.nodes.list(self.lb) expected = mock.call(self.nodes, '/loadbalancers/lbfakeid/nodes', 'nodes') self.assertTrue(mock_list.called) self.assertEqual(mock_list.mock_calls, [expected])