from unittest.mock import patch

import pytest

from app import main


@patch("app.clear_delivery_history_isrcs")
def test_missing_partner_id_arg_exits_with_error(mock_clear_delivery_history_isrcs):
    with patch("sys.argv", ["app.py"]):
        with pytest.raises(SystemExit):
            main()
    assert not mock_clear_delivery_history_isrcs.called


@patch("app.clear_delivery_history_isrcs")
def test_clear_delivery_history_isrcs_called_with_correct_args(mock_clear_delivery_history_isrcs):
    with patch("sys.argv", ["app.py", "1", "abc", "def"]):
        main()
    mock_clear_delivery_history_isrcs.assert_called_once_with(1, ["abc", "def"])