4275cdaedc
Add new handler SnapshotDownload which returns empty response with X-Accel-Redirect header. Nginx reads this header and serves actuall snapshot file. Also changed 'reverse' function location, let it be more available for other modules. Change-Id: Id46b533e8d2a8021d1fbb2e5d4284508e1181704 Implements: blueprint snapshot-download-with-auth
145 lines
4.3 KiB
Python
145 lines
4.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright 2013 Mirantis, Inc.
|
|
#
|
|
# 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 web
|
|
|
|
from mock import patch
|
|
|
|
from nailgun.api.v1.handlers.base import BaseHandler
|
|
from nailgun.api.v1.handlers.base import content
|
|
|
|
from nailgun.test.base import BaseIntegrationTest
|
|
from nailgun.utils import reverse
|
|
|
|
|
|
class TestHandlers(BaseIntegrationTest):
|
|
def test_all_api_urls_404_or_405(self):
|
|
urls = {
|
|
'ClusterHandler': {'obj_id': 1},
|
|
'NodeHandler': {'obj_id': 1},
|
|
'ReleaseHandler': {'obj_id': 1},
|
|
}
|
|
for handler in urls:
|
|
test_url = reverse(handler, urls[handler])
|
|
resp = self.app.get(test_url, expect_errors=True)
|
|
self.assertTrue(resp.status_code in [404, 405])
|
|
resp = self.app.delete(test_url, expect_errors=True)
|
|
self.assertTrue(resp.status_code in [404, 405])
|
|
resp = self.app.put(
|
|
test_url,
|
|
json.dumps({}),
|
|
expect_errors=True
|
|
)
|
|
self.assertTrue(resp.status_code in [404, 405])
|
|
resp = self.app.post(
|
|
test_url,
|
|
json.dumps({}),
|
|
expect_errors=True
|
|
)
|
|
self.assertTrue(resp.status_code in [404, 405])
|
|
|
|
def test_http_response(self):
|
|
web.ctx.headers = []
|
|
|
|
http_codes = (
|
|
(200, 'ok'),
|
|
(201, 'created'),
|
|
(202, 'accepted'),
|
|
(204, 'no content'),
|
|
|
|
(400, 'bad request :('),
|
|
(401, 'unauthorized'),
|
|
(403, 'forbidden'),
|
|
(404, 'not found, try again'),
|
|
(405, 'no method'),
|
|
(406, 'unacceptable'),
|
|
(409, 'ooops, conflict'),
|
|
(415, 'unsupported media type'),
|
|
|
|
(500, 'internal problems'),
|
|
)
|
|
|
|
headers = {
|
|
'Content-Type': 'application/json',
|
|
'ETag': '737060cd8c284d8af7ad3082f209582d',
|
|
}
|
|
|
|
# test response status code and message
|
|
for code, message in http_codes:
|
|
with self.assertRaises(web.HTTPError) as cm:
|
|
raise BaseHandler.http(
|
|
status_code=code,
|
|
msg=message,
|
|
headers=headers
|
|
)
|
|
|
|
self.assertTrue(web.ctx.status.startswith(str(code)))
|
|
self.assertTrue(cm.exception.data, message)
|
|
|
|
for header, value in headers.items():
|
|
self.assertIn((header, value), web.ctx.headers)
|
|
|
|
def test_content_decorator(self):
|
|
|
|
class FakeHandler(object):
|
|
|
|
@content
|
|
def GET(self):
|
|
return {}
|
|
|
|
@content(["text/plain"])
|
|
def POST(self):
|
|
return "Plain Text"
|
|
|
|
web.ctx.headers = []
|
|
web.ctx.env = {"HTTP_ACCEPT": "text/html"}
|
|
|
|
fake_handler = FakeHandler()
|
|
self.assertRaises(
|
|
web.webapi.UnsupportedMediaType,
|
|
fake_handler.GET
|
|
)
|
|
|
|
web.ctx.env = {"HTTP_ACCEPT": "application/json"}
|
|
|
|
with patch("nailgun.api.v1.handlers.base.content_json") as cj:
|
|
fake_handler.GET()
|
|
self.assertEqual(cj.call_count, 1)
|
|
|
|
web.ctx.env = {"HTTP_ACCEPT": "*/*"}
|
|
|
|
with patch("nailgun.api.v1.handlers.base.content_json") as cj:
|
|
fake_handler.GET()
|
|
self.assertEqual(cj.call_count, 1)
|
|
|
|
web.ctx.headers = []
|
|
fake_handler.GET()
|
|
self.assertIn(
|
|
('Content-Type', 'application/json'),
|
|
web.ctx.headers
|
|
)
|
|
|
|
web.ctx.headers = []
|
|
web.ctx.env = {"HTTP_ACCEPT": "text/plain"}
|
|
fake_handler.POST()
|
|
self.assertIn(
|
|
# we don't have plain/text serializer right now
|
|
('Content-Type', 'application/json'),
|
|
web.ctx.headers
|
|
)
|