def setUp(self):
self.cli_emptysettings = api.ClientAPI({})
self.cli_withsettings = api.ClientAPI({
'token': 'abcd1234',
'username': '******',
'password': '******',
'url': 'https://mydomain.com'
})
self.temp_auth_file = tempfile.NamedTemporaryFile()
def test_clientapi_build_token_invalid_is_given(self,
mock_requests_get_func):
"""Test api.ClientAPI._build_token func when given token is not valid"""
# mock an invalid token (response 401)
mock_requests_get_func.return_value.status_code = 401
new_client = api.ClientAPI({})
new_client._build_token(token=AUTH_TOKEN_VALUE)
def test_clientapi_build_token_none_is_given(self, mock_requests_get_func):
"""Test api.ClientAPI._build_token function when none is given"""
# mock a valid token (response 200)
mock_requests_get_func.return_value.status_code = 200
new_client = api.ClientAPI({})
new_client._build_token()
assert_equal(new_client.token, AUTH_TOKEN_VALUE)
def test_clientapi_creation_tokenfile_error(self):
"""Test api.ClientAPI creation when auth token can't be created"""
mycli = api.ClientAPI({})
def test_clientapi_generate_token_conn_error(self, mock_requests_get_func):
"""Test api.ClientAPI._generate_token throws ConnectionError"""
mock_requests_get_func.side_effect = ConnectionError()
new_client = api.ClientAPI({})
new_client._generate_token()
def test_clientapi_generate_token_auth_fails(self, mock_requests_get_func):
"""Test api.ClientAPI._generate_token when authentication fails"""
mock_requests_get_func.return_value.status_code = 401
new_client = api.ClientAPI({})
new_client._generate_token()
def test_clientapi_generate_token_none_password(self):
"""Test api.ClientAPI._generate_token when password is none"""
new_client = api.ClientAPI({'password': None})
new_client._generate_token()
def test_clientapi_generate_token_none_username(self):
"""Test api.ClientAPI._generate_token when username is none"""
new_client = api.ClientAPI({'username': None})
new_client._generate_token()