/usr/share/pyshared/bzrlib/tests/fixtures.py is in python-bzrlib.tests 2.6.0~bzr6526-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 | # Copyright (C) 2010 Canonical Ltd
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
"""Fixtures that can be used within tests.
Fixtures can be created during a test as a way to separate out creation of
objects to test. Fixture objects can hold some state so that different
objects created during a test instance can be related. Normally a fixture
should live only for the duration of a single test, and its tearDown method
should be passed to `addCleanup` on the test.
"""
import itertools
def generate_unicode_names():
"""Generate a sequence of arbitrary unique unicode names.
By default they are not representable in ascii.
>>> gen = generate_unicode_names()
>>> n1 = gen.next()
>>> n2 = gen.next()
>>> type(n1)
<type 'unicode'>
>>> n1 == n2
False
>>> n1.encode('ascii', 'replace') == n1
False
"""
# include a mathematical symbol unlikely to be in 8-bit encodings
return (u"\N{SINE WAVE}%d" % x for x in itertools.count())
interesting_encodings = [
('iso-8859-1', False),
('ascii', False),
('cp850', False),
('utf-8', True),
('ucs-2', True),
]
def generate_unicode_encodings(universal_encoding=None):
"""Return a generator of unicode encoding names.
These can be passed to Python encode/decode/etc.
:param universal_encoding: True/False/None tristate to say whether the
generated encodings either can or cannot encode all unicode
strings.
>>> n1 = generate_unicode_names().next()
>>> enc = generate_unicode_encodings(universal_encoding=True).next()
>>> enc2 = generate_unicode_encodings(universal_encoding=False).next()
>>> n1.encode(enc).decode(enc) == n1
True
>>> try:
... n1.encode(enc2).decode(enc2)
... except UnicodeError:
... print 'fail'
fail
"""
# TODO: check they're supported on this platform?
if universal_encoding is not None:
e = [n for (n, u) in interesting_encodings if u == universal_encoding]
else:
e = [n for (n, u) in interesting_encodings]
return itertools.cycle(iter(e))
class RecordingContextManager(object):
"""A context manager that records."""
def __init__(self):
self._calls = []
def __enter__(self):
self._calls.append('__enter__')
return self # This is bound to the 'as' clause in a with statement.
def __exit__(self, exc_type, exc_val, exc_tb):
self._calls.append('__exit__')
return False # propogate exceptions.
def build_branch_with_non_ancestral_rev(branch_builder):
"""Builds a branch with a rev not in the ancestry of the tip.
This is the revision graph::
rev-2
|
rev-1
|
(null)
The branch tip is 'rev-1'. 'rev-2' is present in the branch's repository,
but is not part of rev-1's ancestry.
:param branch_builder: A BranchBuilder (e.g. from
TestCaseWithMemoryTransport.make_branch_builder).
:returns: the new branch
"""
# Make a sequence of two commits
branch_builder.build_commit(message="Rev 1", rev_id='rev-1')
branch_builder.build_commit(message="Rev 2", rev_id='rev-2')
# Move the branch tip back to the first commit
source = branch_builder.get_branch()
source.set_last_revision_info(1, 'rev-1')
return source
def make_branch_and_populated_tree(testcase):
"""Make a simple branch and tree.
The tree holds some added but uncommitted files.
"""
# TODO: Either accept or return the names of the files, so the caller
# doesn't need to be bound to the particular files created? -- mbp
# 20110705
tree = testcase.make_branch_and_tree('t')
testcase.build_tree_contents([('t/hello', 'hello world')])
tree.add(['hello'], ['hello-id'])
return tree
class TimeoutFixture(object):
"""Kill a test with sigalarm if it runs too long.
Only works on Unix at present.
"""
def __init__(self, timeout_secs):
import signal
self.timeout_secs = timeout_secs
self.alarm_fn = getattr(signal, 'alarm', None)
def setUp(self):
if self.alarm_fn is not None:
self.alarm_fn(self.timeout_secs)
def cleanUp(self):
if self.alarm_fn is not None:
self.alarm_fn(0)
|