designate/designate/tests/unit/api/test_admin_api.py

61 lines
2.0 KiB
Python

# Copyright 2015 Hewlett-Packard Development Company, L.P.
#
# Author: Federico Ceratto <federico.ceratto@hpe.com>
#
# 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 mock
import designate.tests
from designate import exceptions
from designate.api.admin.views import base
class MockRequest(object):
def __init__(self, GET=None):
self.GET = GET
class TestAdminAPI(designate.tests.TestCase):
def setUp(self):
super(TestAdminAPI, self).setUp()
@mock.patch.object(base.BaseView, '_get_collection_href')
@mock.patch.object(base.BaseView, '_get_next_href')
def test_limit_max(self, mock_coll_href, mock_next_href):
# Bug 1494799
# The code being tested should be deduplicated, see bug 1498432
mock_coll_href.return_value = None
mock_next_href.return_value = None
item_list = range(200)
bv = base.BaseView()
request = MockRequest(GET=dict(limit="max"))
links = bv._get_collection_links(request, item_list)
self.assertEqual(dict(self=None), links)
request = MockRequest(GET=dict(limit="MAX"))
links = bv._get_collection_links(request, item_list)
self.assertEqual(dict(self=None), links)
request = MockRequest(GET=dict(limit="200"))
links = bv._get_collection_links(request, item_list)
self.assertEqual(dict(self=None, next=None), links)
request = MockRequest(GET=dict(limit="BOGUS_STRING"))
self.assertRaises(
exceptions.ValueError,
bv._get_collection_links, request, item_list
)