def test_list_commands() -> None: """Test commands that need a sub chosen from a list.""" conf = MagicMock() conf.details = MagicMock() conf.download_queue = MagicMock() conf.enqueue = MagicMock() conf.mark = MagicMock() conf.unmark = MagicMock() # Test a command with a mocked sub choice. main._choose_sub = MagicMock(return_value=1) main._handle_command("download_queue", conf) conf.download_queue.assert_called_once_with(1) main._choose_sub = MagicMock(return_value=10000) main._handle_command("download_queue", conf) conf.download_queue.assert_called_with(10000) # Enqueue. main._sub_list_command_wrapper = MagicMock(return_value=(1, [1])) main._handle_command("enqueue", conf) conf.enqueue.assert_called_once_with(1, [1]) # Mark. main._sub_list_command_wrapper = MagicMock(return_value=(1, [1])) main._handle_command("mark", conf) conf.mark.assert_called_once_with(1, [1]) # Unmark. main._sub_list_command_wrapper = MagicMock(return_value=(1, [1])) main._handle_command("unmark", conf) conf.unmark.assert_called_once_with(1, [1])
def test_simple_commands(): """Test simple commands.""" config = Config.Config.__new__(Config.Config) # Mock config commands. config.update = MagicMock() config.list = MagicMock() # Test simple commands. main._handle_command("update", config, None, None) main._handle_command("list", config, None, None) config.update.assert_called_once_with() config.list.assert_called_once_with()
def test_simple_commands() -> None: """Test simple commands.""" conf = MagicMock() # Mock config commands. conf.update = MagicMock() conf.list = MagicMock() # Test simple commands. main._handle_command("update", conf) main._handle_command("list", conf) main._handle_command("reload_config", conf) conf.update.assert_called_once_with() conf.list.assert_called_once_with() conf.reload_config.assert_called_once_with()
def test_list_commands(): """Test commands that need a sub chosen from a list.""" config = Config.Config.__new__(Config.Config) config.details = MagicMock() config.download_queue = MagicMock() config.enqueue = MagicMock() config.mark = MagicMock() config.unmark = MagicMock() # Test a command with a mocked sub choice. main._choose_sub = MagicMock(return_value=1) main._handle_command("download_queue", config, None, None) config.download_queue.assert_called_once_with(1) main._choose_sub = MagicMock(return_value=10000) main._handle_command("download_queue", config, None, None) config.download_queue.assert_called_with(10000) # Enqueue. main._sub_list_command_wrapper = MagicMock(return_value=(1, [1])) main._handle_command("enqueue", config, None, None) config.enqueue.assert_called_once_with(1, [1]) # Mark. main._sub_list_command_wrapper = MagicMock(return_value=(1, [1])) main._handle_command("mark", config, None, None) config.mark.assert_called_once_with(1, [1]) # Unmark. main._sub_list_command_wrapper = MagicMock(return_value=(1, [1])) main._handle_command("unmark", config, None, None) config.unmark.assert_called_once_with(1, [1]) log = MagicMock() log.error = MagicMock() log.error.assert_not_called() fake_command_options = [{"return": "foo", "prompt": "bar"}, {"return": "foo", "prompt": "bar"}, {"return": "foo", "prompt": "bar"}, {"return": "foo", "prompt": "bar"}] main._handle_command("adsf", None, fake_command_options, log) log.error.call_count = 6