← Back to branch summary

~launchpad-pqm/launchpad/devel

« back to all changes in this revision

Viewing changes to lib/lp/testing/__init__.py

Merged rocketfuel into disco.

Show diffs side-by-side

added added

removed removed

Lines of Context:
lib/lp/testing/__init__.py
29
29
    'logout',
30
30
    'map_branch_contents',
31
31
    'normalize_whitespace',
32
 
    'nonblocking_readline',
33
32
    'oauth_access_token_for',
34
33
    'person_logged_in',
35
34
    'quote_jquery_expression',
70
69
import os
71
70
from pprint import pformat
72
71
import re
73
 
from select import select
74
72
import shutil
75
73
import subprocess
76
74
import sys
1327
1325
def extract_lp_cache(text):
1328
1326
    match = re.search(r'<script>LP.cache = (\{.*\});</script>', text)
1329
1327
    return simplejson.loads(match.group(1))
1330
 
 
1331
 
 
1332
 
def nonblocking_readline(instream, timeout):
1333
 
    """Non-blocking readline.
1334
 
 
1335
 
    Files must provide a valid fileno() method. This is a test helper
1336
 
    as it is inefficient and unlikely useful for production code.
1337
 
    """
1338
 
    result = StringIO()
1339
 
    start = now = time.time()
1340
 
    deadline = start + timeout
1341
 
    while (now < deadline and not result.getvalue().endswith('\n')):
1342
 
        rlist = select([instream], [], [], deadline - now)
1343
 
        if rlist:
1344
 
            # Reading 1 character at a time is inefficient, but means
1345
 
            # we don't need to implement put-back.
1346
 
            next_char = os.read(instream.fileno(), 1)
1347
 
            if next_char == "":
1348
 
                break  # EOF
1349
 
            result.write(next_char)
1350
 
        now = time.time()
1351
 
    return result.getvalue()