← Back to branch summary

~unity-2d-team/unity-2d/Shell-MultiMonitor

« back to all changes in this revision

Viewing changes to grackle/tests/test_model.py

  • Committer: William Grant
  • Date: 2012-01-24 22:54:07 UTC
  • Revision ID: william.grant@canonical.com-20120124225407-8f4cxv3p14njgjnp
Use testtools so we work in python2.6.

Show diffs side-by-side

added added

removed removed

Lines of Context:
grackle/tests/test_model.py
21
21
from email.utils import formatdate
22
22
import os
23
23
from tempfile import _RandomNameSequence
24
 
import unittest
25
24
 
26
25
from dateutil.tz import (
27
26
    tzoffset,
28
27
    tzutc,
29
28
    )
30
29
from pycassa.util import convert_uuid_to_time
 
30
import testtools
31
31
 
32
32
from grackle.model import (
33
33
    CassandraConnection,
51
51
    id='<aaaaaaaaaaaaa@example.com>')
52
52
 
53
53
 
54
 
class TestParseMessage(unittest.TestCase):
 
54
class TestParseMessage(testtools.TestCase):
55
55
 
56
56
    def test_works(self):
57
57
        # _parse_message extracts interesting fields. It also parses the
68
68
            date)
69
69
 
70
70
 
71
 
class TestAddMessage(unittest.TestCase):
 
71
class TestAddMessage(testtools.TestCase):
72
72
 
73
73
    def test_add_message(self):
74
74
        c = CassandraConnection(
101
101
            self.assertEqual(value, cmsg[key])
102
102
 
103
103
 
104
 
class TestGetMessages(unittest.TestCase):
 
104
class TestGetMessages(testtools.TestCase):
105
105
 
106
106
    def assertMessages(self, expected_ids, messages):
107
107
        expected_msgids = [
204
204
        self.assertMessages([5], messages)
205
205
 
206
206
 
207
 
class TestMessageFormatter(unittest.TestCase):
 
207
class TestMessageFormatter(testtools.TestCase):
208
208
 
209
209
    def test_all(self):
210
210
        parsed = _parse_message(TEST_MESSAGE)[1]