# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.

import pytest

from pyrit.converter import AskToDecodeConverter, ConverterResult


async def test_ask_to_decode_converter_with_custom_template():
    """Test converter with a custom template."""
    template = "Decode this: {encoded_text}"
    converter = AskToDecodeConverter(template=template)
    prompt = "SGVsbG8gV29ybGQ="  # Base64 encoded "Hello World"

    result = await converter.convert_async(prompt=prompt, input_type="text")

    assert isinstance(result, ConverterResult)
    assert result.output_text == "Decode this: SGVsbG8gV29ybGQ="
    assert result.output_type == "text"


async def test_ask_to_decode_converter_with_encoding_name_and_template():
    """Test converter with encoding name and custom template."""
    template = "{encoding_name}: {encoded_text}"
    converter = AskToDecodeConverter(template=template, encoding_name="Base64")
    prompt = "SGVsbG8gV29ybGQ="

    result = await converter.convert_async(prompt=prompt, input_type="text")

    assert isinstance(result, ConverterResult)
    assert result.output_text == "Base64: SGVsbG8gV29ybGQ="
    assert result.output_type == "text"


async def test_ask_to_decode_converter_random_template():
    """Test converter with random template selection."""
    converter = AskToDecodeConverter()
    prompt = "test_encoded_text"

    result = await converter.convert_async(prompt=prompt, input_type="text")

    assert isinstance(result, ConverterResult)
    assert result.output_type == "text"
    # The result should contain the encoded text somewhere
    assert "test_encoded_text" in result.output_text


async def test_ask_to_decode_converter_random_template_with_encoding_name():
    """Test converter with random template and encoding name."""
    converter = AskToDecodeConverter(encoding_name="ROT13")
    prompt = "grfg"

    result = await converter.convert_async(prompt=prompt, input_type="text")

    assert isinstance(result, ConverterResult)
    assert result.output_type == "text"
    # The result should contain the encoded text
    assert "grfg" in result.output_text


async def test_ask_to_decode_converter_input_type_not_supported():
    """Test that non-text input types raise ValueError."""
    converter = AskToDecodeConverter()

    with pytest.raises(ValueError, match="Input type not supported"):
        await converter.convert_async(prompt="test", input_type="image_path")


def test_ask_to_decode_converter_input_supported():
    """Test input_supported method."""
    converter = AskToDecodeConverter()
    assert converter.input_supported("text") is True
    assert converter.input_supported("image_path") is False


def test_ask_to_decode_converter_output_supported():
    """Test output_supported method."""
    converter = AskToDecodeConverter()
    assert converter.output_supported("text") is True
    assert converter.output_supported("image_path") is False