95
99
parsed_message = _parse_message(TEST_MESSAGE)[1]
96
100
for key, value in parsed_message.iteritems():
97
101
self.assertEqual(value, cmsg[key])
104
class TestGetMessages(unittest.TestCase):
106
def assertMessages(self, expected_ids, messages):
108
'<message%d@example.com>' % id for id in expected_ids]
109
actual_msgids = [msg['message-id'] for msg in messages]
110
self.assertEqual(expected_msgids, actual_msgids)
112
def makeMessages(self, conn, archive, count):
116
TEMPLATE_MESSAGE.format(
117
date=formatdate(i * 100),
118
id='<message%d@example.com>' % i))
119
for i in range(count)]
121
def makeArchive(self):
122
conn = CassandraConnection(
123
os.environ['GRACKLE_TEST_KEYSPACE'], ['localhost:9160'])
124
archive = next(_RandomNameSequence())
127
def test_single_message(self):
128
conn, archive = self.makeArchive()
129
self.makeMessages(conn, archive, 1)
130
messages = conn.get_messages(archive, 'date', 1, '')[1]
131
self.assertMessages([0], messages)
133
expected_content = TEMPLATE_MESSAGE.format(
134
date=formatdate(0), id='<message0@example.com>')
135
self.assertEqual(expected_content, messages[0]['content'])
136
pmsg = _parse_message(expected_content)[1]
137
self.assertEqual(pmsg['date'], cmsg['date'])
138
self.assertEqual(pmsg['from'], cmsg['from'])
139
self.assertEqual(pmsg['subject'], cmsg['subject'])
140
self.assertEqual(pmsg['message-id'], cmsg['message-id'])
142
def test_limit(self):
143
conn, archive = self.makeArchive()
144
self.makeMessages(conn, archive, 4)
146
[0, 1], conn.get_messages(archive, 'date', 2, '')[1])
148
def test_order(self):
149
conn, archive = self.makeArchive()
150
self.makeMessages(conn, archive, 4)
152
[3, 2], conn.get_messages(archive, '-date', 2, '')[1])
154
def test_batching_forward(self):
155
conn, archive = self.makeArchive()
156
self.makeMessages(conn, archive, 5)
157
prev, messages, next = conn.get_messages(archive, 'date', 2, '')
158
self.assertMessages([0, 1], messages)
159
prev, messages, next = conn.get_messages(archive, 'date', 2, next)
160
self.assertMessages([2, 3], messages)
161
prev, messages, next = conn.get_messages(archive, 'date', 2, next)
162
self.assertMessages([4], messages)
163
prev, messages, next = conn.get_messages(archive, 'date', 2, next)
164
self.assertIs(None, prev)
165
self.assertMessages([], messages)
166
self.assertIs(None, next)
168
def test_batching_backward(self):
169
conn, archive = self.makeArchive()
170
self.makeMessages(conn, archive, 5)
171
prev, messages, next = conn.get_messages(archive, 'date', 2, '')
172
self.assertMessages([0, 1], messages)
173
prev, messages, next = conn.get_messages(archive, 'date', 2, next)
174
self.assertMessages([2, 3], messages)
175
prev, messages, next = conn.get_messages(
176
archive, 'date', 2, prev, backward=True)
177
self.assertMessages([0, 1], messages)
178
prev, messages, next = conn.get_messages(
179
archive, 'date', 2, prev, backward=True)
180
self.assertIs(None, prev)
181
self.assertMessages([], messages)
182
self.assertIs(None, next)
184
def test_date_filter(self):
185
conn, archive = self.makeArchive()
186
self.makeMessages(conn, archive, 10)
187
start = datetime.datetime.utcfromtimestamp(250).replace(
189
finish = datetime.datetime.utcfromtimestamp(500).replace(
191
prev, messages, next = conn.get_messages(
192
archive, 'date', 2, '', start_date=start, finish_date=finish)
193
self.assertMessages([3, 4], messages)
194
prev, messages, next = conn.get_messages(
195
archive, 'date', 2, next, start_date=start, finish_date=finish)
196
self.assertMessages([5], messages)