Skip to content

Commit

Permalink
HTTP-42-BatchRequest - add support for batch request processing in HT…
Browse files Browse the repository at this point in the history
…TP sink #5 tests

Signed-off-by: Krzysztof Chmielewski <[email protected]>
  • Loading branch information
kristoffSC committed Jun 22, 2023
1 parent 627b54a commit 148d155
Showing 1 changed file with 19 additions and 10 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -2,13 +2,15 @@

import java.net.http.HttpClient;
import java.util.Properties;

import java.util.stream.Stream;

import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.Arguments;
import org.junit.jupiter.params.provider.MethodSource;
import org.mockito.Mock;
import org.mockito.MockedStatic;
import org.mockito.junit.jupiter.MockitoExtension;
Expand Down Expand Up @@ -56,22 +58,30 @@ public void setUp() {
when(httpClientBuilder.executor(any())).thenReturn(httpClientBuilder);
}

@Test
public void shouldBuildClientWithoutHeaders() {
private static Stream<Arguments> provideSubmitterFactory() {
return Stream.of(
Arguments.of(new PerRequestRequestSubmitterFactory()),
Arguments.of(new BatchRequestSubmitterFactory(50))
);
}

@ParameterizedTest
@MethodSource("provideSubmitterFactory")
public void shouldBuildClientWithoutHeaders(RequestSubmitterFactory requestSubmitterFactory) {

JavaNetSinkHttpClient client =
new JavaNetSinkHttpClient(
new Properties(),
postRequestCallback,
this.headerPreprocessor,
//TODO HTTP-42 add test for PerRequest submitter
new BatchRequestSubmitterFactory(50)
requestSubmitterFactory
);
assertThat(client.getHeadersAndValues()).isEmpty();
}

@Test
public void shouldBuildClientWithHeaders() {
@ParameterizedTest
@MethodSource("provideSubmitterFactory")
public void shouldBuildClientWithHeaders(RequestSubmitterFactory requestSubmitterFactory) {

// GIVEN
Properties properties = new Properties();
Expand All @@ -95,8 +105,7 @@ public void shouldBuildClientWithHeaders() {
properties,
postRequestCallback,
headerPreprocessor,
//TODO HTTP-42 add test for PerRequest submitter
new BatchRequestSubmitterFactory(50)
requestSubmitterFactory
);
String[] headersAndValues = client.getHeadersAndValues();
assertThat(headersAndValues).hasSize(6);
Expand Down

0 comments on commit 148d155

Please sign in to comment.