← Back to branch summary

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

« back to all changes in this revision

Viewing changes to grackle/tests/test_client.py

  • Committer: Aaron Bentley
  • Date: 2012-01-12 08:52:25 UTC
  • mto: (6.1.30 trunk)
  • mto: This revision was merged to the branch mainline in revision 45.
  • Revision ID: aaron@canonical.com-20120112085225-ajy6igsil634t0g5
Switch hyphens to underscores.

Show diffs side-by-side

added added

removed removed

Lines of Context:
grackle/tests/test_client.py
9
9
from StringIO import StringIO
10
10
from unittest import TestCase
11
11
from urlparse import urlparse
 
12
from urlparse import parse_qs
12
13
 
13
14
from testtools import ExpectedException
14
15
 
58
59
            self.send_error(httplib.BAD_REQUEST)
59
60
 
60
61
    def do_GET(self):
61
 
        scheme, netloc, path, params, query, fragments = urlparse(self.path)
 
62
        scheme, netloc, path, params, query_string, fragments = (
 
63
            urlparse(self.path))
62
64
        archive = os.path.split(path)[1]
 
65
        query = parse_qs(query_string)
 
66
        parameters = simplejson.loads(query['parameters'][0])
63
67
        self.send_response(httplib.OK)
64
68
        self.end_headers()
65
 
        print "path: %s" % self.path
66
 
        self.wfile.write(simplejson.dumps(self.server.messages[archive]))
 
69
        messages = [m for m in self.server.messages[archive] if 'message_ids'
 
70
                    not in parameters or m['message_id'] in
 
71
                    parameters['message_ids']]
 
72
        response = {
 
73
            'messages': messages,
 
74
            'next_memo': None,
 
75
            'previous_memo': None
 
76
            }
 
77
        self.wfile.write(simplejson.dumps(response))
67
78
 
68
79
 
69
80
def fake_grackle_service(client, messages=None):
89
100
        client = GrackleClient('localhost', 8435)
90
101
        with fake_grackle_service(client,
91
102
            {'baz':
92
 
            [{'message-id': 'foo'},
93
 
             {'message-id': 'bar'}]}):
94
 
            response = client.get_messages('baz', message_ids=['a'])
95
 
        self.assertEqual(['bar', 'foo'], sorted(m['message-id'] for m in
96
 
            response))
 
103
            [{'message_id': 'foo'},
 
104
             {'message_id': 'bar'}]}):
 
105
            response = client.get_messages('baz')
 
106
        self.assertEqual(['bar', 'foo'], sorted(m['message_id'] for m in
 
107
            response['messages']))
 
108
        self.assertIs(None, response['next_memo'])
 
109
        self.assertIs(None, response['previous_memo'])
 
110
 
 
111
    def test_get_messages_by_id(self):
 
112
        client = GrackleClient('localhost', 8437)
 
113
        with fake_grackle_service(client,
 
114
            {'baz':
 
115
            [{'message_id': 'foo'},
 
116
             {'message_id': 'bar'}]}):
 
117
            response = client.get_messages('baz', message_ids=['foo'])
 
118
        message, = response['messages']
 
119
        self.assertEqual('foo', message['message_id'])