def setUp(self):
     self.nodes = NodeManager(mock.Mock())
     self.node = mock.Mock(id='fakeid')
     self.lb = mock.Mock(id='lbfakeid')
Example #2
0
 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])
Example #4
0
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])