« back to all changes in this revision
Viewing changes to grackle/tests/test_client.py
- browse files at revision 11
- compare with another revision
- download diff
- download tarball
- view history from revision 11
- Committer: Aaron Bentley
- Date: 2012-01-11 11:18:15 UTC
- Revision ID: aaron@canonical.com-20120111111815-0ecv9fihdao2u68o
Start working on GET.
- files modified:
- grackle/client.py
added
removed
grackle/tests/test_client.py
5
5
import httplib
6
6
import os
7
7
from signal import SIGKILL
8
import simplejson
9
8
from StringIO import StringIO
10
9
from unittest import TestCase
11
from urlparse import urlparse
12
from urlparse import parse_qs
13
10
14
11
from testtools import ExpectedException
15
12
18
15
)
19
16
20
17
21
class ForkedFake:
18
class Forked:
22
19
23
def __init__(self, port, messages=None):
20
def __init__(self, func_or_method, *args):
21
self.func_or_method = func_or_method
24
22
self.pid = None
25
self.port = port
26
self.messages = messages
27
self.read_end, self.write_end = os.pipe()
28
29
def is_ready(self):
30
os.write(self.write_end, 'asdf')
23
self.args = args
31
24
32
25
def __enter__(self):
33
26
pid = os.fork()
34
if pid == 0:
35
self.start_server()
36
self.pid = pid
37
os.read(self.read_end, 1)
38
return
27
if pid != 0:
28
self.pid = pid
29
return
30
self.func_or_method(*self.args)
39
31
40
def start_server(self):
41
service = HTTPServer(('', self.port), FakeGrackleRequestHandler)
42
service.messages = self.messages
43
self.is_ready()
44
service.serve_forever()
45
32
46
33
def __exit__(self, exc_type, exc_val, traceback):
47
34
os.kill(self.pid, SIGKILL)
58
45
else:
59
46
self.send_error(httplib.BAD_REQUEST)
60
47
61
def do_GET(self):
62
scheme, netloc, path, params, query_string, fragments = (
63
urlparse(self.path))
64
archive = os.path.split(path)[1]
65
query = parse_qs(query_string)
66
parameters = simplejson.loads(query['parameters'][0])
67
self.send_response(httplib.OK)
68
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
if 'order' in parameters:
73
messages.sort(key=lambda m: m[parameters['order']])
74
limit = parameters.get('limit', 100)
75
memo = parameters.get('memo')
76
message_id_indices = dict(
77
(m['message_id'], idx) for idx, m in enumerate(messages))
78
if memo is None:
79
start = 0
80
else:
81
start = message_id_indices[memo.encode('rot13')]
82
if start > 0:
83
previous_memo = messages[start - 1]['message_id'].encode('rot13')
84
else:
85
previous_memo = None
86
end = min(start + limit, len(messages))
87
if end < len(messages):
88
next_memo = messages[end]['message_id'].encode('rot13')
89
else:
90
next_memo = None
91
messages = messages[start:end]
92
response = {
93
'messages': messages,
94
'next_memo': next_memo,
95
'previous_memo': previous_memo
96
}
97
self.wfile.write(simplejson.dumps(response))
48
49
def run_service(port, messages):
50
service = HTTPServer(('', port), FakeGrackleRequestHandler)
51
service.messages = messages
52
service.serve_forever()
98
53
99
54
100
55
def fake_grackle_service(client, messages=None):
101
56
if messages is None:
102
57
messages = {}
103
return ForkedFake(client.port, messages)
58
return Forked(run_service, client.port, messages)
104
59
105
60
106
61
class TestPutMessage(TestCase):
107
62
108
63
def test_put_message(self):
109
client = GrackleClient('localhost', 8436)
64
client = GrackleClient('localhost', 8435)
110
65
with fake_grackle_service(client):
66
import time; time.sleep(0.5)
111
67
client.put_message('arch1', 'asdf', StringIO('This is a message'))
112
68
with ExpectedException(Exception, 'wtf'):
113
69
client.put_message('arch1', 'asdf',
116
72
117
73
class TestGetMessages(TestCase):
118
74
119
def assertIDOrder(self, ids, messages):
120
self.assertEqual(ids, [m['message_id'] for m in messages])
121
122
def assertMessageIDs(self, ids, messages):
123
self.assertIDOrder(
124
sorted(ids), sorted(messages, key=lambda m:m['message_id']))
125
126
75
def test_get_messages(self):
127
76
client = GrackleClient('localhost', 8435)
128
77
with fake_grackle_service(client,
129
78
{'baz':
130
[{'message_id': 'foo'},
131
{'message_id': 'bar'}]}):
132
response = client.get_messages('baz')
133
self.assertEqual(['bar', 'foo'], sorted(m['message_id'] for m in
134
response['messages']))
135
self.assertIs(None, response['next_memo'])
136
self.assertIs(None, response['previous_memo'])
137
138
def test_get_messages_by_id(self):
139
client = GrackleClient('localhost', 8437)
140
with fake_grackle_service(client,
141
{'baz':
142
[{'message_id': 'foo'},
143
{'message_id': 'bar'}]}):
144
response = client.get_messages('baz', message_ids=['foo'])
145
message, = response['messages']
146
self.assertEqual('foo', message['message_id'])
147
148
def test_get_messages_batching(self):
149
client = GrackleClient('localhost', 8438)
150
with fake_grackle_service(client,
151
{'baz':
152
[{'message_id': 'foo'},
153
{'message_id': 'bar'}]}):
154
response = client.get_messages('baz', limit=1)
155
self.assertEqual(1, len(response['messages']))
156
messages = response['messages']
157
response = client.get_messages(
158
'baz', limit=1, memo=response['next_memo'])
159
self.assertEqual(1, len(response['messages']))
160
messages.extend(response['messages'])
161
self.assertMessageIDs(['foo', 'bar'], messages)
162
163
def test_get_messages_date_order(self):
164
client = GrackleClient('localhost', 8439)
165
with fake_grackle_service(client,
166
{'baz': [{'message_id': 'foo', 'date': '2011-03-25'},
167
{'message_id': 'bar', 'date': '2011-03-24'}]}):
168
response = client.get_messages('baz')
169
self.assertIDOrder(['foo', 'bar'], response['messages'])
170
response = client.get_messages('baz', order='date')
171
self.assertIDOrder(['bar', 'foo'], response['messages'])
79
[{'message-id': 'foo'},
80
{'message-id': 'bar'}]}):
81
response = client.get_messages('baz')
82
self.assertEqual(['bar', 'foo'], sorted(response.keys()))
Loggerhead is a web-based interface for Breezy
Version: 2.0.1