def test_acs_client(self): with self.assertRaises(ClientException) as ex: AcsClient() self.assertEqual(ex.exception.error_code, "SDK.InvalidCredential") self.assertEqual( ex.exception.message, "Need a ak&secret pair or public_key_id&private_key pair to auth.") client = AcsClient(ak="access_key_id", secret="access_key_secret") self.assertEqual(client.get_access_key(), "access_key_id") self.assertEqual(client.get_access_secret(), "access_key_secret") # region id self.assertEqual(client.get_region_id(), "cn-hangzhou") client.set_region_id('cn-shanghai') self.assertEqual(client.get_region_id(), "cn-shanghai") # auto retry self.assertTrue(client.is_auto_retry()) client.set_auto_retry(False) self.assertFalse(client.is_auto_retry()) # max retry num self.assertEqual(client.get_max_retry_num(), None) client.set_max_retry_num(10) self.assertEqual(client.get_max_retry_num(), 10) # user agent self.assertEqual(client.get_user_agent(), None) client.set_user_agent('new-user-agent') self.assertEqual(client.get_user_agent(), "new-user-agent") # port self.assertEqual(client.get_port(), 80) self.assertIsNone(client.get_location_service())
def test_client_init(self): acs_client = AcsClient(ak, secret, region) assert acs_client acs_client.set_user_agent(agent) acs_client.set_auto_retry(True) acs_client.set_max_retry_num(max_retry_num) assert acs_client.get_user_agent() == agent assert acs_client.get_max_retry_num() == max_retry_num assert acs_client.get_access_key() == ak assert acs_client.get_access_secret() == secret assert acs_client.get_region_id() == region assert acs_client.is_auto_retry()
def __get_client(self, region_info): region_id = region_info.region_id if self.client_dict.get( region_id ) is not None and self.client_dict.get(region_id) != '': return self.client_dict.get(region_id) endpoint = region_info.endpoint vpc = region_info.vpc if endpoint is not None and endpoint.strip() != '': region_provider.add_endpoint(const.PRODUCT_NAME, region_id, endpoint) elif vpc is not None and vpc: region_provider.add_endpoint(const.PRODUCT_NAME, region_id, get_vpc_endpoint(region_id)) client = AcsClient(credential=self.credential, region_id=region_id, verify=False) client.set_user_agent(get_user_agent()) self.client_dict[region_id] = client return self.client_dict.get(region_id)