def test_init(self): conn_params = self.context.to_dict() conn_params['lb_status_timeout'] = 10 res = lbaas.LoadBalancerDriver(conn_params) self.assertEqual(conn_params, res.conn_params) self.assertIsNone(res._nc)
def setUp(self): super(TestOctaviaLBaaSDriver, self).setUp() self.context = utils.dummy_context() self.conn_params = self.context.to_dict() self.lb_driver = lbaas.LoadBalancerDriver(self.conn_params) self.lb_driver.lb_status_timeout = 10 self.patchobject(neutron_v2, 'NeutronClient') self.patchobject(octavia_v2, 'OctaviaClient') self.nc = self.lb_driver.nc() self.oc = self.lb_driver.oc() self.vip = { 'subnet': 'subnet-01', 'address': '192.168.1.100', 'admin_state_up': True, 'protocol': 'HTTP', 'protocol_port': 80, 'connection_limit': 50 } self.pool = { 'lb_method': 'ROUND_ROBIN', 'protocol': 'HTTP', 'admin_state_up': True } self.hm = { "type": "HTTP", "delay": "1", "timeout": 1, "max_retries": 5, "pool_id": "POOL_ID", "admin_state_up": True, "http_method": "GET", "url_path": "/index.html", "expected_codes": "200,201,202" }
def test_nc_initialize(self, mock_neutron_client): conn_params = self.context.to_dict() conn_params['lb_status_timeout'] = 10 fake_nc = mock.Mock() mock_neutron_client.return_value = fake_nc lb_driver = lbaas.LoadBalancerDriver(conn_params) self.assertIsNone(lb_driver._nc) # Create a new NeutronClient res = lb_driver.nc() mock_neutron_client.assert_called_once_with(conn_params) self.assertEqual(fake_nc, res) # Use the existing NeutronClient stored in self._nc fake_nc_new = mock.Mock() mock_neutron_client.return_value = fake_nc_new res1 = lb_driver.nc() mock_neutron_client.assert_called_once_with(conn_params) self.assertNotEqual(fake_nc_new, res1) self.assertEqual(res, res1)