class TrafficWrapperTest(TestCase): def setUp(self): self._wrapper = TrafficWrapper('TOKEN') @responses.activate def test_001_fetch_traffic_info(self): payload = ('{"host1":{"left": 42, "bytes": 12, "links": 12, ' '"limit": "8", "type": "links", "extra": 20, ' '"reset": "daily"}}') responses.add( responses.GET, 'https://api.real-debrid.com/rest/1.0/traffic', status=200, body=payload, content_type='application/json' ) infos = self._wrapper.get_traffic_info() assert_true(isinstance(infos, list)) assert_true(isinstance(infos[0], Traffic)) tf = infos[0] eq_(tf.name, 'host1') eq_(tf.type, 'links')
def setUp(self): self._wrapper = TrafficWrapper('TOKEN')