← Back to branch summary

~didrocks/unity/altf10

« back to all changes in this revision

Viewing changes to grackle/tests/test_client.py

  • Committer: Aaron Bentley
  • Date: 2012-01-11 13:31:08 UTC
  • Revision ID: aaron@canonical.com-20120111133108-9ijvfo7quc49wmm8
Retrieve messages.

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
13
12
 
14
13
from testtools import ExpectedException
15
14
 
59
58
            self.send_error(httplib.BAD_REQUEST)
60
59
 
61
60
    def do_GET(self):
62
 
        scheme, netloc, path, params, query_string, fragments = (
63
 
            urlparse(self.path))
 
61
        scheme, netloc, path, params, query, fragments = urlparse(self.path)
64
62
        archive = os.path.split(path)[1]
65
 
        query = parse_qs(query_string)
66
 
        parameters = simplejson.loads(query['parameters'][0])
67
63
        self.send_response(httplib.OK)
68
64
        self.end_headers()
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
 
        self.wfile.write(simplejson.dumps(messages))
 
65
        print "path: %s" % self.path
 
66
        self.wfile.write(simplejson.dumps(self.server.messages[archive]))
73
67
 
74
68
 
75
69
def fake_grackle_service(client, messages=None):
97
91
            {'baz':
98
92
            [{'message-id': 'foo'},
99
93
             {'message-id': 'bar'}]}):
100
 
            response = client.get_messages('baz')
 
94
            response = client.get_messages('baz', message_ids=['a'])
101
95
        self.assertEqual(['bar', 'foo'], sorted(m['message-id'] for m in
102
96
            response))
103
 
 
104
 
    def test_get_messages_by_id(self):
105
 
        client = GrackleClient('localhost', 8437)
106
 
        with fake_grackle_service(client,
107
 
            {'baz':
108
 
            [{'message-id': 'foo'},
109
 
             {'message-id': 'bar'}]}):
110
 
            message, = client.get_messages('baz', message_ids=['foo'])
111
 
        self.assertEqual('foo', message['message-id'])