def test_failed(self, project_id, cluster_id):
cluster = CtxCluster.create(cluster_id, project_id, token='token')
querier = BcsAPIEnvironmentQuerier(cluster)
with StubPaaSCCClient.get_cluster.mock(
return_value=fake_cc_get_cluster_result_failed):
with pytest.raises(ComponentError):
assert querier.do()
def test_normal(self, project_id, cluster_id):
cluster = CtxCluster.create(cluster_id, project_id, token='token')
querier = BcsAPIEnvironmentQuerier(cluster)
with StubPaaSCCClient.get_cluster.mock(
return_value=fake_cc_get_cluster_result_ok):
api_env_name = querier.do()
assert api_env_name == 'my_stag'
def get_access_cluster_context(self):
"""获取访问集群需要的信息"""
# 获取集群的环境
# TODO: 这一部分逻辑后续直接放到组装kubeconfig中
ctx_cluster = CtxCluster.create(id=self.cluster_id,
project_id=self.project_id,
token=self.access_token)
env_name = BcsAPIEnvironmentQuerier(ctx_cluster).do()
return {
'server_address':
f"{settings.BCS_APIGW_DOMAIN[env_name]}/clusters/{self.cluster_id}",
'identifier': self.cluster_id,
'user_token': settings.BCS_APIGW_TOKEN,
}