package com.ruoyi.order.mapper;

import org.junit.jupiter.api.Test;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.nio.charset.StandardCharsets;
import java.util.stream.Collectors;

import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertTrue;

class AccountSettlementStatementMapperTest {

    @Test
    void usesIcbcPaidAmountsAndKeepsLegacyFieldsForCompatibility() throws IOException {
        String statementMapper = resource("mapper/order/TAccountSettlementStatementMapper.xml");
        assertTrue(statementMapper.contains("AND account_type = 3"));
        assertTrue(statementMapper.contains("SUM(platform_income)"));
        assertTrue(statementMapper.contains("SUM(partner_income)"));
        assertTrue(statementMapper.contains("platform_fee"));
        assertTrue(statementMapper.contains("tripartite_commission"));
        assertTrue(statementMapper.contains("recorded_amount"));

        String orderMapper = resource("mapper/order/TChargingOrderMapper.xml");
        String settlementQuery = orderMapper.substring(
                orderMapper.indexOf("<select id=\"getAccountSettlementStatement\""),
                orderMapper.indexOf("<select id=\"enterpriseInvoiceRows\""));
        assertTrue(settlementQuery.contains("b.order_source = 1"));
        assertTrue(settlementQuery.contains("b.source_name = 'ICBC'"));
        assertTrue(settlementQuery.contains("pay_channel = 'ICBC'"));
        assertTrue(settlementQuery.contains("channel_pay_amount"));
        assertTrue(settlementQuery.contains("channel_service_amount"));
    }

    private String resource(String path) throws IOException {
        InputStream input = getClass().getClassLoader().getResourceAsStream(path);
        assertNotNull(input);
        try (BufferedReader reader = new BufferedReader(new InputStreamReader(input, StandardCharsets.UTF_8))) {
            return reader.lines().collect(Collectors.joining("\n"));
        }
    }
}