Skip to content

D
django-rest-framework
Commit 9bc0319f by homm
Options

Revert tests

parent 60656e91
Showing with 36 additions and 120 deletions
tests/test_pagination.py
...@@ -2,9 +2,8 @@ ...@@ -2,9 +2,8 @@
from __future__ import unicode_literals from __future__ import unicode_literals
from rest_framework import exceptions, generics, pagination, serializers, status, filters from rest_framework import exceptions, generics, pagination, serializers, status, filters
from rest_framework.request import Request from rest_framework.request import Request
from rest_framework.pagination import PageLink, PAGE_BREAK, Cursor from rest_framework.pagination import PageLink, PAGE_BREAK
from rest_framework.test import APIRequestFactory from rest_framework.test import APIRequestFactory
from rest_framework.utils.urls import replace_query_param, remove_query_param
import pytest import pytest
factory = APIRequestFactory() factory = APIRequestFactory()
...@@ -477,75 +476,52 @@ class TestCursorPagination: ...@@ -477,75 +476,52 @@ class TestCursorPagination:
""" """
Unit tests for `pagination.CursorPagination`. Unit tests for `pagination.CursorPagination`.
""" """
class MockObject(object):
def __init__(self, idx):
self.created = idx
class MockQuerySet(object): def setup(self):
def __init__(self, items): class MockObject(object):
self.items = list(items) def __init__(self, idx):
self.created = idx
def filter(self, created__gt=None, created__lt=None):
if created__gt is not None: class MockQuerySet(object):
return type(self)([ def __init__(self, items):
self.items = items
def filter(self, created__gt=None, created__lt=None):
if created__gt is not None:
return MockQuerySet([
item for item in self.items
if item.created > int(created__gt)
])
assert created__lt is not None
return MockQuerySet([
item for item in self.items item for item in self.items
if item.created > int(created__gt) if item.created < int(created__lt)
]) ])
assert created__lt is not None def order_by(self, *ordering):
return type(self)([ if ordering[0].startswith('-'):
item for item in self.items return MockQuerySet(list(reversed(self.items)))
if item.created < int(created__lt) return self
])
def order_by(self, *ordering):
if ordering[0].startswith('-'):
return type(self)(list(reversed(self.items)))
return self
def __getitem__(self, sliced):
return self.items[sliced]
class ExamplePagination(pagination.CursorPagination):
page_size = 5
ordering = 'created'
class CustomCursorPagination(ExamplePagination):
cursor_query_param = 'since'
reverse_query_param = 'before'
def decode_cursor(self, request):
rev = False
if self.reverse_query_param in request.query_params:
rev = True
query_param = self.reverse_query_param
elif self.cursor_query_param in request.query_params:
query_param = self.cursor_query_param
else:
return
return Cursor(0, rev, request.query_params[query_param])
def encode_cursor(self, cursor):
if cursor.reverse:
query_param = self.reverse_query_param
base_url = remove_query_param(self.base_url, self.cursor_query_param)
else:
query_param = self.cursor_query_param
base_url = remove_query_param(self.base_url, self.reverse_query_param)
return replace_query_param(base_url, query_param, cursor.position)
def setup(self): def __getitem__(self, sliced):
self.pagination = self.ExamplePagination() return self.items[sliced]
self.queryset = self.MockQuerySet(
map(self.MockObject, [ class ExamplePagination(pagination.CursorPagination):
page_size = 5
ordering = 'created'
self.pagination = ExamplePagination()
self.queryset = MockQuerySet([
MockObject(idx) for idx in [
1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1, 2, 3, 4, 4, 1, 2, 3, 4, 4,
4, 4, 5, 6, 7, 4, 4, 5, 6, 7,
7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
7, 7, 7, 8, 9, 7, 7, 7, 8, 9,
9, 9, 9, 9, 9 9, 9, 9, 9, 9
]) ]
) ])
def get_pages(self, url): def get_pages(self, url):
""" """
...@@ -668,66 +644,6 @@ class TestCursorPagination: ...@@ -668,66 +644,6 @@ class TestCursorPagination:
assert isinstance(self.pagination.to_html(), type('')) assert isinstance(self.pagination.to_html(), type(''))
def test_custom_cursor_format(self):
# setup
self.pagination = self.CustomCursorPagination()
# The CustomCursorPagination expects unique keys
self.queryset = self.MockQuerySet(
map(self.MockObject, [
1, 2, 4, 8, 10,
11, 12, 13, 14, 15,
18, 33, 35, 36, 37,
38, 39, 40, 41,
])
)
# actual test
(previous, current, next, previous_url, next_url) = self.get_pages('/')
assert previous is None
assert current == [1, 2, 4, 8, 10]
assert next == [11, 12, 13, 14, 15]
assert previous_url is None
assert 'since=10' in next_url
(previous, current, next, previous_url, next_url) = self.get_pages(next_url)
assert previous == [1, 2, 4, 8, 10]
assert current == [11, 12, 13, 14, 15]
assert next == [18, 33, 35, 36, 37]
assert 'before=11' in previous_url
assert 'since' not in previous_url
assert 'since=15' in next_url
(previous, current, next, previous_url, next_url) = self.get_pages(next_url)
(previous, current, next, previous_url, next_url) = self.get_pages(next_url)
assert previous == [18, 33, 35, 36, 37]
assert current == [38, 39, 40, 41]
assert next is None
assert 'before=38' in previous_url
assert 'since' not in previous_url
assert next_url is None
(previous, current, next, previous_url, next_url) = self.get_pages(previous_url)
assert previous == [11, 12, 13, 14, 15]
assert current == [18, 33, 35, 36, 37]
assert next == [38, 39, 40, 41]
assert 'before=18' in previous_url
assert 'since=37' in next_url
assert 'before' not in next_url
(previous, current, next, previous_url, next_url) = self.get_pages(previous_url)
(previous, current, next, previous_url, next_url) = self.get_pages(previous_url)
assert previous is None
assert current == [1, 2, 4, 8, 10]
assert next == [11, 12, 13, 14, 15]
assert previous_url is None
assert 'since=10' in next_url
assert 'before' not in next_url
def test_get_displayed_page_numbers(): def test_get_displayed_page_numbers():
""" """
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment