init

tests/test_email_manager.py

@@ -0,0 +1,108 @@
+import sys
+from pathlib import Path
+
+# Add api directory to path
+api_dir = Path(__file__).parent.parent / 'api'
+sys.path.insert(0, str(api_dir))
+import unittest
+import tempfile
+import shutil
+import os
+from unittest.mock import patch, MagicMock
+from email_manager import EmailManager
+
+class TestEmailManager(unittest.TestCase):
+    def setUp(self):
+        self.test_dir = tempfile.mkdtemp()
+        self.data_dir = os.path.join(self.test_dir, 'data')
+        self.config_dir = os.path.join(self.test_dir, 'config')
+        os.makedirs(self.data_dir, exist_ok=True)
+        os.makedirs(self.config_dir, exist_ok=True)
+        self.manager = EmailManager(data_dir=self.data_dir, config_dir=self.config_dir)
+
+    def tearDown(self):
+        shutil.rmtree(self.test_dir)
+
+    def test_initialization(self):
+        self.assertTrue(os.path.exists(self.manager.email_dir))
+        self.assertTrue(os.path.exists(self.manager.postfix_dir))
+        self.assertTrue(os.path.exists(self.manager.dovecot_dir))
+
+    @patch.object(EmailManager, '_reload_email_services', return_value=True)
+    def test_create_and_delete_email_user(self, mock_reload):
+        result = self.manager.create_email_user('testuser', 'testdomain', 'password')
+        self.assertTrue(result)
+        result = self.manager.delete_email_user('testuser', 'testdomain')
+        self.assertTrue(result)
+
+    def test_list_email_users_empty(self):
+        users = self.manager.list_email_users()
+        self.assertIsInstance(users, list)
+        self.assertEqual(len(users), 0)
+
+    @patch('smtplib.SMTP')
+    def test_send_email(self, mock_smtp):
+        instance = mock_smtp.return_value.__enter__.return_value
+        instance.sendmail.return_value = {}
+        result = self.manager.send_email('from@cell', 'to@cell', 'Subject', 'Body')
+        self.assertTrue(result)
+        # Simulate error by raising in SMTP constructor
+        mock_smtp.side_effect = Exception('SMTP error')
+        result = self.manager.send_email('from@cell', 'to@cell', 'Subject', 'Body')
+        self.assertFalse(result)
+
+    @patch('imaplib.IMAP4_SSL')
+    def test_get_mailbox_info(self, mock_imap):
+        instance = mock_imap.return_value
+        instance.login.return_value = 'OK'
+        instance.select.return_value = ('OK', [b'1'])
+        instance.search.return_value = ('OK', [b'1 2 3'])
+        instance.fetch.return_value = ('OK', [(b'1', b'RFC822')])
+        info = self.manager.get_mailbox_info('testuser', 'testdomain')
+        self.assertIsInstance(info, dict)
+        instance.login.side_effect = Exception('IMAP error')
+        info = self.manager.get_mailbox_info('testuser', 'testdomain')
+        self.assertIn('error', info)
+
+    @patch('subprocess.run')
+    def test_get_email_status(self, mock_run):
+        mock_run.return_value.stdout = 'cell-mail\n'
+        mock_run.return_value.returncode = 0
+        status = self.manager.get_email_status()
+        self.assertIsInstance(status, dict)
+        self.assertIn('postfix_running', status)
+        self.assertIn('dovecot_running', status)
+
+    @patch('requests.get')
+    def test_test_email_connectivity(self, mock_get):
+        mock_get.return_value.status_code = 200
+        result = self.manager.test_email_connectivity()
+        self.assertIsInstance(result, dict)
+        mock_get.side_effect = Exception('HTTP error')
+        result = self.manager.test_email_connectivity()
+        self.assertIn('smtp', result)
+
+    @patch('subprocess.run')
+    def test_get_email_logs(self, mock_run):
+        mock_run.return_value.stdout = 'log line\n'
+        mock_run.return_value.returncode = 0
+        logs = self.manager.get_email_logs('all', 10)
+        self.assertIsInstance(logs, dict)
+        self.assertIn('postfix', logs)
+        self.assertIn('dovecot', logs)
+
+    def test_get_status(self):
+        status = self.manager.get_status()
+        self.assertIsInstance(status, dict)
+        self.assertIn('status', status)
+
+    def test_error_handling(self):
+        # Force errors by passing invalid arguments, should return False or error dict
+        self.assertFalse(self.manager.create_email_user(None, None, None))
+        self.assertFalse(self.manager.delete_email_user(None, None))
+        self.assertFalse(self.manager.send_email(None, None, None, None))
+        info = self.manager.get_mailbox_info(None, None)
+        self.assertIn('error', info)
+
+if __name__ == '__main__':
+    unittest.main()