baiyue-firstleg/common-web/src/test/java/com/baiyue/common/web/GlobalExceptionHandlerTest.java

package com.baiyue.common.web;

import com.baiyue.common.core.api.ResultCode;
import com.baiyue.common.core.exception.BizException;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.core.MethodParameter;
import org.springframework.web.method.annotation.MethodArgumentTypeMismatchException;

import java.lang.reflect.Method;

import static org.assertj.core.api.Assertions.assertThat;

@DisplayName("全局异常响应")
class GlobalExceptionHandlerTest {
    private final GlobalExceptionHandler handler = new GlobalExceptionHandler();

    @Test
    void businessExceptionKeepsBusinessCodeAndCorrelation() {
        MockHttpServletRequest request = new MockHttpServletRequest();
        request.setAttribute(CorrelationContext.REQUEST_ID, "r1");
        request.setAttribute(CorrelationContext.TRACE_ID, "t1");
        var response = handler.handleBusiness(new BizException("inventory is locked"), request);
        assertThat(response.getStatusCode().value()).isEqualTo(400);
        assertThat(response.getBody().code()).isEqualTo(ResultCode.BUSINESS_ERROR.code());
        assertThat(response.getBody().requestId()).isEqualTo("r1");
    }

    @Test
    void invalidQueryParameterTypeMapsToBadRequest() throws Exception {
        MockHttpServletRequest request = new MockHttpServletRequest();
        Method method = GlobalExceptionHandlerTest.class.getDeclaredMethod("paging", long.class);
        MethodParameter parameter = new MethodParameter(method, 0);
        MethodArgumentTypeMismatchException exception = new MethodArgumentTypeMismatchException(
                "abc", Long.class, "page", parameter, new NumberFormatException("abc"));

        var response = handler.handleBadRequest(exception, request);

        assertThat(response.getStatusCode().value()).isEqualTo(400);
        assertThat(response.getBody().code()).isEqualTo(ResultCode.BAD_REQUEST.code());
    }

    private void paging(long page) {
    }
}