package com.estate.expert.admin.stat.report.data;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.estate.expert.admin.domain.exam.eval.fuzzy.FuzzyResult;
import com.estate.expert.admin.domain.exam.eval.fuzzy.FuzzyRiskLevel;
import com.estate.expert.admin.domain.exam.eval.fuzzy.FuzzyValue;
import com.estate.expert.admin.domain.exam.eval.score.ScoreResult;
import com.estate.expert.admin.domain.exam.eval.score.ScoreValue;
import com.estate.expert.admin.domain.project.Project;
import com.estate.expert.admin.domain.project.enums.ProjectType;
import com.estate.expert.admin.domain.report.ReportTemplate;
import com.estate.expert.admin.stat.report.builder.SystemReportBuilder;
import com.estate.expert.admin.stat.report.data.impl.SystemScoreReportData;
import com.estate.expert.admin.utils.tree.TreeNode;
import org.junit.jupiter.api.Test;

import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.List;

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

class SystemReportDataTest {

    @Test
    void scoreReportUsesScoreTreeForHazardSummary() {
        var project = new Project();
        project.setName("测试项目");
        project.setType(ProjectType.HOUSE_CONSTRUCTION);

        var template = new ReportTemplate();
        template.setNumber("TEST");

        var scoreRoot = new TreeNode<>("root", ScoreValue.builder()
                .score(80)
                .totalErrorCounts(new int[]{1, 2, 3, 4})
                .build());
        var fuzzyRoot = new TreeNode<>("root", FuzzyValue.builder()
                .riskLevel(FuzzyRiskLevel.P4)
                .totalErrorCounts(new int[]{9, 9, 9, 9})
                .build());

        var report = SystemScoreReportData.builder()
                .reportId(1)
                .project(project)
                .template(template)
                .devices(List.of())
                .scoreResult(new ScoreResult(scoreRoot))
                .sysResult(new FuzzyResult(fuzzyRoot))
                .build()
                .generate();

        var summary = report.getJSONObject("评估结果").getJSONObject("隐患等级分布");
        assertEquals(10, summary.getIntValue("隐患数量"));
        assertEquals(1, summary.getIntValue("一级隐患数量"));
        assertEquals(4, summary.getIntValue("四级隐患数量"));
    }

    @Test
    void hazardDistributionIncludesManagementBehavior() throws Exception {
        var evalResult = new JSONObject();
        var checkItems = new JSONArray();
        checkItems.add(checkItem("安全检查"));
        checkItems.add(checkItem("管理行为"));
        evalResult.put("检查项", checkItems);

        var data = new JSONObject();
        data.put("评估结果", evalResult);

        Method initChart = SystemReportBuilder.class.getDeclaredMethod(
                "initChart", JSONObject.class, java.util.Map.class);
        initChart.setAccessible(true);
        initChart.invoke(new SystemReportBuilder(null), data, new HashMap<>());

        assertEquals(2, evalResult.getJSONArray("安全检查项").size());
        assertEquals("管理行为", evalResult.getJSONArray("安全检查项")
                .getJSONObject(1).getString("名称"));
    }

    private JSONObject checkItem(String name) {
        var item = new JSONObject();
        item.put("类别", "测试类别");
        item.put("问题数量", 1);
        item.put("问题百分比", "100.00%");

        var category = new JSONObject();
        category.put("分项", new JSONArray(List.of(item)));

        var result = new JSONObject();
        result.put("名称", name);
        result.put("分项评估", new JSONArray(List.of(category)));
        return result;
    }
}