~unity-2d-team/unity-2d/Shell-MultiMonitor
» Revision
6.1.19
(compared to revision 6.1.23)
: grackle/tests/test_client.py
« back to all changes in this revision
Viewing changes to grackle/tests/test_client.py
- browse files at revision 6.1.19
- compare with another revision
- download diff
- download tarball
- view history from revision 6.1.19
- Committer: Aaron Bentley
- Date: 2012-01-13 13:52:09 UTC
- mto: (6.1.30 trunk)
- mto: This revision was merged to the branch mainline in revision 45.
- Revision ID: aaron@canonical.com-20120113135209-3s1l4hu25wjyq6jj
Fix grackle-put-message.
- files modified:
- grackle/client.py
added
removed
grackle/tests/test_client.py
34
34
return threads.values()
35
35
36
36
37
class GrackleStore:
37
class ForkedFake:
38
38
39
def __init__(self, messages):
39
def __init__(self, port, messages=None):
40
self.pid = None
41
self.port = port
40
42
self.messages = messages
41
42
def get_messages(self, archive_id, query_string):
43
self.read_end, self.write_end = os.pipe()
44
45
def is_ready(self):
46
os.write(self.write_end, 'asdf')
47
48
def __enter__(self):
49
pid = os.fork()
50
if pid == 0:
51
self.start_server()
52
self.pid = pid
53
os.read(self.read_end, 1)
54
return
55
56
def start_server(self):
57
service = HTTPServer(('', self.port), FakeGrackleRequestHandler)
58
service.messages = self.messages
59
self.is_ready()
60
service.serve_forever()
61
62
def __exit__(self, exc_type, exc_val, traceback):
63
os.kill(self.pid, SIGKILL)
64
65
66
SUPPORTED_ORDERS = set(
67
['date', 'author', 'subject', 'thread_newest', 'thread_oldest',
68
'thread_subject'])
69
70
71
class FakeGrackleRequestHandler(BaseHTTPRequestHandler):
72
73
def do_POST(self):
74
message = self.rfile.read(int(self.headers['content-length']))
75
if message == 'This is a message':
76
self.send_response(httplib.CREATED)
77
self.end_headers()
78
self.wfile.close()
79
else:
80
self.send_error(httplib.BAD_REQUEST)
81
82
def do_GET(self):
83
scheme, netloc, path, params, query_string, fragments = (
84
urlparse(self.path))
85
archive = os.path.split(path)[1]
43
86
query = parse_qs(query_string)
44
87
parameters = simplejson.loads(query['parameters'][0])
45
88
order = parameters.get('order')
46
messages = self.messages[archive_id]
89
messages = self.server.messages[archive]
47
90
if order is not None :
48
91
if order not in SUPPORTED_ORDERS:
49
raise UnsupportedOrder
92
self.send_response(httplib.BAD_REQUEST)
93
self.wfile.write('Unsupported order')
94
return
50
95
elif order.startswith('thread_'):
51
96
threaded = threaded_messages(messages)
52
97
messages = []
61
106
else:
62
107
messages.sort(key=lambda m: m[order])
63
108
messages = [m for m in messages
64
]
65
new_messages = []
66
for message in messages:
67
if (
68
not parameters['include_hidden']
69
and message.get('hidden', False)):
70
continue
71
72
if ('message_ids' in parameters and
73
message['message_id'] not in parameters['message_ids']):
74
continue
75
message = dict(message)
76
if 'headers' in parameters:
77
headers = dict(
78
(k, v) for k, v in message['headers'].iteritems()
79
if k in parameters['headers'])
80
message['headers'] = headers
81
max_body = parameters.get('max_body_length')
82
if max_body is not None:
83
message['body'] = message['body'][:max_body]
84
new_messages.append(message)
85
messages = new_messages
109
if 'message_ids' not in parameters or
110
m['message_id'] in parameters['message_ids']]
111
self.send_response(httplib.OK)
112
self.end_headers()
86
113
limit = parameters.get('limit', 100)
87
114
memo = parameters.get('memo')
88
115
message_id_indices = dict(
101
128
else:
102
129
next_memo = None
103
130
messages = messages[start:end]
104
105
131
response = {
106
132
'messages': messages,
107
133
'next_memo': next_memo,
108
134
'previous_memo': previous_memo
109
135
}
110
return response
111
112
113
114
class ForkedFake:
115
116
def __init__(self, port, messages=None):
117
self.pid = None
118
self.port = port
119
self.messages = messages
120
self.read_end, self.write_end = os.pipe()
121
122
def is_ready(self):
123
os.write(self.write_end, 'asdf')
124
125
def __enter__(self):
126
pid = os.fork()
127
if pid == 0:
128
self.start_server()
129
self.pid = pid
130
os.read(self.read_end, 1)
131
return
132
133
def start_server(self):
134
service = HTTPServer(('', self.port), FakeGrackleRequestHandler)
135
service.store = GrackleStore(self.messages)
136
for archive_id, messages in service.store.messages.iteritems():
137
for message in messages:
138
message.setdefault('headers', {})
139
self.is_ready()
140
service.serve_forever()
141
142
def __exit__(self, exc_type, exc_val, traceback):
143
os.kill(self.pid, SIGKILL)
144
145
146
SUPPORTED_ORDERS = set(
147
['date', 'author', 'subject', 'thread_newest', 'thread_oldest',
148
'thread_subject'])
149
150
151
class FakeGrackleRequestHandler(BaseHTTPRequestHandler):
152
153
def do_POST(self):
154
message = self.rfile.read(int(self.headers['content-length']))
155
if message == 'This is a message':
156
self.send_response(httplib.CREATED)
157
self.end_headers()
158
self.wfile.close()
159
else:
160
self.send_error(httplib.BAD_REQUEST)
161
162
def do_GET(self):
163
scheme, netloc, path, params, query_string, fragments = (
164
urlparse(self.path))
165
parts = path.split('/')
166
if parts[1] == 'archive':
167
try:
168
response = self.server.store.get_messages(
169
parts[2], query_string)
170
self.send_response(httplib.OK)
171
self.end_headers()
172
self.wfile.write(simplejson.dumps(response))
173
except UnsupportedOrder:
174
self.send_response(httplib.BAD_REQUEST)
175
self.wfile.write('Unsupported order')
176
return
136
self.wfile.write(simplejson.dumps(response))
177
137
178
138
179
139
def fake_grackle_service(client, messages=None):
308
268
{'message_id': 'bar', 'date': '2011-03-24'}]}):
309
269
with ExpectedException(UnsupportedOrder):
310
270
client.get_messages('baz', order='nonsense')
311
312
def test_get_messages_headers_no_headers(self):
313
client = GrackleClient('localhost', 8440)
314
with fake_grackle_service(client,
315
{'baz': [
316
{'message_id': 'foo'}
317
]}):
318
response = client.get_messages('baz', headers=[
319
'Subject', 'Date', 'X-Launchpad-Message-Rationale'])
320
first_message = response['messages'][0]
321
self.assertEqual('foo', first_message['message_id'])
322
self.assertEqual({}, first_message['headers'])
323
324
def test_get_messages_headers_exclude_headers(self):
325
client = GrackleClient('localhost', 8441)
326
with fake_grackle_service(client,
327
{'baz': [
328
{'message_id': 'foo', 'headers': {'From': 'me'}}
329
]}):
330
response = client.get_messages('baz', headers=[
331
'Subject', 'Date', 'X-Launchpad-Message-Rationale'])
332
first_message = response['messages'][0]
333
self.assertEqual('foo', first_message['message_id'])
334
self.assertEqual({}, first_message['headers'])
335
336
def test_get_messages_headers_include_headers(self):
337
client = GrackleClient('localhost', 8442)
338
with fake_grackle_service(client,
339
{'baz': [
340
{'message_id': 'foo', 'headers': {'From': 'me', 'To': 'you'}}
341
]}):
342
response = client.get_messages('baz', headers=[
343
'From', 'To'])
344
first_message = response['messages'][0]
345
self.assertEqual('foo', first_message['message_id'])
346
self.assertEqual({'From': 'me', 'To': 'you'}, first_message['headers'])
347
348
def test_get_messages_max_body_length(self):
349
client = GrackleClient('localhost', 8443)
350
with fake_grackle_service(client,
351
{'baz': [
352
{'message_id': 'foo', 'body': u'abcdefghi'}
353
]}):
354
response = client.get_messages('baz', max_body_length=3)
355
first_message = response['messages'][0]
356
self.assertEqual('abc', first_message['body'])
357
358
def test_include_hidden(self):
359
client = GrackleClient('localhost', 8444)
360
with fake_grackle_service(client,
361
{'baz': [
362
{'message_id': 'foo', 'hidden': True},
363
{'message_id': 'bar', 'hidden': False}
364
]}):
365
response = client.get_messages('baz', include_hidden=True)
366
self.assertMessageIDs(['bar', 'foo'], response['messages'])
367
response = client.get_messages('baz', include_hidden=False)
368
self.assertMessageIDs(['bar'], response['messages'])
369
Loggerhead is a web-based interface for Breezy
Version: 2.0.1