/*
* Copyright 2004-2014 SmartBear Software
*
* Licensed under the EUPL, Version 1.1 or - as soon as they will be approved by the European Commission - subsequent
* versions of the EUPL (the "Licence");
* You may not use this work except in compliance with the Licence.
* You may obtain a copy of the Licence at:
*
* http://ec.europa.eu/idabc/eupl
*
* Unless required by applicable law or agreed to in writing, software distributed under the Licence is
* distributed on an "AS IS" basis, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
* express or implied. See the Licence for the specific language governing permissions and limitations
* under the Licence.
*/
package com.eviware.soapui.impl.rest.actions.oauth;
import com.eviware.soapui.config.OAuth2ProfileConfig;
import com.eviware.soapui.impl.rest.OAuth2Profile;
import com.eviware.soapui.support.SoapUIException;
import com.eviware.soapui.utils.ModelItemFactory;
import org.apache.oltu.oauth2.common.exception.OAuthProblemException;
import org.apache.oltu.oauth2.common.exception.OAuthSystemException;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;
import java.net.MalformedURLException;
import java.net.URISyntaxException;
import static org.mockito.Matchers.any;
import static org.mockito.Mockito.doAnswer;
import static org.mockito.Mockito.mock;
/**
*
*/
public class OAuth2TestUtils {
public static String AUTHORIZATION_CODE = "some_code";
public static String ACCESS_TOKEN = "expected_access_token";
public static String REFRESH_TOKEN = "expected_refresh_token";
public static final String OAUTH_2_OOB_URN = "urn:ietf:wg:oauth:2.0:oob";
public static OAuth2Profile getOAuthProfileWithDefaultValues() throws SoapUIException {
OAuth2ProfileConfig configuration = OAuth2ProfileConfig.Factory.newInstance();
OAuth2Profile profile = new OAuth2Profile(ModelItemFactory.makeOAuth2ProfileContainer(), configuration);
profile.setName("OAuth 2 -Profile");
profile.setAuthorizationURI("http://localhost:8080/authorize");
profile.setAccessTokenURI("http://localhost:8080/accesstoken");
profile.setRedirectURI("http://localhost:8080/redirect");
profile.setClientID("ClientId");
profile.setClientSecret("ClientSecret");
profile.setScope("ReadOnly");
return profile;
}
public static OAuth2Profile getOAuth2ProfileWithOnlyAccessToken() throws SoapUIException {
OAuth2ProfileConfig configuration = OAuth2ProfileConfig.Factory.newInstance();
OAuth2Profile profileWithOnlyAccessToken = new OAuth2Profile(ModelItemFactory.makeOAuth2ProfileContainer(),
configuration);
profileWithOnlyAccessToken.setAccessToken(ACCESS_TOKEN);
return profileWithOnlyAccessToken;
}
public static OAuth2Profile getOAuthProfileWithRefreshToken() throws SoapUIException {
OAuth2Profile profile = getOAuthProfileWithDefaultValues();
profile.setRefreshToken("REFRESH#TOKEN");
return profile;
}
public static OAuth2TokenExtractor mockOAuth2TokenExtractor(final OAuth2Profile profile)
throws OAuthSystemException, MalformedURLException, URISyntaxException, OAuthProblemException {
OAuth2TokenExtractor oAuth2TokenExtractor = mock(OAuth2TokenExtractor.class);
doAnswer(new Answer<Object>() {
@Override
public Object answer(InvocationOnMock invocationOnMock) throws Throwable {
profile.setAccessToken(OAuth2TestUtils.ACCESS_TOKEN);
return profile;
}
}).when(oAuth2TokenExtractor).extractAccessToken(any(OAuth2Parameters.class));
doAnswer(new Answer() {
@Override
public Object answer(InvocationOnMock invocationOnMock) throws Throwable {
profile.setAccessToken(OAuth2TestUtils.ACCESS_TOKEN);
return profile;
}
}).when(oAuth2TokenExtractor).refreshAccessToken(any(OAuth2Parameters.class));
return oAuth2TokenExtractor;
}
public static OltuOAuth2ClientFacade getOltuOAuth2ClientFacadeWithMockedTokenExtractor(final OAuth2Profile profile) {
return new OltuOAuth2ClientFacade() {
@Override
protected OAuth2TokenExtractor getOAuth2TokenExtractor() {
try {
return mockOAuth2TokenExtractor(profile);
} catch (Exception e) {
e.printStackTrace();
return null;
}
}
};
}
}