← 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-11 11:18:15 UTC
  • Revision ID: aaron@canonical.com-20120111111815-0ecv9fihdao2u68o
Start working on GET.

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)
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]))
 
48
 
 
49
def run_service(port, messages):
 
50
    service = HTTPServer(('', port), FakeGrackleRequestHandler)
 
51
    service.messages = messages
 
52
    service.serve_forever()
67
53
 
68
54
 
69
55
def fake_grackle_service(client, messages=None):
70
56
    if messages is None:
71
57
        messages = {}
72
 
    return ForkedFake(client.port, messages)
 
58
    return Forked(run_service, client.port, messages)
73
59
 
74
60
 
75
61
class TestPutMessage(TestCase):
76
62
 
77
63
    def test_put_message(self):
78
 
        client = GrackleClient('localhost', 8436)
 
64
        client = GrackleClient('localhost', 8435)
79
65
        with fake_grackle_service(client):
 
66
            import time; time.sleep(0.5)
80
67
            client.put_message('arch1', 'asdf', StringIO('This is a message'))
81
68
            with ExpectedException(Exception, 'wtf'):
82
69
                client.put_message('arch1', 'asdf',
91
78
            {'baz':
92
79
            [{'message-id': 'foo'},
93
80
             {'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))
 
81
            response = client.get_messages('baz')
 
82
        self.assertEqual(['bar', 'foo'], sorted(response.keys()))