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

from pyrit.executor.attack import RTASystemPromptPaths
from pyrit.models import SeedPrompt


def test_system_prompt_from_file():
    strategy_path = RTASystemPromptPaths.TEXT_GENERATION.value
    with open(strategy_path, encoding="utf-8") as strategy_file:
        strategy = strategy_file.read()
        string_before_template = "value: |\n  "
        strategy_template = strategy[strategy.find(string_before_template) + len(string_before_template) :]
        strategy_template = strategy_template.replace("\n  ", "\n")
        strategy_template = strategy_template.rstrip()
        seed_prompt = SeedPrompt.from_yaml_file(strategy_path)
    assert strategy_template.replace("{{ objective }}", "my objective") == str(
        seed_prompt.render_template_value(objective="my objective")
    )