raboof/pekko
1
0
Fork 0
Code Issues Pull requests Projects Releases 6 Packages Wiki Activity Actions 10

+htc #19237 entity#withSizeLimit for Java + add withoutSizeLimit

Browse source
This commit is contained in:
Konrad Malawski 2016-01-05 02:43:59 +01:00
parent cdc55d71ff
commit 544a4deed0
3 changed files with 152 additions and 59 deletions
Download patch file Download diff file Expand all files Collapse all files

93
akka-http-tests/src/test/java/akka/http/javadsl/server/directives/RouteDirectivesTest.java
View file

@ -4,26 +4,95 @@
package akka.http.javadsl.server.directives;
import akka.dispatch.Futures;
import akka.dispatch.Mapper;
import akka.http.javadsl.model.HttpRequest;
import akka.http.javadsl.model.RequestEntity;
import akka.http.javadsl.model.StatusCodes;
import akka.http.javadsl.model.Uri;
import akka.http.javadsl.model.headers.Location;
import akka.http.javadsl.server.RequestContext;
import akka.http.javadsl.server.RouteResult;
import akka.http.javadsl.testkit.JUnitRouteTest;
import akka.http.javadsl.testkit.TestRoute;
import akka.japi.function.Function;
import akka.stream.javadsl.Sink;
import akka.util.ByteString;
import org.junit.Test;
import scala.concurrent.ExecutionContext$;
import scala.concurrent.forkjoin.ForkJoinPool;
public class RouteDirectivesTest extends JUnitRouteTest {
@Test
public void testRedirection() {
Uri targetUri = Uri.create("http://example.com");
TestRoute route =
testRoute(
redirect(targetUri, StatusCodes.FOUND)
);
@Test
public void testRedirection() {
Uri targetUri = Uri.create("http://example.com");
TestRoute route =
testRoute(
redirect(targetUri, StatusCodes.FOUND)
);
route
.run(HttpRequest.create())
.assertStatusCode(302)
.assertHeaderExists(Location.create(targetUri));
}
route
.run(HttpRequest.create())
.assertStatusCode(302)
.assertHeaderExists(Location.create(targetUri));
}
@Test
public void testEntitySizeLimit() {
TestRoute route =
testRoute(
path("no-limit")
.route(
handleWith(new Function<RequestContext, RouteResult>() {
@Override
public RouteResult apply(final RequestContext ctx) throws Exception {
final RequestEntity entity = ctx.request().entity();
return ctx.completeWith(
entity
.withoutSizeLimit()
.getDataBytes()
.runWith(Sink.<ByteString>head(), ctx.materializer())
.map(new Mapper<ByteString, RouteResult>() {
@Override
public RouteResult apply(ByteString s) {
return ctx.complete(s.utf8String());
}
}, ctx.executionContext()));
}
})),
path("limit-5")
.route(
handleWith(new Function<RequestContext, RouteResult>() {
@Override
public RouteResult apply(final RequestContext ctx) throws Exception {
final RequestEntity entity = ctx.request().entity();
return ctx.completeWith(
entity
.withSizeLimit(5)
.getDataBytes()
.runWith(Sink.<ByteString>head(), ctx.materializer())
.map(new Mapper<ByteString, RouteResult>() {
@Override
public RouteResult apply(ByteString s) {
return ctx.complete(s.utf8String());
}
}, ctx.executionContext()));
}
}))
);
route
.run(HttpRequest.create("/no-limit").withEntity("1234567890"))
.assertStatusCode(200)
.assertEntity("1234567890");
route
.run(HttpRequest.create("/limit-5").withEntity("12345"))
.assertStatusCode(200)
.assertEntity("12345");
route
.run(HttpRequest.create("/limit-5").withEntity("1234567890"))
.assertStatusCode(500)
.assertEntity("There was an internal server error.");
}
}