From eba96c29b83ded600b290f28f4a18f369ce4b4f2 Mon Sep 17 00:00:00 2001
From: Hammy <sgoudham@gmail.com>
Date: Sat, 3 Jul 2021 06:37:21 +0100
Subject: [PATCH] Add tests for MyWaifuClient methods

---
 .../java/me/goudham/MyWaifuClientTest.java    | 152 ++++++++++++++++++
 1 file changed, 152 insertions(+)
 create mode 100644 src/test/java/me/goudham/MyWaifuClientTest.java

diff --git a/src/test/java/me/goudham/MyWaifuClientTest.java b/src/test/java/me/goudham/MyWaifuClientTest.java
new file mode 100644
index 0000000..c4e523f
--- /dev/null
+++ b/src/test/java/me/goudham/MyWaifuClientTest.java
@@ -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;
+            }
+        };
+    }
+}
+