aljaz/tiingo-python
1
1
Fork 0
mirror of https://github.com/hydrosquall/tiingo-python.git synced 2026-02-01 09:04:18 +01:00
Code Issues Packages Projects Releases Wiki Activity

use unittest.mock to remove TIINGO_API_KEY

Browse Source
This commit is contained in:
mohamedemad4
2021-07-07 13:16:52 +02:00
parent 0be8171d0c
commit b162391cef
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_wsclient.py
View File

@@ -1,5 +1,5 @@
import os
from unittest import TestCase
from unittest import TestCase,mock
from tiingo.wsclient import TiingoWebsocketClient
from tiingo.exceptions import MissingRequiredArgumentError
@@ -38,7 +38,7 @@ class TestRestClientWithSession(TestCase):
# test for missing API keys in config dict and in os env
def test_missing_api_key(self):
del os.environ['TIINGO_API_KEY']
with self.assertRaises(RuntimeError) as ex:
TiingoWebsocketClient(config={},endpoint='iex',on_msg_cb=self.cb)
self.assertTrue(type(ex.exception)==RuntimeError)
with mock.patch.dict(os.environ, {}, clear=True): #clear env vars including the TIINGO_API_KEY
with self.assertRaises(RuntimeError) as ex:
TiingoWebsocketClient(config={},endpoint='iex',on_msg_cb=self.cb)
self.assertTrue(type(ex.exception)==RuntimeError)