def test_register_from_endpoints_empty_cfg():
"""Pass an empty config to _register_from_endpoints."""
res = kubernetes._register_from_endpoints(
ns='default',
cfg={}
)
assert res == []
def test_one_endpoint_multiple_ports_invalid():
"""Endpoint has multiple ports, none of which has the name 'http'."""
# mock out the 'load incluster config' fn so it does nothing
kubernetes.kubernetes.config.load_incluster_config = lambda: None
# mock out the CoreV1Api object's list_namespaced_endpoints
# to return no endpoints.
class MockCoreV1Api:
def list_namespaced_endpoints(self, *args, **kwargs):
return k8s.client.V1EndpointsList(items=[
k8s.client.V1Endpoints(
metadata=k8s.client.V1ObjectMeta(
name='test-ep'
),
subsets=[
k8s.client.V1EndpointSubset(
addresses=[
k8s.client.V1EndpointAddress(
hostname='foobar',
ip='127.0.0.1',
node_name='test',
target_ref=k8s.client.V1ObjectReference(
kind='Pod',
name='test-pod',
namespace='default'
)
)
],
ports=[
k8s.client.V1EndpointPort(
name='foo',
port=7788,
protocol='TCP'
),
k8s.client.V1EndpointPort(
name='bar',
port=4321,
protocol='TCP'
)
]
)
]
)
])
kubernetes.kubernetes.client.CoreV1Api = MockCoreV1Api
res = kubernetes._register_from_endpoints(
ns='default',
cfg={'labels': {'foo': 'bar'}}
)
assert res == []
def test_endpoint_no_addr_match():
"""The endpoint address does not match the criteria (pod, with target_ref)"""
# mock out the 'load incluster config' fn so it does nothing
kubernetes.kubernetes.config.load_incluster_config = lambda: None
# mock out the CoreV1Api object's list_namespaced_endpoints
# to return no endpoints.
class MockCoreV1Api:
def list_namespaced_endpoints(self, *args, **kwargs):
return k8s.client.V1EndpointsList(items=[
k8s.client.V1Endpoints(
metadata=k8s.client.V1ObjectMeta(
name='test-ep'
),
subsets=[
k8s.client.V1EndpointSubset(
addresses=[
k8s.client.V1EndpointAddress(
hostname='foobar',
ip='127.0.0.1',
node_name='test',
target_ref=k8s.client.V1ObjectReference(
kind='Service',
name='test-service',
namespace='default'
)
)
],
ports=[
k8s.client.V1EndpointPort(
name='foo',
port=7766,
protocol='TCP'
)
]
)
]
)
])
kubernetes.kubernetes.client.CoreV1Api = MockCoreV1Api
res = kubernetes._register_from_endpoints(
ns='default',
cfg={'labels': {'foo': 'bar'}}
)
assert res == []
def test_no_endpoints():
"""The specified labels result in no endpoints being found."""
# mock out the 'load incluster config' fn so it does nothing
kubernetes.kubernetes.config.load_incluster_config = lambda: None
# mock out the CoreV1Api object's list_namespaced_endpoints
# to return no endpoints.
class MockCoreV1Api:
def list_namespaced_endpoints(self, *args, **kwargs):
return k8s.client.V1EndpointsList(items=[])
kubernetes.kubernetes.client.CoreV1Api = MockCoreV1Api
res = kubernetes._register_from_endpoints(
ns='default',
cfg={'labels': {'foo': 'bar'}}
)
assert res == []
def test_one_endpoint_no_ports():
"""The endpoint has a single subset with address but no port."""
# mock out the 'load incluster config' fn so it does nothing
kubernetes.kubernetes.config.load_incluster_config = lambda: None
# mock out the CoreV1Api object's list_namespaced_endpoints
# to return no endpoints.
class MockCoreV1Api:
def list_namespaced_endpoints(self, *args, **kwargs):
return k8s.client.V1EndpointsList(items=[
k8s.client.V1Endpoints(
metadata=k8s.client.V1ObjectMeta(
name='test-ep'
),
subsets=[
k8s.client.V1EndpointSubset(
addresses=[
k8s.client.V1EndpointAddress(
hostname='foobar',
ip='127.0.0.1',
node_name='test',
target_ref=k8s.client.V1ObjectReference(
kind='Pod',
name='test-pod',
namespace='default'
)
)
],
ports=[]
)
]
)
])
kubernetes.kubernetes.client.CoreV1Api = MockCoreV1Api
res = kubernetes._register_from_endpoints(
ns='default',
cfg={'labels': {'foo': 'bar'}}
)
assert res == []
def test_one_endpoint_multiple_addrs():
"""Endpoint has multiple addresses, some ok, others not what we are looking
for, and a single port.
"""
# mock out the 'load incluster config' fn so it does nothing
kubernetes.kubernetes.config.load_incluster_config = lambda: None
# mock out the CoreV1Api object's list_namespaced_endpoints
# to return no endpoints.
class MockCoreV1Api:
def list_namespaced_endpoints(self, *args, **kwargs):
return k8s.client.V1EndpointsList(items=[
k8s.client.V1Endpoints(
metadata=k8s.client.V1ObjectMeta(
name='test-ep'
),
subsets=[
k8s.client.V1EndpointSubset(
addresses=[
# ok
k8s.client.V1EndpointAddress(
hostname='foobar',
ip='127.0.0.1',
node_name='test',
target_ref=k8s.client.V1ObjectReference(
kind='Pod',
name='test-pod',
namespace='default'
)
),
# no target_ref
k8s.client.V1EndpointAddress(
hostname='foobar',
ip='127.0.0.2',
node_name='test'
),
# not a pod
k8s.client.V1EndpointAddress(
hostname='foobar',
ip='127.0.0.3',
node_name='test',
target_ref=k8s.client.V1ObjectReference(
kind='Service',
name='test-service',
namespace='default'
)
),
# ok
k8s.client.V1EndpointAddress(
hostname='foobar',
ip='127.0.0.4',
node_name='test',
target_ref=k8s.client.V1ObjectReference(
kind='Pod',
name='test-pod',
namespace='default'
)
)
],
ports=[
k8s.client.V1EndpointPort(
name='foo',
port=7766,
protocol='TCP'
)
]
)
]
)
])
kubernetes.kubernetes.client.CoreV1Api = MockCoreV1Api
res = kubernetes._register_from_endpoints(
ns='default',
cfg={'labels': {'foo': 'bar'}}
)
assert res == ['127.0.0.1:7766', '127.0.0.4:7766']
def test_register_from_endpoints_no_ns():
"""An empty namespace string is provided to the _register_from_endpoints function."""
with pytest.raises(ValueError):
kubernetes._register_from_endpoints('', {})
