serverscanning/capnproto-java-rpc
16
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

TestCase.java

Browse source
This commit is contained in:
David Renshaw 2014-06-20 15:13:36 -04:00
parent fd1d82bd7c
commit ec1142b8a1
2 changed files with 21 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
benchmark/src/main/java/org/capnproto/benchmark/CarSales.java
View file

@ -5,7 +5,8 @@ import org.capnproto.StructList;
import org.capnproto.Text; import org.capnproto.Text;
import org.capnproto.benchmark.CarSalesSchema.*; import org.capnproto.benchmark.CarSalesSchema.*;
public class CarSales { public class CarSales
implements TestCase<ParkingLot.Builder, ParkingLot.Reader, TotalValue.Builder, TotalValue.Reader, Long> {
static final long carValue(Car.Reader car) { static final long carValue(Car.Reader car) {
long result = 0; long result = 0;
@ -88,7 +89,7 @@ public class CarSales {
} }
public static long setupRequest(Common.FastRand rng, ParkingLot.Builder request) { public Long setupRequest(Common.FastRand rng, ParkingLot.Builder request) {
long result = 0; long result = 0;
StructList.Builder<Car.Builder> cars = request.initCars(rng.nextLessThan(200)); StructList.Builder<Car.Builder> cars = request.initCars(rng.nextLessThan(200));
for (int i = 0; i < cars.size(); ++i) { for (int i = 0; i < cars.size(); ++i) {
@ -100,7 +101,7 @@ public class CarSales {
} }
public static void handleRequest(ParkingLot.Reader request, TotalValue.Builder response) { public void handleRequest(ParkingLot.Reader request, TotalValue.Builder response) {
long result = 0; long result = 0;
StructList.Reader<Car.Reader> cars = request.getCars(); StructList.Reader<Car.Reader> cars = request.getCars();
for (int i =0; i < cars.size(); ++i) { for (int i =0; i < cars.size(); ++i) {
@ -109,7 +110,7 @@ public class CarSales {
response.setAmount(result); response.setAmount(result);
} }
public static boolean checkResponse(TotalValue.Reader response, long expected) { public boolean checkResponse(TotalValue.Reader response, Long expected) {
return response.getAmount() == expected; return response.getAmount() == expected;
} }
@ -117,14 +118,17 @@ public class CarSales {
public static void main(String[] args) { public static void main(String[] args) {
Common.FastRand rng = new Common.FastRand(); Common.FastRand rng = new Common.FastRand();
TestCase<ParkingLot.Builder, ParkingLot.Reader,
TotalValue.Builder, TotalValue.Reader, Long> testCase = new CarSales();
for (int i = 0; i < 50000; ++i) { for (int i = 0; i < 50000; ++i) {
MessageBuilder requestMessage = new MessageBuilder(); MessageBuilder requestMessage = new MessageBuilder();
MessageBuilder responseMessage = new MessageBuilder(); MessageBuilder responseMessage = new MessageBuilder();
ParkingLot.Builder request = requestMessage.initRoot(ParkingLot.Builder.factory); ParkingLot.Builder request = requestMessage.initRoot(ParkingLot.Builder.factory);
long expected = setupRequest(rng, request); long expected = testCase.setupRequest(rng, request);
TotalValue.Builder response = responseMessage.initRoot(TotalValue.Builder.factory); TotalValue.Builder response = responseMessage.initRoot(TotalValue.Builder.factory);
handleRequest(request.asReader(), response); testCase.handleRequest(request.asReader(), response);
if (!checkResponse(response.asReader(), expected)) { if (!testCase.checkResponse(response.asReader(), expected)) {
System.out.println("mismatch!"); System.out.println("mismatch!");
} }
} }

10
benchmark/src/main/java/org/capnproto/benchmark/TestCase.java Normal file
View file

@ -0,0 +1,10 @@
package org.capnproto.benchmark;
import org.capnproto.FromStructReader;
import org.capnproto.FromStructBuilder;
public interface TestCase<RequestBuilder, RequestReader, ResponseBuilder, ResponseReader, Expectation> {
public Expectation setupRequest(Common.FastRand rng, RequestBuilder request);
public void handleRequest(RequestReader request, ResponseBuilder response);
public boolean checkResponse(ResponseReader response, Expectation expected);
}