def test_get_qe_deployments(kube_config, expected_api, expected_query): config = KubernetesConfig(**kube_config) url_hit = [False] @urlmatch(netloc=r"www.customhost.com") def handler(request, _): assert request.path == expected_api assert request.query == expected_query url_hit[0] = True return response(200, "{}") with HTTMock(handler): KubernetesAccessorSingleton._instance = None assert KubernetesAccessorSingleton.get_instance(config).get_qe_deployments() is not None assert url_hit[0]
def test_cycle_qe_deployments(kube_config, deployment_names, expected_api_hits): KubernetesAccessorSingleton._instance = None config = KubernetesConfig(**kube_config) url_hit = [False] * len(expected_api_hits) i = [0] @urlmatch(netloc=r"www.customhost.com", method="PATCH") def handler(request, _): assert request.path == expected_api_hits[i[0]] url_hit[i[0]] = True i[0] += 1 return response(200, "{}") with HTTMock(handler): KubernetesAccessorSingleton.get_instance(config).cycle_qe_deployments(deployment_names) assert all(url_hit)
def __init__(self, kube_config=None): self.kube_config = kube_config if kube_config is None: self.kube_config = KubernetesConfig.from_env() KubernetesAccessorSingleton._instance = self