def test_smoke(self):
with mock.patch.dict(os.environ, {}):
with mock.patch(
'ilabs.aclient.get_secret.configparser',
mock.Mock(ConfigParser=mock.Mock(return_value=mock.Mock(
get=mock.Mock(return_value=None))))):
user_key = get_secret().get('ilabs_user_key')
self.assertIsNone(user_key)
def __init__(self,
user_key=None,
user_agent=ILABS_USER_AGENT,
endpoint='https://ilabs-api.innodata.com'):
self._endpoint = endpoint
self._user_key = user_key or get_secret().get('ilabs_user_key')
if self._user_key is None:
raise RuntimeError('Could not find credentials')
self._user_agent = user_agent
def test_file(self):
config = '''
[default]
ilabs_user_key=foo
'''
os.environ.pop('ILABS_USER_KEY', None)
with mock.patch('ilabs.aclient.get_secret.configparser.open',
mock_fs({'/etc/ilabs.conf': config})):
with mock.patch.dict(os.environ, {}):
user_key = get_secret().get('ilabs_user_key')
self.assertEqual(user_key, 'foo')
def test_file_precedence(self):
config1 = '''
[default]
ilabs_user_key=bar
'''
config2 = '''
[default]
ilabs_user_key=foo
'''
os.environ.pop('ILABS_USER_KEY', None)
with mock.patch(
'ilabs.aclient.get_secret.configparser.open',
mock_fs({
'/etc/ilabs.conf':
config2,
os.path.expanduser(r'~/.config/ilabs/ilabs.conf'):
config1
})):
user_key = get_secret().get('ilabs_user_key')
self.assertEqual(user_key, 'bar')
def test_env(self):
with mock.patch.dict(os.environ, {'ILABS_USER_KEY': 'blah'}):
user_key = get_secret().get('ilabs_user_key')
self.assertEqual(user_key, 'blah')