src/tests/test_helper_sent.py

@@ -0,0 +1,76 @@
+"""Test cases for helper_sent class"""
+
+import unittest
+from pybitmessage.helper_sent import insert, delete, trash, retrieve_message_details
+
+try:
+    # Python 3
+    from unittest.mock import patch
+except ImportError:
+    # Python 2
+    from mock import patch
+
+
+class TestHelperSent(unittest.TestCase):
+    """Test class for helper_sent"""
+
+    @patch("pybitmessage.helper_sent.sqlExecute")
+    def test_insert_valid_address(self, mock_sql_execute):
+        """Test insert with valid address"""
+        VALID_ADDRESS = "BM-2cUGaEcGz9Zft1SPAo8FJtfzyADTpEgU9U"
+        ackdata = insert(
+            msgid="123456",
+            toAddress="[Broadcast subscribers]",
+            fromAddress=VALID_ADDRESS,
+            subject="Test Subject",
+            message="Test Message",
+            status="msgqueued",
+            sentTime=1234567890,
+            lastActionTime=1234567890,
+            sleeptill=0,
+            retryNumber=0,
+            encoding=2,
+            ttl=3600,
+            folder="sent",
+        )
+        mock_sql_execute.assert_called_once()
+        self.assertIsNotNone(ackdata)
+
+    def test_insert_invalid_address(self):
+        """Test insert with invalid address"""
+        INVALID_ADDRESS = "TEST@1245.780"
+        ackdata = insert(toAddress=INVALID_ADDRESS)
+        self.assertIsNone(ackdata)
+
+    @patch("pybitmessage.helper_sent.sqlExecute")
+    def test_delete(self, mock_sql_execute):
+        """Test delete function"""
+        delete("ack_data")
+        self.assertTrue(mock_sql_execute.called)
+        mock_sql_execute.assert_called_once_with(
+            "DELETE FROM sent WHERE ackdata = ?", "ack_data"
+        )
+
+    @patch("pybitmessage.helper_sent.sqlQuery")
+    def test_retrieve_valid_message_details(self, mock_sql_query):
+        """Test retrieving valid message details"""
+        return_data = [
+            (
+                "to@example.com",
+                "from@example.com",
+                "Test Subject",
+                "Test Message",
+                "2022-01-01",
+            )
+        ]
+        mock_sql_query.return_value = return_data
+        result = retrieve_message_details("12345")
+        self.assertEqual(result, return_data)
+
+    @patch("pybitmessage.helper_sent.sqlExecute")
+    def test_trash(self, mock_sql_execute):
+        """Test marking a message as 'trash'"""
+        ackdata = "ack_data"
+        mock_sql_execute.return_value = 1
+        rowcount = trash(ackdata)
+        self.assertEqual(rowcount, 1)