← Back to branch summary

~unity-2d-team/unity-2d/Shell-MultiMonitor

« back to all changes in this revision

Viewing changes to grackle/tests/test_client.py

  • Committer: Aaron Bentley
  • Date: 2012-01-10 11:06:36 UTC
  • Revision ID: aaron@canonical.com-20120110110636-fztfdo4gn1qfqty3
Add skeleton grackle-put-message.

Show diffs side-by-side

added added

removed removed

Lines of Context:
grackle/tests/test_client.py
1
 
from BaseHTTPServer import (
2
 
    HTTPServer,
3
 
    BaseHTTPRequestHandler,
4
 
    )
5
 
import httplib
6
 
import os
7
 
from signal import SIGKILL
8
 
import simplejson
9
 
from StringIO import StringIO
10
 
from unittest import TestCase
11
 
from urlparse import urlparse
12
 
from urlparse import parse_qs
13
 
 
14
 
from testtools import ExpectedException
15
 
 
16
 
from grackle.client import (
17
 
    GrackleClient,
18
 
    )
19
 
 
20
 
 
21
 
class ForkedFake:
22
 
 
23
 
    def __init__(self, port, messages=None):
24
 
        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
 
 
32
 
    def __enter__(self):
33
 
        pid = os.fork()
34
 
        if pid == 0:
35
 
            self.start_server()
36
 
        self.pid = pid
37
 
        os.read(self.read_end, 1)
38
 
        return
39
 
 
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
 
 
46
 
    def __exit__(self, exc_type, exc_val, traceback):
47
 
        os.kill(self.pid, SIGKILL)
48
 
 
49
 
 
50
 
class FakeGrackleRequestHandler(BaseHTTPRequestHandler):
51
 
 
52
 
    def do_POST(self):
53
 
        message = self.rfile.read(int(self.headers['content-length']))
54
 
        if message == 'This is a message':
55
 
            self.send_response(httplib.CREATED)
56
 
            self.end_headers()
57
 
            self.wfile.close()
58
 
        else:
59
 
            self.send_error(httplib.BAD_REQUEST)
60
 
 
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
 
        self.wfile.write(simplejson.dumps(messages))
73
 
 
74
 
 
75
 
def fake_grackle_service(client, messages=None):
76
 
    if messages is None:
77
 
        messages = {}
78
 
    return ForkedFake(client.port, messages)
79
 
 
80
 
 
81
 
class TestPutMessage(TestCase):
82
 
 
83
 
    def test_put_message(self):
84
 
        client = GrackleClient('localhost', 8436)
85
 
        with fake_grackle_service(client):
86
 
            client.put_message('arch1', 'asdf', StringIO('This is a message'))
87
 
            with ExpectedException(Exception, 'wtf'):
88
 
                client.put_message('arch1', 'asdf',
89
 
                    StringIO('This is not a message'))
90
 
 
91
 
 
92
 
class TestGetMessages(TestCase):
93
 
 
94
 
    def test_get_messages(self):
95
 
        client = GrackleClient('localhost', 8435)
96
 
        with fake_grackle_service(client,
97
 
            {'baz':
98
 
            [{'message-id': 'foo'},
99
 
             {'message-id': 'bar'}]}):
100
 
            response = client.get_messages('baz')
101
 
        self.assertEqual(['bar', 'foo'], sorted(m['message-id'] for m in
102
 
            response))
103
 
 
104
 
    def test_get_messages_by_id(self):
105
 
        client = GrackleClient('localhost', 8437)
106
 
        with fake_grackle_service(client,
107
 
            {'baz':
108
 
            [{'message-id': 'foo'},
109
 
             {'message-id': 'bar'}]}):
110
 
            message, = client.get_messages('baz', message_ids=['foo'])
111
 
        self.assertEqual('foo', message['message-id'])