← Back to branch summary

~launchpad-pqm/launchpad/devel

« back to all changes in this revision

Viewing changes to utilities/migrater/migrater.py

  • Committer: Curtis Hovey
  • Date: 2011-12-23 23:50:31 UTC
  • mto: This revision was merged to the branch mainline in revision 14599.
  • Revision ID: curtis.hovey@canonical.com-20111223235031-uexud7yr0mrigivw
Remove doctest verifying a method that is not used out side of layers.
Removed doc.

Show diffs side-by-side

added added

removed removed

Lines of Context:
utilities/migrater/migrater.py
296
296
        test_doc = test_doc_file.read()
297
297
    finally:
298
298
        test_doc_file.close()
 
299
    if doctests is not None:
 
300
        test_doc = test_doc.replace('special = {}', get_special(doctests))
299
301
    test_doc_path = os.path.join(dir_path, file_name)
300
302
    if os.path.isfile(test_doc_path):
301
303
        # This harness was made in a previous run.
309
311
    bzr_add([test_doc_path])
310
312
 
311
313
 
 
314
def get_special(doctests):
 
315
    """extract the special setups from test_system_documentation."""
 
316
    system_doc_lines = []
 
317
    special_lines = []
 
318
    doctest_pattern = re.compile(r"^    '(%s)[^']*':" % '|'.join(doctests))
 
319
    system_doc_path = os.path.join(
 
320
        OLD_TOP, 'ftests', 'test_system_documentation.py')
 
321
    system_doc = open(system_doc_path)
 
322
    try:
 
323
        in_special = False
 
324
        for line in system_doc:
 
325
            match = doctest_pattern.match(line)
 
326
            if match is not None:
 
327
                in_special = True
 
328
                print '    * Extracting special test for %s' % match.group(1)
 
329
            if in_special:
 
330
                special_lines.append(line.replace('        ', '    '))
 
331
            else:
 
332
                system_doc_lines.append(line)
 
333
            if in_special and '),' in line:
 
334
                in_special = False
 
335
    finally:
 
336
        system_doc.close()
 
337
    if len(special_lines) == 0:
 
338
        # There was nothing to extract.
 
339
        return 'special = {}'
 
340
    # Get the setup and teardown functions.
 
341
    special_lines.insert(0, 'special = {\n')
 
342
    special_lines.append('    }')
 
343
    code = ''.join(special_lines)
 
344
    helper_pattern = re.compile(r'\b(setUp|tearDown)=(\w*)\b')
 
345
    helpers = set(match.group(2) for match in helper_pattern.finditer(code))
 
346
    if 'setUp' in helpers:
 
347
        helpers.remove('setUp')
 
348
    if 'tearDown' in helpers:
 
349
        helpers.remove('tearDown')
 
350
    # Extract the setup and teardown functions.
 
351
    lines = list(system_doc_lines)
 
352
    system_doc_lines = []
 
353
    helper_lines = []
 
354
    helper_pattern = re.compile(r'^def (%s)\b' % '|'.join(helpers))
 
355
    in_helper = False
 
356
    for line in lines:
 
357
        if in_helper and len(line) > 1 and line[0] != ' ':
 
358
            in_helper = False
 
359
        match = helper_pattern.match(line)
 
360
        if match is not None:
 
361
            in_helper = True
 
362
            print '    * Extracting special function for %s' % match.group(1)
 
363
        if in_helper:
 
364
            helper_lines.append(line)
 
365
        else:
 
366
            system_doc_lines.append(line)
 
367
    if len(helper_lines) > 0:
 
368
        code = ''.join(helper_lines) + code
 
369
    # Write the smaller test_system_documentation.py.
 
370
    system_doc = open(system_doc_path, 'w')
 
371
    try:
 
372
        system_doc.write(''.join(system_doc_lines))
 
373
    finally:
 
374
        system_doc.close()
 
375
    # Return the local app's specials code.
 
376
    special_lines.insert(0, 'special = {\n')
 
377
    special_lines.append('    }')
 
378
    return code
 
379
 
 
380
 
312
381
def handle_py_file(old_path, new_path, subdir):
313
382
    """Migrate python files."""
314
383
    if subdir in APP_DIRECTORIES: