def test_get_server(self, mock_get_service_by_name, mock_get_node_ip,
mock_get_pod_by_name):
class Service(object):
def __init__(self):
self.name = 'yardstick'
self.node_port = 30000
class Services(object):
def __init__(self):
self.ports = [Service()]
class Status(object):
def __init__(self):
self.pod_ip = '172.16.10.131'
class Pod(object):
def __init__(self):
self.status = Status()
k8s_context = KubernetesContext()
k8s_context.init(context_cfg)
mock_get_service_by_name.return_value = Services()
mock_get_pod_by_name.return_value = Pod()
mock_get_node_ip.return_value = '172.16.10.131'
server = k8s_context._get_server('server')
self.assertIsNotNone(server)
def test_get_server(self, mock_get_pod_list):
k8s_context = KubernetesContext()
k8s_context.init(context_cfg)
mock_get_pod_list.return_value.items = []
server = k8s_context._get_server('server')
self.assertIsNone(server)