def test_init(self, m_get_svc_sg_driver, m_get_lbaas_driver, m_get_project_driver, m_get_subnets_driver, m_get_sg_driver, m_get_vif_driver, m_get_vif_pool_driver, m_set_vifs_driver, m_get_multi_vif_drivers, m_is_network_policy_enabled): project_driver = mock.sentinel.project_driver subnets_driver = mock.sentinel.subnets_driver sg_driver = mock.sentinel.sg_driver vif_driver = mock.sentinel.vif_driver vif_pool_driver = mock.Mock(spec=drivers.VIFPoolDriver) multi_vif_drivers = [mock.MagicMock(spec=drivers.MultiVIFDriver)] lbaas_driver = mock.sentinel.lbaas_driver svc_sg_driver = mock.Mock(spec=drivers.ServiceSecurityGroupsDriver) m_get_project_driver.return_value = project_driver m_get_subnets_driver.return_value = subnets_driver m_get_sg_driver.return_value = sg_driver m_get_vif_driver.return_value = vif_driver m_get_vif_pool_driver.return_value = vif_pool_driver m_get_multi_vif_drivers.return_value = multi_vif_drivers m_get_lbaas_driver.return_value = lbaas_driver m_get_svc_sg_driver.return_value = svc_sg_driver m_is_network_policy_enabled.return_value = True handler = h_vif.VIFHandler() self.assertEqual(project_driver, handler._drv_project) self.assertEqual(subnets_driver, handler._drv_subnets) self.assertEqual(sg_driver, handler._drv_sg) self.assertEqual(vif_pool_driver, handler._drv_vif_pool) self.assertEqual(multi_vif_drivers, handler._drv_multi_vif) self.assertEqual(lbaas_driver, handler._drv_lbaas) self.assertEqual(svc_sg_driver, handler._drv_svc_sg)
def test_init(self, m_get_lbaas_driver, m_get_project_driver, m_get_subnets_driver, m_get_sg_driver, m_get_vif_driver, m_get_vif_pool_driver, m_set_vifs_driver, m_get_multi_vif_drivers): project_driver = mock.sentinel.project_driver subnets_driver = mock.sentinel.subnets_driver sg_driver = mock.sentinel.sg_driver vif_driver = mock.sentinel.vif_driver vif_pool_driver = mock.Mock(spec=drivers.VIFPoolDriver) multi_vif_drivers = [mock.MagicMock(spec=drivers.MultiVIFDriver)] lbaas_driver = mock.sentinel.lbaas_driver m_get_project_driver.return_value = project_driver m_get_subnets_driver.return_value = subnets_driver m_get_sg_driver.return_value = sg_driver m_get_vif_driver.return_value = vif_driver m_get_vif_pool_driver.return_value = vif_pool_driver m_get_multi_vif_drivers.return_value = multi_vif_drivers m_get_lbaas_driver.return_value = lbaas_driver handler = h_vif.VIFHandler() self.assertEqual(project_driver, handler._drv_project) self.assertEqual(subnets_driver, handler._drv_subnets) self.assertEqual(sg_driver, handler._drv_sg) self.assertEqual(vif_pool_driver, handler._drv_vif_pool) self.assertEqual(multi_vif_drivers, handler._drv_multi_vif) self.assertEqual(lbaas_driver, handler._drv_lbaas)
def __init__(self): super(KuryrK8sService, self).__init__() pipeline = h_pipeline.ControllerPipeline(self.tg) self.watcher = watcher.Watcher(pipeline, self.tg) # TODO(ivc): pluggable resource/handler registration for resource in ["pods", "services", "endpoints"]: self.watcher.add("%s/%s" % (constants.K8S_API_BASE, resource)) pipeline.register(h_vif.VIFHandler())
def __init__(self): super(KuryrK8sService, self).__init__() objects.register_locally_defined_vifs() pipeline = h_pipeline.ControllerPipeline(self.tg) self.watcher = watcher.Watcher(pipeline, self.tg) self.health_manager = health.HealthServer() # TODO(ivc): pluggable resource/handler registration for resource in ["pods", "services", "endpoints"]: self.watcher.add("%s/%s" % (constants.K8S_API_BASE, resource)) pipeline.register(h_vif.VIFHandler()) pipeline.register(h_lbaas.LBaaSSpecHandler()) pipeline.register(h_lbaas.LoadBalancerHandler())
def test_init(self, m_get_project_driver, m_get_subnets_driver, m_get_sg_driver, m_get_vif_driver): project_driver = mock.sentinel.project_driver subnets_driver = mock.sentinel.subnets_driver sg_driver = mock.sentinel.sg_driver vif_driver = mock.sentinel.vif_driver m_get_project_driver.return_value = project_driver m_get_subnets_driver.return_value = subnets_driver m_get_sg_driver.return_value = sg_driver m_get_vif_driver.return_value = vif_driver handler = h_vif.VIFHandler() self.assertEqual(project_driver, handler._drv_project) self.assertEqual(subnets_driver, handler._drv_subnets) self.assertEqual(sg_driver, handler._drv_sg) self.assertEqual(vif_driver, handler._drv_vif)