← 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 12:05:22 UTC
  • Revision ID: aaron@canonical.com-20120110120522-433wzsf5lw1eo0qq
Initial test.

Show diffs side-by-side

added added

removed removed

Lines of Context:
grackle/tests/test_client.py
2
2
    HTTPServer,
3
3
    BaseHTTPRequestHandler,
4
4
    )
5
 
import httplib
 
5
import contextlib
6
6
import os
7
 
from signal import SIGKILL
8
7
from StringIO import StringIO
9
8
from unittest import TestCase
10
9
 
11
 
from testtools import ExpectedException
12
 
 
13
10
from grackle import client
14
11
 
15
12
 
16
 
class Forked:
17
 
 
18
 
    def __init__(self, func_or_method):
19
 
        self.func_or_method = func_or_method
20
 
        self.pid = None
21
 
 
22
 
    def __enter__(self):
23
 
        pid = os.fork()
24
 
        if pid != 0:
25
 
            self.pid = pid
26
 
            return
27
 
        self.func_or_method()
28
 
 
29
 
 
30
 
    def __exit__(self, exc_type, exc_val, traceback):
31
 
        os.kill(self.pid, SIGKILL)
32
 
 
33
 
 
34
 
class FakeGrackleRequestHandler(BaseHTTPRequestHandler):
35
 
 
36
 
    def do_POST(self):
37
 
        message = self.rfile.read(int(self.headers['content-length']))
38
 
        if message == 'This is a message':
39
 
            self.send_response(httplib.CREATED)
40
 
            self.end_headers()
41
 
            self.wfile.close()
42
 
        else:
43
 
            self.send_error(httplib.BAD_REQUEST)
44
 
 
45
 
 
46
 
def run_service():
47
 
    service = HTTPServer(('', 8435), FakeGrackleRequestHandler)
48
 
    service.serve_forever()
49
 
 
 
13
def fake_grackle_service():
 
14
    pid = os.fork()
 
15
    if pid != 0:
 
16
        return pid
 
17
    service = HTTPServer(('', 8435), BaseHTTPRequestHandler)
 
18
    service.handle_request()
 
19
    os._exit(0)
50
20
 
51
21
 
52
22
class TestPutMessage(TestCase):
53
23
 
54
24
    def test_put_message(self):
55
 
        with Forked(run_service):
56
 
            client.put_message('arch1', StringIO('This is a message'))
57
 
            with ExpectedException(Exception, 'wtf'):
58
 
                client.put_message('arch1', StringIO('This is not a message'))
 
25
        service = fake_grackle_service()
 
26
        client.put_message('arch1', StringIO('This is a message'))