def test_extend_query_with_batching_no_batching(self): self.expect(self.config.batching_enabled).result(False) self.expect(self.config.lazy).result(False) source = BaseManagementTableSource(self.config, self.request) self.replay() result = source.extend_query_with_batching({}) self.assertEquals(result, {'batching': False})
def test_extend_query_with_batching(self): self.expect(self.config.batching_enabled).result(True) self.expect(self.config.lazy).result(False) self.expect(self.config.batching_pagesize).result(3) self.expect(self.config.batching_current_page).result(4) source = BaseManagementTableSource(self.config, self.request) self.replay() result = source.extend_query_with_batching({}) self.assertEquals( result, { 'pagesize': 3, 'current_page': 4, 'batching': True, })