def _enqueue_delete_service(self, ports, srv_meta, wait=True):
srv_spec = {'type': None, 'ports': ports}
srv_del_event = self.create_event('Service', srv_spec, srv_meta,
'DELETED')
ServiceKM.delete(srv_meta['uid'])
self.enqueue_event(srv_del_event)
if wait:
self.wait_for_all_tasks_done()
def _enqueue_delete_service(self, ports, srv_meta, wait=True):
srv_spec = {'type': None, 'ports': ports}
srv_del_event = self.create_event('Service', srv_spec, srv_meta,
'DELETED')
ServiceKM.delete(srv_meta['uid'])
self.enqueue_event(srv_del_event)
if wait:
self.wait_for_all_tasks_done()
def tearDownClass(cls):
for pod in list(PodKM):
PodKM.delete(pod)
for service in list(ServiceKM):
ServiceKM.delete(service)
for namespace in list(NamespaceKM):
NamespaceKM.delete(namespace)
super(VncEndpointsTestBase, cls).tearDownClass()
def tearDownClass(cls):
for pod in list(PodKM):
PodKM.delete(pod)
for service in list(ServiceKM):
ServiceKM.delete(service)
for namespace in list(NamespaceKM):
NamespaceKM.delete(namespace)
super(VncEndpointsTestBase, cls).tearDownClass()
def _delete_service(self, srv_uid, srv_name, namespace):
srv_meta = {
'uid': srv_uid,
'name': srv_name,
'namespace': namespace
}
srv_delete_event = self.create_event('Service', {}, srv_meta, 'DELETED')
ServiceKM.delete(srv_uid)
self.enqueue_event(srv_delete_event)
def _delete_service(self, srv_uid, srv_name, namespace):
srv_meta = {
'uid': srv_uid,
'name': srv_name,
'namespace': namespace
}
srv_delete_event = self.create_event('Service', {}, srv_meta, 'DELETED')
ServiceKM.delete(srv_uid)
self.enqueue_event(srv_delete_event)
def _add_service(self, namespace, srv_name, srv_spec):
srv_meta = {
'name': srv_name,
'uid': str(uuid.uuid4()),
'namespace': namespace
}
srv_add_event = self.create_event('Service', srv_spec, srv_meta,
'ADDED')
ServiceKM.locate(srv_meta['uid'], srv_add_event['object'])
self.enqueue_event(srv_add_event)
self.kube_mock.get_resource.return_value = {'metadata': srv_meta}
return srv_meta['uid']
def _add_service(self, namespace, srv_name, srv_spec):
srv_meta = {
'name': srv_name,
'uid': str(uuid.uuid4()),
'namespace': namespace
}
srv_add_event = self.create_event(
'Service', srv_spec, srv_meta, 'ADDED')
ServiceKM.locate(srv_meta['uid'], srv_add_event['object'])
self.enqueue_event(srv_add_event)
self.kube_mock.get_resource.return_value = {'metadata': srv_meta}
return srv_meta['uid']
def _enqueue_add_service(self,
namespace_name,
srv_name='test-service',
srv_type='ClusterIP'):
srv_uuid = str(uuid.uuid4())
srv_meta = {'name': srv_name, 'uid': srv_uuid,
'namespace': namespace_name}
ports = [{'name': 'http', 'protocol': 'TCP', 'port': 80}]
srv_spec = {'type': srv_type, 'ports': ports,
'externalIPs': [self.external_ip]}
srv_add_event = self.create_event('Service', srv_spec, srv_meta,
'ADDED')
ServiceKM.locate(srv_uuid, srv_add_event['object'])
self.enqueue_event(srv_add_event)
return ports, srv_meta, srv_uuid
def _sync_service_lb(self):
lb_uuid_set = set(LoadbalancerKM.keys())
service_uuid_set = set(ServiceKM.keys())
deleted_uuid_set = lb_uuid_set - service_uuid_set
for uuid in deleted_uuid_set:
lb = LoadbalancerKM.get(uuid)
if not lb:
continue
if not lb.annotations:
continue
owner = None
kind = None
cluster = None
for kvp in lb.annotations['key_value_pair'] or []:
if kvp['key'] == 'cluster':
cluster = kvp['value']
elif kvp['key'] == 'owner':
owner = kvp['value']
elif kvp['key'] == 'kind':
kind = kvp['value']
if cluster == vnc_kube_config.cluster_name() and \
owner == 'k8s' and \
kind == self._k8s_event_type:
self._create_service_event('delete', uuid, lb)
break
return
def _sync_service_lb(self):
lb_uuid_set = set(LoadbalancerKM.keys())
service_uuid_set = set(ServiceKM.keys())
deleted_uuid_set = lb_uuid_set - service_uuid_set
for uuid in deleted_uuid_set:
lb = LoadbalancerKM.get(uuid)
if not lb:
continue
if not lb.annotations:
continue
owner = None
kind = None
cluster = None
for kvp in lb.annotations['key_value_pair'] or []:
if kvp['key'] == 'cluster':
cluster = kvp['value']
elif kvp['key'] == 'owner':
owner = kvp['value']
elif kvp['key'] == 'kind':
kind = kvp['value']
if cluster == vnc_kube_config.cluster_name() and \
owner == 'k8s' and \
kind == self._k8s_event_type:
self._create_service_event('delete', uuid, lb)
break
return
def _enqueue_add_service(self,
namespace_name,
srv_name='test-service',
srv_type='ClusterIP',
wait=True):
srv_uuid = str(uuid.uuid4())
srv_meta = {
'name': srv_name,
'uid': srv_uuid,
'namespace': namespace_name
}
if srv_name == 'kubernetes':
ports = [{
'name': 'https',
'protocol': 'TCP',
'port': 443
}, {
'name': 'dns',
'protocol': 'UDP',
'port': 53
}, {
'name': 'dns-tcp',
'protocol': 'TCP',
'port': 53
}]
else:
ports = [{'name': 'http', 'protocol': 'TCP', 'port': 80}]
srv_spec = {
'type': srv_type,
'ports': ports,
'externalIPs': [self.external_ip]
}
srv_add_event = self.create_event('Service', srv_spec, srv_meta,
'ADDED')
ServiceKM.locate(srv_uuid, srv_add_event['object'])
self.enqueue_event(srv_add_event)
if wait:
self.wait_for_all_tasks_done()
return ports, srv_meta, srv_uuid
def _enqueue_add_service(self,
namespace_name,
srv_name='test-service',
srv_type='ClusterIP', wait=True):
srv_uuid = str(uuid.uuid4())
srv_meta = {'name': srv_name, 'uid': srv_uuid,
'namespace': namespace_name}
if srv_name == 'kubernetes':
ports = [
{'name': 'https', 'protocol': 'TCP', 'port': 443},
{'name': 'dns', 'protocol': 'UDP', 'port': 53},
{'name': 'dns-tcp', 'protocol': 'TCP', 'port': 53}
]
else:
ports = [{'name': 'http', 'protocol': 'TCP', 'port': 80}]
srv_spec = {'type': srv_type, 'ports': ports,
'externalIPs': [self.external_ip]}
srv_add_event = self.create_event('Service', srv_spec, srv_meta,
'ADDED')
ServiceKM.locate(srv_uuid, srv_add_event['object'])
self.enqueue_event(srv_add_event)
if wait:
self.wait_for_all_tasks_done()
return ports, srv_meta, srv_uuid
def _sync_service_lb(self):
lb_uuid_list = list(LoadbalancerKM.keys())
service_uuid_list = list(ServiceKM.keys())
for uuid in lb_uuid_list:
if uuid in service_uuid_list:
continue
lb = LoadbalancerKM.get(uuid)
if not lb:
continue
if not lb.annotations:
continue
for kvp in lb.annotations['key_value_pair'] or []:
if kvp['key'] == 'owner' \
and kvp['value'] == 'k8s':
self._create_service_event('delete', uuid, lb)
break
return
def _enqueue_delete_service(self, ports, srv_meta):
srv_spec = {'type': None, 'ports': ports}
srv_del_event = self.create_event('Service', srv_spec, srv_meta,
'DELETED')
ServiceKM.delete(srv_meta['uid'])
self.enqueue_event(srv_del_event)
