← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_source.py

Merge the 2.0 branch, resolve one conflict.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_source.py
372
372
            self.fail(
373
373
                "these files contain an assert statement and should not:\n%s"
374
374
                % '\n'.join(badfiles))
 
375
 
 
376
    def test_extension_exceptions(self):
 
377
        """Extension functions should propagate exceptions.
 
378
 
 
379
        Either they should return an object, have an 'except' clause, or have a
 
380
        "# cannot_raise" to indicate that we've audited them and defined them as not
 
381
        raising exceptions.
 
382
        """
 
383
        both_exc_and_no_exc = []
 
384
        missing_except = []
 
385
        class_re = re.compile(r'^(cdef\s+)?class (\w+).*:', re.MULTILINE)
 
386
        except_re = re.compile(r'cdef\s*' # start with cdef
 
387
                               r'([\w *]*?)\s*' # this is the return signature
 
388
                               r'(\w+)\s*\(' # the function name
 
389
                               r'[^)]*\)\s*' # parameters
 
390
                               r'(.*)\s*:' # the except clause
 
391
                               r'\s*(#\s*cannot[- _]raise)?' # cannot raise comment
 
392
                              )
 
393
        for fname, text in self.get_source_file_contents(
 
394
                extensions=('.pyx',)):
 
395
            known_classes = set([m[1] for m in class_re.findall(text)])
 
396
            cdefs = except_re.findall(text)
 
397
            for sig, func, exc_clause, no_exc_comment in cdefs:
 
398
                if not sig or sig in known_classes:
 
399
                    sig = 'object'
 
400
                if exc_clause and no_exc_comment:
 
401
                    both_exc_and_no_exc.append((fname, func))
 
402
                if sig != 'object' and not (exc_clause or no_exc_comment):
 
403
                    missing_except.append((fname, func))
 
404
        error_msg = []
 
405
        if both_exc_and_no_exc:
 
406
            error_msg.append('The following functions had "cannot raise" comments'
 
407
                             ' but did have an except clause set:')
 
408
            for fname, func in both_exc_and_no_exc:
 
409
                error_msg.append('%s:%s' % (fname, func))
 
410
            error_msg.extend(('', ''))
 
411
        if missing_except:
 
412
            error_msg.append('The following functions have fixed return types,'
 
413
                             ' but no except clause.')
 
414
            error_msg.append('Either add an except or append "# cannot_raise".')
 
415
            for fname, func in missing_except:
 
416
                error_msg.append('%s:%s' % (fname, func))
 
417
            error_msg.extend(('', ''))
 
418
        if error_msg:
 
419
            self.fail('\n'.join(error_msg))