mirror of
https://github.com/getredash/redash.git
synced 2025-12-19 17:37:19 -05:00
cbde237b12
* removed explicit object inheritance * minor fix * pr comments --------- Co-authored-by: Andrew Chubatiuk <andrew.chubatiuk@motional.com>
36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
from unittest import TestCase
|
|
|
|
from mock import MagicMock
|
|
from werkzeug.exceptions import BadRequest
|
|
|
|
from redash.handlers.base import paginate
|
|
|
|
|
|
class DummyResults:
|
|
items = [i for i in range(25)]
|
|
|
|
|
|
dummy_results = DummyResults()
|
|
|
|
|
|
class TestPaginate(TestCase):
|
|
def setUp(self):
|
|
self.query_set = MagicMock()
|
|
self.query_set.count = MagicMock(return_value=102)
|
|
self.query_set.paginate = MagicMock(return_value=dummy_results)
|
|
|
|
def test_returns_paginated_results(self):
|
|
page = paginate(self.query_set, 1, 25, lambda x: x)
|
|
self.assertEqual(page["page"], 1)
|
|
self.assertEqual(page["page_size"], 25)
|
|
self.assertEqual(page["count"], 102)
|
|
self.assertEqual(page["results"], dummy_results.items)
|
|
|
|
def test_raises_error_for_bad_page(self):
|
|
self.assertRaises(BadRequest, lambda: paginate(self.query_set, -1, 25, lambda x: x))
|
|
self.assertRaises(BadRequest, lambda: paginate(self.query_set, 6, 25, lambda x: x))
|
|
|
|
def test_raises_error_for_bad_page_size(self):
|
|
self.assertRaises(BadRequest, lambda: paginate(self.query_set, 1, 251, lambda x: x))
|
|
self.assertRaises(BadRequest, lambda: paginate(self.query_set, 1, -1, lambda x: x))
|