package cn.iocoder.yudao.module.iot.gateway.protocol.modbus.tcpclient.manager;

import cn.iocoder.yudao.module.iot.core.biz.dto.IotModbusDeviceConfigRespDTO;
import cn.iocoder.yudao.module.iot.core.biz.dto.IotModbusPointRespDTO;
import cn.iocoder.yudao.module.iot.gateway.protocol.modbus.tcpclient.manager.IotModbusTcpClientPollScheduler.ReadSegment;
import org.junit.jupiter.api.Test;

import java.util.Arrays;
import java.util.Collections;
import java.util.List;

import static cn.iocoder.yudao.module.iot.gateway.protocol.modbus.common.utils.IotModbusCommonUtils.FC_READ_COILS;
import static cn.iocoder.yudao.module.iot.gateway.protocol.modbus.common.utils.IotModbusCommonUtils.FC_READ_HOLDING_REGISTERS;
import static cn.iocoder.yudao.module.iot.gateway.protocol.modbus.common.utils.IotModbusCommonUtils.FC_READ_INPUT_REGISTERS;
import static org.junit.jupiter.api.Assertions.*;

/**
 * {@link IotModbusTcpClientPollScheduler} 的单元测试
 *
 * @author 芋道源码
 */
public class IotModbusTcpClientPollSchedulerTest {

    @Test
    public void testBuildReadSegments_mergeContinuousPoints() {
        // 准备参数
        IotModbusPointRespDTO point01 = randomPoint(1L, FC_READ_HOLDING_REGISTERS, 0, 1, 1000);
        IotModbusPointRespDTO point02 = randomPoint(2L, FC_READ_HOLDING_REGISTERS, 1, 2, 1000);
        IotModbusPointRespDTO point03 = randomPoint(3L, FC_READ_HOLDING_REGISTERS, 4, 1, 1000);
        IotModbusDeviceConfigRespDTO config = randomConfig(point01, point02, point03);

        // 调用
        List<ReadSegment> segments = IotModbusTcpClientPollScheduler.buildReadSegments(config);

        // 断言
        assertEquals(2, segments.size());
        assertEquals(0, segments.get(0).getStartAddress());
        assertEquals(3, segments.get(0).getRegisterCount());
        assertIterableEquals(Arrays.asList(point01, point02), segments.get(0).getPoints());
        assertEquals(4, segments.get(1).getStartAddress());
        assertEquals(1, segments.get(1).getRegisterCount());
        assertIterableEquals(Collections.singletonList(point03), segments.get(1).getPoints());
    }

    @Test
    public void testBuildReadSegments_mergeOverlappingPoints() {
        // 准备参数
        IotModbusPointRespDTO point01 = randomPoint(1L, FC_READ_HOLDING_REGISTERS, 0, 2, 1000);
        IotModbusPointRespDTO point02 = randomPoint(2L, FC_READ_HOLDING_REGISTERS, 1, 1, 1000);
        IotModbusDeviceConfigRespDTO config = randomConfig(point01, point02);

        // 调用
        List<ReadSegment> segments = IotModbusTcpClientPollScheduler.buildReadSegments(config);

        // 断言
        assertEquals(1, segments.size());
        assertEquals(0, segments.get(0).getStartAddress());
        assertEquals(2, segments.get(0).getRegisterCount());
        assertIterableEquals(Arrays.asList(point01, point02), segments.get(0).getPoints());
    }

    @Test
    public void testBuildReadSegments_notMergeDifferentFunctionCodeOrPollInterval() {
        // 准备参数
        IotModbusPointRespDTO point01 = randomPoint(1L, FC_READ_HOLDING_REGISTERS, 0, 1, 1000);
        IotModbusPointRespDTO point02 = randomPoint(2L, FC_READ_INPUT_REGISTERS, 1, 1, 1000);
        IotModbusPointRespDTO point03 = randomPoint(3L, FC_READ_HOLDING_REGISTERS, 1, 1, 2000);
        IotModbusDeviceConfigRespDTO config = randomConfig(point01, point02, point03);

        // 调用
        List<ReadSegment> segments = IotModbusTcpClientPollScheduler.buildReadSegments(config);

        // 断言
        assertEquals(3, segments.size());
        assertEquals(FC_READ_HOLDING_REGISTERS, segments.get(0).getFunctionCode());
        assertEquals(1000, segments.get(0).getPollInterval());
        assertEquals(FC_READ_HOLDING_REGISTERS, segments.get(1).getFunctionCode());
        assertEquals(2000, segments.get(1).getPollInterval());
        assertEquals(FC_READ_INPUT_REGISTERS, segments.get(2).getFunctionCode());
        assertEquals(1000, segments.get(2).getPollInterval());
    }

    @Test
    public void testBuildReadSegments_splitWhenExceedsMaxRegisterCount() {
        // 准备参数
        IotModbusPointRespDTO point01 = randomPoint(1L, FC_READ_HOLDING_REGISTERS, 0, 100, 1000);
        IotModbusPointRespDTO point02 = randomPoint(2L, FC_READ_HOLDING_REGISTERS, 100, 30, 1000);
        IotModbusDeviceConfigRespDTO config = randomConfig(point01, point02);

        // 调用
        List<ReadSegment> segments = IotModbusTcpClientPollScheduler.buildReadSegments(config);

        // 断言
        assertEquals(2, segments.size());
        assertEquals(0, segments.get(0).getStartAddress());
        assertEquals(100, segments.get(0).getRegisterCount());
        assertEquals(100, segments.get(1).getStartAddress());
        assertEquals(30, segments.get(1).getRegisterCount());
    }

    @Test
    public void testExtractPointRawValues() {
        // 准备参数
        IotModbusPointRespDTO point = randomPoint(1L, FC_READ_HOLDING_REGISTERS, 12, 2, 1000);
        ReadSegment segment = new ReadSegment(FC_READ_HOLDING_REGISTERS, 1000, 10, 4, Collections.singletonList(point));
        int[] rawValues = new int[]{100, 200, 300, 400};

        // 调用
        int[] pointRawValues = IotModbusTcpClientPollScheduler.extractPointRawValues(rawValues, segment, point);

        // 断言
        assertArrayEquals(new int[]{300, 400}, pointRawValues);
    }

    @Test
    public void testExtractPointRawValuesForCoils() {
        // 准备参数
        IotModbusPointRespDTO point = randomPoint(1L, FC_READ_COILS, 3, 2, 1000);
        ReadSegment segment = new ReadSegment(FC_READ_COILS, 1000, 0, 5, Collections.singletonList(point));
        int[] rawValues = new int[]{1, 0, 1, 1, 0, 0, 0, 0}; // 线圈响应可能按字节补齐

        // 调用
        int[] pointRawValues = IotModbusTcpClientPollScheduler.extractPointRawValues(rawValues, segment, point);

        // 断言
        assertArrayEquals(new int[]{1, 0}, pointRawValues);
    }

    private static IotModbusDeviceConfigRespDTO randomConfig(IotModbusPointRespDTO... points) {
        IotModbusDeviceConfigRespDTO config = new IotModbusDeviceConfigRespDTO();
        config.setDeviceId(1L);
        config.setPoints(Arrays.asList(points));
        return config;
    }

    private static IotModbusPointRespDTO randomPoint(Long id, Integer functionCode, Integer registerAddress,
                                                     Integer registerCount, Integer pollInterval) {
        IotModbusPointRespDTO point = new IotModbusPointRespDTO();
        point.setId(id);
        point.setFunctionCode(functionCode);
        point.setRegisterAddress(registerAddress);
        point.setRegisterCount(registerCount);
        point.setPollInterval(pollInterval);
        return point;
    }

}