Add tests for MyWaifuClient methods

pull/16/head
Hammy 4 years ago
parent b9ccaf159f
commit eba96c29b8

@ -0,0 +1,152 @@
package me.goudham;
import me.goudham.domain.waifu.FilteredWaifu;
import me.goudham.domain.waifu.Waifu;
import me.goudham.exception.APIMapperException;
import me.goudham.exception.APIResponseException;
import me.goudham.util.TestEntity;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import javax.net.ssl.SSLSession;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.URI;
import java.net.http.HttpClient;
import java.net.http.HttpHeaders;
import java.net.http.HttpRequest;
import java.net.http.HttpResponse;
import java.nio.charset.StandardCharsets;
import java.time.Duration;
import java.util.List;
import java.util.Objects;
import java.util.Optional;
import java.util.stream.Collectors;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.is;
import static org.mockito.Mockito.*;
class MyWaifuClientTest {
@Mock
private HttpClient httpClient;
private APIWrapper apiWrapper;
private final String apiKey = "ValidAPIKey";
private MyWaifuClient sut;
@BeforeEach
void setUp() {
MockitoAnnotations.openMocks(this);
sut = MyWaifuClient.createDefault(apiKey);
apiWrapper = new APIWrapper(apiKey, httpClient);
sut.setAPIWrapper(apiWrapper);
}
@Test
void successfullyGetWaifu() throws IOException, InterruptedException, APIMapperException, APIResponseException {
HttpRequest expectedHttpRequest = buildHttpGetRequest(apiKey, "waifu/1");
int expectedStatusCode = 200;
String expectedBody = getJsonAsString("getWaifu.json");
Waifu expectedWaifu = TestEntity.getExpectedWaifu();
HttpResponse<String> expectedHttpResponse = buildHttpResponse(expectedStatusCode, expectedBody);
doReturn(expectedHttpResponse).when(httpClient).send(expectedHttpRequest, HttpResponse.BodyHandlers.ofString());
Response<Waifu> actualWaifuResponse = sut.getWaifu(1);
assertThat(actualWaifuResponse.getStatusCode(), is(expectedStatusCode));
assertThat(actualWaifuResponse.getBody(), is(expectedBody));
assertThat(actualWaifuResponse.getModel(), is(expectedWaifu));
verify(httpClient, times(1)).send(expectedHttpRequest, HttpResponse.BodyHandlers.ofString());
verifyNoMoreInteractions(httpClient);
}
@Test
void successfullyGetBestWaifus() throws IOException, InterruptedException, APIMapperException, APIResponseException {
HttpRequest expectedHttpRequest = buildHttpGetRequest(apiKey, "airing/best");
int expectedStatusCode = 200;
String expectedBody = getJsonAsString("getBestWaifus.json");
List<FilteredWaifu> expectedBestWaifus = TestEntity.getBestWaifus();
HttpResponse<String> expectedHttpResponse = buildHttpResponse(expectedStatusCode, expectedBody);
doReturn(expectedHttpResponse).when(httpClient).send(expectedHttpRequest, HttpResponse.BodyHandlers.ofString());
Response<List<FilteredWaifu>> actualWaifuResponse = sut.getBestWaifus();
assertThat(actualWaifuResponse.getStatusCode(), is(expectedStatusCode));
assertThat(actualWaifuResponse.getBody(), is(expectedBody));
assertThat(actualWaifuResponse.getModel(), is(expectedBestWaifus));
verify(httpClient, times(1)).send(expectedHttpRequest, HttpResponse.BodyHandlers.ofString());
verifyNoMoreInteractions(httpClient);
}
private String getJsonAsString(String filename) {
InputStream resourceAsStream = this.getClass().getClassLoader().getResourceAsStream(filename);
return new BufferedReader(new InputStreamReader(Objects.requireNonNull(resourceAsStream), StandardCharsets.UTF_8))
.lines()
.collect(Collectors.joining("\n"));
}
private HttpRequest buildHttpGetRequest(String apiKey, String param) {
return HttpRequest.newBuilder()
.uri(URI.create("https://mywaifulist.moe/api/v1/" + param))
.timeout(Duration.ofSeconds(20))
.headers("Content-Type", "application/json", "apikey", apiKey)
.GET()
.build();
}
private HttpResponse<String> buildHttpResponse(int statusCode, String body) {
return new HttpResponse<>() {
@Override
public int statusCode() {
return statusCode;
}
@Override
public HttpRequest request() {
return null;
}
@Override
public Optional<HttpResponse<String>> previousResponse() {
return Optional.empty();
}
@Override
public HttpHeaders headers() {
return null;
}
@Override
public String body() {
return body;
}
@Override
public Optional<SSLSession> sslSession() {
return Optional.empty();
}
@Override
public URI uri() {
return null;
}
@Override
public HttpClient.Version version() {
return null;
}
};
}
}
Loading…
Cancel
Save