Skip to content

Commit

Permalink
fix: update AreaUtil tests
Browse files Browse the repository at this point in the history
  • Loading branch information
@BulatRuslanovich
BulatRuslanovich committed Oct 15, 2024
1 parent 7e1f818 commit 8d5aade
Showing 1 changed file with 138 additions and 139 deletions.
277 changes: 138 additions & 139 deletions node/src/test/java/com/bipbup/utils/AreaUtilTest.java
View file
Original file line number Diff line number Diff line change
@@ -1,161 +1,160 @@
package com.bipbup.utils;

import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import java.io.IOException;
import java.net.http.HttpClient;
import java.net.http.HttpRequest;
import java.net.http.HttpResponse;
import java.util.List;
import java.util.Map;
import lombok.SneakyThrows;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertNull;
import static org.junit.jupiter.api.Assertions.assertThrows;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import static org.mockito.ArgumentMatchers.anyString;
import static java.net.http.HttpResponse.BodyHandlers.ofString;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;


import static java.net.http.HttpResponse.BodyHandlers.ofString;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertNull;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.mockito.Mockito.any;
import static org.mockito.Mockito.eq;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import org.mockito.MockitoAnnotations;

class AreaUtilTest {

@Mock
private ObjectMapper objectMapper;

@Mock
private HttpClient client;

@InjectMocks
private AreaUtil areaUtil;

@BeforeEach
void setUp() {
MockitoAnnotations.openMocks(this);
}

private HttpResponse<String> mockHttpResponse(int statusCode, String body)
throws IOException, InterruptedException {
HttpResponse<String> response = mock(HttpResponse.class);
when(response.statusCode()).thenReturn(statusCode);
when(response.body()).thenReturn(body);
when(client.send(any(HttpRequest.class), eq(ofString()))).thenReturn(response);
return response;
}

@Test
@DisplayName("Should return area ID for valid area name")
void testGetAreaIdFromApi_ValidAreaName() throws Exception {
// Arrange
String areaName = "Moscow";
String jsonResponse = "[{\"id\":\"1\", \"name\":\"Moscow\"}]";
mockHttpResponse(200, jsonResponse);

when(objectMapper.readValue(eq(jsonResponse), any(TypeReference.class))).thenReturn(
List.of(Map.of("id", "1", "name", "Moscow")));
// Act
Integer areaId = areaUtil.getAreaIdFromApi(areaName);

// Assert
assertEquals(1, areaId);
}

@Test
@DisplayName("Should return null for invalid area name")
void testGetAreaIdFromApi_InvalidAreaName() throws Exception {
// Arrange
String areaName = "Unknown";
String jsonResponse = "[{\"id\":\"1\", \"name\":\"Moscow\"}]";
mockHttpResponse(200, jsonResponse);

when(objectMapper.readValue(anyString(), any(TypeReference.class))).thenReturn(
List.of(Map.of("id", "1", "name", "Moscow")));

// Act
Integer areaId = areaUtil.getAreaIdFromApi(areaName);

// Assert
assertNull(areaId);
}

@Test
@DisplayName("Should return null for empty area list")
void testGetAreaIdFromApi_EmptyAreaList() throws Exception {
// Arrange
String areaName = "Moscow";
String jsonResponse = "[]"; // Пустой JSON массив
mockHttpResponse(200, jsonResponse);

when(objectMapper.readValue(anyString(), any(TypeReference.class))).thenReturn(List.of());

// Act
Integer areaId = areaUtil.getAreaIdFromApi(areaName);

// Assert
assertNull(areaId);
}

@SneakyThrows
@Test
@DisplayName("Should return null for API error response")
void testGetAreaIdFromApi_ApiError() {
// Arrange
String areaName = "Moscow";
mockHttpResponse(500, "Internal Server Error");

// Act
Integer areaId = areaUtil.getAreaIdFromApi(areaName);

// Assert
assertNull(areaId);
}

@Test
@DisplayName("Should return null for null area name")
void testGetAreaIdFromApi_NullAreaName() {
// Act
Integer areaId = areaUtil.getAreaIdFromApi(null);

// Assert
assertNull(areaId);
}

@Test
@DisplayName("Should return null for IOException during API call")
void testGetAreaIdFromApi_IOException() throws Exception {
// Arrange
String areaName = "Moscow";
when(client.send(any(HttpRequest.class), eq(HttpResponse.BodyHandlers.ofString()))).thenThrow(
new IOException("Network error"));

// Act
assertThrows(IOException.class, () -> areaUtil.getAreaIdFromApi(areaName));
}

@Test
@DisplayName("Should return area ID for nested area structure")
void testGetAreaIdFromApi_NestedAreas() throws Exception {
// Arrange
String areaName = "SubMoscow";
String jsonResponse =
"[{\"id\":\"1\", \"name\":\"Moscow\", \"areas\":[{\"id\":\"2\", \"name\":\"SubMoscow\"}]}]";
mockHttpResponse(200, jsonResponse);

when(objectMapper.readValue(anyString(), any(TypeReference.class))).thenReturn(
List.of(Map.of("id", "1", "name", "Moscow", "areas", List.of(Map.of("id", "2", "name", "SubMoscow")))));

// Act
Integer areaId = areaUtil.getAreaIdFromApi(areaName);

// Assert
assertEquals(2, areaId);
}
ObjectMapper realObjectMapper = new ObjectMapper();
@Mock
private ObjectMapper objectMapper;
@Mock
private HttpClient client;
@InjectMocks
private AreaUtil areaUtil;

@BeforeEach
void setUp() {
MockitoAnnotations.openMocks(this);
}

private HttpResponse<String> mockHttpResponse(int statusCode, String body)
throws IOException, InterruptedException {
HttpResponse<String> response = mock(HttpResponse.class);
when(response.statusCode()).thenReturn(statusCode);
when(response.body()).thenReturn(body);
when(client.send(any(HttpRequest.class), eq(ofString()))).thenReturn(response);
return response;
}

@Test
@DisplayName("Should return area ID for valid area name")
void testGetAreaIdFromApi_ValidAreaName() throws Exception {
// Arrange
String areaName = "Moscow";
String jsonResponse = "[{\"id\":\"1\", \"name\":\"Moscow\"}]";
mockHttpResponse(200, jsonResponse);

JsonNode jsonNode = realObjectMapper.readTree(jsonResponse);
when(objectMapper.readTree(jsonResponse)).thenReturn(jsonNode);

// Act
Integer areaId = areaUtil.getAreaIdFromApi(areaName);

// Assert
assertEquals(1, areaId);
}

@Test
@DisplayName("Should return null for invalid area name")
void testGetAreaIdFromApi_InvalidAreaName() throws Exception {
// Arrange
String areaName = "Unknown";
String jsonResponse = "[{\"id\":\"1\", \"name\":\"Moscow\"}]";
var response = mockHttpResponse(200, jsonResponse);

JsonNode jsonNode = realObjectMapper.readTree(jsonResponse);
when(objectMapper.readTree(response.body())).thenReturn(jsonNode);

// Act
Integer areaId = areaUtil.getAreaIdFromApi(areaName);

// Assert
assertNull(areaId);
}

@Test
@DisplayName("Should return null for empty area list")
void testGetAreaIdFromApi_EmptyAreaList() throws Exception {
// Arrange
String areaName = "Moscow";
String jsonResponse = "[]"; // Пустой JSON массив
mockHttpResponse(200, jsonResponse);

JsonNode jsonNode = realObjectMapper.readTree(jsonResponse);
when(objectMapper.readTree(jsonResponse)).thenReturn(jsonNode);

// Act
Integer areaId = areaUtil.getAreaIdFromApi(areaName);

// Assert
assertNull(areaId);
}

@SneakyThrows
@Test
@DisplayName("Should return null for API error response")
void testGetAreaIdFromApi_ApiError() {
// Arrange
String areaName = "Moscow";
mockHttpResponse(500, "Internal Server Error");

// Act
Integer areaId = areaUtil.getAreaIdFromApi(areaName);

// Assert
assertNull(areaId);
}

@Test
@DisplayName("Should return null for null area name")
void testGetAreaIdFromApi_NullAreaName() {
// Act
Integer areaId = areaUtil.getAreaIdFromApi(null);

// Assert
assertNull(areaId);
}

@Test
@DisplayName("Should return null for IOException during API call")
void testGetAreaIdFromApi_IOException() throws Exception {
// Arrange
String areaName = "Moscow";
when(client.send(any(HttpRequest.class), eq(HttpResponse.BodyHandlers.ofString()))).thenThrow(new IOException("Network error"));

// Act
assertThrows(IOException.class, () -> areaUtil.getAreaIdFromApi(areaName));
}

@Test
@DisplayName("Should return area ID for nested area structure")
void testGetAreaIdFromApi_NestedAreas() throws Exception {
// Arrange
String areaName = "SubMoscow";
String jsonResponse =
"[{\"id\":\"1\", \"name\":\"Moscow\", \"areas\":[{\"id\":\"2\", \"name\":\"SubMoscow\"}]}]";
mockHttpResponse(200, jsonResponse);

JsonNode jsonNode = realObjectMapper.readTree(jsonResponse);
when(objectMapper.readTree(jsonResponse)).thenReturn(jsonNode);

// Act
Integer areaId = areaUtil.getAreaIdFromApi(areaName);

// Assert
assertEquals(2, areaId);
}

}

0 comments on commit 8d5aade

Please sign in to comment.