from unittest.mock import patch import pytest import app def test_missing_channel_id_argument(): with patch("sys.argv", ["app.py"]): with pytest.raises(SystemExit): app.main() def test_correct_args_passed_to_update(): with patch("sys.argv", ["app.py", "UCD9aayaoRKSzV8QtUvq8ehw", "1"]): with patch("app.update_cms_account_for_channel") as mock_update: app.main() mock_update.assert_called_once_with("UCD9aayaoRKSzV8QtUvq8ehw", 1)