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')