test_api.py 文件源码

def test_items(self):
        dwarves = [{'name': 'Dori'}, {'name': 'Nori'}, {'name': 'Ori'},
                   {'name': 'Bifur'}, {'name': 'Bofur'}, {'name': 'Bombur'}]

        def request_callback(request, uri, headers):
            if uri.endswith('dwarves') or uri.endswith('?page=1'):
                items = dwarves[:3]
                page_id = 1
            elif uri.endswith('?page=2'):
                items = dwarves[3:]
                page_id = 2
            else:
                items = []
                page_id = uri.split('=')[-1]
            return 200, headers, json.dumps({'page': page_id, 'dwarves': items, 'total_pages': 2})

        self.register_url(self.GET, 'dwarves', body=request_callback)

        api = QuizletAPI()
        self.assertEqual(list(api.dwarves.items()), dwarves)