Add file processing notifications feature

Co-authored-by: christianlouis <361235+christianlouis@users.noreply.github.com>

tests/test_notifications.py

@@ -0,0 +1,175 @@
+import os
+import unittest
+from unittest.mock import patch, MagicMock
+from app.utils.notification import notify_file_processed, send_notification
+from app.config import settings
+
+
+class TestFileProcessedNotification(unittest.TestCase):
+    """Test file processing notification functionality"""
+    
+    @patch('app.utils.notification.send_notification')
+    def test_notify_file_processed_when_enabled(self, mock_send):
+        """Test that notification is sent when NOTIFY_ON_FILE_PROCESSED is True"""
+        # Arrange
+        mock_send.return_value = True
+        original_value = settings.notify_on_file_processed
+        settings.notify_on_file_processed = True
+        
+        filename = "test_document.pdf"
+        file_size = 1024 * 1024  # 1 MB
+        metadata = {
+            'document_type': 'Invoice',
+            'tags': ['financial', 'urgent']
+        }
+        destinations = ['Dropbox', 'Google Drive']
+        
+        try:
+            # Act
+            result = notify_file_processed(filename, file_size, metadata, destinations)
+            
+            # Assert
+            self.assertTrue(result)
+            mock_send.assert_called_once()
+            
+            # Check the call arguments
+            call_args = mock_send.call_args
+            self.assertIn("test_document.pdf", call_args[1]['title'])
+            self.assertIn("Invoice", call_args[1]['message'])
+            self.assertIn("financial, urgent", call_args[1]['message'])
+            self.assertIn("Dropbox, Google Drive", call_args[1]['message'])
+            self.assertEqual(call_args[1]['notification_type'], "success")
+        finally:
+            settings.notify_on_file_processed = original_value
+    
+    @patch('app.utils.notification.send_notification')
+    def test_notify_file_processed_when_disabled(self, mock_send):
+        """Test that notification is not sent when NOTIFY_ON_FILE_PROCESSED is False"""
+        # Arrange
+        original_value = settings.notify_on_file_processed
+        settings.notify_on_file_processed = False
+        
+        filename = "test_document.pdf"
+        file_size = 1024 * 1024
+        metadata = {'document_type': 'Invoice', 'tags': []}
+        destinations = ['Dropbox']
+        
+        try:
+            # Act
+            result = notify_file_processed(filename, file_size, metadata, destinations)
+            
+            # Assert
+            self.assertFalse(result)
+            mock_send.assert_not_called()
+        finally:
+            settings.notify_on_file_processed = original_value
+    
+    @patch('app.utils.notification.send_notification')
+    def test_notify_file_processed_with_no_destinations(self, mock_send):
+        """Test notification when no destinations are configured"""
+        # Arrange
+        mock_send.return_value = True
+        original_value = settings.notify_on_file_processed
+        settings.notify_on_file_processed = True
+        
+        filename = "test_document.pdf"
+        file_size = 512 * 1024  # 512 KB
+        metadata = {
+            'document_type': 'Receipt',
+            'tags': []
+        }
+        destinations = []
+        
+        try:
+            # Act
+            result = notify_file_processed(filename, file_size, metadata, destinations)
+            
+            # Assert
+            self.assertTrue(result)
+            mock_send.assert_called_once()
+            
+            # Check that message indicates no destinations
+            call_args = mock_send.call_args
+            self.assertIn("None configured", call_args[1]['message'])
+        finally:
+            settings.notify_on_file_processed = original_value
+    
+    @patch('app.utils.notification.send_notification')
+    def test_notify_file_processed_formats_file_size_mb(self, mock_send):
+        """Test that file size is formatted correctly for MB"""
+        # Arrange
+        mock_send.return_value = True
+        original_value = settings.notify_on_file_processed
+        settings.notify_on_file_processed = True
+        
+        filename = "large_document.pdf"
+        file_size = 5 * 1024 * 1024  # 5 MB
+        metadata = {'document_type': 'Contract', 'tags': []}
+        destinations = ['Dropbox']
+        
+        try:
+            # Act
+            result = notify_file_processed(filename, file_size, metadata, destinations)
+            
+            # Assert
+            self.assertTrue(result)
+            call_args = mock_send.call_args
+            self.assertIn("5.00 MB", call_args[1]['message'])
+        finally:
+            settings.notify_on_file_processed = original_value
+    
+    @patch('app.utils.notification.send_notification')
+    def test_notify_file_processed_formats_file_size_kb(self, mock_send):
+        """Test that file size is formatted correctly for KB"""
+        # Arrange
+        mock_send.return_value = True
+        original_value = settings.notify_on_file_processed
+        settings.notify_on_file_processed = True
+        
+        filename = "small_document.pdf"
+        file_size = 512 * 1024  # 512 KB
+        metadata = {'document_type': 'Note', 'tags': []}
+        destinations = ['Dropbox']
+        
+        try:
+            # Act
+            result = notify_file_processed(filename, file_size, metadata, destinations)
+            
+            # Assert
+            self.assertTrue(result)
+            call_args = mock_send.call_args
+            self.assertIn("512.00 KB", call_args[1]['message'])
+        finally:
+            settings.notify_on_file_processed = original_value
+    
+    @patch('app.utils.notification.send_notification')
+    def test_notify_file_processed_with_missing_metadata_fields(self, mock_send):
+        """Test that notification handles missing metadata fields gracefully"""
+        # Arrange
+        mock_send.return_value = True
+        original_value = settings.notify_on_file_processed
+        settings.notify_on_file_processed = True
+        
+        filename = "document.pdf"
+        file_size = 1024 * 1024
+        metadata = {}  # Empty metadata
+        destinations = ['Dropbox']
+        
+        try:
+            # Act
+            result = notify_file_processed(filename, file_size, metadata, destinations)
+            
+            # Assert
+            self.assertTrue(result)
+            mock_send.assert_called_once()
+            
+            # Check that defaults are used
+            call_args = mock_send.call_args
+            self.assertIn("Unknown", call_args[1]['message'])  # Default document type
+            self.assertIn("None", call_args[1]['message'])  # No tags
+        finally:
+            settings.notify_on_file_processed = original_value
+
+
+if __name__ == '__main__':
+    unittest.main()