aljaz/tiingo-python
1
1
Fork 0
mirror of https://github.com/hydrosquall/tiingo-python.git synced 2025-12-17 20:04:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

Use VCR.py to Mock API Responses

Browse Source
This commit is contained in:
Dmitry Budaev
2017-10-08 13:14:41 +02:00
parent ab21be291c
commit c847da3aa9
15 changed files with 455 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tests/test_restclient.py
View File

@@ -3,6 +3,9 @@
from unittest import TestCase
import vcr
from tiingo.restclient import (RestClient, RestClientError)
# Tests of basic REST API functionality
@@ -25,12 +28,14 @@ class TestRestClient(TestCase):
# Test valid page
def test_valid_url(self):
response = self._client._request('GET', "")
with vcr.use_cassette('tests/fixtures/valid_url.yaml'):
response = self._client._request('GET', "")
self.assertEqual(response.status_code, 200)
# Test 404 error
def test_invalid_url(self):
with self.assertRaisesRegexp(RestClientError, "404"):
with self.assertRaisesRegexp(RestClientError, "404"),\
vcr.use_cassette('tests/fixtures/invalid_url.yaml'):
# Should return 404 error
self._client._request('GET', "bing_is_great")
@@ -55,11 +60,13 @@ class TestRestClientWithSession(TestCase):
# Test valid page
def test_valid_url(self):
response = self._client._request('GET', "")
with vcr.use_cassette('tests/fixtures/valid_url.yaml'):
response = self._client._request('GET', "")
self.assertEqual(response.status_code, 200)
# Test 404 error
def test_invalid_url(self):
with self.assertRaisesRegexp(RestClientError, "404"):
with self.assertRaisesRegexp(RestClientError, "404"),\
vcr.use_cassette('tests/fixtures/invalid_url.yaml'):
# Should return 404 error
self._client._request('GET', "bing_is_great")