3 files changed, 102 insertions, 9 deletions
diff --git a/kittystore/test/__init__.py b/kittystore/test/__init__.py
index 9a10e1f..2a2b228 100644
--- a/kittystore/test/__init__.py
+++ b/kittystore/test/__init__.py
@@ -6,3 +6,13 @@ import os
 def get_test_file(*fileparts):
     return os.path.join(os.path.dirname(__file__), "testdata", *fileparts)
 get_test_file.__test__ = False
+
+
+class FakeList(object):
+    # pylint: disable=R0903
+    # (Too few public methods)
+    def __init__(self, name):
+        self.fqdn_listname = name
+        self.display_name = None
+
+
diff --git a/kittystore/test/test_storm_model.py b/kittystore/test/test_storm_model.py
new file mode 100644
index 0000000..38ba6c2
--- /dev/null
+++ b/kittystore/test/test_storm_model.py
@@ -0,0 +1,91 @@
+# -*- coding: utf-8 -*-
+# pylint: disable=R0904,C0103
+# - Too many public methods
+# - Invalid name XXX (should match YYY)
+
+import unittest
+
+from mailman.email.message import Message
+
+from kittystore.storm import get_storm_store
+from kittystore.storm.model import Email, Thread
+
+from kittystore.test import get_test_file, FakeList
+
+
+class TestStormModel(unittest.TestCase):
+
+    def setUp(self):
+        self.store = get_storm_store("sqlite:")
+        #self.store = get_storm_store("postgres://kittystore:kittystore@localhost/kittystore_test", True)
+        #self.store = get_storm_store("mysql://kittystore:kittystore@localhost/kittystore_test", True)
+
+    def tearDown(self):
+        self.store.close()
+
+    def test_starting_message_1(self):
+        # A basic thread: msg2 replies to msg1
+        ml = FakeList("example-list")
+        msg1 = Message()
+        msg1["From"] = "sender1@example.com"
+        msg1["Message-ID"] = "<msg1>"
+        msg1.set_payload("message 1")
+        self.store.add_to_list(ml, msg1)
+        msg2 = Message()
+        msg2["From"] = "sender2@example.com"
+        msg2["Message-ID"] = "<msg2>"
+        msg2.set_payload("message 2")
+        msg2["In-Reply-To"] = msg1["Message-ID"]
+        self.store.add_to_list(ml, msg2)
+        thread = self.store.db.find(Thread).one()
+        self.assertEqual(thread.starting_email.message_id, "msg1")
+
+    def test_starting_message_2(self):
+        # A partially-imported thread: msg1 replies to something we don't have
+        ml = FakeList("example-list")
+        msg1 = Message()
+        msg1["From"] = "sender1@example.com"
+        msg1["Message-ID"] = "<msg1>"
+        msg1["In-Reply-To"] = "<msg0>"
+        msg1.set_payload("message 1")
+        self.store.add_to_list(ml, msg1)
+        msg2 = Message()
+        msg2["From"] = "sender2@example.com"
+        msg2["Message-ID"] = "<msg2>"
+        msg2["In-Reply-To"] = msg1["Message-ID"]
+        msg2.set_payload("message 2")
+        self.store.add_to_list(ml, msg2)
+        thread = self.store.db.find(Thread).one()
+        self.assertEqual(thread.starting_email.message_id, "msg1")
+
+    def test_starting_message_3(self):
+        # A thread where the reply has an anterior date to the first email
+        # (the In-Reply-To header must win over the date sort)
+        ml = FakeList("example-list")
+        msg1 = Message()
+        msg1["From"] = "sender1@example.com"
+        msg1["Message-ID"] = "<msg1>"
+        msg1["Date"] = "Fri, 02 Nov 2012 16:07:54 +0000"
+        msg1.set_payload("message 1")
+        self.store.add_to_list(ml, msg1)
+        msg2 = Message()
+        msg2["From"] = "sender2@example.com"
+        msg2["Message-ID"] = "<msg2>"
+        msg2["Date"] = "Fri, 01 Nov 2012 16:07:54 +0000"
+        msg2.set_payload("message 2")
+        msg2["In-Reply-To"] = msg1["Message-ID"]
+        self.store.add_to_list(ml, msg2)
+        thread = self.store.db.find(Thread).one()
+        self.assertEqual(thread.starting_email.message_id, "msg1")
+
+    def test_subject(self):
+        ml = FakeList("example-list")
+        msg = Message()
+        msg["From"] = "sender@example.com"
+        msg["Message-ID"] = "<dummymsg>"
+        msg["Date"] = "Fri, 02 Nov 2012 16:07:54 +0000"
+        msg["Subject"] = "Dummy subject"
+        msg.set_payload("Dummy message")
+        self.store.add_to_list(ml, msg)
+        thread = self.store.db.find(Thread).one()
+        self.assertEqual(thread.subject, "Dummy subject")
diff --git a/kittystore/test/test_storm_store.py b/kittystore/test/test_storm_store.py
index f89feb1..fb7d847 100644
--- a/kittystore/test/test_storm_store.py
+++ b/kittystore/test/test_storm_store.py
@@ -13,15 +13,7 @@ from mailman.email.message import Message
 from kittystore.storm import get_storm_store
 from kittystore.storm.model import Email, Attachment
 
-from kittystore.test import get_test_file
-
-
-class FakeList(object):
-    # pylint: disable=R0903
-    # (Too few public methods)
-    def __init__(self, name):
-        self.fqdn_listname = name
-        self.display_name = None
+from kittystore.test import get_test_file, FakeList
 
 
 class TestStormStore(unittest.TestCase):