← 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 10:05:32 UTC
  • mto: (6.1.30 trunk)
  • mto: This revision was merged to the branch mainline in revision 45.
  • Revision ID: aaron@canonical.com-20120112100532-gljok11rzp7i3y11
Test unsupported orders.

Show diffs side-by-side

added added

removed removed

Lines of Context:
grackle/tests/test_client.py
5
5
import httplib
6
6
import os
7
7
from signal import SIGKILL
 
8
import simplejson
8
9
from StringIO import StringIO
9
10
from unittest import TestCase
 
11
from urlparse import urlparse
 
12
from urlparse import parse_qs
10
13
 
11
14
from testtools import ExpectedException
12
15
 
13
 
from grackle import client
14
 
 
15
 
 
16
 
class Forked:
17
 
 
18
 
    def __init__(self, func_or_method):
19
 
        self.func_or_method = func_or_method
 
16
from grackle.client import (
 
17
    GrackleClient,
 
18
    UnsupportedOrder,
 
19
    )
 
20
 
 
21
 
 
22
class ForkedFake:
 
23
 
 
24
    def __init__(self, port, messages=None):
20
25
        self.pid = None
 
26
        self.port = port
 
27
        self.messages = messages
 
28
        self.read_end, self.write_end = os.pipe()
 
29
 
 
30
    def is_ready(self):
 
31
        os.write(self.write_end, 'asdf')
21
32
 
22
33
    def __enter__(self):
23
34
        pid = os.fork()
24
 
        if pid != 0:
25
 
            self.pid = pid
26
 
            return
27
 
        self.func_or_method()
 
35
        if pid == 0:
 
36
            self.start_server()
 
37
        self.pid = pid
 
38
        os.read(self.read_end, 1)
 
39
        return
28
40
 
 
41
    def start_server(self):
 
42
        service = HTTPServer(('', self.port), FakeGrackleRequestHandler)
 
43
        service.messages = self.messages
 
44
        self.is_ready()
 
45
        service.serve_forever()
29
46
 
30
47
    def __exit__(self, exc_type, exc_val, traceback):
31
48
        os.kill(self.pid, SIGKILL)
32
49
 
33
50
 
 
51
SUPPORTED_ORDERS = set(['date'])
 
52
 
 
53
 
34
54
class FakeGrackleRequestHandler(BaseHTTPRequestHandler):
35
55
 
36
56
    def do_POST(self):
42
62
        else:
43
63
            self.send_error(httplib.BAD_REQUEST)
44
64
 
45
 
 
46
 
def run_service():
47
 
    service = HTTPServer(('', 8435), FakeGrackleRequestHandler)
48
 
    service.serve_forever()
49
 
 
 
65
    def do_GET(self):
 
66
        scheme, netloc, path, params, query_string, fragments = (
 
67
            urlparse(self.path))
 
68
        archive = os.path.split(path)[1]
 
69
        query = parse_qs(query_string)
 
70
        parameters = simplejson.loads(query['parameters'][0])
 
71
        messages = [m for m in self.server.messages[archive] if 'message_ids'
 
72
                    not in parameters or m['message_id'] in
 
73
                    parameters['message_ids']]
 
74
        if 'order' in parameters:
 
75
            if parameters['order'] not in SUPPORTED_ORDERS:
 
76
                self.send_response(httplib.BAD_REQUEST)
 
77
                self.wfile.write('Unsupported order')
 
78
                return
 
79
            messages.sort(key=lambda m: m[parameters['order']])
 
80
        self.send_response(httplib.OK)
 
81
        self.end_headers()
 
82
        limit = parameters.get('limit', 100)
 
83
        memo = parameters.get('memo')
 
84
        message_id_indices = dict(
 
85
            (m['message_id'], idx) for idx, m in enumerate(messages))
 
86
        if memo is None:
 
87
            start = 0
 
88
        else:
 
89
            start = message_id_indices[memo.encode('rot13')]
 
90
        if start > 0:
 
91
            previous_memo = messages[start - 1]['message_id'].encode('rot13')
 
92
        else:
 
93
            previous_memo = None
 
94
        end = min(start + limit, len(messages))
 
95
        if end < len(messages):
 
96
            next_memo = messages[end]['message_id'].encode('rot13')
 
97
        else:
 
98
            next_memo = None
 
99
        messages = messages[start:end]
 
100
        response = {
 
101
            'messages': messages,
 
102
            'next_memo': next_memo,
 
103
            'previous_memo': previous_memo
 
104
            }
 
105
        self.wfile.write(simplejson.dumps(response))
 
106
 
 
107
 
 
108
def fake_grackle_service(client, messages=None):
 
109
    if messages is None:
 
110
        messages = {}
 
111
    return ForkedFake(client.port, messages)
50
112
 
51
113
 
52
114
class TestPutMessage(TestCase):
53
115
 
54
116
    def test_put_message(self):
55
 
        with Forked(run_service):
56
 
            client.put_message('arch1', StringIO('This is a message'))
 
117
        client = GrackleClient('localhost', 8436)
 
118
        with fake_grackle_service(client):
 
119
            client.put_message('arch1', 'asdf', StringIO('This is a message'))
57
120
            with ExpectedException(Exception, 'wtf'):
58
 
                client.put_message('arch1', StringIO('This is not a message'))
 
121
                client.put_message('arch1', 'asdf',
 
122
                    StringIO('This is not a message'))
 
123
 
 
124
 
 
125
class TestGetMessages(TestCase):
 
126
 
 
127
    def assertIDOrder(self, ids, messages):
 
128
        self.assertEqual(ids, [m['message_id'] for m in messages])
 
129
 
 
130
    def assertMessageIDs(self, ids, messages):
 
131
        self.assertIDOrder(
 
132
            sorted(ids), sorted(messages, key=lambda m:m['message_id']))
 
133
 
 
134
    def test_get_messages(self):
 
135
        client = GrackleClient('localhost', 8435)
 
136
        with fake_grackle_service(client,
 
137
            {'baz':
 
138
            [{'message_id': 'foo'},
 
139
             {'message_id': 'bar'}]}):
 
140
            response = client.get_messages('baz')
 
141
        self.assertEqual(['bar', 'foo'], sorted(m['message_id'] for m in
 
142
            response['messages']))
 
143
        self.assertIs(None, response['next_memo'])
 
144
        self.assertIs(None, response['previous_memo'])
 
145
 
 
146
    def test_get_messages_by_id(self):
 
147
        client = GrackleClient('localhost', 8437)
 
148
        with fake_grackle_service(client,
 
149
            {'baz':
 
150
            [{'message_id': 'foo'},
 
151
             {'message_id': 'bar'}]}):
 
152
            response = client.get_messages('baz', message_ids=['foo'])
 
153
        message, = response['messages']
 
154
        self.assertEqual('foo', message['message_id'])
 
155
 
 
156
    def test_get_messages_batching(self):
 
157
        client = GrackleClient('localhost', 8438)
 
158
        with fake_grackle_service(client,
 
159
            {'baz':
 
160
            [{'message_id': 'foo'},
 
161
             {'message_id': 'bar'}]}):
 
162
            response = client.get_messages('baz', limit=1)
 
163
            self.assertEqual(1, len(response['messages']))
 
164
            messages = response['messages']
 
165
            response = client.get_messages(
 
166
                'baz', limit=1, memo=response['next_memo'])
 
167
            self.assertEqual(1, len(response['messages']))
 
168
            messages.extend(response['messages'])
 
169
            self.assertMessageIDs(['foo', 'bar'], messages)
 
170
 
 
171
    def test_get_messages_date_order(self):
 
172
        client = GrackleClient('localhost', 8439)
 
173
        with fake_grackle_service(client,
 
174
                {'baz': [{'message_id': 'foo', 'date': '2011-03-25'},
 
175
                 {'message_id': 'bar', 'date': '2011-03-24'}]}):
 
176
            response = client.get_messages('baz')
 
177
            self.assertIDOrder(['foo', 'bar'], response['messages'])
 
178
            response = client.get_messages('baz', order='date')
 
179
            self.assertIDOrder(['bar', 'foo'], response['messages'])
 
180
 
 
181
    def test_get_messages_unsupported_order(self):
 
182
        client = GrackleClient('localhost', 8439)
 
183
        with fake_grackle_service(client,
 
184
                {'baz': [{'message_id': 'foo', 'date': '2011-03-25'},
 
185
                 {'message_id': 'bar', 'date': '2011-03-24'}]}):
 
186
            with ExpectedException(UnsupportedOrder):
 
187
                client.get_messages('baz', order='nonsense')