def test_retrieve_k8s_master_url_without_baymodel_apiserver_port( self, mock_baymodel_get_by_uuid, mock_bay_get_by_uuid): expected_context = 'context' expected_api_address = 'api_address' expected_baymodel_id = 'e74c40e0-d825-11e2-a28f-0800200c9a61' expected_protocol = cfg.CONF.kubernetes.k8s_protocol expected_apiserver_port = cfg.CONF.kubernetes.k8s_port resource = self.mock_pod() resource.bay_uuid = 'bay_uuid' bay = self.mock_bay() bay.api_address = expected_api_address bay.baymodel_id = expected_baymodel_id baymodel = self.mock_baymodel() baymodel.apiserver_port = None mock_bay_get_by_uuid.return_value = bay mock_baymodel_get_by_uuid.return_value = baymodel actual_api_address = k8s_conductor._retrieve_k8s_master_url( expected_context, resource) self.assertEqual("%s://%s:%d" % (expected_protocol, expected_api_address, expected_apiserver_port), actual_api_address)
def test_retrieve_k8s_master_url_from_pod( self, mock_baymodel_get_by_uuid, mock_bay_get_by_uuid): expected_context = 'context' expected_api_address = 'api_address' expected_baymodel_id = 'e74c40e0-d825-11e2-a28f-0800200c9a61' expected_apiserver_port = 9999 pod = self.mock_pod() pod.bay_uuid = 'bay_uuid' bay = self.mock_bay() bay.api_address = expected_api_address bay.baymodel_id = expected_baymodel_id baymodel = self.mock_baymodel() baymodel.apiserver_port = expected_apiserver_port mock_bay_get_by_uuid.return_value = bay mock_baymodel_get_by_uuid.return_value = baymodel actual_api_address = k8s_conductor._retrieve_k8s_master_url( expected_context, pod) self.assertEqual("http://%s:%d" % (expected_api_address, expected_apiserver_port), actual_api_address)