def token_mock(url, request):
params = parse_qs(url.query)
if params['grant_type'][0] == 'authorization_code':
auth = text_(
base64.b64encode(bytes_("%s:%s" % (client_id, api_key))))
self.assertEqual(request.headers['Authorization'],
"Basic %s" % auth)
if params['code'][0] == code:
body = {
"access_token": access_token,
"refresh_token": refresh_token,
"expires_in": 1200
}
return {"status_code": 200, "content": body}
elif params['grant_type'][0] == 'refresh_token':
auth = text_(
base64.b64encode(bytes_("%s:%s" % (client_id, api_key))))
self.assertEqual(request.headers['Authorization'],
"Basic %s" % auth)
if params['refresh_token'][0] == refresh_token:
body = {
"access_token": access_token,
"refresh_token": refresh_token,
"expires_in": 1200
}
return {"status_code": 200, "content": body}
return {"status_code": 403, "content": {}}
def _post(href, *args, **kwargs):
if href == "https://login.eveonline.com/oauth/token":
self.assertIn('headers', kwargs)
if kwargs['params']['grant_type'] == 'authorization_code':
auth = text_(
base64.b64encode(bytes_("%s:%s" %
(client_id, api_key))))
self.assertEqual(kwargs['headers']['Authorization'],
"Basic %s" % auth)
if kwargs['params']['code'] == code:
body = {
"access_token": access_token,
"refresh_token": refresh_token,
"expires_in": 1200
}
res = mock_resp()
res.status_code = 200
res.json.return_value = body
return res
else:
res = mock_resp()
res.status_code = 403
return res
elif kwargs['params']['grant_type'] == 'refresh_token':
auth = text_(
base64.b64encode(bytes_("%s:%s" %
(client_id, api_key))))
self.assertEqual(kwargs['headers']['Authorization'],
"Basic %s" % auth)
if kwargs['params']['refresh_token'] == refresh_token:
body = {
"access_token": access_token,
"refresh_token": refresh_token,
"expires_in": 1200
}
res = mock_resp()
res.status_code = 200
res.json.return_value = body
return res
else:
res = mock_resp()
res.status_code = 403
return res
else:
res = mock_resp()
res.status_code = 403
res.json.return_value = {}
return res
else:
res = mock_resp()
res.status_code = 404
res.json.return_value = {}
return res
def _authorize(self, params):
auth = text_(base64.b64encode(bytes_("%s:%s" % (self.client_id, self.api_key))))
headers = {"Authorization": "Basic %s" % auth}
res = self._session.post("%s/token" % self._oauth_endpoint, params=params, headers=headers)
if res.status_code != 200:
raise APIException("Got unexpected status code from API: %i" % res.status_code)
return res.json()
def _authorize(self, params):
auth = text_(
base64.b64encode(bytes_("%s:%s" % (self.client_id, self.api_key))))
headers = {"Authorization": "Basic %s" % auth}
resource = "%s/token" % self._oauth_endpoint
res = self._session.post(resource, params=params, headers=headers)
if res.status_code != 200:
raise APIException(resource, res.status_code, res.json())
return res.json()
def _authorize(self, params):
auth = text_(
base64.b64encode(bytes_("%s:%s" % (self.client_id, self.api_key))))
headers = {"Authorization": "Basic %s" % auth}
res = self._session.post("%s/token" % self._oauth_endpoint,
params=params,
headers=headers)
if res.status_code != 200:
raise APIException("Got unexpected status code from API: %i" %
res.status_code)
return res.json()
def token_mock(url, request):
params = parse_qs(url.query)
if params['grant_type'][0] == 'authorization_code':
auth = text_(base64.b64encode(bytes_("%s:%s" % (client_id, api_key))))
self.assertEqual(request.headers['Authorization'], "Basic %s" % auth)
if params['code'][0] == code:
body = {
"access_token": access_token,
"refresh_token": refresh_token,
"expires_in": 1200
}
return {"status_code": 200, "content": body}
elif params['grant_type'][0] == 'refresh_token':
auth = text_(base64.b64encode(bytes_("%s:%s" % (client_id, api_key))))
self.assertEqual(request.headers['Authorization'], "Basic %s" % auth)
if params['refresh_token'][0] == refresh_token:
body = {
"access_token": access_token,
"refresh_token": refresh_token,
"expires_in": 1200
}
return {"status_code": 200, "content": body}
return {"status_code": 403, "content": {}}
def authorize(self, code):
auth = text_(
base64.b64encode(bytes_("%s:%s" % (self.client_id, self.api_key))))
headers = {"Authorization": "Basic %s" % auth}
params = {"grant_type": "authorization_code", "code": code}
res = requests.post("%s/token" % self._oauth_endpoint,
params=params,
headers=headers)
if res.status_code != 200:
raise APIException("Got unexpected status code from API: %i" %
res.status_code)
return AuthedConnection(res.json(), self._authed_endpoint,
self._oauth_endpoint, self.client_id,
self.api_key)
def _authorize(self, params):
auth = text_(
base64.b64encode(
bytes_(
"%s:%s" %
(self.client_id, self.api_key))))
headers = {"Authorization": "Basic %s" % auth}
resource = "%s/token" % self._oauth_endpoint
res = self._session.post(
resource,
params=params,
headers=headers)
if res.status_code != 200:
raise APIException(
resource,
res.status_code,
res.json()
)
return res.json()
