def test_request_with_bad_auth_acl(mock, req_mock, auth_mock): mock.url = 'http://domain.com' mock.headers = {'www-authenticate': 'acsjwt'} mock.status_code = 401 auth_mock.return_value = http.DCOSAcsAuth("token") req_mock.return_value = mock with pytest.raises(DCOSException) as e: http._request_with_auth(mock, "method", mock.url) assert e.exconly().split(':')[1].strip() == "Authentication failed"
def test_request_with_bad_auth_basic(mock, req_mock, auth_mock): mock.url = 'http://domain.com' mock.headers = {'www-authenticate': 'Basic realm="Restricted"'} mock.status_code = 401 auth_mock.return_value = HTTPBasicAuth("username", "password") req_mock.return_value = mock with pytest.raises(DCOSException) as e: http._request_with_auth(mock, "method", mock.url) assert e.exconly().split(':')[1].strip() == "Authentication failed"
def test_request_with_bad_oauth(mock, req_mock, auth_mock): mock.url = 'http://domain.com' mock.headers = {'www-authenticate': 'oauthjwt'} mock.status_code = 401 auth_mock.return_value = http.DCOSAcsAuth("token") req_mock.return_value = mock with pytest.raises(DCOSException) as e: http._request_with_auth(mock, "method", mock.url) msg = "Your core.dcos_acs_token is invalid. Please run: `dcos auth login`" assert e.exconly().split(':', 1)[1].strip() == msg
def test_request_with_auth_acl(mock, req_mock, auth_mock): mock.url = 'http://domain.com' mock.headers = {'www-authenticate': 'acsjwt'} mock.status_code = 401 auth = http.DCOSAcsAuth("token") auth_mock.return_value = auth mock2 = copy.deepcopy(mock) mock2.status_code = 200 req_mock.return_value = mock2 response = http._request_with_auth(mock, "method", mock.url) assert response.status_code == 200
def test_request_with_auth_basic(mock, req_mock, auth_mock): mock.url = 'http://domain.com' mock.headers = {'www-authenticate': 'Basic realm="Restricted"'} mock.status_code = 401 auth = HTTPBasicAuth("username", "password") auth_mock.return_value = auth mock2 = copy.deepcopy(mock) mock2.status_code = 200 req_mock.return_value = mock2 response = http._request_with_auth(mock, "method", mock.url) assert response.status_code == 200