package com.superbiz.agent.graph.diagnosis;

import com.alibaba.cloud.ai.graph.OverAllState;
import com.alibaba.cloud.ai.graph.RunnableConfig;
import org.junit.jupiter.api.Test;

import java.util.List;
import java.util.Map;

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

class VerifiedInputNodeTest {

    @Test
    @SuppressWarnings("unchecked")
    void projectsOnlyClaimsWithExactlyMatchedPassedBindings() throws Exception {
        Map<String, Object> passedBinding = Map.of(
                "tool_name", "query_metrics",
                "source_invocation_id", 11L,
                "raw_path", "$.alerts[0]",
                "evidence_excerpt", "active=50 max=50");
        Map<String, Object> failedBinding = Map.of(
                "tool_name", "query_logs",
                "source_invocation_id", 12L,
                "raw_path", "$.logs[0]",
                "evidence_excerpt", "suspected leak");
        Map<String, Object> executorOutput = Map.of(
                "answer_version", "executor_evidence_v2",
                "claims", List.of(
                        Map.of("claim_id", "c1", "claim_text", "连接数达到上限",
                                "claim_type", "symptom", "support_level", "direct",
                                "evidence_bindings", List.of(passedBinding, failedBinding)),
                        Map.of("claim_id", "c2", "claim_text", "存在连接泄漏",
                                "claim_type", "root_cause", "support_level", "indirect",
                                "evidence_bindings", List.of(failedBinding))),
                "hypotheses", List.of(Map.of("hypothesis_text", "must-not-leak")));
        Map<String, Object> gatekeeper = Map.of(
                "status", "fail",
                "severity", "low_confid",
                "checked_bindings", List.of(
                        Map.of("claim_id", "c1", "status", "pass",
                                "tool_name", "query_metrics",
                                "source_invocation_id", 11L,
                                "raw_path", "$.alerts[0]",
                                "matched_text", "active=50 max=50"),
                        Map.of("claim_id", "c1", "status", "fail",
                                "tool_name", "query_logs",
                                "source_invocation_id", 12L,
                                "raw_path", "$.logs[0]"),
                        Map.of("claim_id", "c3", "status", "pass",
                                "tool_name", "query_metrics",
                                "source_invocation_id", 99L,
                                "raw_path", "$.alerts[9]",
                                "matched_text", "unreferenced")));
        VerifiedInputNode node = new VerifiedInputNode();

        Map<String, Object> update = node.apply(new OverAllState(Map.of(
                        DiagnosisGraphState.EXECUTOR_OUTPUT, executorOutput,
                        DiagnosisGraphState.GATEKEEPER_RESULT, gatekeeper,
                        DiagnosisGraphState.EVIDENCE_RETRY_COUNT, 0)),
                RunnableConfig.builder().threadId("run-verified").build()).get();

        Map<String, Object> verifiedOutput = (Map<String, Object>) update.get(
                DiagnosisGraphState.VERIFIED_EXECUTOR_OUTPUT);
        List<Map<String, Object>> claims =
                (List<Map<String, Object>>) verifiedOutput.get("claims");
        assertEquals(List.of("c1"),
                claims.stream().map(claim -> claim.get("claim_id")).toList());
        assertEquals(List.of(passedBinding), claims.get(0).get("evidence_bindings"));
        assertFalse(verifiedOutput.containsKey("hypotheses"));
        List<Map<String, Object>> evidence =
                (List<Map<String, Object>>) update.get(
                        DiagnosisGraphState.VERIFIED_EVIDENCE);
        assertEquals(List.of("active=50 max=50"),
                evidence.stream().map(item -> item.get("matched_text")).toList());
        assertEquals(1, update.get(DiagnosisGraphState.VERIFIED_BINDING_COUNT));
    }
}