58 lines
2.1 KiB
Java
58 lines
2.1 KiB
Java
package org.owasp.webgoat;
|
|
|
|
import io.restassured.RestAssured;
|
|
import io.restassured.response.Response;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.concurrent.Callable;
|
|
import java.util.concurrent.ExecutionException;
|
|
import java.util.concurrent.ExecutorService;
|
|
import java.util.concurrent.Executors;
|
|
import java.util.stream.Collectors;
|
|
import java.util.stream.IntStream;
|
|
import org.assertj.core.api.Assertions;
|
|
import org.junit.jupiter.api.Test;
|
|
|
|
public class ProgressRaceConditionIntegrationTest extends IntegrationTest {
|
|
|
|
@Test
|
|
public void runTests() throws InterruptedException {
|
|
int NUMBER_OF_CALLS = 40;
|
|
int NUMBER_OF_PARALLEL_THREADS = 5;
|
|
startLesson("Challenge1");
|
|
|
|
Callable<Response> call =
|
|
() -> {
|
|
// System.out.println("thread "+Thread.currentThread().getName());
|
|
return RestAssured.given()
|
|
.when()
|
|
.relaxedHTTPSValidation()
|
|
.cookie("JSESSIONID", getWebGoatCookie())
|
|
.formParams(Map.of("flag", "test"))
|
|
.post(url("/challenge/flag/"));
|
|
};
|
|
ExecutorService executorService = Executors.newWorkStealingPool(NUMBER_OF_PARALLEL_THREADS);
|
|
List<? extends Callable<Response>> flagCalls =
|
|
IntStream.range(0, NUMBER_OF_CALLS).mapToObj(i -> call).collect(Collectors.toList());
|
|
var responses = executorService.invokeAll(flagCalls);
|
|
|
|
// A certain amount of parallel calls should fail as optimistic locking in DB is applied
|
|
long countStatusCode500 =
|
|
responses.stream()
|
|
.filter(
|
|
r -> {
|
|
try {
|
|
// System.err.println(r.get().getStatusCode());
|
|
return r.get().getStatusCode() != 200;
|
|
} catch (InterruptedException | ExecutionException e) {
|
|
// System.err.println(e);
|
|
throw new IllegalStateException(e);
|
|
}
|
|
})
|
|
.count();
|
|
System.err.println("counted status 500: " + countStatusCode500);
|
|
Assertions.assertThat(countStatusCode500)
|
|
.isLessThanOrEqualTo((NUMBER_OF_CALLS - (NUMBER_OF_CALLS / NUMBER_OF_PARALLEL_THREADS)));
|
|
}
|
|
}
|