test_translate.py

import pytest
from unittest.mock import patch, MagicMock

from translate import translate_signal


# ---------- Hindi ----------

@patch("translate.translate_client")
def test_hindi_translation(mock_client):
    mock_client.translate_text.return_value = {
        "SourceLanguageCode": "hi",
        "TranslatedText": "There is a big pothole on the road"
    }

    result = translate_signal("सड़क पर बड़ा गड्ढा है")

    assert result["detected_language"] == "hi"
    assert result["translated_content"] == "There is a big pothole on the road"
    assert result["was_translated"] is True


# ---------- Tamil ----------

@patch("translate.translate_client")
def test_tamil_translation(mock_client):
    mock_client.translate_text.return_value = {
        "SourceLanguageCode": "ta",
        "TranslatedText": "There is a large pothole"
    }

    result = translate_signal("சாலையில் பெரிய குழி உள்ளது")

    assert result["detected_language"] == "ta"
    assert result["translated_content"] == "There is a large pothole"
    assert result["was_translated"] is True


# ---------- Telugu ----------

@patch("translate.translate_client")
def test_telugu_translation(mock_client):
    mock_client.translate_text.return_value = {
        "SourceLanguageCode": "te",
        "TranslatedText": "Big pothole on the road"
    }

    result = translate_signal("రోడ్డుపై పెద్ద గుంత ఉంది")

    assert result["detected_language"] == "te"
    assert result["translated_content"] == "Big pothole on the road"
    assert result["was_translated"] is True


# ---------- English (no translation) ----------

@patch("translate.translate_client")
def test_english_passthrough(mock_client):
    mock_client.translate_text.return_value = {
        "SourceLanguageCode": "en",
        "TranslatedText": "Huge pothole on MG Road"
    }

    result = translate_signal("Huge pothole on MG Road")

    assert result["detected_language"] == "en"
    assert result["translated_content"] == "Huge pothole on MG Road"
    assert result["was_translated"] is False


# ---------- Empty Content ----------

def test_empty_content_passthrough():
    result = translate_signal("")

    assert result["detected_language"] == "unknown"
    assert result["translated_content"] == ""
    assert result["was_translated"] is False