Jatin K MalikandGitHub
642be732db
* Improved test coverage for audio_feedback, main, and tray_indicator modules * Fix test suite and improve code stability * Black formatter changes * Fixing import sorting
128 lines
4.8 KiB
Python
128 lines
4.8 KiB
Python
"""
|
|
Tests for the main module.
|
|
"""
|
|
|
|
import logging
|
|
import unittest
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from src.main import main, parse_arguments
|
|
|
|
|
|
class TestMainModule(unittest.TestCase):
|
|
"""Test cases for the main module."""
|
|
|
|
def test_parse_arguments_defaults(self):
|
|
"""Test argument parsing with defaults."""
|
|
# Test with no arguments
|
|
with patch("sys.argv", ["vocalinux"]):
|
|
args = parse_arguments()
|
|
self.assertFalse(args.debug)
|
|
self.assertEqual(args.model, "small")
|
|
self.assertEqual(args.engine, "vosk")
|
|
self.assertFalse(args.wayland)
|
|
|
|
def test_parse_arguments_custom(self):
|
|
"""Test argument parsing with custom values."""
|
|
# Test with custom arguments
|
|
with patch(
|
|
"sys.argv",
|
|
[
|
|
"vocalinux",
|
|
"--debug",
|
|
"--model",
|
|
"large",
|
|
"--engine",
|
|
"whisper",
|
|
"--wayland",
|
|
],
|
|
):
|
|
args = parse_arguments()
|
|
self.assertTrue(args.debug)
|
|
self.assertEqual(args.model, "large")
|
|
self.assertEqual(args.engine, "whisper")
|
|
self.assertTrue(args.wayland)
|
|
|
|
@patch("src.main.recognition_manager.SpeechRecognitionManager")
|
|
@patch("src.main.text_injector.TextInjector")
|
|
@patch("src.main.tray_indicator.TrayIndicator")
|
|
@patch("src.main.logging")
|
|
def test_main_initializes_components(
|
|
self, mock_logging, mock_tray, mock_text, mock_speech
|
|
):
|
|
"""Test that main initializes all the required components."""
|
|
# Mock objects
|
|
mock_speech_instance = MagicMock()
|
|
mock_text_instance = MagicMock()
|
|
mock_tray_instance = MagicMock()
|
|
|
|
# Setup return values
|
|
mock_speech.return_value = mock_speech_instance
|
|
mock_text.return_value = mock_text_instance
|
|
mock_tray.return_value = mock_tray_instance
|
|
|
|
# Mock the arguments
|
|
with patch("src.main.parse_arguments") as mock_parse:
|
|
mock_args = MagicMock()
|
|
mock_args.debug = False
|
|
mock_args.model = "medium"
|
|
mock_args.engine = "vosk"
|
|
mock_args.wayland = True
|
|
mock_parse.return_value = mock_args
|
|
|
|
# Call main function
|
|
main()
|
|
|
|
# Verify components were initialized correctly
|
|
mock_speech.assert_called_once_with(engine="vosk", model_size="medium")
|
|
mock_text.assert_called_once_with(wayland_mode=True)
|
|
mock_tray.assert_called_once_with(
|
|
speech_engine=mock_speech_instance, text_injector=mock_text_instance
|
|
)
|
|
|
|
# Verify callbacks were registered
|
|
mock_speech_instance.register_text_callback.assert_called_once_with(
|
|
mock_text_instance.inject_text
|
|
)
|
|
|
|
# Verify the tray indicator was started
|
|
mock_tray_instance.run.assert_called_once()
|
|
|
|
def test_main_with_debug_enabled(self):
|
|
"""Test that debug mode enables debug logging."""
|
|
# Test a simpler approach - just check that when args.debug is True,
|
|
# we call setLevel with DEBUG
|
|
with patch("src.main.parse_arguments") as mock_parse:
|
|
mock_args = MagicMock()
|
|
mock_args.debug = True
|
|
mock_args.model = "small"
|
|
mock_args.engine = "vosk"
|
|
mock_args.wayland = False
|
|
mock_parse.return_value = mock_args
|
|
|
|
# Skip actually running main to avoid complex mocking
|
|
# Just verify that the debug flag works correctly
|
|
with patch("src.main.logging.getLogger") as mock_get_logger:
|
|
with patch("src.main.logging.DEBUG", logging.DEBUG):
|
|
# Need to mock the rest of the components to avoid actual execution
|
|
with patch("src.main.recognition_manager.SpeechRecognitionManager"):
|
|
with patch("src.main.text_injector.TextInjector"):
|
|
with patch("src.main.tray_indicator.TrayIndicator"):
|
|
# These mock loggers will be returned by successive calls to getLogger()
|
|
root_logger = MagicMock()
|
|
named_logger = MagicMock()
|
|
mock_get_logger.side_effect = [
|
|
root_logger,
|
|
named_logger,
|
|
]
|
|
|
|
# Call main
|
|
main()
|
|
|
|
# Verify root logger had setLevel called with DEBUG
|
|
root_logger.setLevel.assert_called_once_with(
|
|
logging.DEBUG
|
|
)
|