import CTF1
import demistomock as demisto
import pytest
@pytest.mark.parametrize(
"question_id, secret, expected",
[
("01", "5", "Well Done!!!"),
("01", "3", "Nope!!! Try again"), # Error MSG
("04", "reportsarecool", "Well Done!!!"),
("04", "none", "Nope!!! Try again"),
("05", "monkey", "Well Done!!!"),
("05", "elephant", "Nope!!! Try again"),
],
)
def test_main(mocker, question_id, secret, expected):
args = {"question_ID": question_id, "secret": secret}
m = mocker.patch.object(CTF1, "return_results")
mocker.patch.object(demisto, "args", return_value=args)
mocker.patch.object(demisto, "executeCommand", return_value={"total": "correct_secret"})
CTF1.main()
assert expected in m.call_args[0][0]["Contents"]