diff --git a/test/test_daemon_mode.py b/test/test_daemon_mode.py index 0f2ab30..db08d97 100644 --- a/test/test_daemon_mode.py +++ b/test/test_daemon_mode.py @@ -1,5 +1,6 @@ import unittest from unittest.mock import patch, MagicMock +import sys import aider_gitea.__main__ as main_module class TestDaemonMode(unittest.TestCase): @@ -23,7 +24,8 @@ class TestDaemonMode(unittest.TestCase): mock_db = MockSeenIssuesDB.return_value mock_db.has_seen.return_value = False - with patch('time.sleep', return_value=None): + with patch('sys.argv', ['main.py', '--gitea-url', 'http://example.com', '--owner', 'owner', '--repo', 'repo']), \ + patch('time.sleep', return_value=None): with self.assertRaises(SystemExit): main_module.main()