CoreTerminateCausality

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 CoreTerminateCausality import main

    args = {"k": "v"}
    mocker.patch("CoreTerminateCausality.demisto.args", return_value=args)
    mock_execute_polling_command = mocker.patch("CoreTerminateCausality.execute_polling_command", return_value=[])

    main()

    assert mock_execute_polling_command.call_count == 1
    assert mock_execute_polling_command.call_args[0] == ("core-terminate-causality", args)