/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/tests/matchers.py

  • Committer: Andrew Bennetts
  • Date: 2011-06-09 07:38:32 UTC
  • mto: This revision was merged to the branch mainline in revision 5964.
  • Revision ID: andrew.bennetts@canonical.com-20110609073832-dt6oww033iexli4l
Fix thinko in wording regarding stacking invariants and revisions with multiple parents.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/matchers.py
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
 
"""Matchers for breezy.
 
17
"""Matchers for bzrlib.
18
18
 
19
 
Primarily test support, Matchers are used by self.assertThat in the breezy
 
19
Primarily test support, Matchers are used by self.assertThat in the bzrlib
20
20
test suite. A matcher is a stateful test helper which can be used to determine
21
21
if a passed object 'matches', much like a regex. If the object does not match
22
22
the mismatch can be described in a human readable fashion. assertThat then
27
27
"""
28
28
 
29
29
__all__ = [
30
 
    'HasLayout',
31
 
    'HasPathRelations',
32
 
    'MatchesAncestry',
33
30
    'ReturnsUnlockable',
34
 
    'RevisionHistoryMatches',
35
31
    ]
36
32
 
37
 
from .. import (
38
 
    osutils,
39
 
    revision as _mod_revision,
40
 
    )
41
 
 
42
 
from ..tree import InterTree
43
 
 
44
 
from testtools.matchers import Equals, Mismatch, Matcher
 
33
from testtools.matchers import Mismatch, Matcher
45
34
 
46
35
 
47
36
class ReturnsUnlockable(Matcher):
59
48
        self.lockable_thing = lockable_thing
60
49
 
61
50
    def __str__(self):
62
 
        return ('ReturnsUnlockable(lockable_thing=%s)' %
63
 
                self.lockable_thing)
 
51
        return ('ReturnsUnlockable(lockable_thing=%s)' % 
 
52
            self.lockable_thing)
64
53
 
65
54
    def match(self, lock_method):
66
55
        lock_method().unlock()
77
66
 
78
67
    def describe(self):
79
68
        return "%s is locked" % self.lockable_thing
80
 
 
81
 
 
82
 
class _AncestryMismatch(Mismatch):
83
 
    """Ancestry matching mismatch."""
84
 
 
85
 
    def __init__(self, tip_revision, got, expected):
86
 
        self.tip_revision = tip_revision
87
 
        self.got = got
88
 
        self.expected = expected
89
 
 
90
 
    def describe(self):
91
 
        return "mismatched ancestry for revision %r was %r, expected %r" % (
92
 
            self.tip_revision, self.got, self.expected)
93
 
 
94
 
 
95
 
class MatchesAncestry(Matcher):
96
 
    """A matcher that checks the ancestry of a particular revision.
97
 
 
98
 
    :ivar graph: Graph in which to check the ancestry
99
 
    :ivar revision_id: Revision id of the revision
100
 
    """
101
 
 
102
 
    def __init__(self, repository, revision_id):
103
 
        Matcher.__init__(self)
104
 
        self.repository = repository
105
 
        self.revision_id = revision_id
106
 
 
107
 
    def __str__(self):
108
 
        return ('MatchesAncestry(repository=%r, revision_id=%r)' % (
109
 
            self.repository, self.revision_id))
110
 
 
111
 
    def match(self, expected):
112
 
        with self.repository.lock_read():
113
 
            graph = self.repository.get_graph()
114
 
            got = [r for r, p in graph.iter_ancestry([self.revision_id])]
115
 
            if _mod_revision.NULL_REVISION in got:
116
 
                got.remove(_mod_revision.NULL_REVISION)
117
 
        if sorted(got) != sorted(expected):
118
 
            return _AncestryMismatch(self.revision_id, sorted(got),
119
 
                                     sorted(expected))
120
 
 
121
 
 
122
 
class HasLayout(Matcher):
123
 
    """A matcher that checks if a tree has a specific layout.
124
 
 
125
 
    :ivar entries: List of expected entries, as (path, file_id) pairs.
126
 
    """
127
 
 
128
 
    def __init__(self, entries):
129
 
        Matcher.__init__(self)
130
 
        self.entries = entries
131
 
 
132
 
    def get_tree_layout(self, tree, include_file_ids):
133
 
        """Get the (path, file_id) pairs for the current tree."""
134
 
        with tree.lock_read():
135
 
            for path, ie in tree.iter_entries_by_dir():
136
 
                if path != u'':
137
 
                    path += ie.kind_character()
138
 
                if include_file_ids:
139
 
                    yield (path, ie.file_id)
140
 
                else:
141
 
                    yield path
142
 
 
143
 
    @staticmethod
144
 
    def _strip_unreferenced_directories(entries):
145
 
        """Strip all directories that don't (in)directly contain any files.
146
 
 
147
 
        :param entries: List of path strings or (path, ie) tuples to process
148
 
        """
149
 
        directories = []
150
 
        for entry in entries:
151
 
            if isinstance(entry, str):
152
 
                path = entry
153
 
            else:
154
 
                path = entry[0]
155
 
            if not path or path[-1] == "/":
156
 
                # directory
157
 
                directories.append((path, entry))
158
 
            else:
159
 
                # Yield the referenced parent directories
160
 
                for dirpath, direntry in directories:
161
 
                    if osutils.is_inside(dirpath, path):
162
 
                        yield direntry
163
 
                directories = []
164
 
                yield entry
165
 
 
166
 
    def __str__(self):
167
 
        return 'HasLayout(%r)' % self.entries
168
 
 
169
 
    def match(self, tree):
170
 
        include_file_ids = self.entries and not isinstance(
171
 
            self.entries[0], str)
172
 
        actual = list(self.get_tree_layout(
173
 
            tree, include_file_ids=include_file_ids))
174
 
        if not tree.has_versioned_directories():
175
 
            entries = list(self._strip_unreferenced_directories(self.entries))
176
 
        else:
177
 
            entries = self.entries
178
 
        return Equals(entries).match(actual)
179
 
 
180
 
 
181
 
class HasPathRelations(Matcher):
182
 
    """Matcher verifies that paths have a relation to those in another tree.
183
 
 
184
 
    :ivar previous_tree: tree to compare to
185
 
    :ivar previous_entries: List of expected entries, as (path, previous_path) pairs.
186
 
    """
187
 
 
188
 
    def __init__(self, previous_tree, previous_entries):
189
 
        Matcher.__init__(self)
190
 
        self.previous_tree = previous_tree
191
 
        self.previous_entries = previous_entries
192
 
 
193
 
    def get_path_map(self, tree):
194
 
        """Get the (path, previous_path) pairs for the current tree."""
195
 
        previous_intertree = InterTree.get(self.previous_tree, tree)
196
 
        with tree.lock_read(), self.previous_tree.lock_read():
197
 
            for path, ie in tree.iter_entries_by_dir():
198
 
                if tree.supports_rename_tracking():
199
 
                    previous_path = previous_intertree.find_source_path(path)
200
 
                else:
201
 
                    if self.previous_tree.is_versioned(path):
202
 
                        previous_path = path
203
 
                    else:
204
 
                        previous_path = None
205
 
                if previous_path:
206
 
                    kind = self.previous_tree.kind(previous_path)
207
 
                    if kind == 'directory':
208
 
                        previous_path += '/'
209
 
                if path == u'':
210
 
                    yield (u"", previous_path)
211
 
                else:
212
 
                    yield (path + ie.kind_character(), previous_path)
213
 
 
214
 
    @staticmethod
215
 
    def _strip_unreferenced_directories(entries):
216
 
        """Strip all directories that don't (in)directly contain any files.
217
 
 
218
 
        :param entries: List of path strings or (path, previous_path) tuples to process
219
 
        """
220
 
        directory_used = set()
221
 
        directories = []
222
 
        for (path, previous_path) in entries:
223
 
            if not path or path[-1] == "/":
224
 
                # directory
225
 
                directories.append((path, previous_path))
226
 
            else:
227
 
                # Yield the referenced parent directories
228
 
                for direntry in directories:
229
 
                    if osutils.is_inside(direntry[0], path):
230
 
                        directory_used.add(direntry[0])
231
 
        for (path, previous_path) in entries:
232
 
            if (not path.endswith("/")) or path in directory_used:
233
 
                yield (path, previous_path)
234
 
 
235
 
    def __str__(self):
236
 
        return 'HasPathRelations(%r, %r)' % (self.previous_tree, self.previous_entries)
237
 
 
238
 
    def match(self, tree):
239
 
        actual = list(self.get_path_map(tree))
240
 
        if not tree.has_versioned_directories():
241
 
            entries = list(self._strip_unreferenced_directories(
242
 
                self.previous_entries))
243
 
        else:
244
 
            entries = self.previous_entries
245
 
        if not tree.supports_rename_tracking():
246
 
            entries = [
247
 
                (path, path if self.previous_tree.is_versioned(path) else None)
248
 
                for (path, previous_path) in entries]
249
 
        return Equals(entries).match(actual)
250
 
 
251
 
 
252
 
class RevisionHistoryMatches(Matcher):
253
 
    """A matcher that checks if a branch has a specific revision history.
254
 
 
255
 
    :ivar history: Revision history, as list of revisions. Oldest first.
256
 
    """
257
 
 
258
 
    def __init__(self, history):
259
 
        Matcher.__init__(self)
260
 
        self.expected = history
261
 
 
262
 
    def __str__(self):
263
 
        return 'RevisionHistoryMatches(%r)' % self.expected
264
 
 
265
 
    def match(self, branch):
266
 
        with branch.lock_read():
267
 
            graph = branch.repository.get_graph()
268
 
            history = list(graph.iter_lefthand_ancestry(
269
 
                branch.last_revision(), [_mod_revision.NULL_REVISION]))
270
 
            history.reverse()
271
 
        return Equals(self.expected).match(history)