extracts get_lastuid and adds tests for create_mail uid consistency.

-- Issue #334

1 files changed, 25 insertions, 0 deletions

diff --git a/service/test/unit/adapter/test_soledad_querier.py b/service/test/unit/adapter/test_soledad_querier.py
index e977b5dc..e5ea457d 100644
--- a/service/test/unit/adapter/test_soledad_querier.py
+++ b/service/test/unit/adapter/test_soledad_querier.py
@@ -123,3 +123,28 @@ class SoledadQuerierTest(unittest.TestCase):
         call_with_bad_parameters(querier.get_recent_by_mbox)
         call_with_bad_parameters(querier.idents_by_mailbox)
         call_with_bad_parameters(querier.get_mbox)
+
+    def test_get_lastuid(self):
+        soledad = mock()
+        mbox = mock()
+        mbox.content = {'lastuid': 0}
+        when(soledad).get_from_index('by-type-and-mbox', 'mbox', 'INBOX').thenReturn([mbox])
+        querier = SoledadQuerier(soledad)
+
+        self.assertEquals(querier.get_lastuid(querier.get_mbox('INBOX')[0]), 0)
+        mbox.content = {'lastuid': 1}
+        self.assertEquals(querier.get_lastuid(querier.get_mbox('INBOX')[0]), 1)
+
+    def test_create_mail_increments_uid(self):
+        soledad = mock()
+        mbox = mock()
+        mail = mock()
+        when(mail).get_for_save(next_uid=any(), mailbox='INBOX').thenReturn([])
+        mbox.content = {'lastuid': 0}
+        when(soledad).get_from_index('by-type-and-mbox', 'mbox', 'INBOX').thenReturn([mbox])
+        querier = SoledadQuerier(soledad)
+        when(querier).mail(any()).thenReturn([])
+
+        self.assertEquals(querier.get_lastuid(querier.get_mbox('INBOX')[0]), 0)
+        querier.create_mail(mail, 'INBOX')
+        self.assertEquals(querier.get_lastuid(querier.get_mbox('INBOX')[0]), 1)