コード例 #1
0
class TestRequests(unittest.TestCase):
    """
    Test HTTP requests for POST and GET.
    """
    def setUp(self):
        self.req = Requests('https://www.bitstamp.net/api/')

    def tearDown(self):
        pass

    def test_init(self):
        """
        Test __init__() - test setup.
        """
        self.assertEqual(self.req.baseURL, 'https://www.bitstamp.net/api/')
        self.assertEqual(self.req.URLError, None)
        self.assertEqual(self.req.HTTPError, None)


    def test_json_decode(self):
        """
        Test json_decode() - JSON decoder.
        """
        url = 'https://www.bitstamp.net/api/ticker/'
        json = urllib.request.urlopen(url)
        decode = self.req.json_decode(json)
        self.assertEqual(type(decode), dict)

    def test_create_url(self):
        """
        Test create_url() - creates a url.
        """
        ext = 'order_book/'
        params = {"group": 0}
        self.assertEqual(self.req.create_url(ext), {'url': 'https://www.bitstamp.net/api/order_book/'})
        self.assertEqual(self.req.create_url(ext, params), {'url': 'https://www.bitstamp.net/api/order_book/', 'params': b'group=0'})

    def test_post(self):
        """
        Test get() - HTTP GET request.
        """
        ext = 'order_book/'
        params = {"group": 0}
        # get request with just extension.
        get1 = self.req.request(ext)
        self.assertEqual(type(get1), dict)
        # get request with extension and parameters. Should give different result then last get.
        get2 = self.req.request(ext, params)
        self.assertEqual(type(get2), dict)
        self.assertNotEqual(get1, get2)
        # bad url extension. Should give warning.
        ext_bad = 'bad/'
        get3 = self.req.request(ext_bad)
        self.assertEqual(get3, None)
        self.assertEqual(self.req.URLError, 'NOT FOUND')
コード例 #2
0
 def setUp(self):
     self.req = Requests('https://www.bitstamp.net/api/')