Add documentation and basic tests

tests/test_basic.py

@@ -0,0 +1,77 @@
+"""
+Basic tests for the Ubuntu Voice Typing application.
+"""
+
+import unittest
+from unittest.mock import MagicMock, patch
+
+from src.speech_recognition.recognition_manager import SpeechRecognitionManager, RecognitionState
+from src.text_injection.text_injector import TextInjector, DesktopEnvironment
+
+
+class TestSpeechRecognition(unittest.TestCase):
+    """Test cases for the speech recognition manager."""
+    
+    @patch('src.speech_recognition.recognition_manager.os.makedirs')
+    def test_init_state(self, mock_makedirs):
+        """Test the initial state of the speech recognition manager."""
+        manager = SpeechRecognitionManager(engine="vosk", model_size="small")
+        self.assertEqual(manager.state, RecognitionState.IDLE)
+        self.assertEqual(manager.engine, "vosk")
+        self.assertEqual(manager.model_size, "small")
+    
+    @patch('src.speech_recognition.recognition_manager.os.makedirs')
+    def test_callbacks(self, mock_makedirs):
+        """Test callback registration and invocation."""
+        manager = SpeechRecognitionManager(engine="vosk", model_size="small")
+        
+        # Mock callbacks
+        text_callback = MagicMock()
+        state_callback = MagicMock()
+        
+        # Register callbacks
+        manager.register_text_callback(text_callback)
+        manager.register_state_callback(state_callback)
+        
+        # Verify callbacks were registered
+        self.assertIn(text_callback, manager.text_callbacks)
+        self.assertIn(state_callback, manager.state_callbacks)
+        
+        # Test state update
+        manager._update_state(RecognitionState.LISTENING)
+        state_callback.assert_called_once_with(RecognitionState.LISTENING)
+
+
+class TestTextInjection(unittest.TestCase):
+    """Test cases for the text injector."""
+    
+    @patch('src.text_injection.text_injector.shutil.which')
+    @patch('src.text_injection.text_injector.TextInjector._detect_environment')
+    def test_init_x11(self, mock_detect, mock_which):
+        """Test initialization in X11 environment."""
+        mock_detect.return_value = DesktopEnvironment.X11
+        mock_which.return_value = '/usr/bin/xdotool'  # Simulate xdotool exists
+        
+        injector = TextInjector()
+        self.assertEqual(injector.environment, DesktopEnvironment.X11)
+    
+    @patch('src.text_injection.text_injector.subprocess.run')
+    @patch('src.text_injection.text_injector.shutil.which')
+    @patch('src.text_injection.text_injector.TextInjector._detect_environment')
+    def test_inject_text_x11(self, mock_detect, mock_which, mock_run):
+        """Test text injection in X11 environment."""
+        mock_detect.return_value = DesktopEnvironment.X11
+        mock_which.return_value = '/usr/bin/xdotool'
+        
+        injector = TextInjector()
+        injector.inject_text("Hello, world!")
+        
+        mock_run.assert_called_once()
+        args, kwargs = mock_run.call_args
+        self.assertEqual(args[0][0], "xdotool")
+        self.assertEqual(args[0][2], "--clearmodifiers")
+        self.assertEqual(args[0][3], "Hello, world!")
+
+
+if __name__ == '__main__':
+    unittest.main()