34
34
return threads.values()
39
def __init__(self, messages):
39
def __init__(self, port, messages=None):
40
42
self.messages = messages
42
def get_messages(self, archive_id, query_string):
43
self.read_end, self.write_end = os.pipe()
46
os.write(self.write_end, 'asdf')
53
os.read(self.read_end, 1)
56
def start_server(self):
57
service = HTTPServer(('', self.port), FakeGrackleRequestHandler)
58
service.messages = self.messages
60
service.serve_forever()
62
def __exit__(self, exc_type, exc_val, traceback):
63
os.kill(self.pid, SIGKILL)
66
SUPPORTED_ORDERS = set(
67
['date', 'author', 'subject', 'thread_newest', 'thread_oldest',
71
class FakeGrackleRequestHandler(BaseHTTPRequestHandler):
74
message = self.rfile.read(int(self.headers['content-length']))
75
if message == 'This is a message':
76
self.send_response(httplib.CREATED)
80
self.send_error(httplib.BAD_REQUEST)
83
scheme, netloc, path, params, query_string, fragments = (
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')
50
95
elif order.startswith('thread_'):
51
96
threaded = threaded_messages(messages)
60
105
messages.extend(thread)
62
107
messages.sort(key=lambda m: m[order])
64
for message in messages:
66
not parameters['include_hidden']
67
and message.get('hidden', False)):
70
if ('message_ids' in parameters and
71
message['message_id'] not in parameters['message_ids']):
73
message = dict(message)
74
if 'headers' in parameters:
76
(k, v) for k, v in message['headers'].iteritems()
77
if k in parameters['headers'])
78
message['headers'] = headers
79
max_body = parameters.get('max_body_length')
80
if max_body is not None:
81
message['body'] = message['body'][:max_body]
82
new_messages.append(message)
83
messages = new_messages
108
messages = [m for m in messages
109
if 'message_ids' not in parameters or
110
m['message_id'] in parameters['message_ids']]
111
self.send_response(httplib.OK)
84
113
limit = parameters.get('limit', 100)
85
114
memo = parameters.get('memo')
86
115
message_id_indices = dict(
101
130
messages = messages[start:end]
104
132
'messages': messages,
105
133
'next_memo': next_memo,
106
134
'previous_memo': previous_memo
114
def __init__(self, port, messages=None):
117
self.messages = messages
118
self.read_end, self.write_end = os.pipe()
121
os.write(self.write_end, 'asdf')
128
os.read(self.read_end, 1)
131
def start_server(self):
132
service = HTTPServer(('', self.port), FakeGrackleRequestHandler)
133
service.store = GrackleStore(self.messages)
134
for archive_id, messages in service.store.messages.iteritems():
135
for message in messages:
136
message.setdefault('headers', {})
138
service.serve_forever()
140
def __exit__(self, exc_type, exc_val, traceback):
141
os.kill(self.pid, SIGKILL)
144
SUPPORTED_ORDERS = set(
145
['date', 'author', 'subject', 'thread_newest', 'thread_oldest',
149
class FakeGrackleRequestHandler(BaseHTTPRequestHandler):
152
message = self.rfile.read(int(self.headers['content-length']))
153
if message == 'This is a message':
154
self.send_response(httplib.CREATED)
158
self.send_error(httplib.BAD_REQUEST)
161
scheme, netloc, path, params, query_string, fragments = (
163
parts = path.split('/')
164
if parts[1] == 'archive':
166
response = self.server.store.get_messages(
167
parts[2], query_string)
168
self.send_response(httplib.OK)
170
self.wfile.write(simplejson.dumps(response))
171
except UnsupportedOrder:
172
self.send_response(httplib.BAD_REQUEST)
173
self.wfile.write('Unsupported order')
136
self.wfile.write(simplejson.dumps(response))
177
139
def fake_grackle_service(client, messages=None):
306
268
{'message_id': 'bar', 'date': '2011-03-24'}]}):
307
269
with ExpectedException(UnsupportedOrder):
308
270
client.get_messages('baz', order='nonsense')
310
def test_get_messages_headers_no_headers(self):
311
client = GrackleClient('localhost', 8440)
312
with fake_grackle_service(client,
314
{'message_id': 'foo'}
316
response = client.get_messages('baz', headers=[
317
'Subject', 'Date', 'X-Launchpad-Message-Rationale'])
318
first_message = response['messages'][0]
319
self.assertEqual('foo', first_message['message_id'])
320
self.assertEqual({}, first_message['headers'])
322
def test_get_messages_headers_exclude_headers(self):
323
client = GrackleClient('localhost', 8441)
324
with fake_grackle_service(client,
326
{'message_id': 'foo', 'headers': {'From': 'me'}}
328
response = client.get_messages('baz', headers=[
329
'Subject', 'Date', 'X-Launchpad-Message-Rationale'])
330
first_message = response['messages'][0]
331
self.assertEqual('foo', first_message['message_id'])
332
self.assertEqual({}, first_message['headers'])
334
def test_get_messages_headers_include_headers(self):
335
client = GrackleClient('localhost', 8442)
336
with fake_grackle_service(client,
338
{'message_id': 'foo', 'headers': {'From': 'me', 'To': 'you'}}
340
response = client.get_messages('baz', headers=[
342
first_message = response['messages'][0]
343
self.assertEqual('foo', first_message['message_id'])
344
self.assertEqual({'From': 'me', 'To': 'you'}, first_message['headers'])
346
def test_get_messages_max_body_length(self):
347
client = GrackleClient('localhost', 8443)
348
with fake_grackle_service(client,
350
{'message_id': 'foo', 'body': u'abcdefghi'}
352
response = client.get_messages('baz', max_body_length=3)
353
first_message = response['messages'][0]
354
self.assertEqual('abc', first_message['body'])
356
def test_include_hidden(self):
357
client = GrackleClient('localhost', 8444)
358
with fake_grackle_service(client,
360
{'message_id': 'foo', 'hidden': True},
361
{'message_id': 'bar', 'hidden': False}
363
response = client.get_messages('baz', include_hidden=True)
364
self.assertMessageIDs(['bar', 'foo'], response['messages'])
365
response = client.get_messages('baz', include_hidden=False)
366
self.assertMessageIDs(['bar'], response['messages'])