package com.ruoyi.order.service.impl;

import com.ruoyi.order.dto.SixChargingDto;
import org.junit.jupiter.api.Test;

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

import static org.junit.jupiter.api.Assertions.assertEquals;

class MonthlyIncomeFallbackTest {

    @Test
    void addsOnlyOrderMonthsMissingFromSettlementIncome() {
        SixChargingDto settlementApril = month("04");
        SixChargingDto orderMarch = month("03");
        SixChargingDto orderApril = month("04");

        List<SixChargingDto> merged = TChargingOrderServiceImpl.mergeMissingMonthlyIncome(
                Arrays.asList(settlementApril), Arrays.asList(orderMarch, orderApril));

        assertEquals(Arrays.asList("04", "03"), Arrays.asList(
                merged.get(0).getMonth(), merged.get(1).getMonth()));
    }

    private SixChargingDto month(String value) {
        SixChargingDto dto = new SixChargingDto();
        dto.setMonth(value);
        return dto;
    }
}