1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
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")
|