~unity-2d-team/unity-2d/Shell-MultiMonitor
» Revision
13
(compared to revision 6.1.13)
: grackle/tests/test_client.py
« back to all changes in this revision
Viewing changes to grackle/tests/test_client.py
- browse files at revision 13
- compare with another revision
- download diff
- download tarball
- view history from revision 13
- Committer: William Grant
- Date: 2012-01-22 08:27:01 UTC
- Revision ID: william.grant@canonical.com-20120122082701-b7iiqxdxsbpj3krq
Test add_message a little.
- files added:
- grackle-create-instance
- grackle/server
- grackle/server/testing
- grackle/server/tests
- files modified:
- Makefile
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
16
from grackle.client import (
17
GrackleClient,
18
)
19
20
21
class ForkedFake:
22
23
def __init__(self, port, messages=None):
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
24
20
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')
31
21
32
22
def __enter__(self):
33
23
pid = os.fork()
34
if pid == 0:
35
self.start_server()
36
self.pid = pid
37
os.read(self.read_end, 1)
38
return
24
if pid != 0:
25
self.pid = pid
26
return
27
self.func_or_method()
39
28
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
29
46
30
def __exit__(self, exc_type, exc_val, traceback):
47
31
os.kill(self.pid, SIGKILL)
58
42
else:
59
43
self.send_error(httplib.BAD_REQUEST)
60
44
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
limit = parameters.get('limit', 100)
73
memo = parameters.get('memo')
74
message_id_indices = dict(
75
(m['message_id'], idx) for idx, m in enumerate(messages))
76
if memo is None:
77
start = 0
78
else:
79
start = message_id_indices[memo.encode('rot13')]
80
if start > 0:
81
previous_memo = messages[start - 1]['message_id'].encode('rot13')
82
else:
83
previous_memo = None
84
end = min(start + limit, len(messages))
85
if end < len(messages):
86
next_memo = messages[end]['message_id'].encode('rot13')
87
else:
88
next_memo = None
89
messages = messages[start:end]
90
response = {
91
'messages': messages,
92
'next_memo': next_memo,
93
'previous_memo': previous_memo
94
}
95
self.wfile.write(simplejson.dumps(response))
96
97
98
def fake_grackle_service(client, messages=None):
99
if messages is None:
100
messages = {}
101
return ForkedFake(client.port, messages)
45
46
def run_service():
47
service = HTTPServer(('', 8435), FakeGrackleRequestHandler)
48
service.serve_forever()
49
102
50
103
51
104
52
class TestPutMessage(TestCase):
105
53
106
54
def test_put_message(self):
107
client = GrackleClient('localhost', 8436)
108
with fake_grackle_service(client):
109
client.put_message('arch1', 'asdf', StringIO('This is a message'))
55
with Forked(run_service):
56
client.put_message('arch1', StringIO('This is a message'))
110
57
with ExpectedException(Exception, 'wtf'):
111
client.put_message('arch1', 'asdf',
112
StringIO('This is not a message'))
113
114
115
class TestGetMessages(TestCase):
116
117
def assertMessageIDs(self, ids, messages):
118
self.assertEqual(
119
sorted(ids), sorted(m['message_id'] for m in messages))
120
121
def test_get_messages(self):
122
client = GrackleClient('localhost', 8435)
123
with fake_grackle_service(client,
124
{'baz':
125
[{'message_id': 'foo'},
126
{'message_id': 'bar'}]}):
127
response = client.get_messages('baz')
128
self.assertEqual(['bar', 'foo'], sorted(m['message_id'] for m in
129
response['messages']))
130
self.assertIs(None, response['next_memo'])
131
self.assertIs(None, response['previous_memo'])
132
133
def test_get_messages_by_id(self):
134
client = GrackleClient('localhost', 8437)
135
with fake_grackle_service(client,
136
{'baz':
137
[{'message_id': 'foo'},
138
{'message_id': 'bar'}]}):
139
response = client.get_messages('baz', message_ids=['foo'])
140
message, = response['messages']
141
self.assertEqual('foo', message['message_id'])
142
143
def test_get_messages_batching(self):
144
client = GrackleClient('localhost', 8437)
145
with fake_grackle_service(client,
146
{'baz':
147
[{'message_id': 'foo'},
148
{'message_id': 'bar'}]}):
149
response = client.get_messages('baz', limit=1)
150
self.assertEqual(1, len(response['messages']))
151
messages = response['messages']
152
response = client.get_messages(
153
'baz', limit=1, memo=response['next_memo'])
154
self.assertEqual(1, len(response['messages']))
155
messages.extend(response['messages'])
156
self.assertMessageIDs(['foo', 'bar'], messages)
58
client.put_message('arch1', StringIO('This is not a message'))
Loggerhead is a web-based interface for Breezy
Version: 2.0.1