50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
# Copyright (c) 2016 Johns Hopkins University Applied Physics Laboratory
|
|
#
|
|
# 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
|
|
|
|
from urllib import parse as urllib
|
|
|
|
from tempest import config
|
|
from tempest.lib.common import rest_client
|
|
|
|
CONF = config.CONF
|
|
|
|
|
|
class ConsumerClient(rest_client.RestClient):
|
|
|
|
def list_consumers_in_container(self, container_id, **kwargs):
|
|
uri = "/v1/containers/%s/consumers" % container_id
|
|
if kwargs:
|
|
uri += "?%s" % urllib.urlencode(kwargs)
|
|
|
|
response, body = self.get(uri)
|
|
self.expected_success(200, response.status)
|
|
return json.loads(body.decode("utf-8"))
|
|
|
|
def add_consumer_to_container(self, container_id, **kwargs):
|
|
uri = "/v1/containers/%s/consumers" % container_id
|
|
|
|
response, body = self.post(uri, json.dumps(kwargs))
|
|
self.expected_success(200, response.status)
|
|
return json.loads(body.decode("utf-8"))
|
|
|
|
def delete_consumer_from_container(self, container_id, **kwargs):
|
|
uri = "/v1/containers/%s/consumers" % container_id
|
|
|
|
response, body = self.delete(uri, body=json.dumps(kwargs))
|
|
self.expected_success(200, response.status)
|
|
return json.loads(body.decode("utf-8"))
|