# -*- coding: utf-8 -*-
"""Tests for HeartbeatHook."""
import tempfile
from pathlib import Path

import pytest

from backend.agents.base.hooks import HeartbeatHook


class TestHeartbeatHook:
    """Tests for HeartbeatHook._read_heartbeat_content."""

    def test_read_heartbeat_content_with_content(self, tmp_path):
        """Test reading HEARTBEAT.md when it exists and has content."""
        ws_dir = tmp_path / "analyst_workspace"
        ws_dir.mkdir()
        hb_file = ws_dir / "HEARTBEAT.md"
        hb_file.write_text("# 定期主动检查\n\n- [ ] 持仓是否健康\n", encoding="utf-8")

        hook = HeartbeatHook(workspace_dir=ws_dir)
        content = hook._read_heartbeat_content()

        assert content is not None
        assert "# 定期主动检查" in content
        assert "持仓是否健康" in content

    def test_read_heartbeat_content_absent(self, tmp_path):
        """Test reading when HEARTBEAT.md does not exist."""
        ws_dir = tmp_path / "analyst_workspace"
        ws_dir.mkdir()

        hook = HeartbeatHook(workspace_dir=ws_dir)
        content = hook._read_heartbeat_content()

        assert content is None

    def test_read_heartbeat_content_empty(self, tmp_path):
        """Test reading when HEARTBEAT.md is empty."""
        ws_dir = tmp_path / "analyst_workspace"
        ws_dir.mkdir()
        hb_file = ws_dir / "HEARTBEAT.md"
        hb_file.write_text("", encoding="utf-8")

        hook = HeartbeatHook(workspace_dir=ws_dir)
        content = hook._read_heartbeat_content()

        assert content is None

    def test_read_heartbeat_content_whitespace_only(self, tmp_path):
        """Test reading when HEARTBEAT.md contains only whitespace."""
        ws_dir = tmp_path / "analyst_workspace"
        ws_dir.mkdir()
        hb_file = ws_dir / "HEARTBEAT.md"
        hb_file.write_text("   \n\n  ", encoding="utf-8")

        hook = HeartbeatHook(workspace_dir=ws_dir)
        content = hook._read_heartbeat_content()

        assert content is None

    def test_completed_flag_path(self, tmp_path):
        """Test that completion flag is placed in workspace directory."""
        ws_dir = tmp_path / "analyst_workspace"
        ws_dir.mkdir()

        hook = HeartbeatHook(workspace_dir=ws_dir)

        assert hook._completed_flag == ws_dir / ".heartbeat_completed"