Пример #1
0
    def test_write(self):
        request = DummyRequest()
        response_mock = ResponseMock()
        request.response = response_mock
        sorting = Sorting('id', 'desc')
        sorting_writer = CookieSortingWriter(
            request,
            PropertyNamingMock(
                'page_number',
                'page_size',
                'sort_property',
                'sort_direction'
            )
        )
        sorting_writer(sorting)
        self.assertEqual(response_mock.get_cookie('sort_property'), 'id')
        self.assertEqual(response_mock.get_cookie('sort_direction'), 'desc')
        self.assertEqual(response_mock.call_count(), 2)

        request = DummyRequest()
        response_mock = ResponseMock()
        request.response = response_mock
        sorting = Sorting('id', 'desc')
        sorting_writer = CookieSortingWriter(
            request,
            PropertyNamingMock(
                'cookie_prefix.page_number',
                'cookie_prefix.page_size',
                'cookie_prefix.sort_property',
                'cookie_prefix.sort_direction'
            )
        )
        sorting_writer(sorting)
        self.assertEqual(response_mock.get_cookie('cookie_prefix.sort_property'), 'id')
        self.assertEqual(response_mock.get_cookie('cookie_prefix.sort_direction'), 'desc')
        self.assertEqual(response_mock.call_count(), 2)
Пример #2
0
 def test_write(self):
     request = DummyRequest()
     response_mock = ResponseMock()
     request.response = response_mock
     pagination_request = PagingRequest(3, 12)
     pagination_request_writer = CookiePagingRequestWriter(
         request,
         PropertyNamingMock(
             'page_number',
             'page_size',
             'sort_property',
             'sort_direction'
         )
     )
     pagination_request_writer(pagination_request)
     self.assertEqual(response_mock.get_cookie('page_number'), '3')
     self.assertEqual(response_mock.get_cookie('page_size'), '12')
     self.assertEqual(response_mock.call_count(), 2)