GRAD2-3037 - adds test coverage

bcgov · Dec 14, 2024 · 7356b82 · 7356b82
1 parent a33902d
commit 7356b82
Show file tree

Hide file tree

Showing 2 changed files with 170 additions and 0 deletions.
diff --git a/api/src/test/java/ca/bc/gov/educ/api/grad/report/service/v2/RESTServiceGETTest.java b/api/src/test/java/ca/bc/gov/educ/api/grad/report/service/v2/RESTServiceGETTest.java
@@ -0,0 +1,86 @@
+package ca.bc.gov.educ.api.grad.report.service.v2;
+
+import ca.bc.gov.educ.api.grad.report.exception.ServiceException;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Qualifier;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.boot.test.mock.mockito.MockBean;
+import org.springframework.security.oauth2.client.registration.ClientRegistrationRepository;
+import org.springframework.security.oauth2.client.web.OAuth2AuthorizedClientRepository;
+import org.springframework.test.context.junit4.SpringRunner;
+import org.springframework.web.reactive.function.client.WebClient;
+import reactor.core.publisher.Mono;
+
+import java.util.function.Consumer;
+
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.when;
+
+@SpringBootTest
+@RunWith(SpringRunner.class)
+public class RESTServiceGETTest {
+
+    @Autowired
+    private RESTService restService;
+
+    @MockBean
+    private WebClient.RequestHeadersSpec requestHeadersMock;
+    @MockBean
+    private WebClient.RequestHeadersUriSpec requestHeadersUriMock;
+    @MockBean
+    private WebClient.RequestBodySpec requestBodyMock;
+    @MockBean
+    private WebClient.RequestBodyUriSpec requestBodyUriMock;
+    @MockBean
+    private WebClient.ResponseSpec responseMock;
+    @MockBean(name = "webClient")
+    WebClient webClient;
+
+    @MockBean
+    @Qualifier("graduationReportClient")
+    WebClient getGraduationReportClient;
+
+    @MockBean
+    private ClientRegistrationRepository clientRegistrationRepositoryMock;
+
+    @MockBean
+    private OAuth2AuthorizedClientRepository oAuth2AuthorizedClientRepositoryMock;
+
+    private static final String TEST_URL_200 = "https://httpstat.us/200";
+    private static final String TEST_URL_403 = "https://httpstat.us/403";
+    private static final String TEST_URL_503 = "https://httpstat.us/503";
+    private static final String OK_RESPONSE = "200 OK";
+
+    @Before
+    public void setUp(){
+        when(this.webClient.get()).thenReturn(this.requestHeadersUriMock);
+        when(this.getGraduationReportClient.get()).thenReturn(this.requestHeadersUriMock);
+        when(this.requestHeadersUriMock.uri(any(String.class))).thenReturn(this.requestHeadersMock);
+        when(this.requestHeadersMock.headers(any(Consumer.class))).thenReturn(this.requestHeadersMock);
+        when(this.requestHeadersMock.retrieve()).thenReturn(this.responseMock);
+        when(this.responseMock.onStatus(any(), any())).thenReturn(this.responseMock);
+    }
+
+    @Test
+    public void testGetOverride_GivenProperData_Expect200Response(){
+        when(this.responseMock.bodyToMono(String.class)).thenReturn(Mono.just(OK_RESPONSE));
+        String response = this.restService.get(TEST_URL_200, String.class);
+        Assert.assertEquals(OK_RESPONSE, response);
+    }
+
+    @Test(expected = ServiceException.class)
+    public void testGetOverride_Given5xxErrorFromService_ExpectServiceError(){
+        when(this.responseMock.bodyToMono(ServiceException.class)).thenReturn(Mono.just(new ServiceException("Error", 500)));
+        this.restService.get(TEST_URL_503, String.class);
+    }
+
+    @Test(expected = ServiceException.class)
+    public void testGetOverride_Given4xxErrorFromService_ExpectServiceError(){
+        when(this.responseMock.bodyToMono(ServiceException.class)).thenReturn(Mono.just(new ServiceException("Error", 500)));
+        this.restService.get(TEST_URL_403, String.class);
+    }
+}
diff --git a/api/src/test/java/ca/bc/gov/educ/api/grad/report/service/v2/RESTServicePOSTTest.java b/api/src/test/java/ca/bc/gov/educ/api/grad/report/service/v2/RESTServicePOSTTest.java
@@ -0,0 +1,84 @@
+package ca.bc.gov.educ.api.grad.report.service.v2;
+
+import ca.bc.gov.educ.api.grad.report.exception.ServiceException;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.junit.runner.RunWith;
+import org.mockito.junit.jupiter.MockitoExtension;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Qualifier;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.boot.test.mock.mockito.MockBean;
+import org.springframework.security.oauth2.client.registration.ClientRegistrationRepository;
+import org.springframework.security.oauth2.client.web.OAuth2AuthorizedClientRepository;
+import org.springframework.test.context.junit4.SpringRunner;
+import org.springframework.web.reactive.function.BodyInserter;
+import org.springframework.web.reactive.function.client.WebClient;
+import reactor.core.publisher.Mono;
+
+import java.util.function.Consumer;
+
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.when;
+
+@SpringBootTest
+@RunWith(SpringRunner.class)
+@ExtendWith(MockitoExtension.class)
+public class RESTServicePOSTTest {
+
+    @Autowired
+    private RESTService restService;
+
+    @MockBean
+    private WebClient.RequestHeadersSpec requestHeadersMock;
+    @MockBean
+    private WebClient.RequestBodySpec requestBodyMock;
+    @MockBean
+    private WebClient.RequestBodyUriSpec requestBodyUriMock;
+    @MockBean
+    private WebClient.ResponseSpec responseMock;
+    @MockBean(name = "webClient")
+    WebClient webClient;
+
+    @MockBean
+    @Qualifier("graduationReportClient")
+    WebClient getGraduationReportClient;
+
+    @MockBean
+    ClientRegistrationRepository clientRegistrationRepository;
+
+    @MockBean
+    OAuth2AuthorizedClientRepository oAuth2AuthorizedClientRepository;
+
+    private static final byte[] TEST_BYTES = "The rain in Spain stays mainly on the plain.".getBytes();
+    private static final String TEST_BODY = "{test:test}";
+    private static final String TEST_URL = "https://fake.url.com";
+
+    @Before
+    public void setUp(){
+        when(this.webClient.post()).thenReturn(this.requestBodyUriMock);
+        when(this.getGraduationReportClient.post()).thenReturn(this.requestBodyUriMock);
+        when(this.requestBodyUriMock.uri(any(String.class))).thenReturn(this.requestBodyUriMock);
+        when(this.requestBodyUriMock.headers(any(Consumer.class))).thenReturn(this.requestBodyMock);
+        when(this.requestBodyMock.contentType(any())).thenReturn(this.requestBodyMock);
+        when(this.requestBodyMock.body(any(BodyInserter.class))).thenReturn(this.requestHeadersMock);
+        when(this.requestHeadersMock.retrieve()).thenReturn(this.responseMock);
+        when(this.responseMock.bodyToMono(byte[].class)).thenReturn(Mono.just(TEST_BYTES));
+    }
+
+    @Test
+    public void testPostOverride_GivenProperData_Expect200Response(){
+        when(this.responseMock.onStatus(any(), any())).thenReturn(this.responseMock);
+        byte[] response = this.restService.post(TEST_URL, TEST_BODY, byte[].class);
+        Assert.assertArrayEquals(TEST_BYTES, response);
+    }
+
+    @Test(expected = ServiceException.class)
+    public void testPostOverride_Given4xxErrorFromService_ExpectServiceError() {
+        when(this.responseMock.onStatus(any(), any())).thenThrow(new ServiceException("Error", 500));
+        this.restService.post(TEST_URL, TEST_BODY, byte[].class);
+    }
+
+}