200 lines
8.1 KiB
Python
200 lines
8.1 KiB
Python
import httplib2
|
|
import json
|
|
import mock
|
|
import urlparse
|
|
|
|
|
|
from novaclient.v1_1 import client
|
|
from novaclient import exceptions
|
|
from tests import utils
|
|
|
|
|
|
def to_http_response(resp_dict):
|
|
"""Handy converts dict {'status':status, 'body': body, 'headers':headers} to httplib response."""
|
|
resp = httplib2.Response(resp_dict)
|
|
for k, v in resp_dict['headers'].items():
|
|
resp[k] = v
|
|
return resp
|
|
|
|
|
|
class AuthenticateAgainstKeystoneTests(utils.TestCase):
|
|
def test_authenticate_success(self):
|
|
cs = client.Client("username", "password", "project_id", "auth_url/v2.0")
|
|
resp = {"access":
|
|
{"token": {"expires": "12345", "id": "FAKE_ID"},
|
|
"serviceCatalog": [
|
|
{"adminURL": "http://localhost:8774/v1.1",
|
|
"type": "compute",
|
|
"endpoints" : [{
|
|
"region": "RegionOne",
|
|
"internalURL": "http://localhost:8774/v1.1",
|
|
"publicURL": "http://localhost:8774/v1.1/"},]},]}}
|
|
auth_response = httplib2.Response({
|
|
"status": 200,
|
|
"body": json.dumps(resp),
|
|
})
|
|
|
|
mock_request = mock.Mock(return_value=(auth_response,
|
|
json.dumps(resp)))
|
|
|
|
@mock.patch.object(httplib2.Http, "request", mock_request)
|
|
def test_auth_call():
|
|
cs.client.authenticate()
|
|
headers = {'User-Agent': cs.client.USER_AGENT,
|
|
'Content-Type': 'application/json', }
|
|
body = {'auth': {
|
|
'passwordCredentials': {'username': cs.client.user,
|
|
'password': cs.client.password},
|
|
'tenantName': cs.client.projectid, }}
|
|
|
|
token_url = urlparse.urljoin(cs.client.auth_url, "tokens")
|
|
mock_request.assert_called_with(token_url, "POST",
|
|
headers=headers,
|
|
body=json.dumps(body))
|
|
|
|
self.assertEqual(cs.client.management_url,
|
|
resp["access"]["serviceCatalog"][0]
|
|
['endpoints'][0]["publicURL"])
|
|
self.assertEqual(cs.client.auth_token, resp["access"]["token"]["id"])
|
|
|
|
test_auth_call()
|
|
|
|
def test_authenticate_failure(self):
|
|
cs = client.Client("username", "password", "project_id", "auth_url/v2.0")
|
|
resp = {"unauthorized": {"message": "Unauthorized", "code": "401"}}
|
|
auth_response = httplib2.Response({
|
|
"status": 401,
|
|
"body": json.dumps(resp),
|
|
})
|
|
|
|
mock_request = mock.Mock(return_value=(auth_response,
|
|
json.dumps(resp)))
|
|
|
|
@mock.patch.object(httplib2.Http, "request", mock_request)
|
|
def test_auth_call():
|
|
self.assertRaises(exceptions.Unauthorized, cs.client.authenticate)
|
|
|
|
test_auth_call()
|
|
|
|
def test_auth_redirect(self):
|
|
cs = client.Client("username", "password", "project_id", "auth_url/v1.0")
|
|
dict_correct_response = {"access": {"token": {"expires": "12345", "id": "FAKE_ID"},
|
|
"serviceCatalog": [{
|
|
"type": "compute",
|
|
"endpoints": [{"adminURL": "http://localhost:8774/v1.1",
|
|
"region": "RegionOne",
|
|
"internalURL": "http://localhost:8774/v1.1",
|
|
"publicURL": "http://localhost:8774/v1.1/"},]},]}}
|
|
correct_response = json.dumps(dict_correct_response)
|
|
dict_responses = [
|
|
{"headers": {'location':'http://127.0.0.1:5001'},
|
|
"status": 305,
|
|
"body": "Use proxy"},
|
|
# Configured on admin port, nova redirects to v2.0 port.
|
|
# When trying to connect on it, keystone auth succeed by v1.0
|
|
# protocol (through headers) but tokens are being returned in
|
|
# body (looks like keystone bug). Leaved for compatibility.
|
|
{"headers": {},
|
|
"status": 200,
|
|
"body": correct_response},
|
|
{"headers": {},
|
|
"status": 200,
|
|
"body": correct_response}
|
|
]
|
|
|
|
responses = [ (to_http_response(resp), resp['body']) for resp in dict_responses ]
|
|
|
|
def side_effect(*args, **kwargs):
|
|
return responses.pop(0)
|
|
|
|
mock_request = mock.Mock(side_effect=side_effect)
|
|
|
|
@mock.patch.object(httplib2.Http, "request", mock_request)
|
|
def test_auth_call():
|
|
cs.client.authenticate()
|
|
headers = {'User-Agent': cs.client.USER_AGENT,
|
|
'Content-Type': 'application/json',}
|
|
body = {'auth': {
|
|
'passwordCredentials': {'username': cs.client.user,
|
|
'password': cs.client.password},
|
|
'tenantName': cs.client.projectid,}}
|
|
|
|
token_url = urlparse.urljoin(cs.client.auth_url, "tokens")
|
|
mock_request.assert_called_with(token_url, "POST",
|
|
headers=headers,
|
|
body=json.dumps(body))
|
|
|
|
resp = dict_correct_response
|
|
self.assertEqual(cs.client.management_url,
|
|
resp["access"]["serviceCatalog"][0]
|
|
['endpoints'][0]["publicURL"])
|
|
self.assertEqual(cs.client.auth_token, resp["access"]["token"]["id"])
|
|
|
|
test_auth_call()
|
|
|
|
|
|
class AuthenticationTests(utils.TestCase):
|
|
def test_authenticate_success(self):
|
|
cs = client.Client("username", "password", "project_id", "auth_url")
|
|
management_url = 'https://servers.api.rackspacecloud.com/v1.1/443470'
|
|
auth_response = httplib2.Response({
|
|
'status': 204,
|
|
'x-server-management-url': management_url,
|
|
'x-auth-token': '1b751d74-de0c-46ae-84f0-915744b582d1',
|
|
})
|
|
mock_request = mock.Mock(return_value=(auth_response, None))
|
|
|
|
@mock.patch.object(httplib2.Http, "request", mock_request)
|
|
def test_auth_call():
|
|
cs.client.authenticate()
|
|
headers = {
|
|
'X-Auth-User': 'username',
|
|
'X-Auth-Key': 'password',
|
|
'X-Auth-Project-Id': 'project_id',
|
|
'User-Agent': cs.client.USER_AGENT
|
|
}
|
|
mock_request.assert_called_with(cs.client.auth_url, 'GET',
|
|
headers=headers)
|
|
self.assertEqual(cs.client.management_url,
|
|
auth_response['x-server-management-url'])
|
|
self.assertEqual(cs.client.auth_token,
|
|
auth_response['x-auth-token'])
|
|
|
|
test_auth_call()
|
|
|
|
def test_authenticate_failure(self):
|
|
cs = client.Client("username", "password", "project_id", "auth_url")
|
|
auth_response = httplib2.Response({'status': 401})
|
|
mock_request = mock.Mock(return_value=(auth_response, None))
|
|
|
|
@mock.patch.object(httplib2.Http, "request", mock_request)
|
|
def test_auth_call():
|
|
self.assertRaises(exceptions.Unauthorized, cs.client.authenticate)
|
|
|
|
test_auth_call()
|
|
|
|
def test_auth_automatic(self):
|
|
cs = client.Client("username", "password", "project_id", "auth_url")
|
|
http_client = cs.client
|
|
http_client.management_url = ''
|
|
mock_request = mock.Mock(return_value=(None, None))
|
|
|
|
@mock.patch.object(http_client, 'request', mock_request)
|
|
@mock.patch.object(http_client, 'authenticate')
|
|
def test_auth_call(m):
|
|
http_client.get('/')
|
|
m.assert_called()
|
|
mock_request.assert_called()
|
|
|
|
test_auth_call()
|
|
|
|
def test_auth_manual(self):
|
|
cs = client.Client("username", "password", "project_id", "auth_url")
|
|
|
|
@mock.patch.object(cs.client, 'authenticate')
|
|
def test_auth_call(m):
|
|
cs.authenticate()
|
|
m.assert_called()
|
|
|
|
test_auth_call()
|