from __future__ import annotations

from app.exercises.dead_bug.rules import detect_diagonal_extension, has_required_visibility, is_ready_position
from app.exercises.dead_bug.types import DeadBugMetrics, Point

class TestDeadBugRules:
    """死虫式规则函数单元测试"""

    def _make_landmark(self, x=0.5, y=0.5, z=0.0, visibility=1.0):
        """创建测试用Point对象"""
        return Point(x, y, z, visibility)

    def _make_visible_landmarks(self):
        """创建33个全可见的测试用关键点"""
        return [self._make_landmark() for _ in range(33)]

    def test_has_required_visibility_all_visible(self):
        """测试：所有关键点可见时应返回True"""
        lm = self._make_visible_landmarks()
        indices = (11, 12, 13, 14, 15, 16, 23, 24, 25, 26, 27, 28)
        assert has_required_visibility(lm, indices, 0.45)

    def test_has_required_visibility_low(self):
        """测试：关键点可见度过低时应返回False"""
        lm = self._make_visible_landmarks()
        lm[11] = self._make_landmark(visibility=0.1)
        indices = (11, 12, 13, 14, 15, 16, 23, 24, 25, 26, 27, 28)
        assert not has_required_visibility(lm, indices, 0.45)

    def test_detect_diagonal_extension_none(self):
        """测试：四肢均未伸展时应返回None"""
        metrics = DeadBugMetrics(
            left_arm_extended=False, right_arm_extended=False,
            left_leg_extended=False, right_leg_extended=False,
            left_elbow_angle=90, right_elbow_angle=90,
            left_knee_angle=90, right_knee_angle=90,
            feedback=[],
        )
        assert detect_diagonal_extension(metrics) is None

    def test_detect_diagonal_extension_left_arm_right_leg(self):
        """测试：左臂+右腿对角伸展检测"""
        metrics = DeadBugMetrics(
            left_arm_extended=True, right_arm_extended=False,
            left_leg_extended=False, right_leg_extended=True,
            left_elbow_angle=160, right_elbow_angle=90,
            left_knee_angle=90, right_knee_angle=160,
            feedback=[],
        )
        assert detect_diagonal_extension(metrics) == "left_arm_right_leg"

    def test_detect_diagonal_extension_allows_ready_arm_overlap(self):
        """测试：准备位双臂上举时，单腿对侧伸展仍应识别"""
        metrics = DeadBugMetrics(
            left_arm_extended=True, right_arm_extended=True,
            left_leg_extended=False, right_leg_extended=True,
            left_elbow_angle=160, right_elbow_angle=160,
            left_knee_angle=90, right_knee_angle=160,
            feedback=[],
        )
        assert detect_diagonal_extension(metrics) == "left_arm_right_leg"

    def test_detect_diagonal_extension_rejects_both_legs(self):
        """测试：双腿同时伸展不应识别为可计数对角伸展"""
        metrics = DeadBugMetrics(
            left_arm_extended=True, right_arm_extended=True,
            left_leg_extended=True, right_leg_extended=True,
            left_elbow_angle=160, right_elbow_angle=160,
            left_knee_angle=160, right_knee_angle=160,
            feedback=[],
        )
        assert detect_diagonal_extension(metrics) is None

    def test_is_ready_position(self):
        """测试：膝盖弯曲且四肢收缩应识别为准备姿态"""
        metrics = DeadBugMetrics(
            left_arm_extended=False, right_arm_extended=False,
            left_leg_extended=False, right_leg_extended=False,
            left_elbow_angle=90, right_elbow_angle=90,
            left_knee_angle=100, right_knee_angle=100,
            feedback=[],
        )
        assert is_ready_position(metrics)

    def test_is_ready_allows_arms_extended(self):
        """测试：dead bug 准备位允许双臂上举"""
        metrics = DeadBugMetrics(
            left_arm_extended=True, right_arm_extended=True,
            left_leg_extended=False, right_leg_extended=False,
            left_elbow_angle=160, right_elbow_angle=160,
            left_knee_angle=100, right_knee_angle=100,
            feedback=[],
        )
        assert is_ready_position(metrics)

    def test_is_not_ready_legs_extended(self):
        """测试：腿部伸展时不识别为准备姿态"""
        metrics = DeadBugMetrics(
            left_arm_extended=False, right_arm_extended=False,
            left_leg_extended=True, right_leg_extended=False,
            left_elbow_angle=90, right_elbow_angle=90,
            left_knee_angle=100, right_knee_angle=100,
            feedback=[],
        )
        assert not is_ready_position(metrics)