picasso/laos/tests/functional/client.py

113 lines
3.4 KiB
Python

# All Rights Reserved.
#
# 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 as jsonlib
from aiohttp import test_utils
class AppsV1(object):
# /v1/{project_id}/apps
apps_path = "/v1/{}/apps"
# /v1/{project_id}/apps/{app}
app_path = apps_path + "/{}"
def __init__(self, test_client):
self.client = test_client
async def list(self):
return await self.client.list(self.apps_path)
async def show(self, app_name):
return await self.client.show(self.app_path, app_name)
async def create(self, app_name):
data = {
"app": {
"name": app_name
}
}
return await self.client.create(self.apps_path, data)
async def delete(self, app_name):
return await self.client.remove(self.app_path, app_name)
class RoutesV1(object):
# /v1/{project_id}/apps/{app}/routes
routes_path = "/v1/{}/apps/{}/routes"
# /v1/{project_id}/apps/{app}/routes{}
route_path = routes_path + "{}"
def __init__(self, test_client):
self.client = test_client
async def list(self, app_name):
return await self.client.list(
self.routes_path, app_name)
async def show(self, app_name, path):
return await self.client.show(
self.route_path, app_name, path)
async def create(self, app_name, **data):
return await self.client.create(
self.routes_path, {"route": data}, app_name)
async def delete(self, app_name, path):
return await self.client.remove(
self.route_path, app_name, path)
class ProjectBoundLaosTestClient(test_utils.TestClient):
def __init__(self, app_or_server, project_id):
super(ProjectBoundLaosTestClient, self).__init__(app_or_server)
self.project_id = project_id
self.headers = {
"Content-Type": "application/json"
}
self.apps = AppsV1(self)
self.routes = RoutesV1(self)
async def create(self, route, data, *parts):
resp = await self.post(
route.format(self.project_id, *parts),
data=jsonlib.dumps(data), headers=self.headers)
json = await resp.json()
return json, resp.status
async def list(self, route, *parts):
resp = await self.get(
route.format(self.project_id, *parts),
headers=self.headers)
json = await resp.json()
return json, resp.status
async def show(self, route, resource_name, *parts):
resp = await self.get(
route.format(self.project_id, resource_name, *parts),
headers=self.headers)
json = await resp.json()
return json, resp.status
async def remove(self, route, resource_name, *parts):
resp = await self.delete(
route.format(self.project_id, resource_name, *parts),
headers=self.headers)
json = await resp.json()
return json, resp.status