17
17
__metaclass__ = type
21
from grackle.server.testing.cassandra import TemporaryDB
22
from grackle.server.model import CassandraConnection
21
from tempfile import _RandomNameSequence
25
from dateutil.tz import (
29
from pycassa.util import convert_uuid_to_time
31
from grackle.model import (
37
TEMPLATE_MESSAGE = """\
26
38
From: sysadmin@example.com
27
39
To: developer@example.com
28
40
Subject: Everything is broken
29
Date: Sat, 1 Jan 2000 11:02:34 +1100
30
Message-Id: <aaaaaaaaaaaaa@example.com>
32
44
Help, everything has just broken.
36
class TestModel(fixtures.TestWithFixtures):
38
def test_add_message(self):
39
keyspace = self.useFixture(TemporaryDB()).keyspace
40
c = CassandraConnection(keyspace, ['localhost:9160'])
41
c.add_message('foo', TEST_MESSAGE)
44
class TestModel(fixtures.TestWithFixtures):
46
def test_add_message(self):
47
keyspace = self.useFixture(TemporaryDB()).keyspace
48
c = CassandraConnection(keyspace, ['localhost:9160'])
49
key = c.add_message('foo', TEST_MESSAGE)
47
TEST_MESSAGE = TEMPLATE_MESSAGE.format(
48
date='Sat, 1 Jan 2000 11:02:34 +1100',
49
id='<aaaaaaaaaaaaa@example.com>')
52
class TestParseMessage(unittest.TestCase):
55
# _parse_message extracts interesting fields. It also parses the
56
# date and returns it separately.
57
date, msg = _parse_message(TEST_MESSAGE)
58
self.assertEqual('sysadmin@example.com', msg['from'])
59
self.assertEqual('developer@example.com', msg['to'])
60
self.assertEqual('Everything is broken', msg['subject'])
61
self.assertEqual('2000-01-01T11:02:34+11:00', msg['date'])
62
self.assertEqual('<aaaaaaaaaaaaa@example.com>', msg['message-id'])
65
2000, 1, 1, 11, 2, 34, tzinfo=tzoffset('', 39600)),
69
class TestAddMessage(unittest.TestCase):
71
def test_add_message(self):
72
c = CassandraConnection(
73
os.environ['GRACKLE_TEST_KEYSPACE'], ['localhost:9160'])
74
archive = next(_RandomNameSequence())
76
# Write the message out to Cassandra, and read it back in.
77
key = c.add_message(archive, TEST_MESSAGE)
50
78
cmsg = c.messages.get(key)
80
# The archive should contain a single message, a reference to
82
archive_messages = c.archive_messages.get(archive).items()
83
self.assertEqual(1, len(archive_messages))
84
self.assertEqual(key, archive_messages[0][1])
86
# The key in archive_message is a TimeUUID for the Date field in
88
utctime = time.mktime(datetime.datetime(
89
2000, 1, 1, 0, 2, 34, tzinfo=tzutc()).timetuple())
92
convert_uuid_to_time(archive_messages[0][0]))
94
# The stored message contains the full original text of the
95
# message, as well as interesting fields parsed out.
51
96
self.assertEqual(TEST_MESSAGE, cmsg['content'])
52
self.assertEqual('sysadmin@example.com', cmsg['from'])
53
self.assertEqual('developer@example.com', cmsg['to'])
54
self.assertEqual('Everything is broken', cmsg['subject'])
55
self.assertEqual('2000-01-01T11:02:34+11:00', cmsg['date'])
56
self.assertEqual('<aaaaaaaaaaaaa@example.com>', cmsg['message-id'])
97
parsed_message = _parse_message(TEST_MESSAGE)[1]
98
for key, value in parsed_message.iteritems():
99
self.assertEqual(value, cmsg[key])