show pk in str

wooyek · May 7, 2018 · 29b72a3 · 29b72a3
1 parent 204181f
commit 29b72a3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/django_email_queue/models.py b/src/django_email_queue/models.py
@@ -58,7 +58,7 @@ class QueuedEmailMessage(models.Model):
     # TODO: attachments, headers, alternatives
 
     def __str__(self):
-        return u"{}:{}:{}".format(self.__class__.__name__, self.subject, self.to)
+        return u"{}:{}:{}:{}".format(self.__class__.__name__, self.pk, self.subject, self.to)
 
     @classmethod
     def _create(cls, email_message, commit=True):

diff --git a/tests/test_backend.py b/tests/test_backend.py
@@ -123,7 +123,11 @@ def test_no_recipients(self):
 
     def test_str(self):
         o = QueuedEmailMessage(subject="ążśźćńółĄŻŚŹĘĆŃÓŁ", to="b")
-        self.assertEqual(u"QueuedEmailMessage:ążśźćńółĄŻŚŹĘĆŃÓŁ:b", six.text_type(o))
+        self.assertEqual(u"QueuedEmailMessage:None:ążśźćńółĄŻŚŹĘĆŃÓŁ:b", six.text_type(o))
+
+    def test_str2(self):
+        o = QueuedEmailMessage(subject="ążśźćńółĄŻŚŹĘĆŃÓŁ", to="b", pk=123)
+        self.assertEqual(u"QueuedEmailMessage:123:ążśźćńółĄŻŚŹĘĆŃÓŁ:b", six.text_type(o))
 
     @override_settings(EMAIL_QUEUE_DISCARD_HOURS=2)
     @patch('django_email_queue.models.QueuedEmailMessage._send')