MicrosoftAtpStopAndQuarantineFile

from pytest_mock import MockerFixture


def test_main(mocker: MockerFixture):
    """
    Given:
        Command args.
    When:
        Calling `main`.
    Assert:
        Ensure `execute_polling_command` is called once with the correct command name and args.
    """
    from MicrosoftAtpStopAndQuarantineFile import main

    args = {"machine_id": "machineA", "file_hash": "TEST", "commnet": "TEST"}
    mocker.patch("MicrosoftAtpStopAndQuarantineFile.demisto.args", return_value=args)
    mock_execute_polling_command = mocker.patch("MicrosoftAtpStopAndQuarantineFile.execute_polling_command", return_value=[])

    main()

    assert mock_execute_polling_command.call_count == 1
    assert mock_execute_polling_command.call_args[0] == ("microsoft-atp-stop-and-quarantine-file", args)