← 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-11 14:04:41 UTC
  • mto: (6.1.30 trunk)
  • mto: This revision was merged to the branch mainline in revision 45.
  • Revision ID: aaron@canonical.com-20120111140441-l4sanxq1en07oblx
Test filtering by message-id.

Show diffs side-by-side

added added

removed removed

Lines of Context:
grackle/tests/test_client.py
5
5
import httplib
6
6
import os
7
7
from signal import SIGKILL
 
8
import simplejson
8
9
from StringIO import StringIO
9
10
from unittest import TestCase
 
11
from urlparse import urlparse
 
12
from urlparse import parse_qs
10
13
 
11
14
from testtools import ExpectedException
12
15
 
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
 
16
from grackle.client import (
 
17
    GrackleClient,
 
18
    )
 
19
 
 
20
 
 
21
class ForkedFake:
 
22
 
 
23
    def __init__(self, port, messages=None):
20
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')
21
31
 
22
32
    def __enter__(self):
23
33
        pid = os.fork()
24
 
        if pid != 0:
25
 
            self.pid = pid
26
 
            return
27
 
        self.func_or_method()
 
34
        if pid == 0:
 
35
            self.start_server()
 
36
        self.pid = pid
 
37
        os.read(self.read_end, 1)
 
38
        return
28
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()
29
45
 
30
46
    def __exit__(self, exc_type, exc_val, traceback):
31
47
        os.kill(self.pid, SIGKILL)
42
58
        else:
43
59
            self.send_error(httplib.BAD_REQUEST)
44
60
 
45
 
 
46
 
def run_service():
47
 
    service = HTTPServer(('', 8435), FakeGrackleRequestHandler)
48
 
    service.serve_forever()
49
 
 
 
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)
50
79
 
51
80
 
52
81
class TestPutMessage(TestCase):
53
82
 
54
83
    def test_put_message(self):
55
 
        with Forked(run_service):
56
 
            client.put_message('arch1', StringIO('This is a message'))
 
84
        client = GrackleClient('localhost', 8436)
 
85
        with fake_grackle_service(client):
 
86
            client.put_message('arch1', 'asdf', StringIO('This is a message'))
57
87
            with ExpectedException(Exception, 'wtf'):
58
 
                client.put_message('arch1', StringIO('This is not a message'))
 
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'])