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, }