/*
* Copyright 2014 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.springframework.social.security.test;
import org.mockito.ArgumentMatcher;
import org.mockito.Matchers;
import org.springframework.social.oauth1.AuthorizedRequestToken;
import org.springframework.social.oauth1.OAuthToken;
import org.springframework.social.oauth2.OAuth2Parameters;
public class ArgMatchers {
private ArgMatchers() {
}
public static OAuth2Parameters oAuth2Parameters(final String redirectUri) {
return oAuth2Parameters(redirectUri, null, null);
}
public static OAuth2Parameters oAuth2Parameters(final String redirectUri, final String scope, final String state) {
return Matchers.argThat(new ArgumentMatcher<OAuth2Parameters>() {
public boolean matches(Object item) {
if (item instanceof OAuth2Parameters) {
OAuth2Parameters params = (OAuth2Parameters) item;
return eq(state, params.getState()) && eq(scope, params.getScope())
&& eq(redirectUri, params.getRedirectUri());
}
return false;
}
});
}
public static OAuthToken oAuthToken(OAuthToken token) {
return oAuthToken(token.getValue(), token.getSecret());
}
public static OAuthToken oAuthToken(final String value, final String secret) {
return Matchers.argThat(new ArgumentMatcher<OAuthToken>() {
public boolean matches(Object item) {
if (item instanceof OAuthToken) {
OAuthToken token = (OAuthToken) item;
return eq(value, token.getValue()) && eq(secret, token.getSecret());
}
return false;
}
});
}
public static AuthorizedRequestToken authorizedRequestToken(OAuthToken token, String verifier) {
return authorizedRequestToken(token.getValue(), token.getSecret(), verifier);
}
public static AuthorizedRequestToken authorizedRequestToken(final String value, final String secret,
final String verifier) {
return Matchers.argThat(new ArgumentMatcher<AuthorizedRequestToken>() {
public boolean matches(Object item) {
if (item instanceof AuthorizedRequestToken) {
AuthorizedRequestToken token = (AuthorizedRequestToken) item;
return eq(value, token.getValue()) && eq(secret, token.getSecret())
&& eq(verifier, token.getVerifier());
}
return false;
}
});
}
private static boolean eq(Object a, Object b) {
return (a == null) ? b == null : a.equals(b);
}
}