Skip to content

Commit a1b0798

Browse files
committed
Added awaitInitialization call in all spark unit tests
1 parent 2a5c8e9 commit a1b0798

File tree

3 files changed

+5
-2
lines changed

3 files changed

+5
-2
lines changed

aws-serverless-java-container-spark/src/test/java/com/amazonaws/serverless/proxy/spark/HelloWorldSparkTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ public static void initializeServer() {
3737
handler = SparkLambdaContainerHandler.getAwsProxyHandler();
3838

3939
configureRoutes();
40-
40+
Spark.awaitInitialization();
4141
} catch (RuntimeException | ContainerInitializationException e) {
4242
e.printStackTrace();
4343
fail();

aws-serverless-java-container-spark/src/test/java/com/amazonaws/serverless/proxy/spark/InitExceptionHandlerTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ public void initException_mockException_expectHandlerToRun() {
4444
serverFactory);
4545

4646
configureRoutes();
47-
47+
Spark.awaitInitialization();
4848
} catch (Exception e) {
4949
e.printStackTrace();
5050
fail("Error while mocking server");

aws-serverless-java-container-spark/src/test/java/com/amazonaws/serverless/proxy/spark/SparkLambdaContainerHandlerTest.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,8 @@ public void filters_onStartupMethod_executeFilters() {
5050

5151
configureRoutes();
5252

53+
Spark.awaitInitialization();
54+
5355
AwsProxyRequest req = new AwsProxyRequestBuilder().method("GET").path("/header-filter").build();
5456
AwsProxyResponse response = handler.proxy(req, new MockLambdaContext());
5557

@@ -84,6 +86,7 @@ public void filters_unauthenticatedFilter_stopRequestProcessing() {
8486
});
8587

8688
configureRoutes();
89+
Spark.awaitInitialization();
8790

8891
// first we test without the custom header, we expect request processing to complete
8992
// successfully

0 commit comments

Comments
 (0)