Merge "Add tempest clients for limits"
diff --git a/keystone_tempest_plugin/clients.py b/keystone_tempest_plugin/clients.py
index 97b5b14..93c0753 100644
--- a/keystone_tempest_plugin/clients.py
+++ b/keystone_tempest_plugin/clients.py
@@ -19,6 +19,7 @@
from keystone_tempest_plugin.services.identity.v3 import (
service_providers_client)
from keystone_tempest_plugin.services.identity.v3 import auth_client
+from keystone_tempest_plugin.services.identity.v3 import limits_client
from keystone_tempest_plugin.services.identity.v3 import saml2_client
from tempest import clients
@@ -29,7 +30,10 @@
def __init__(self, credentials):
super(Manager, self).__init__(credentials)
+ # keystone auth client
self.auth_client = auth_client.AuthClient(self.auth_provider)
+
+ # federation clients
self.identity_providers_client = (
identity_providers_client.IdentityProvidersClient(
self.auth_provider))
@@ -40,3 +44,8 @@
self.service_providers_client = (
service_providers_client.ServiceProvidersClient(
self.auth_provider))
+
+ # unified limits clients
+ self.registered_limits_client = limits_client.RegisteredLimitsClient(
+ self.auth_provider)
+ self.limits_client = limits_client.LimitsClient(self.auth_provider)
diff --git a/keystone_tempest_plugin/services/identity/v3/limits_client.py b/keystone_tempest_plugin/services/identity/v3/limits_client.py
new file mode 100644
index 0000000..6e95b9d
--- /dev/null
+++ b/keystone_tempest_plugin/services/identity/v3/limits_client.py
@@ -0,0 +1,156 @@
+# Copyright 2020 SUSE 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 json
+
+import http.client
+from tempest.lib.common import rest_client
+
+from keystone_tempest_plugin.services.identity import clients
+
+
+class RegisteredLimitsClient(clients.Identity):
+
+ subpath_prefix = 'registered_limits'
+ collection_url = subpath_prefix
+ entity_url = subpath_prefix + '/%s'
+
+ def create_registered_limits(self, payload):
+ """Create a list of registered limits.
+
+ :param body: A list of registered limits objects.
+ """
+ post_body = json.dumps({'registered_limits': payload})
+ resp, body = super(RegisteredLimitsClient, self).post(
+ self.collection_url, post_body)
+ self.expected_success(http.client.CREATED, resp.status)
+ body = json.loads(body.decode('utf-8'))
+ return rest_client.ResponseBody(resp, body)
+
+ def list_registered_limits(self, **kwargs):
+ """List registered limits.
+
+ :param kwargs: Filter by service_id, region_id, or resource_name
+ """
+ resp, body = super(RegisteredLimitsClient, self).get(
+ self.collection_url, **kwargs)
+ self.expected_success(http.client.OK, resp.status)
+ body = json.loads(body.decode('utf-8'))
+ return rest_client.ResponseBody(resp, body)
+
+ def update_registered_limit(self, registered_limit_id, registered_limit):
+ """Update a registered limit.
+
+ :param registered_limit_id: ID of registered limit to update
+ :param registered_limit: new registered limit object
+ """
+ patch_body = json.dumps({'registered_limit': registered_limit})
+ resp, body = super(RegisteredLimitsClient, self).patch(
+ self.entity_url % registered_limit_id, patch_body)
+ self.expected_success(http.client.OK, resp.status)
+ body = json.loads(body.decode('utf-8'))
+ return rest_client.ResponseBody(resp, body)
+
+ def show_registered_limit(self, registered_limit_id):
+ """Get a registered limit.
+
+ :param registered_limit_id: ID of registered limit to show
+ """
+ resp, body = super(RegisteredLimitsClient, self).get(
+ self.entity_url % registered_limit_id)
+ self.expected_success(http.client.OK, resp.status)
+ body = json.loads(body.decode('utf-8'))
+ return rest_client.ResponseBody(resp, body)
+
+ def delete_registered_limit(self, registered_limit_id):
+ """Delete a registered limit.
+
+ :param registered_limit_id: ID of registered limit to delete.
+ """
+ resp, body = super(RegisteredLimitsClient, self).delete(
+ self.entity_url % registered_limit_id)
+ self.expected_success(http.client.NO_CONTENT, resp.status)
+ return rest_client.ResponseBody(resp, body)
+
+
+class LimitsClient(clients.Identity):
+
+ subpath_prefix = 'limits'
+ collection_url = subpath_prefix
+ entity_url = subpath_prefix + '/%s'
+
+ def limits_model(self):
+ """Get limits model from server."""
+ url = self.entity_url % 'model'
+ resp, body = super(LimitsClient, self).get(url)
+ self.expected_success(http.client.OK, resp.status)
+ body = json.loads(body.decode('utf-8'))
+ return rest_client.ResponseBody(resp, body)
+
+ def create_limits(self, payload):
+ """Create a list of project limits.
+
+ :param body: A list of project limits objects.
+ """
+ post_body = json.dumps({'limits': payload})
+ resp, body = super(LimitsClient, self).post(
+ self.collection_url, post_body)
+ self.expected_success(http.client.CREATED, resp.status)
+ body = json.loads(body.decode('utf-8'))
+ return rest_client.ResponseBody(resp, body)
+
+ def list_limits(self, **kwargs):
+ """List project limits.
+
+ :param kwargs: Filter by service_id, region_id, resource_name,
+ or project/domain ID
+ """
+ resp, body = super(LimitsClient, self).get(
+ self.collection_url, **kwargs)
+ self.expected_success(http.client.OK, resp.status)
+ body = json.loads(body.decode('utf-8'))
+ return rest_client.ResponseBody(resp, body)
+
+ def update_limit(self, limit_id, limit):
+ """Update a project limit.
+
+ :param limit_id: ID of project limit to update
+ :param limit: new project limit object
+ """
+ patch_body = json.dumps({'limit': limit})
+ resp, body = super(LimitsClient, self).patch(
+ self.entity_url % limit_id, patch_body)
+ self.expected_success(http.client.OK, resp.status)
+ body = json.loads(body.decode('utf-8'))
+ return rest_client.ResponseBody(resp, body)
+
+ def show_limit(self, limit_id):
+ """Get a project limit.
+
+ :param limit_id: ID of project limit to show
+ """
+ resp, body = super(LimitsClient, self).get(self.entity_url % limit_id)
+ self.expected_success(http.client.OK, resp.status)
+ body = json.loads(body.decode('utf-8'))
+ return rest_client.ResponseBody(resp, body)
+
+ def delete_limit(self, limit_id):
+ """Delete a project limit.
+
+ :param limit_id: ID of project limit to delete.
+ """
+ resp, body = super(LimitsClient, self).delete(
+ self.entity_url % limit_id)
+ self.expected_success(http.client.NO_CONTENT, resp.status)
+ return rest_client.ResponseBody(resp, body)
diff --git a/keystone_tempest_plugin/tests/base.py b/keystone_tempest_plugin/tests/base.py
index 16f0092..77dcafe 100644
--- a/keystone_tempest_plugin/tests/base.py
+++ b/keystone_tempest_plugin/tests/base.py
@@ -42,3 +42,6 @@
cls.tokens_client = cls.keystone_manager.token_v3_client
cls.consumers_client = cls.keystone_manager.oauth_consumers_client
cls.oauth_token_client = cls.keystone_manager.oauth_token_client
+ cls.registered_limits_client = (
+ cls.keystone_manager.registered_limits_client)
+ cls.limits_client = cls.keystone_manager.limits_client