Removed perf package from testsuite/integration
This commit is contained in:
Stian Thorgersen
parent
4aa102fcab
commit
a6af72cbbc
1 changed files with 0 additions and 272 deletions
|
|
@ -1,272 +0,0 @@
|
|||
/*
|
||||
* Copyright 2016 Red Hat, Inc. and/or its affiliates
|
||||
* and other contributors as indicated by the @author tags.
|
||||
*
|
||||
* 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.keycloak.testsuite.perf;
|
||||
|
||||
import org.apache.http.NameValuePair;
|
||||
import org.apache.http.client.CookieStore;
|
||||
import org.apache.http.client.utils.URLEncodedUtils;
|
||||
import org.apache.http.cookie.Cookie;
|
||||
import org.apache.http.impl.client.DefaultHttpClient;
|
||||
import org.jboss.resteasy.client.jaxrs.ResteasyClient;
|
||||
import org.jboss.resteasy.client.jaxrs.ResteasyClientBuilder;
|
||||
import org.jboss.resteasy.client.jaxrs.engines.ApacheHttpClient4Engine;
|
||||
import org.junit.Assert;
|
||||
import org.junit.ClassRule;
|
||||
import org.junit.Test;
|
||||
import org.keycloak.OAuth2Constants;
|
||||
import org.keycloak.adapters.HttpClientBuilder;
|
||||
import org.keycloak.protocol.oidc.OIDCLoginProtocolService;
|
||||
import org.keycloak.services.resources.LoginActionsService;
|
||||
import org.keycloak.testsuite.Constants;
|
||||
import org.keycloak.testsuite.OAuthClient;
|
||||
import org.keycloak.testsuite.OAuthClient.AccessTokenResponse;
|
||||
import org.keycloak.testsuite.rule.KeycloakRule;
|
||||
import org.keycloak.testsuite.rule.WebRule;
|
||||
import org.keycloak.util.BasicAuthHelper;
|
||||
import org.openqa.selenium.WebDriver;
|
||||
|
||||
import javax.ws.rs.client.Entity;
|
||||
import javax.ws.rs.core.Form;
|
||||
import javax.ws.rs.core.HttpHeaders;
|
||||
import javax.ws.rs.core.Response;
|
||||
import javax.ws.rs.core.UriBuilder;
|
||||
import java.net.URI;
|
||||
import java.util.Collections;
|
||||
import java.util.Date;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.concurrent.atomic.AtomicLong;
|
||||
import java.util.regex.Matcher;
|
||||
import java.util.regex.Pattern;
|
||||
|
||||
/**
|
||||
* @author <a href="mailto:sthorger@redhat.com">Stian Thorgersen</a>
|
||||
*/
|
||||
public class AccessTokenPerfTest {
|
||||
|
||||
@ClassRule
|
||||
public static KeycloakRule keycloakRule = new KeycloakRule();
|
||||
|
||||
public static class BrowserLogin implements Runnable
|
||||
{
|
||||
|
||||
private WebDriver driver;
|
||||
|
||||
public BrowserLogin() {
|
||||
driver = WebRule.createWebDriver();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void run() {
|
||||
driver.manage().deleteAllCookies();
|
||||
OAuthClient oauth = new OAuthClient(driver);
|
||||
oauth.doLogin("test-user@localhost", "password");
|
||||
String code = oauth.getCurrentQuery().get(OAuth2Constants.CODE);
|
||||
AccessTokenResponse response = oauth.doAccessTokenRequest(code, "password");
|
||||
Assert.assertEquals(200, response.getStatusCode());
|
||||
count.incrementAndGet();
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
public static AtomicLong count = new AtomicLong(0);
|
||||
|
||||
public static class JaxrsClientLogin implements Runnable
|
||||
{
|
||||
ResteasyClient client;
|
||||
|
||||
private String baseUrl = Constants.AUTH_SERVER_ROOT;
|
||||
|
||||
private String realm = "test";
|
||||
|
||||
private String responseType = OAuth2Constants.CODE;
|
||||
|
||||
private String grantType = "authorization_code";
|
||||
|
||||
private String clientId = "test-app";
|
||||
|
||||
private String redirectUri = "http://localhost:8081/app/auth";
|
||||
|
||||
|
||||
public JaxrsClientLogin() {
|
||||
DefaultHttpClient httpClient = (DefaultHttpClient) new HttpClientBuilder().build();
|
||||
httpClient.setCookieStore(new CookieStore() {
|
||||
@Override
|
||||
public void addCookie(Cookie cookie) {
|
||||
//To change body of implemented methods use File | Settings | File Templates.
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<Cookie> getCookies() {
|
||||
return Collections.emptyList();
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean clearExpired(Date date) {
|
||||
return false; //To change body of implemented methods use File | Settings | File Templates.
|
||||
}
|
||||
|
||||
@Override
|
||||
public void clear() {
|
||||
//To change body of implemented methods use File | Settings | File Templates.
|
||||
}
|
||||
});
|
||||
ApacheHttpClient4Engine engine = new ApacheHttpClient4Engine(httpClient);
|
||||
this.client = new ResteasyClientBuilder().httpEngine(engine).build();
|
||||
}
|
||||
|
||||
public String getLoginFormUrl(String state) {
|
||||
UriBuilder b = OIDCLoginProtocolService.authUrl(UriBuilder.fromUri(baseUrl));
|
||||
if (responseType != null) {
|
||||
b.queryParam(OAuth2Constants.RESPONSE_TYPE, responseType);
|
||||
}
|
||||
if (clientId != null) {
|
||||
b.queryParam(OAuth2Constants.CLIENT_ID, clientId);
|
||||
}
|
||||
if (redirectUri != null) {
|
||||
b.queryParam(OAuth2Constants.REDIRECT_URI, redirectUri);
|
||||
}
|
||||
if (state != null) {
|
||||
b.queryParam(OAuth2Constants.STATE, state);
|
||||
}
|
||||
return b.build(realm).toString();
|
||||
}
|
||||
|
||||
static Pattern actionParser = Pattern.compile("action=\"([^\"]+)\"");
|
||||
|
||||
public void run() {
|
||||
//this.client = new ResteasyClientBuilder().build();
|
||||
String state = "42";
|
||||
String loginFormUrl = getLoginFormUrl(state);
|
||||
String html = client.target(loginFormUrl).request().get(String.class);
|
||||
Matcher matcher = actionParser.matcher(html);
|
||||
matcher.find();
|
||||
String actionUrl = matcher.group(1);
|
||||
if (!actionUrl.startsWith("http")) {
|
||||
actionUrl = UriBuilder.fromUri(actionUrl).scheme("http").host("localhost").port(8081).build().toString();
|
||||
}
|
||||
Form form = new Form();
|
||||
form.param("username", "test-user@localhost");
|
||||
form.param("password", "password");
|
||||
Response response = client.target(actionUrl).request().post(Entity.form(form));
|
||||
URI uri = null;
|
||||
Assert.assertEquals(302, response.getStatus());
|
||||
uri = response.getLocation();
|
||||
if (response.getStatus() == 302) {
|
||||
while (uri.toString().contains("login-actions/")) {
|
||||
response = client.target(uri).request().get();
|
||||
Assert.assertEquals(302, response.getStatus());
|
||||
uri = response.getLocation();
|
||||
}
|
||||
}
|
||||
|
||||
for (String header : response.getHeaders().keySet()) {
|
||||
for (Object value : response.getHeaders().get(header)) {
|
||||
System.out.println(header + ": " + value);
|
||||
}
|
||||
}
|
||||
response.close();
|
||||
|
||||
Assert.assertNotNull(uri);
|
||||
String code = getCode(uri);
|
||||
Assert.assertNotNull(code);
|
||||
|
||||
form = new Form();
|
||||
form.param(OAuth2Constants.GRANT_TYPE, grantType)
|
||||
.param(OAuth2Constants.CODE, code)
|
||||
.param(OAuth2Constants.REDIRECT_URI, redirectUri);
|
||||
|
||||
String authorization = BasicAuthHelper.createHeader(clientId, "password");
|
||||
|
||||
String res = client.target(OIDCLoginProtocolService.tokenUrl(UriBuilder.fromUri(baseUrl)).build(realm)).request()
|
||||
.header(HttpHeaders.AUTHORIZATION, authorization)
|
||||
.post(Entity.form(form), String.class);
|
||||
count.incrementAndGet();
|
||||
//client.close();
|
||||
}
|
||||
|
||||
public String getCode(URI uri) {
|
||||
Map<String, String> m = new HashMap<String, String>();
|
||||
List<NameValuePair> pairs = URLEncodedUtils.parse(uri, "UTF-8");
|
||||
for (NameValuePair p : pairs) {
|
||||
if (p.getName().equals("code")) return p.getValue();
|
||||
m.put(p.getName(), p.getValue());
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
|
||||
public void close()
|
||||
{
|
||||
client.close();
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
public void perfJaxrsClientLogin()
|
||||
{
|
||||
long ITERATIONS = 3;
|
||||
JaxrsClientLogin login = new JaxrsClientLogin();
|
||||
long start = System.currentTimeMillis();
|
||||
for (int i = 0; i < ITERATIONS; i++) {
|
||||
//System.out.println("*************************");
|
||||
login.run();
|
||||
}
|
||||
long end = System.currentTimeMillis() - start;
|
||||
System.out.println("took: " + end);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void perfBrowserLogin()
|
||||
{
|
||||
long ITERATIONS = 3;
|
||||
long start = System.currentTimeMillis();
|
||||
BrowserLogin login = new BrowserLogin();
|
||||
for (int i = 0; i < ITERATIONS; i++) {
|
||||
//System.out.println("----------------------------------");
|
||||
login.run();
|
||||
}
|
||||
long end = System.currentTimeMillis() - start;
|
||||
System.out.println("took: " + end);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void multiThread() throws Exception {
|
||||
int num_threads = 20;
|
||||
Thread[] threads = new Thread[num_threads];
|
||||
for (int i = 0; i < num_threads; i++) {
|
||||
threads[i] = new Thread(new Runnable() {
|
||||
@Override
|
||||
public void run() {
|
||||
perfJaxrsClientLogin();
|
||||
}
|
||||
});
|
||||
}
|
||||
long start = System.currentTimeMillis();
|
||||
for (int i = 0; i < num_threads; i++) {
|
||||
threads[i].start();
|
||||
}
|
||||
for (int i = 0; i < num_threads; i++) {
|
||||
threads[i].join();
|
||||
}
|
||||
long end = System.currentTimeMillis() - start;
|
||||
System.out.println(count.toString() + " took: " + end);
|
||||
System.out.println(count.floatValue() / ((float)end) * 1000+ " logins/s");
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in a new issue