import CTF2BF
import demistomock as demisto
import pytest


@pytest.mark.parametrize(
    "question_id, secret, expected",
    [
        ("01", "no", "Well Done!!!"),
        ("01", "yes", "Nope!!! Try again"),  # Error MSG
        ("04", "2017", "Well Done!!!"),
        ("04", "1990", "Nope!!! Try again"),
        ("07", "blocked", "Well Done!!!"),
        ("07", "done", "Nope!!! Try again"),
    ],
)
def test_main(mocker, question_id, secret, expected):
    args = {"question_ID": question_id, "secret": secret}

    m = mocker.patch.object(CTF2BF, "return_results")
    mocker.patch.object(demisto, "args", return_value=args)
    mocker.patch.object(demisto, "executeCommand", return_value={"total": "correct_secret"})

    CTF2BF.main()
    assert expected in m.call_args[0][0]["Contents"]