#!/usr/bin/env python3
"""
Weather Chatbot Application Tests
==================================

Author: RSK World (https://rskworld.in)
Year: 2026

Description: Unit tests for the Weather Chatbot Flask application
"""

import pytest
import os
import sys
from unittest.mock import patch, MagicMock

# Add parent directory to path
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))

from app import app, chatbot

@pytest.fixture
def client():
    """Create a test client for the Flask application."""
    app.config['TESTING'] = True
    app.config['SECRET_KEY'] = 'test-secret-key'
    with app.test_client() as client:
        yield client

def test_index_route(client):
    """Test the index route."""
    response = client.get('/')
    assert response.status_code == 200
    assert b'Weather Chatbot' in response.data

def test_health_check(client):
    """Test the health check endpoint."""
    response = client.get('/health')
    assert response.status_code == 200
    data = response.get_json()
    assert data['status'] == 'healthy'
    assert 'timestamp' in data

def test_chat_route_empty_message(client):
    """Test chat route with empty message."""
    response = client.post('/chat', data={'message': ''})
    assert response.status_code == 200
    data = response.get_json()
    assert 'error' in data

def test_chat_route_with_message(client):
    """Test chat route with a message."""
    with patch.object(chatbot, 'chat', return_value={'city': 'London', 'temperature': 15}):
        response = client.post('/chat', data={'message': 'What is the weather in London?'})
        assert response.status_code == 200
        data = response.get_json()
        assert 'city' in data or 'error' in data

def test_weather_route(client):
    """Test weather endpoint."""
    with patch.object(chatbot, 'get_weather_by_city', return_value={'city': 'London', 'temperature': 15}):
        response = client.get('/weather/London')
        assert response.status_code == 200
        data = response.get_json()
        assert 'city' in data or 'error' in data

def test_forecast_route(client):
    """Test forecast endpoint."""
    with patch.object(chatbot, 'get_forecast_by_city', return_value={'city': 'London', 'forecasts': []}):
        response = client.get('/forecast/London')
        assert response.status_code == 200
        data = response.get_json()
        assert 'city' in data or 'error' in data

def test_alerts_route(client):
    """Test alerts endpoint."""
    with patch.object(chatbot, 'get_weather_alerts', return_value={'city': 'London', 'alerts': []}):
        response = client.get('/alerts/London')
        assert response.status_code == 200
        data = response.get_json()
        assert 'city' in data or 'error' in data

def test_404_error(client):
    """Test 404 error handling."""
    response = client.get('/nonexistent')
    assert response.status_code == 404

def test_500_error_handling(client):
    """Test 500 error handling."""
    with patch.object(chatbot, 'chat', side_effect=Exception('Test error')):
        response = client.post('/chat', data={'message': 'test'})
        assert response.status_code == 200
        data = response.get_json()
        assert 'error' in data