package com.battery.business.web.device;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import java.lang.reflect.Method;
import org.junit.jupiter.api.Test;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
import org.springframework.test.web.servlet.result.MockMvcResultMatchers;
import org.springframework.test.web.servlet.setup.MockMvcBuilders;
import com.ruoyi.common.annotation.Anonymous;
import com.ruoyi.common.exception.ServiceException;
import com.battery.business.service.DeviceAdapterErrorCodes;
import com.battery.business.service.DeviceIngestErrorCodes;
import com.battery.business.service.IDeviceIngestService;
import com.battery.business.service.IDeviceIngestService.IngestResult;

/**
 * 接收端点的应答契约：拒绝一律用非 2xx 状态码——取数程序只在收到 2xx 后才推进自己的读取位置，
 * 所以"响应 200 但体内是错误码"是不能接受的：那会让程序把没落库的数据当成已送达。
 */
class DeviceIngestControllerTest
{
    private final IDeviceIngestService ingest = mock(IDeviceIngestService.class);
    private final MockMvc mvc = MockMvcBuilders.standaloneSetup(new DeviceIngestController(ingest)).build();

    @Test
    void acceptedBatchAnswersOkSoTheProgramMayAdvanceItsWatermark() throws Exception
    {
        when(ingest.ingest(any(), any(), any())).thenReturn(new IngestResult(2, 0, 0));

        mvc.perform(MockMvcRequestBuilders.post("/device-ingest/j22")
                        .header(DeviceIngestController.TOKEN_HEADER, "token").content("{}"))
                .andExpect(MockMvcResultMatchers.status().isOk())
                .andExpect(MockMvcResultMatchers.jsonPath("$.accepted").value(2));
    }

    @Test
    void rejectionsAnswerWithNonOkStatusesSoTheProgramKeepsItsPosition() throws Exception
    {
        assertStatus(DeviceIngestErrorCodes.TOKEN_INVALID, 401);
        assertStatus(DeviceIngestErrorCodes.SOURCE_DISABLED, 403);
        assertStatus(DeviceAdapterErrorCodes.UNSUPPORTED_TYPE, 403);
        assertStatus(DeviceIngestErrorCodes.PAYLOAD_INVALID, 400);
        assertStatus(DeviceAdapterErrorCodes.NOT_FOUND, 404);
        assertStatus(59999, 500);
    }

    /** 端点匿名放行（取数程序没有登录态），凭据由请求头里的来源令牌自己校验。 */
    @Test
    void endpointIsAnonymousAndAuthenticatesWithTheSourceTokenHeader() throws Exception
    {
        Method receive = DeviceIngestController.class.getMethod("receive", String.class, String.class, String.class);
        assertTrue(DeviceIngestController.class.isAnnotationPresent(Anonymous.class));
        assertEquals("X-Device-Ingest-Token", DeviceIngestController.TOKEN_HEADER);
        assertTrue(receive.getParameters()[1].toString().contains("token"));
    }

    private void assertStatus(int code, int expected) throws Exception
    {
        // 用 doThrow/when 形式重新打桩：直接调 mock 方法会被上一次的桩立即抛异常。
        org.mockito.Mockito.doThrow(new ServiceException("拒绝", code)).when(ingest).ingest(any(), any(), any());

        mvc.perform(MockMvcRequestBuilders.post("/device-ingest/j22").content("{}"))
                .andExpect(MockMvcResultMatchers.status().is(expected));
    }
}