from app.clients.modelhub import ModelHubClient
from app.settings import Settings


def test_strategy_id_payload_field_configurable():
    settings = Settings(
        auth_token="a", hf_token="h", contest_api_token="c", email="e", user_id="1",
        contributors="u", strategy_id="sid", contest_task_strategy_field="strategy_id"
    )
    payload = ModelHubClient(settings).build_contest_payload("owner/model", "hygon_k100-ai", "cfg")
    assert payload["strategy_id"] == "sid"
    assert payload["taskType"] == "text-generation"


def test_strategy_id_payload_can_disable_injection():
    settings = Settings(
        auth_token="a", hf_token="h", contest_api_token="c", email="e", user_id="1",
        contributors="u", strategy_id="sid", inject_strategy_id=False
    )
    payload = ModelHubClient(settings).build_contest_payload("owner/model", "hygon_k100-ai", "cfg")
    assert "strategyId" not in payload