def setUp(self): generate_credentials_file() self.creds_provider = PropertiesCredentialsProvider( ).set_properties_file(fake_credentials_file) self.base = 'http://localhost:' + str(8080) generate_properties_file(self.base, fake_credentials_file) self.token_provider = None
def testRealCloudGetAuthorizationStringAndToken(self): generate_properties_file(idcs_url(), credentials_file) self.token_provider = DefaultAccessTokenProvider( idcs_props_file=properties_file) # get authorization string for ListTablesRequest result = self.token_provider.get_authorization_string( ListTablesRequest()) self.assertIsNotNone(result) # get authorization string for TableRequest result = self.token_provider.get_authorization_string( TableRequest()) self.assertIsNotNone(result) # get account access token result = self.token_provider.get_account_access_token() self.assertIsNotNone(result) # get service access token result = self.token_provider.get_service_access_token() self.assertIsNotNone(result)
def testAccessTokenProviderGetAccountAccessToken(self): account_at = 'account-at' class TokenHandler(SimpleHTTPRequestHandler): def do_GET(self): rawpath = self.path.split('?')[0] if rawpath == APP_ENDPOINT: self.send_response(codes.ok) self.send_header('Content-Type', 'application/json') self.send_header('Content-Length', str(len(CLIENT_INFO))) self.end_headers() self.wfile.write(CLIENT_INFO.encode()) def do_POST(self): rawpath = self.path.split('?')[0] if rawpath == TOKEN_ENDPOINT: res = str.format(TOKEN_RESULT, account_at) self.send_response(codes.ok) self.send_header('Content-Type', 'application/json') self.send_header('Content-Length', str(len(res))) self.end_headers() self.wfile.write(res.encode()) httpd, port = self.__find_port_start_server(TokenHandler) # connect to illegal idcs url self.base = 'http://localhost:80' generate_properties_file(self.base, fake_credentials_file) self.token_provider = DefaultAccessTokenProvider(properties_file) self.assertRaises(ConnectionError, self.token_provider.get_account_access_token) self.token_provider.close() self.base = 'http://localhost:' + str(port) # connect to legal idcs url generate_properties_file(self.base, fake_credentials_file) self.token_provider = DefaultAccessTokenProvider(properties_file) result = self.token_provider.get_account_access_token() self.assertIsNotNone(result) self.assertEqual(result, account_at) self.__stop_server(httpd)
def testAccessTokenProviderGetAccountAccessToken(self): global GET_INFO, POST_INFO GET_INFO = CLIENT_INFO POST_INFO = None httpd, port = self._find_port_start_server(TokenHandler) # connect to illegal idcs url self.base = 'http://localhost:80' generate_properties_file(self.base, fake_credentials_file) self.token_provider = DefaultAccessTokenProvider(properties_file) self.assertRaises(ConnectionError, self.token_provider.get_account_access_token) self.token_provider.close() self.base = 'http://localhost:' + str(port) # connect to legal idcs url generate_properties_file(self.base, fake_credentials_file) self.token_provider = DefaultAccessTokenProvider(properties_file) result = self.token_provider.get_account_access_token() self.assertIsNotNone(result) self.assertEqual(result, ACCOUNT_AT) self._stop_server(httpd)