def test_limit_offset_query(self): ds = MagicMock() ds.get_resource.return_value = {'offset': 5, 'limit': 5, 'items': []} rl = CollectionResource(client=MagicMock(data_store=ds), href='/') rl2 = rl.query(offset=5, limit=5) self.assertEqual(ds.get_resource.call_count, 0) list(rl2) ds.get_resource.assert_called_once_with('/', params={ 'offset': 5, 'limit': 5 }) self.assertEqual(rl2.offset, 5) self.assertEqual(rl2.limit, 5)
def test_limit_offset_query(self): ds = MagicMock() ds.get_resource.return_value = { 'offset': 5, 'limit': 5, 'items': [] } rl = CollectionResource(client=MagicMock(data_store=ds), href='/') rl2 = rl.query(offset=5, limit=5) self.assertEqual(ds.get_resource.call_count, 0) list(rl2) ds.get_resource.assert_called_once_with('/', params={ 'offset': 5, 'limit': 5}) self.assertEqual(rl2.offset, 5) self.assertEqual(rl2.limit, 5)