def test_view_with_all_pagination_attrs(self):
self.kwargs['view_instance'].page_kwarg = 'page'
self.kwargs['view_instance'].paginate_by_param = 'page_size'
self.assertEqual(PaginationKeyBit().get_data(**self.kwargs), {
'page_size': u'10',
'page': u'1'
})
def test_view_with_offset_pagination_attrs(self):
self.kwargs['view_instance'].paginator.limit_query_param = 'limit'
self.kwargs['view_instance'].paginator.offset_query_param = 'offset'
self.assertEqual(PaginationKeyBit().get_data(**self.kwargs), {
'limit': '5',
'offset': '15'
})
class CustomListKeyConstructor(DefaultKeyConstructor):
list_sql = ListSqlQueryKeyBit()
pagination = PaginationKeyBit()
updated_at = UpdatedAtKeyBit()
print("List SQL", list_sql)
print("pagination", pagination)
print("updated_at", updated_at)
class CustomListKeyConstructor(DefaultKeyConstructor):
"""
Used to compute cache key for a list of objects.
"""
list_sql = ListSqlQueryKeyBit()
pagination = PaginationKeyBit()
updated_at = UpdatedAtKeyBit()
user = UserKeyBit()
all_query_params = QueryParamsKeyBit()
def test_view_with_cursor_pagination_attrs(self):
self.kwargs['view_instance'].paginator.cursor_query_param = 'cursor'
self.assertEqual(PaginationKeyBit().get_data(**self.kwargs), {'cursor': 'foo'})
def test_view_with_all_pagination_attrs__without_query_params(self):
self.kwargs['view_instance'].paginator.page_query_param = 'page'
self.kwargs['view_instance'].paginator.page_size_query_param = 'page_size'
self.kwargs['request'] = factory.get('')
self.assertEqual(PaginationKeyBit().get_data(**self.kwargs), {})
def test_view_with_paginate_by_param(self):
self.kwargs['view_instance'].paginator.page_query_param = None
self.kwargs['view_instance'].paginator.page_size_query_param = 'page_size'
self.assertEqual(PaginationKeyBit().get_data(**self.kwargs), {'page_size': '10'})
def test_view_with_empty_pagination_arguments(self):
self.kwargs['view_instance'].paginator.page_query_param = None
self.kwargs['view_instance'].paginator.page_size_query_param = None
self.assertEqual(PaginationKeyBit().get_data(**self.kwargs), {})
def test_view_without_pagination_arguments(self):
self.kwargs['view_instance'] = Mock(spec_set=[])
self.assertEqual(PaginationKeyBit().get_data(**self.kwargs), {})
class CustomListKeyConstructor(DefaultKeyConstructor):
list_sql = ListSqlQueryKeyBit()
pagination = PaginationKeyBit()
updated_at = UpdatedAtKeyBit()
def test_view_with_page_kwarg(self):
self.kwargs['view_instance'].page_kwarg = 'page'
self.kwargs['view_instance'].paginate_by_param = None
self.assertEqual(PaginationKeyBit().get_data(**self.kwargs),
{'page': u'1'})
class UpdatedAtDayForFeed(DefaultKeyConstructor):
updated_at = UpdatedAtKeyBit()
data_day = UpdatedAtDay()
pagination = PaginationKeyBit()
query = QueryParamsKeyBit()
