← Back to branch summary

~didrocks/unity/altf10

« back to all changes in this revision

Viewing changes to grackle/tests/test_client.py

  • Committer: Aaron Bentley
  • Date: 2012-01-10 15:36:31 UTC
  • Revision ID: aaron@canonical.com-20120110153631-vvqlfg122u08wsrt
Accept arbitrary permalink urls.

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
9
8
from StringIO import StringIO
10
9
from unittest import TestCase
11
 
from urlparse import urlparse
12
10
 
13
11
from testtools import ExpectedException
14
12
 
17
15
    )
18
16
 
19
17
 
20
 
class ForkedFake:
 
18
class Forked:
21
19
 
22
 
    def __init__(self, port, messages=None):
 
20
    def __init__(self, func_or_method, *args):
 
21
        self.func_or_method = func_or_method
23
22
        self.pid = None
24
 
        self.port = port
25
 
        self.messages = messages
26
 
        self.read_end, self.write_end = os.pipe()
27
 
 
28
 
    def is_ready(self):
29
 
        os.write(self.write_end, 'asdf')
 
23
        self.args = args
30
24
 
31
25
    def __enter__(self):
32
26
        pid = os.fork()
33
 
        if pid == 0:
34
 
            self.start_server()
35
 
        self.pid = pid
36
 
        os.read(self.read_end, 1)
37
 
        return
 
27
        if pid != 0:
 
28
            self.pid = pid
 
29
            return
 
30
        self.func_or_method(*self.args)
38
31
 
39
 
    def start_server(self):
40
 
        service = HTTPServer(('', self.port), FakeGrackleRequestHandler)
41
 
        service.messages = self.messages
42
 
        self.is_ready()
43
 
        service.serve_forever()
44
32
 
45
33
    def __exit__(self, exc_type, exc_val, traceback):
46
34
        os.kill(self.pid, SIGKILL)
48
36
 
49
37
class FakeGrackleRequestHandler(BaseHTTPRequestHandler):
50
38
 
51
 
    def do_POST(self):
 
39
    def do_PUT(self):
52
40
        message = self.rfile.read(int(self.headers['content-length']))
53
41
        if message == 'This is a message':
54
42
            self.send_response(httplib.CREATED)
57
45
        else:
58
46
            self.send_error(httplib.BAD_REQUEST)
59
47
 
60
 
    def do_GET(self):
61
 
        scheme, netloc, path, params, query, fragments = urlparse(self.path)
62
 
        archive = os.path.split(path)[1]
63
 
        self.send_response(httplib.OK)
64
 
        self.end_headers()
65
 
        print "path: %s" % self.path
66
 
        self.wfile.write(simplejson.dumps(self.server.messages[archive]))
67
 
 
68
 
 
69
 
def fake_grackle_service(client, messages=None):
70
 
    if messages is None:
71
 
        messages = {}
72
 
    return ForkedFake(client.port, messages)
 
48
 
 
49
def run_service(port):
 
50
    service = HTTPServer(('', port), FakeGrackleRequestHandler)
 
51
    service.serve_forever()
 
52
 
73
53
 
74
54
 
75
55
class TestPutMessage(TestCase):
76
56
 
77
57
    def test_put_message(self):
78
 
        client = GrackleClient('localhost', 8436)
79
 
        with fake_grackle_service(client):
 
58
        client = GrackleClient('localhost', 8435)
 
59
        with Forked(run_service, client.port):
80
60
            client.put_message('arch1', 'asdf', StringIO('This is a message'))
81
61
            with ExpectedException(Exception, 'wtf'):
82
62
                client.put_message('arch1', 'asdf',
83
63
                    StringIO('This is not a message'))
84
 
 
85
 
 
86
 
class TestGetMessages(TestCase):
87
 
 
88
 
    def test_get_messages(self):
89
 
        client = GrackleClient('localhost', 8435)
90
 
        with fake_grackle_service(client,
91
 
            {'baz':
92
 
            [{'message-id': 'foo'},
93
 
             {'message-id': 'bar'}]}):
94
 
            response = client.get_messages('baz', message_ids=['a'])
95
 
        self.assertEqual(['bar', 'foo'], sorted(m['message-id'] for m in
96
 
            response))