OpenApiControllerTest.java
3.18 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package com.diligrp.rider.controller;
import com.diligrp.rider.common.exception.BizException;
import com.diligrp.rider.common.result.Result;
import com.diligrp.rider.dto.OpenRouteCalcDTO;
import com.diligrp.rider.service.DeliveryFeeService;
import com.diligrp.rider.service.OpenAppService;
import com.diligrp.rider.service.amap.AmapRouteClient;
import com.diligrp.rider.util.GeoUtil;
import com.diligrp.rider.vo.RouteCalcVO;
import org.junit.jupiter.api.Test;
import java.util.Optional;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertNull;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
class OpenApiControllerTest {
private final DeliveryFeeService deliveryFeeService = mock(DeliveryFeeService.class);
private final OpenAppService openAppService = mock(OpenAppService.class);
private final AmapRouteClient amapRouteClient = mock(AmapRouteClient.class);
private final OpenApiController controller =
new OpenApiController(deliveryFeeService, openAppService, amapRouteClient);
@Test
void routeCalcShouldReturnAmapValuesWhenHit() {
OpenRouteCalcDTO dto = dto("121.4737", "31.2304", "121.5067", "31.2454");
when(amapRouteClient.getElectrobikeRoute(
dto.getStartLng(), dto.getStartLat(), dto.getEndLng(), dto.getEndLat()))
.thenReturn(Optional.of(new AmapRouteClient.AmapRoute(4830, 1080)));
Result<RouteCalcVO> result = controller.routeCalc(dto);
assertEquals(0, result.getCode());
assertEquals(4830, result.getData().getDistance());
assertEquals(1080, result.getData().getDuration());
}
@Test
void routeCalcShouldFallbackToGeoUtilWhenAmapMisses() {
OpenRouteCalcDTO dto = dto("121.4737", "31.2304", "121.4737", "31.2574");
when(amapRouteClient.getElectrobikeRoute(any(), any(), any(), any()))
.thenReturn(Optional.empty());
Result<RouteCalcVO> result = controller.routeCalc(dto);
int expected = (int) Math.round(
GeoUtil.calcDistanceKm(31.2304, 121.4737, 31.2574, 121.4737) * 1000);
assertEquals(0, result.getCode());
assertNotNull(result.getData().getDistance());
assertEquals(expected, result.getData().getDistance());
assertNull(result.getData().getDuration());
}
@Test
void routeCalcShouldThrowOnInvalidCoordinatesWhenAmapMisses() {
OpenRouteCalcDTO dto = dto("not-a-number", "31.2304", "121.5067", "31.2454");
when(amapRouteClient.getElectrobikeRoute(any(), any(), any(), any()))
.thenReturn(Optional.empty());
assertThrows(BizException.class, () -> controller.routeCalc(dto));
}
private OpenRouteCalcDTO dto(String startLng, String startLat, String endLng, String endLat) {
OpenRouteCalcDTO dto = new OpenRouteCalcDTO();
dto.setStartLng(startLng);
dto.setStartLat(startLat);
dto.setEndLng(endLng);
dto.setEndLat(endLat);
return dto;
}
}