46
46
self.send_error(httplib.BAD_REQUEST)
49
def run_service(port):
49
def run_service(port, messages):
50
50
service = HTTPServer(('', port), FakeGrackleRequestHandler)
51
service.messages = messages
51
52
service.serve_forever()
55
def fake_grackle_service(client, messages=None):
58
return Forked(run_service, client.port, messages)
55
61
class TestPutMessage(TestCase):
57
63
def test_put_message(self):
58
64
client = GrackleClient('localhost', 8435)
59
with Forked(run_service, client.port):
65
with fake_grackle_service(client):
66
import time; time.sleep(0.5)
60
67
client.put_message('arch1', 'asdf', StringIO('This is a message'))
61
68
with ExpectedException(Exception, 'wtf'):
62
69
client.put_message('arch1', 'asdf',
63
70
StringIO('This is not a message'))
73
class TestGetMessages(TestCase):
75
def test_get_messages(self):
76
client = GrackleClient('localhost', 8435)
77
with fake_grackle_service(client,
79
[{'message-id': 'foo'},
80
{'message-id': 'bar'}]}):
81
response = client.get_messages('baz')
82
self.assertEqual(['bar', 'foo'], sorted(response.keys()))