package com.xiaopeng.automaticPilot.util;
|
|
import com.xiaopeng.api.dto.automaticPilot.MqttPublishDto;
|
import com.xiaopeng.api.feignClient.automaticPilot.VmpOpenServiceClient;
|
import com.xiaopeng.api.feignClient.automaticPilot.XPUClient;
|
import com.xiaopeng.api.vo.automaticPilot.MqttPublishVo;
|
import com.xiaopeng.api.vo.automaticPilot.protobuf.*;
|
import org.junit.jupiter.api.Assertions;
|
import org.junit.jupiter.api.BeforeEach;
|
import org.junit.jupiter.api.Test;
|
import org.mockito.ArgumentCaptor;
|
import org.mockito.Mockito;
|
|
import java.lang.reflect.Field;
|
import java.lang.reflect.Method;
|
import java.util.HashMap;
|
import java.util.Map;
|
|
public class XPUUtilTest {
|
|
private XPUUtil xpuUtil;
|
private XPUClient xpuClient;
|
private VmpOpenServiceClient vmpClient;
|
|
@BeforeEach
|
public void setUp() throws Exception {
|
xpuUtil = new XPUUtil();
|
|
// Mock VmpOpenServiceClient
|
vmpClient = Mockito.mock(VmpOpenServiceClient.class);
|
Map<String, String> mapping = new HashMap<>();
|
mapping.put(XPUUtil.XPU_ID_TYPE, "1234567890ABCDEFGHIJKL");
|
Mockito.when(vmpClient.getVehicleBindInfoByVin(Mockito.anyString(), Mockito.anyList()))
|
.thenReturn(mapping);
|
|
// Mock XPUClient
|
xpuClient = Mockito.mock(XPUClient.class);
|
Mockito.when(xpuClient.publish(Mockito.any(MqttPublishVo.class)))
|
.thenReturn(new MqttPublishDto());
|
|
// Inject mocks
|
Field vmpField = XPUUtil.class.getDeclaredField("vmpOpenServiceClient");
|
vmpField.setAccessible(true);
|
vmpField.set(xpuUtil, vmpClient);
|
|
Field xpuClientField = XPUUtil.class.getDeclaredField("xpuClient");
|
xpuClientField.setAccessible(true);
|
xpuClientField.set(xpuUtil, xpuClient);
|
|
Field topicField = XPUUtil.class.getDeclaredField("topicPrefix");
|
topicField.setAccessible(true);
|
topicField.set(xpuUtil, "Robotaxi/Msg/Xpu");
|
}
|
|
@Test
|
public void testBuildTopicByVin() throws Exception {
|
Method buildTopic = XPUUtil.class.getDeclaredMethod("buildTopicByVin", String.class);
|
buildTopic.setAccessible(true);
|
String result = (String) buildTopic.invoke(xpuUtil, "VIN123");
|
Assertions.assertEquals("Robotaxi/Msg/Xpu/ABCDEFGHIJKL00000000", result);
|
}
|
|
@Test
|
public void testCloudCreateOrder() {
|
String orderId = "1881347710714060802";
|
|
RtTask.Builder taskBuilder = RtTask.newBuilder();
|
taskBuilder.setTaskId("task-123");
|
taskBuilder.setOrderId(orderId);
|
taskBuilder.setTaskType(OrderType.ORDER_TYPE_ROBOTAXI);
|
|
Waypoint.Builder pointBuilder = Waypoint.newBuilder();
|
pointBuilder.setPointType(PointType.START_POINT);
|
pointBuilder.setActionType(ActionType.PICK_UP);
|
pointBuilder.setSequence(1);
|
|
GeoPoint.Builder geoPointBuilder = GeoPoint.newBuilder();
|
geoPointBuilder.setLatitude(30.0);
|
geoPointBuilder.setLongitude(120.0);
|
// geoPointBuilder.setPoiName("Start Point"); // GeoPoint might not have poi_name
|
pointBuilder.setPoint(geoPointBuilder.build());
|
|
taskBuilder.addPoints(pointBuilder);
|
|
RtTaskList.Builder taskListBuilder = RtTaskList.newBuilder();
|
taskListBuilder.addTasks(taskBuilder.build());
|
|
String vin = "VIN12345678900000";
|
Long sequence = 1L;
|
|
xpuUtil.cloudCreateOrder(taskListBuilder.build(), vin, sequence);
|
|
ArgumentCaptor<MqttPublishVo> captor = ArgumentCaptor.forClass(MqttPublishVo.class);
|
Mockito.verify(xpuClient).publish(captor.capture());
|
MqttPublishVo capturedVo = captor.getValue();
|
|
Assertions.assertEquals("Robotaxi/Msg/Xpu/ABCDEFGHIJKL00000000", capturedVo.getTopic());
|
// Note: Current implementation serializes enum as ordinal.
|
Assertions.assertTrue(capturedVo.getPayload().contains("\"cmd\": 10003"), "Payload should contain cmd CLOUD_CREATE_ORDER (10003), but was: " + capturedVo.getPayload());
|
}
|
|
@Test
|
public void testCloudOrderOperation() {
|
String orderId = "1881347710714060802";
|
|
OrderOperation.Builder operationBuilder = OrderOperation.newBuilder();
|
operationBuilder.setOrderId(orderId);
|
operationBuilder.setTaskId("task-123");
|
operationBuilder.setOperationType(OrderOperationType.CLOUD_CREATE_ORDER);
|
operationBuilder.setOrderType(OrderType.ORDER_TYPE_ROBOTAXI);
|
|
Waypoint.Builder waypointBuilder = Waypoint.newBuilder();
|
waypointBuilder.setWaypointId("wp-1");
|
waypointBuilder.setPointType(PointType.WAY_POINT);
|
waypointBuilder.setActionType(ActionType.PICK_UP); // Use valid enum
|
waypointBuilder.setSequence(2);
|
|
GeoPoint.Builder geoPointBuilder = GeoPoint.newBuilder();
|
geoPointBuilder.setLatitude(30.1);
|
geoPointBuilder.setLongitude(120.1);
|
// geoPointBuilder.setPoiName("Way Point");
|
waypointBuilder.setPoint(geoPointBuilder.build());
|
|
operationBuilder.addWaypoints(waypointBuilder.build());
|
|
// TargetPoint targetPoint = new TargetPoint(); // OrderOperation uses Waypoint for target_point in proto?
|
// Check proto: Waypoint target_point = 6;
|
Waypoint.Builder targetPointBuilder = Waypoint.newBuilder();
|
GeoPoint.Builder endGeoPointBuilder = GeoPoint.newBuilder();
|
endGeoPointBuilder.setLatitude(30.2);
|
endGeoPointBuilder.setLongitude(120.2);
|
targetPointBuilder.setPoint(endGeoPointBuilder.build());
|
operationBuilder.setTargetPoint(targetPointBuilder.build());
|
|
String vin = "VIN12345678900000";
|
Long sequence = 2L;
|
|
xpuUtil.cloudOrderOperation(operationBuilder.build(), vin, sequence);
|
|
ArgumentCaptor<MqttPublishVo> captor = ArgumentCaptor.forClass(MqttPublishVo.class);
|
Mockito.verify(xpuClient).publish(captor.capture());
|
MqttPublishVo capturedVo = captor.getValue();
|
|
Assertions.assertEquals("Robotaxi/Msg/Xpu/ABCDEFGHIJKL00000000", capturedVo.getTopic());
|
// Note: CLOUD_ORDER_OPERATION is 10001
|
Assertions.assertTrue(capturedVo.getPayload().contains("\"cmd\": 10001"), "Payload should contain cmd CLOUD_ORDER_OPERATION (10001), but was: " + capturedVo.getPayload());
|
}
|
|
@Test
|
public void testResponseMessage() {
|
ResponseMessage.Builder responseMessageBuilder = ResponseMessage.newBuilder();
|
responseMessageBuilder.setRequestSeq(12345L);
|
responseMessageBuilder.setCode(StatusCode.OK);
|
responseMessageBuilder.setTimestamp(System.currentTimeMillis());
|
|
|
String vin = "VIN12345678900000";
|
xpuUtil.responseMessage(responseMessageBuilder.build(), vin);
|
|
ArgumentCaptor<MqttPublishVo> captor = ArgumentCaptor.forClass(MqttPublishVo.class);
|
Mockito.verify(xpuClient).publish(captor.capture());
|
MqttPublishVo capturedVo = captor.getValue();
|
|
Assertions.assertEquals("Robotaxi/Msg/Xpu/ABCDEFGHIJKL00000000", capturedVo.getTopic());
|
// Note: CLOUD_TASK_CONTROL (used in responseMessage method? check code)
|
// In XPUUtil.java: headerBuilder.setCmd(RTCmdType.CLOUD_TASK_CONTROL.getValue());
|
// CLOUD_TASK_CONTROL is 10002
|
Assertions.assertTrue(capturedVo.getPayload().contains("\"cmd\": 10002"), "Payload should contain cmd CLOUD_TASK_CONTROL (10002), but was: " + capturedVo.getPayload());
|
|
}
|
|
@Test
|
public void testCloudPushNavigation() {
|
RestoreNavigationData.Builder navigationDataBuilder = RestoreNavigationData.newBuilder();
|
navigationDataBuilder.setRequestId("req-123");
|
navigationDataBuilder.setRouteId("5939fa259ad6479cbdf8871ceed96b72");
|
String content = "{\"startPoint\":{\"location\":\"116.481028,39.989643\"},\"endPoint\":{\"location\":\"116.465302,40.004717\"},\"routeID\":\"xxxxxxxxxxxxx\",\"strategy\":17,\"type\":1,\"ferry\":0,\"disableEndDispatch\":1,\"viaInfo\":[{},{}],\"routeUserType\":2,\"bizScene\":2,\"plate\":\"京A12345\",\"vehicleType\":1}";
|
navigationDataBuilder.setContent(content);
|
navigationDataBuilder.setTimestamp(System.currentTimeMillis());
|
|
String xpuId = "1234567890ABCDEFGHIJKL";
|
Long sequence = 3L;
|
|
xpuUtil.cloudPushNavigation(navigationDataBuilder.build(), xpuId, sequence);
|
|
ArgumentCaptor<MqttPublishVo> captor = ArgumentCaptor.forClass(MqttPublishVo.class);
|
Mockito.verify(xpuClient).publish(captor.capture());
|
MqttPublishVo capturedVo = captor.getValue();
|
|
Assertions.assertEquals("Robotaxi/Msg/Xpu/ABCDEFGHIJKL00000000", capturedVo.getTopic());
|
// Note: CLOUD_NAVIGATION_RESPONSE is 10006
|
Assertions.assertTrue(capturedVo.getPayload().contains("\"cmd\": 10006"), "Payload should contain cmd CLOUD_NAVIGATION_RESPONSE (10006), but was: " + capturedVo.getPayload());
|
}
|
}
|
