package com.jetdrone.vertx.yoke.test.middleware;
import com.jetdrone.vertx.yoke.Middleware;
import com.jetdrone.vertx.yoke.Yoke;
import com.jetdrone.vertx.yoke.middleware.AuthHandler;
import com.jetdrone.vertx.yoke.middleware.YokeRequest;
import com.jetdrone.vertx.yoke.test.Response;
import com.jetdrone.vertx.yoke.test.YokeTester;
import org.jetbrains.annotations.NotNull;
import org.junit.Test;
import org.vertx.java.core.http.CaseInsensitiveMultiMap;
import org.vertx.java.core.Handler;
import org.vertx.java.core.MultiMap;
import org.vertx.java.core.json.JsonObject;
import org.vertx.testtools.TestVerticle;
import static org.vertx.testtools.VertxAssert.*;
public class BasicAuth extends TestVerticle {
@Test
public void testBasicAuth() {
final Yoke yoke = new Yoke(this);
yoke.use(new com.jetdrone.vertx.yoke.middleware.BasicAuth("Aladdin", "open sesame"));
yoke.use(new Middleware() {
@Override
public void handle(@NotNull YokeRequest request, @NotNull Handler<Object> next) {
request.response().end();
}
});
final YokeTester yokeAssert = new YokeTester(yoke);
// first time is forbidden
yokeAssert.request("GET", "/", new Handler<Response>() {
@Override
public void handle(Response resp) {
assertEquals(401, resp.getStatusCode());
assertNotNull(resp.headers.get("www-authenticate"));
// second time send the authorization header
MultiMap headers = new CaseInsensitiveMultiMap();
headers.add("authorization", "Basic QWxhZGRpbjpvcGVuIHNlc2FtZQ==");
yokeAssert.request("GET", "/", headers, new Handler<Response>() {
@Override
public void handle(Response resp) {
assertEquals(200, resp.getStatusCode());
testComplete();
}
});
}
});
}
@Test
public void testEmptyPassword() {
final Yoke yoke = new Yoke(this);
yoke.use(new com.jetdrone.vertx.yoke.middleware.BasicAuth(new AuthHandler() {
@Override
public void handle(String username, String password, Handler<JsonObject> result) {
boolean success = username.equals("Aladdin") && password == null;
if (success) {
result.handle(new JsonObject().putString("username", username));
} else {
result.handle(null);
}
}
}));
yoke.use(new Middleware() {
@Override
public void handle(@NotNull YokeRequest request, @NotNull Handler<Object> next) {
request.response().end();
}
});
final YokeTester yokeAssert = new YokeTester(yoke);
// first time is forbidden
yokeAssert.request("GET", "/", new Handler<Response>() {
@Override
public void handle(Response resp) {
assertEquals(401, resp.getStatusCode());
assertNotNull(resp.headers.get("www-authenticate"));
// second time send the authorization header
MultiMap headers = new CaseInsensitiveMultiMap();
headers.add("authorization", "Basic QWxhZGRpbjo=");
yokeAssert.request("GET", "/", headers, new Handler<Response>() {
@Override
public void handle(Response resp) {
assertEquals(200, resp.getStatusCode());
testComplete();
}
});
}
});
}
}