Exemple #1
0
    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)