"""brief 按矩阵分组收窄。"""
from __future__ import annotations

from django.test import SimpleTestCase

from pipeline.reporting.brief_strategy_scope import (
    filter_brief_for_strategy_matrix_group,
    list_matrix_groups_for_api,
    resolve_strategy_matrix_group_index,
)


class BriefStrategyScopeTests(SimpleTestCase):
    def _sample_brief(self) -> dict:
        return {
            "schema_version": 1,
            "keyword": "低GI",
            "matrix_by_group": [
                {
                    "group": "饮料",
                    "sku_count": 1,
                    "skus": [
                        {
                            "brand": "A",
                            "shop": "S1",
                            "category": "水饮 > 茶",
                            "list_price_show": "10",
                        }
                    ],
                },
                {
                    "group": "饼干",
                    "sku_count": 2,
                    "skus": [
                        {
                            "brand": "B",
                            "shop": "S2",
                            "category": "休闲食品 > 饼干 > 粗粮饼干",
                            "list_price_show": "20",
                        },
                        {
                            "brand": "B",
                            "shop": "S2",
                            "category": "休闲食品 > 饼干 > 苏打饼干",
                            "list_price_show": "22",
                        },
                    ],
                },
            ],
            "consumer_feedback_by_matrix_group": [
                {
                    "group": "饮料",
                    "comment_rows": 5,
                    "effective_comment_text_units": 5,
                    "focus_keyword_hits": [{"word": "甜", "count": 2}],
                    "scenarios_top": [
                        {
                            "scenario": "解渴",
                            "count": 2,
                            "share_of_text_units": 0.4,
                        }
                    ],
                },
                {
                    "group": "饼干",
                    "comment_rows": 8,
                    "effective_comment_text_units": 8,
                    "focus_keyword_hits": [{"word": "脆", "count": 3}],
                    "scenarios_top": [
                        {
                            "scenario": "早餐",
                            "count": 4,
                            "share_of_text_units": 0.5,
                        }
                    ],
                },
            ],
            "usage_scenarios_by_matrix_group": [
                {"group": "饮料", "scenarios": []},
                {"group": "饼干", "scenarios": [{"scenario": "早餐", "count": 4}]},
            ],
            "notes": ["原注"],
        }

    def test_list_matrix_groups(self) -> None:
        b = self._sample_brief()
        m = list_matrix_groups_for_api(b)
        self.assertEqual(len(m), 2)
        self.assertEqual(m[1]["group"], "饼干")
        self.assertEqual(m[1]["index"], 1)

    def test_resolve_by_label(self) -> None:
        b = self._sample_brief()
        idx, err = resolve_strategy_matrix_group_index(
            b, matrix_group_label="饼干"
        )
        self.assertIsNone(err)
        self.assertEqual(idx, 1)

    def test_resolve_by_index(self) -> None:
        b = self._sample_brief()
        idx, err = resolve_strategy_matrix_group_index(
            b, matrix_group_index=1
        )
        self.assertIsNone(err)
        self.assertEqual(idx, 1)

    def test_resolve_mismatch(self) -> None:
        b = self._sample_brief()
        idx, err = resolve_strategy_matrix_group_index(
            b, matrix_group_index=0, matrix_group_label="饼干"
        )
        self.assertIsNotNone(err)
        self.assertIsNone(idx)

    def test_filter_keeps_only_group(self) -> None:
        b = self._sample_brief()
        out = filter_brief_for_strategy_matrix_group(b, matrix_group_index=1)
        self.assertEqual(len(out["matrix_by_group"]), 1)
        self.assertEqual(out["matrix_by_group"][0]["group"], "饼干")
        self.assertEqual(len(out["matrix_by_group"][0]["skus"]), 2)
        self.assertEqual(len(out["consumer_feedback_by_matrix_group"]), 1)
        self.assertEqual(out["comment_focus_keywords"][0]["word"], "脆")
        self.assertEqual(out["price_stats_source"], "strategy_scope_matrix_group_skus")
        self.assertIn("strategy_scope_applied", out)
        self.assertEqual(out["strategy_scope_applied"]["group"], "饼干")