test_intelligent_replies.py

"""
Test script for intelligent reply functionality
"""

import asyncio
import sys
from pathlib import Path

# Add project root to path
sys.path.insert(0, str(Path(__file__).parent))

from app.ai.agent.reply_intelligence import (
    determine_reply_to_message,
    build_reply_context_for_response
)


def test_determine_reply_to_message():
    """Test the reply message determination logic"""
    print("\n" + "="*60)
    print("TEST 1: Simple conversation - should quote latest user message")
    print("="*60)

    conversation_history = [
        {
            "id": "msg-003",
            "sender_id": "user-123",
            "sender_name": "John Doe",
            "content": "I need 2 bedrooms",
            "is_ai": False,
            "timestamp": "2026-02-15T10:03:00Z"
        },
        {
            "id": "msg-002",
            "sender_id": "aida",
            "sender_name": "AIDA",
            "content": "Hello! How can I help?",
            "is_ai": True,
            "timestamp": "2026-02-15T10:02:00Z"
        },
        {
            "id": "msg-001",
            "sender_id": "user-123",
            "sender_name": "John Doe",
            "content": "Hi",
            "is_ai": False,
            "timestamp": "2026-02-15T10:01:00Z"
        }
    ]

    reply_to_id = determine_reply_to_message(conversation_history)

    print(f"\n✅ Result: reply_to_id = {reply_to_id}")
    print(f"Expected: msg-003 (latest user message)")
    assert reply_to_id == "msg-003", f"Expected msg-003, got {reply_to_id}"
    print("✅ TEST PASSED")


def test_build_reply_context():
    """Test the reply context building logic"""
    print("\n" + "="*60)
    print("TEST 2: Build reply context for response")
    print("="*60)

    conversation_history = [
        {
            "id": "msg-003",
            "sender_id": "user-123",
            "sender_name": "John Doe",
            "content": "I need 2 bedrooms in Cotonou",
            "is_ai": False,
            "timestamp": "2026-02-15T10:03:00Z"
        },
        {
            "id": "msg-002",
            "sender_id": "aida",
            "sender_name": "AIDA",
            "content": "Hello! How can I help?",
            "is_ai": True,
            "timestamp": "2026-02-15T10:02:00Z"
        }
    ]

    reply_context = build_reply_context_for_response(
        conversation_history=conversation_history,
        reply_to_id="msg-003"
    )

    print(f"\n✅ Result:")
    print(f"  - id: {reply_context['id']}")
    print(f"  - sender_name: {reply_context['sender_name']}")
    print(f"  - content: {reply_context['content']}")

    assert reply_context['id'] == "msg-003"
    assert reply_context['sender_name'] == "John Doe"
    assert reply_context['content'] == "I need 2 bedrooms in Cotonou"
    print("\n✅ TEST PASSED")


def test_empty_conversation():
    """Test with empty conversation history"""
    print("\n" + "="*60)
    print("TEST 3: Empty conversation - should return None")
    print("="*60)

    reply_to_id = determine_reply_to_message([])

    print(f"\n✅ Result: reply_to_id = {reply_to_id}")
    print(f"Expected: None")
    assert reply_to_id is None
    print("✅ TEST PASSED")


def test_only_ai_messages():
    """Test with only AI messages - should return None"""
    print("\n" + "="*60)
    print("TEST 4: Only AI messages - should return None")
    print("="*60)

    conversation_history = [
        {
            "id": "msg-002",
            "sender_id": "aida",
            "sender_name": "AIDA",
            "content": "Hello!",
            "is_ai": True,
            "timestamp": "2026-02-15T10:02:00Z"
        },
        {
            "id": "msg-001",
            "sender_id": "aida",
            "sender_name": "AIDA",
            "content": "How can I help?",
            "is_ai": True,
            "timestamp": "2026-02-15T10:01:00Z"
        }
    ]

    reply_to_id = determine_reply_to_message(conversation_history)

    print(f"\n✅ Result: reply_to_id = {reply_to_id}")
    print(f"Expected: None")
    assert reply_to_id is None
    print("✅ TEST PASSED")


def test_with_reply_context():
    """Test when user swipes to reply to a specific message"""
    print("\n" + "="*60)
    print("TEST 5: User swipes to reply - should still quote latest user message")
    print("="*60)

    conversation_history = [
        {
            "id": "msg-005",
            "sender_id": "user-123",
            "sender_name": "John Doe",
            "content": "I need 2 bedrooms",
            "is_ai": False,
            "timestamp": "2026-02-15T10:05:00Z"
        },
        {
            "id": "msg-004",
            "sender_id": "aida",
            "sender_name": "AIDA",
            "content": "I can help with that!",
            "is_ai": True,
            "timestamp": "2026-02-15T10:04:00Z"
        },
        {
            "id": "msg-003",
            "sender_id": "user-123",
            "sender_name": "John Doe",
            "content": "Show me apartments",
            "is_ai": False,
            "timestamp": "2026-02-15T10:03:00Z"
        },
        {
            "id": "msg-002",
            "sender_id": "aida",
            "sender_name": "AIDA",
            "content": "Hello! How can I help?",
            "is_ai": True,
            "timestamp": "2026-02-15T10:02:00Z"
        },
        {
            "id": "msg-001",
            "sender_id": "user-123",
            "sender_name": "John Doe",
            "content": "Hi",
            "is_ai": False,
            "timestamp": "2026-02-15T10:01:00Z"
        }
    ]

    # User swiped "Hi" but the latest user message is "I need 2 bedrooms"
    reply_context = {"message_id": "msg-001", "content": "Hi"}

    reply_to_id = determine_reply_to_message(
        conversation_history=conversation_history,
        reply_context=reply_context
    )

    print(f"\n✅ Result: reply_to_id = {reply_to_id}")
    print(f"Expected: msg-005 (latest user message, NOT msg-001 which was swiped)")
    assert reply_to_id == "msg-005", f"Expected msg-005, got {reply_to_id}"
    print("✅ TEST PASSED")


def main():
    """Run all tests"""
    print("\n" + "="*60)
    print("INTELLIGENT REPLY TESTS")
    print("="*60)

    try:
        test_determine_reply_to_message()
        test_build_reply_context()
        test_empty_conversation()
        test_only_ai_messages()
        test_with_reply_context()

        print("\n" + "="*60)
        print("✅ ALL TESTS PASSED!")
        print("="*60 + "\n")

    except AssertionError as e:
        print(f"\n❌ TEST FAILED: {e}\n")
        sys.exit(1)
    except Exception as e:
        print(f"\n❌ ERROR: {e}\n")
        import traceback
        traceback.print_exc()
        sys.exit(1)


if __name__ == "__main__":
    main()