openstack
/
keystone
Code Issues Proposed changes
keystone/keystone/test/unit/test_common.py

586 lines
22 KiB
Python
Executable File
Raw Blame History

# vim: tabstop=4 shiftwidth=4 softtabstop=4
# Copyright (c) 2010-2011 OpenStack, LLC.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
# implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import httplib2
import json
from lxml import etree
import os
import sys
sys.path.append(os.path.abspath(os.path.join(os.path.abspath(__file__),
'..', '..', '..', '..', '..', 'keystone')))
import unittest
URL_V1 = 'http://localhost:5000/v1.0/'
URL_V2 = 'http://localhost:5001/v2.0/'
def get_token(user, pswd, tenant_id, kind=''):
header = httplib2.Http(".cache")
url = '%stokens' % URL_V2
if not tenant_id:
body = {"passwordCredentials": {"username": user,
"password": pswd}}
else:
body = {"passwordCredentials": {"username": user,
"password": pswd,
"tenantId": tenant_id}}
resp, content = header.request(url, "POST", body=json.dumps(body),
headers={"Content-Type": "application/json"})
if int(resp['status']) == 200:
content = json.loads(content)
token = str(content['auth']['token']['id'])
else:
token = None
if kind == 'token':
return token
else:
return (resp, content)
def get_token_legacy(user, pswd, kind=''):
header = httplib2.Http(".cache")
url = URL_V1
resp, content = header.request(url, "GET", '',
headers={"Content-Type": "application/json",
"X-Auth-User": user,
"X-Auth-Key": pswd})
if int(resp['status']) == 204:
token = resp['x-auth-token']
else:
token = None
if kind == 'token':
return token
else:
return (resp, content)
def delete_token(token, auth_token):
header = httplib2.Http(".cache")
url = '%stoken/%s' % (URL_V2, token)
resp, content = header.request(url, "DELETE", body='',
headers={"Content-Type": "application/json",
"X-Auth-Token": auth_token})
return (resp, content)
def create_tenant(tenantid, auth_token):
header = httplib2.Http(".cache")
url = '%stenants' % (URL_V2)
body = {"tenant": {"id": tenantid,
"description": "A description ...",
"enabled": True}}
resp, content = header.request(url, "PUT", body=json.dumps(body),
headers={"Content-Type": "application/json",
"X-Auth-Token": auth_token})
return (resp, content)
def delete_tenant(tenantid, auth_token):
header = httplib2.Http(".cache")
url = '%stenants/%s' % (URL_V2, tenantid)
resp, _content = header.request(url, "DELETE", body='{}',
headers={"Content-Type": "application/json",
"X-Auth-Token": auth_token})
return resp
def get_token_xml(user, pswd, tenant_id, return_type=''):
header = httplib2.Http(".cache")
url = '%stokens' % URL_V2
body = '<?xml version="1.0" encoding="UTF-8"?> \
<passwordCredentials \
xmlns="http://docs.openstack.org/identity/api/v2.0" \
password="%s" username="%s" \
tenantId="%s"/> ' % (pswd, user, tenant_id)
resp, content = header.request(url, "POST", body=body,
headers={"Content-Type": "application/xml",
"ACCEPT": "application/xml"})
if int(resp['status']) == 200:
dom = etree.fromstring(content)
root = dom.find("{http://docs.openstack.org/" \
"identity/api/v2.0}token")
token_root = root.attrib
token = str(token_root['id'])
else:
token = None
if return_type == 'token':
return token
else:
return (resp, content)
def delete_token_xml(token, auth_token):
header = httplib2.Http(".cache")
url = '%stoken/%s' % (URL_V2, token)
resp, content = header.request(url, "DELETE", body='',
headers={"Content-Type": "application/xml",
"X-Auth-Token": auth_token,
"ACCEPT": "application/xml"})
return (resp, content)
def create_tenant_xml(tenantid, auth_token):
header = httplib2.Http(".cache")
url = '%stenants' % (URL_V2)
body = '<?xml version="1.0" encoding="UTF-8"?> \
<tenant xmlns="http://docs.openstack.org/identity/api/v2.0" \
enabled="true" id="%s"> \
<description>A description...</description> \
</tenant>' % tenantid
resp, content = header.request(url, "POST", body=body,
headers={"Content-Type": "application/xml",
"X-Auth-Token": auth_token,
"ACCEPT": "application/xml"})
return (resp, content)
def delete_tenant_xml(tenantid, auth_token):
header = httplib2.Http(".cache")
url = '%stenants/%s' % (URL_V2, tenantid)
resp, _content = header.request(url, "DELETE", body='',
headers={"Content-Type": "application/xml",
"X-Auth-Token": auth_token,
"ACCEPT": "application/xml"})
return resp
def create_user(tenantid, userid, auth_token, email=None, password = 'secrete'):
header = httplib2.Http(".cache")
url = '%susers' % (URL_V2)
if email is not None:
email_id = email
else:
email_id = "%s@openstack.org" % userid
body = {"user": {"password": password,
"id": userid,
"tenantId": tenantid,
"email": "%s" % email_id,
"enabled": True}}
resp, content = header.request(url, "PUT", body=json.dumps(body),
headers={"Content-Type": "application/json",
"X-Auth-Token": auth_token})
return (resp, content)
def delete_user(userid, auth_token):
header = httplib2.Http(".cache")
url = '%susers/%s' % (URL_V2, userid)
resp, _content = header.request(url, "DELETE", body='{}',
headers={"Content-Type": "application/json",
"X-Auth-Token": auth_token})
return resp
def create_user_xml(tenantid, userid, auth_token, email=None, password = 'secrete'):
header = httplib2.Http(".cache")
url = '%susers' % (URL_V2)
if email is not None:
email_id = email
else:
email_id = userid
body = '<?xml version="1.0" encoding="UTF-8"?> \
<user xmlns="http://docs.openstack.org/identity/api/v2.0" \
email="%s" \
tenantId="%s" id="%s" \
enabled="true" password="%s"/>' % (email_id, tenantid, userid, password)
resp, content = header.request(url, "PUT", body=body,
headers={"Content-Type": "application/xml",
"X-Auth-Token": auth_token,
"ACCEPT": "application/xml"})
return (resp, content)
def delete_user_xml(userid, auth_token):
header = httplib2.Http(".cache")
url = '%susers/%s' % (URL_V2, userid)
resp, _content = header.request(url, "DELETE", body='',
headers={"Content-Type": "application/xml",
"X-Auth-Token": auth_token,
"ACCEPT": "application/xml"})
return resp
def add_user_xml(userid, auth_token):
header = httplib2.Http(".cache")
url = '%susers/%s/add' % (URL_V2, userid)
resp, content = header.request(url, "PUT", body='{}',
headers={"Content-Type": "application/xml",
"X-Auth-Token": auth_token,
"ACCEPT": "application/xml"})
return (resp, content)
def add_user_json(auth_token):
header = httplib2.Http(".cache")
url = '%susers/' % (URL_V2)
resp, content = header.request(url, "PUT", body='{}',
headers={"Content-Type": "application/json",
"X-Auth-Token": auth_token})
return (resp, content)
def user_set_enabled(auth_token, user_id, enabled=True):
header = httplib2.Http(".cache")
url = '%susers/%s/enabled' % (URL_V2, user_id)
data = '{"user": { "enabled": %s}}' % ("true" if enabled else "false")
resp, content = header.request(url, "PUT", body=data,
headers={"Content-Type": "application/json",
"X-Auth-Token": auth_token})
return (resp, content)
def user_update_json(auth_token, user_id, email=None):
header = httplib2.Http(".cache")
url = '%susers/%s' % (URL_V2, user_id)
if email is None:
new_email = "updatedjoeuser@openstack.org"
else:
new_email = email
data = '{"user": { "email": "%s"}}' % (new_email)
resp, content = header.request(url, "PUT", body=data,
headers={"Content-Type": "application/json",
"X-Auth-Token": auth_token})
return (resp, content)
def user_update_xml(auth_token, user_id, email=None):
header = httplib2.Http(".cache")
url = '%susers/%s' % (URL_V2, user_id)
if email is None:
new_email = "updatedjoeuser@openstack.org"
else:
new_email = email
data = '<?xml version="1.0" encoding="UTF-8"?> \
<user xmlns="http://docs.openstack.org/identity/api/v2.0" \
email="%s" />' % (new_email)
resp, content = header.request(url, "PUT", body=data,
headers={"Content-Type": "application/xml",
"X-Auth-Token": auth_token,
"ACCEPT": "application/xml"})
return (resp, content)
def user_get_json(user_id, auth_token):
header = httplib2.Http(".cache")
url = '%susers/%s' % (URL_V2, user_id)
#test for Content-Type = application/json
resp, content = header.request(url, "GET", body='{}',
headers={"Content-Type": "application/json",
"X-Auth-Token": auth_token})
return (resp, content)
def user_password_json(user_id, auth_token):
header = httplib2.Http(".cache")
url = '%susers/%s/password' % (URL_V2, user_id)
data = '{"user": { "password": "p@ssword"}}'
resp, content = header.request(url, "PUT", body=data,
headers={"Content-Type": "application/json",
"X-Auth-Token": auth_token})
return (resp, content)
def user_password_xml(user_id, auth_token):
header = httplib2.Http(".cache")
url = '%susers/%s/password' % (URL_V2, user_id)
data = '<?xml version="1.0" encoding="UTF-8"?> \
<user xmlns="http://docs.openstack.org/identity/api/v2.0" \
password="p@ssword" />'
resp, content = header.request(url, "PUT", body=data,
headers={"Content-Type": "application/xml",
"X-Auth-Token": auth_token,
"ACCEPT": "application/xml"})
return (resp, content)
def user_enabled_json(user_id, auth_token):
header = httplib2.Http(".cache")
url = '%susers/%s/enabled' % (URL_V2, user_id)
data = {"user": {"enabled": True}}
resp, content = header.request(url, "PUT", body=json.dumps(data),
headers={"Content-Type": "application/json",
"X-Auth-Token": auth_token})
return (resp, content)
def user_enabled_xml(user_id, auth_token):
header = httplib2.Http(".cache")
url = '%susers/%s/enabled' % (URL_V2, user_id)
data = '<?xml version="1.0" encoding="UTF-8"?> \
<user xmlns="http://docs.openstack.org/identity/api/v2.0" \
enabled="true" />'
resp, content = header.request(url, "PUT", body=data,
headers={"Content-Type": "application/xml",
"X-Auth-Token": auth_token,
"ACCEPT": "application/xml"})
return (resp, content)
def user_tenant_update_json(user_id, tenant_id, auth_token):
header = httplib2.Http(".cache")
url = '%susers/%s/tenant' % (URL_V2, user_id)
data = {"user": {"tenantId": tenant_id}}
resp, content = header.request(url, "PUT", body=json.dumps(data),
headers={"Content-Type": "application/json",
"X-Auth-Token": auth_token})
return (resp, content)
def user_tenant_update_xml(user_id, tenant_id, auth_token):
header = httplib2.Http(".cache")
url = '%susers/%s/tenant' % (URL_V2, user_id)
data = '<?xml version="1.0" encoding="UTF-8"?> \
<user xmlns="http://docs.openstack.org/identity/api/v2.0" \
tenantId="%s" />' % (tenant_id)
resp, content = header.request(url, "PUT", body=data,
headers={"Content-Type": "application/xml",
"X-Auth-Token": auth_token,
"ACCEPT": "application/xml"})
return (resp, content)
def user_get_xml(user_id, auth_token):
header = httplib2.Http(".cache")
url = '%susers/%s' % (URL_V2, user_id)
resp, content = header.request(url, "GET", body='{}',
headers={"Content-Type": "application/xml",
"X-Auth-Token": auth_token,
"ACCEPT": "application/xml"})
return (resp, content)
def users_get_json(user_id, auth_token):
header = httplib2.Http(".cache")
url = '%susers/%s' % (URL_V2, user_id)
resp, content = header.request(url, "GET", body='{}',
headers={"Content-Type": "application/json",
"X-Auth-Token": auth_token})
return (resp, content)
def users_get_all_xml(auth_token):
header = httplib2.Http(".cache")
url = '%susers' % (URL_V2)
resp, content = header.request(url, "GET", body='{}',
headers={"Content-Type": "application/xml",
"X-Auth-Token": auth_token,
"ACCEPT": "application/xml"})
return (resp, content)
def users_get_all_json(auth_token):
header = httplib2.Http(".cache")
url = '%susers' % (URL_V2)
resp, content = header.request(url, "GET", body='{}',
headers={"Content-Type": "application/json",
"X-Auth-Token": auth_token})
return (resp, content)
def users_get_xml(tenant_id, auth_token):
header = httplib2.Http(".cache")
url = '%susers/%s' % (URL_V2, tenant_id)
resp, content = header.request(url, "GET", body='{}',
headers={"Content-Type": "application/xml",
"X-Auth-Token": auth_token,
"ACCEPT": "application/xml"})
return (resp, content)
def get_userid():
return 'test_user11'
def get_password():
return 'secrete'
def get_email():
return 'joetest@openstack.org'
def get_tenant():
return '1234'
def get_another_tenant():
return '4321'
def get_user():
return 'test_user'
def get_userdisabled():
return 'disabled'
def get_auth_token():
return '999888777666'
def get_exp_auth_token():
return '000999'
def get_none_token():
return ''
def get_non_existing_token():
return 'invalid_token'
def get_disabled_token():
return '999888777'
def content_type(resp):
return resp['content-type'].split(';')[0]
def get_global_tenant():
return 'GlobalTenant'
def handle_user_resp(self, content, respvalue, resptype):
if respvalue == 200:
if resptype == 'application/json':
content = json.loads(content)
if 'tenantId' in content['user']:
self.tenant = content['user']['tenantId']
self.userid = content['user']['id']
if resptype == 'application/xml':
content = etree.fromstring(content)
self.tenant = content.get("tenantId")
self.id = content.get("id")
if respvalue == 500:
self.fail('Identity Fault')
elif respvalue == 503:
self.fail('Service Not Available')
def create_role(roleid, auth_token):
header = httplib2.Http(".cache")
url = '%sroles' % (URL_V2)
body = {"role": {"id": roleid,
"description": "A description ..."}}
resp, content = header.request(url, "POST", body=json.dumps(body),
headers={"Content-Type": "application/json",
"X-Auth-Token": auth_token})
return (resp, content)
def create_role_ref(user_id, role_id, tenant_id, auth_token):
header = httplib2.Http(".cache")
url = '%susers/%s/roleRefs' % (URL_V2, user_id)
body = {"roleRef": {"tenantId": tenant_id,
"roleId": role_id}}
resp, content = header.request(url, "POST", body=json.dumps(body),
headers={"Content-Type": "application/json",
"X-Auth-Token": auth_token})
return (resp, content)
def create_role_ref_xml(user_id, role_id, tenant_id, auth_token):
header = httplib2.Http(".cache")
url = '%susers/%s/roleRefs' % (URL_V2, user_id)
body = '<?xml version="1.0" encoding="UTF-8"?>\
<roleRef xmlns="http://docs.openstack.org/identity/api/v2.0" \
tenantId="%s" roleId="%s"/>\
' % (tenant_id, role_id)
resp, content = header.request(url, "POST", body=body,
headers={"Content-Type": "application/xml",
"X-Auth-Token": auth_token,
"ACCEPT": "application/xml"})
return (resp, content)
def delete_role_ref(user, role_ref_id, auth_token):
header = httplib2.Http(".cache")
url = '%susers/%s/roleRefs/%s' % (URL_V2, user, role_ref_id)
resp, content = header.request(url, "DELETE", body='',
headers={"Content-Type": "application/json",
"X-Auth-Token": str(auth_token)})
return (resp, content)
def create_role_xml(role_id, auth_token):
header = httplib2.Http(".cache")
url = '%sroles' % (URL_V2)
body = '<?xml version="1.0" encoding="UTF-8"?>\
<role xmlns="http://docs.openstack.org/identity/api/v2.0" \
id="%s" description="A Description of the role"/>\
' % role_id
resp, content = header.request(url, "POST", body=body,
headers={"Content-Type": "application/xml",
"X-Auth-Token": auth_token,
"ACCEPT": "application/xml"})
return (resp, content)
def create_endpoint(tenant_id, endpoint_templates_id, auth_token):
header = httplib2.Http(".cache")
url = '%stenants/%s/endpoints' % (URL_V2, tenant_id)
body = {"endpointTemplate": {"id": endpoint_templates_id}}
resp, content = header.request(url, "POST", body=json.dumps(body),
headers={"Content-Type": "application/json",
"X-Auth-Token": auth_token})
return (resp, content)
def create_endpoint_xml(tenant_id, endpoint_templates_id, auth_token):
header = httplib2.Http(".cache")
url = '%stenants/%s/endpoints' % (URL_V2, tenant_id)
body = '<?xml version="1.0" encoding="UTF-8"?>\
<endpointTemplate xmlns="http://docs.openstack.org/identity/api/v2.0" \
id="%s"/>' % (endpoint_templates_id)
resp, content = header.request(url, "POST", body=body,
headers={"Content-Type": "application/xml",
"X-Auth-Token": auth_token,
"ACCEPT": "application/xml"})
return (resp, content)
def delete_all_endpoint(tenant_id, auth_token):
header = httplib2.Http(".cache")
url = '%stenants/%s/endpoints' % (URL_V2, tenant_id)
#test for Content-Type = application/json
resp, content = header.request(url, "GET", body='{}',
headers={"Content-Type": "application/json",
"X-Auth-Token": auth_token})
if int(resp['status']) == 500:
assert False
# self.fail('Identity Fault')
elif int(resp['status']) == 503:
assert False
# self.fail('Service Not Available')
#verify content
obj = json.loads(content)
try:
endpoints = obj["endpoints"]["values"]
except KeyError:
pass
else:
for endpoint in endpoints:
url = '%stenants/%s/endpoints/%s' % (
URL_V2, tenant_id, endpoint["id"])
header.request(url, "DELETE", body='', headers={
"Content-Type": "application/json",
"X-Auth-Token": str(auth_token)})
if __name__ == '__main__':
unittest.main()
View Git Blame Copy Permalink