← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/per_inventory/basics.py

  • Committer: Andrew Bennetts
  • Date: 2010-01-12 03:53:21 UTC
  • mfrom: (4948 +trunk)
  • mto: This revision was merged to the branch mainline in revision 4964.
  • Revision ID: andrew.bennetts@canonical.com-20100112035321-hofpz5p10224ryj3
Merge lp:bzr, resolving conflicts.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/per_inventory/basics.py
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
"""Tests for different inventory implementations"""
18
18
 
21
21
 
22
22
from bzrlib import (
23
23
        errors,
 
24
        inventory,
 
25
        osutils,
24
26
        )
25
27
 
26
28
from bzrlib.inventory import (
28
30
        InventoryEntry,
29
31
        InventoryFile,
30
32
        InventoryLink,
31
 
        ROOT_ID,
32
33
        TreeReference,
33
34
        )
34
35
 
35
 
from bzrlib.tests import (
36
 
        TestCase,
37
 
        multiply_tests_from_modules,
38
 
        )
39
 
 
40
 
 
41
 
class TestInventoryBasics(TestCase):
42
 
    # Most of these were moved the rather old bzrlib.tests.test_inv module
43
 
    
44
 
    def make_inventory(self, root_id):
45
 
        return self.inventory_class(root_id=root_id)
46
 
 
47
 
    def test_creation_from_root_id(self):
48
 
        # iff a root id is passed to the constructor, a root directory is made
49
 
        inv = self.make_inventory(root_id='tree-root')
50
 
        self.assertNotEqual(None, inv.root)
51
 
        self.assertEqual('tree-root', inv.root.file_id)
52
 
 
53
 
    def test_add_path_of_root(self):
54
 
        # if no root id is given at creation time, there is no root directory
55
 
        inv = self.make_inventory(root_id=None)
56
 
        self.assertIs(None, inv.root)
57
 
        # add a root entry by adding its path
58
 
        ie = inv.add_path("", "directory", "my-root")
59
 
        self.assertEqual("my-root", ie.file_id)
60
 
        self.assertIs(ie, inv.root)
61
 
 
62
 
    def test_add_path(self):
63
 
        inv = self.make_inventory(root_id='tree_root')
64
 
        ie = inv.add_path('hello', 'file', 'hello-id')
65
 
        self.assertEqual('hello-id', ie.file_id)
66
 
        self.assertEqual('file', ie.kind)
67
 
 
68
 
    def test_copy(self):
69
 
        """Make sure copy() works and creates a deep copy."""
70
 
        inv = self.make_inventory(root_id='some-tree-root')
71
 
        ie = inv.add_path('hello', 'file', 'hello-id')
72
 
        inv2 = inv.copy()
73
 
        inv.root.file_id = 'some-new-root'
74
 
        ie.name = 'file2'
75
 
        self.assertEqual('some-tree-root', inv2.root.file_id)
76
 
        self.assertEqual('hello', inv2['hello-id'].name)
77
 
 
78
 
    def test_copy_empty(self):
79
 
        """Make sure an empty inventory can be copied."""
80
 
        inv = self.make_inventory(root_id=None)
81
 
        inv2 = inv.copy()
82
 
        self.assertIs(None, inv2.root)
83
 
 
84
 
    def test_copy_copies_root_revision(self):
85
 
        """Make sure the revision of the root gets copied."""
86
 
        inv = self.make_inventory(root_id='someroot')
87
 
        inv.root.revision = 'therev'
88
 
        inv2 = inv.copy()
89
 
        self.assertEquals('someroot', inv2.root.file_id)
90
 
        self.assertEquals('therev', inv2.root.revision)
 
36
from bzrlib.tests.per_inventory import TestCaseWithInventory
 
37
 
 
38
 
 
39
 
 
40
class TestInventory(TestCaseWithInventory):
 
41
 
 
42
    def make_init_inventory(self):
 
43
        inv = inventory.Inventory('tree-root')
 
44
        inv.revision = 'initial-rev'
 
45
        inv.root.revision = 'initial-rev'
 
46
        return self.inv_to_test_inv(inv)
 
47
 
 
48
    def make_file(self, file_id, name, parent_id, content='content\n',
 
49
                  revision='new-test-rev'):
 
50
        ie = InventoryFile(file_id, name, parent_id)
 
51
        ie.text_sha1 = osutils.sha_string(content)
 
52
        ie.text_size = len(content)
 
53
        ie.revision = revision
 
54
        return ie
 
55
 
 
56
    def make_link(self, file_id, name, parent_id, target='link-target\n'):
 
57
        ie = InventoryLink(file_id, name, parent_id)
 
58
        ie.symlink_target = target
 
59
        return ie
 
60
 
 
61
    def prepare_inv_with_nested_dirs(self):
 
62
        inv = inventory.Inventory('tree-root')
 
63
        for args in [('src', 'directory', 'src-id'),
 
64
                     ('doc', 'directory', 'doc-id'),
 
65
                     ('src/hello.c', 'file', 'hello-id'),
 
66
                     ('src/bye.c', 'file', 'bye-id'),
 
67
                     ('zz', 'file', 'zz-id'),
 
68
                     ('src/sub/', 'directory', 'sub-id'),
 
69
                     ('src/zz.c', 'file', 'zzc-id'),
 
70
                     ('src/sub/a', 'file', 'a-id'),
 
71
                     ('Makefile', 'file', 'makefile-id')]:
 
72
            ie = inv.add_path(*args)
 
73
            if args[1] == 'file':
 
74
                ie.text_sha1 = osutils.sha_string('content\n')
 
75
                ie.text_size = len('content\n')
 
76
        return self.inv_to_test_inv(inv)
 
77
 
 
78
 
 
79
class TestInventoryCreateByApplyDelta(TestInventory):
 
80
    """A subset of the inventory delta application tests.
 
81
 
 
82
    See test_inv which has comprehensive delta application tests for
 
83
    inventories, dirstate, and repository based inventories.
 
84
    """
 
85
    def test_add(self):
 
86
        inv = self.make_init_inventory()
 
87
        inv = inv.create_by_apply_delta([
 
88
            (None, "a", "a-id", self.make_file('a-id', 'a', 'tree-root')),
 
89
            ], 'new-test-rev')
 
90
        self.assertEqual('a', inv.id2path('a-id'))
 
91
 
 
92
    def test_delete(self):
 
93
        inv = self.make_init_inventory()
 
94
        inv = inv.create_by_apply_delta([
 
95
            (None, "a", "a-id", self.make_file('a-id', 'a', 'tree-root')),
 
96
            ], 'new-rev-1')
 
97
        self.assertEqual('a', inv.id2path('a-id'))
 
98
        inv = inv.create_by_apply_delta([
 
99
            ("a", None, "a-id", None),
 
100
            ], 'new-rev-2')
 
101
        self.assertRaises(errors.NoSuchId, inv.id2path, 'a-id')
 
102
 
 
103
    def test_rename(self):
 
104
        inv = self.make_init_inventory()
 
105
        inv = inv.create_by_apply_delta([
 
106
            (None, "a", "a-id", self.make_file('a-id', 'a', 'tree-root')),
 
107
            ], 'new-rev-1')
 
108
        self.assertEqual('a', inv.id2path('a-id'))
 
109
        a_ie = inv['a-id']
 
110
        b_ie = self.make_file(a_ie.file_id, "b", a_ie.parent_id)
 
111
        inv = inv.create_by_apply_delta([("a", "b", "a-id", b_ie)], 'new-rev-2')
 
112
        self.assertEqual("b", inv.id2path('a-id'))
 
113
 
 
114
    def test_illegal(self):
 
115
        # A file-id cannot appear in a delta more than once
 
116
        inv = self.make_init_inventory()
 
117
        self.assertRaises(errors.InconsistentDelta, inv.create_by_apply_delta, [
 
118
            (None, "a", "id-1", self.make_file('id-1', 'a', 'tree-root')),
 
119
            (None, "b", "id-1", self.make_file('id-1', 'b', 'tree-root')),
 
120
            ], 'new-rev-1')
 
121
 
 
122
 
 
123
class TestInventoryReads(TestInventory):
91
124
 
92
125
    def test_is_root(self):
93
126
        """Ensure our root-checking code is accurate."""
94
 
        inv = self.make_inventory('TREE_ROOT')
95
 
        self.assertTrue(inv.is_root('TREE_ROOT'))
 
127
        inv = self.make_init_inventory()
 
128
        self.assertTrue(inv.is_root('tree-root'))
96
129
        self.assertFalse(inv.is_root('booga'))
97
 
        inv.root.file_id = 'booga'
 
130
        ie = inv['tree-root'].copy()
 
131
        ie.file_id = 'booga'
 
132
        inv = inv.create_by_apply_delta([("", None, "tree-root", None),
 
133
                                         (None, "", "booga", ie)], 'new-rev-2')
98
134
        self.assertFalse(inv.is_root('TREE_ROOT'))
99
135
        self.assertTrue(inv.is_root('booga'))
100
 
        # works properly even if no root is set
101
 
        inv.root = None
102
 
        self.assertFalse(inv.is_root('TREE_ROOT'))
103
 
        self.assertFalse(inv.is_root('booga'))
104
 
 
105
 
    def test_create_tree_reference(self):
106
 
        inv = self.make_inventory('tree-root-123')
107
 
        inv.add(TreeReference('nested-id', 'nested', parent_id='tree-root-123',
108
 
                              revision='rev', reference_revision='rev2'))
109
 
 
110
 
    def test_error_encoding(self):
111
 
        inv = self.make_inventory('tree-root')
112
 
        inv.add(InventoryFile('a-id', u'\u1234', 'tree-root'))
113
 
        try:
114
 
            inv.add(InventoryFile('b-id', u'\u1234', 'tree-root'))
115
 
        except errors.BzrError, e:
116
 
            self.assertContainsRe(str(e), u'\u1234'.encode('utf-8'))
117
 
        else:
118
 
            self.fail('BzrError not raised')
119
136
 
120
137
    def test_ids(self):
121
138
        """Test detection of files within selected directories."""
122
 
        inv = self.make_inventory(ROOT_ID)
 
139
        inv = inventory.Inventory('TREE_ROOT')
123
140
        for args in [('src', 'directory', 'src-id'),
124
141
                     ('doc', 'directory', 'doc-id'),
125
142
                     ('src/hello.c', 'file'),
126
143
                     ('src/bye.c', 'file', 'bye-id'),
127
144
                     ('Makefile', 'file')]:
128
 
            inv.add_path(*args)
 
145
            ie = inv.add_path(*args)
 
146
            if args[1] == 'file':
 
147
                ie.text_sha1 = osutils.sha_string('content\n')
 
148
                ie.text_size = len('content\n')
 
149
        inv = self.inv_to_test_inv(inv)
129
150
        self.assertEqual(inv.path2id('src'), 'src-id')
130
151
        self.assertEqual(inv.path2id('src/bye.c'), 'bye-id')
131
 
        self.assert_('src-id' in inv)
 
152
        self.assertTrue('src-id' in inv)
132
153
 
133
154
    def test_non_directory_children(self):
134
155
        """Test path2id when a parent directory has no children"""
135
 
        inv = self.make_inventory('tree_root')
136
 
        inv.add(InventoryFile('file-id','file',
137
 
                                        parent_id='tree_root'))
138
 
        inv.add(InventoryLink('link-id','link',
139
 
                                        parent_id='tree_root'))
 
156
        inv = inventory.Inventory('tree-root')
 
157
        inv.add(self.make_file('file-id','file', 'tree-root'))
 
158
        inv.add(self.make_link('link-id','link', 'tree-root'))
140
159
        self.assertIs(None, inv.path2id('file/subfile'))
141
160
        self.assertIs(None, inv.path2id('link/subfile'))
142
161
 
143
162
    def test_iter_entries(self):
144
 
        inv = self.make_inventory('tree-root')
145
 
        for args in [('src', 'directory', 'src-id'),
146
 
                     ('doc', 'directory', 'doc-id'),
147
 
                     ('src/hello.c', 'file', 'hello-id'),
148
 
                     ('src/bye.c', 'file', 'bye-id'),
149
 
                     ('Makefile', 'file', 'makefile-id')]:
150
 
            inv.add_path(*args)
 
163
        inv = self.prepare_inv_with_nested_dirs()
 
164
 
 
165
        # Test all entries
151
166
        self.assertEqual([
152
167
            ('', 'tree-root'),
153
168
            ('Makefile', 'makefile-id'),
155
170
            ('src', 'src-id'),
156
171
            ('src/bye.c', 'bye-id'),
157
172
            ('src/hello.c', 'hello-id'),
 
173
            ('src/sub', 'sub-id'),
 
174
            ('src/sub/a', 'a-id'),
 
175
            ('src/zz.c', 'zzc-id'),
 
176
            ('zz', 'zz-id'),
158
177
            ], [(path, ie.file_id) for path, ie in inv.iter_entries()])
159
178
 
 
179
        # Test a subdirectory
 
180
        self.assertEqual([
 
181
            ('bye.c', 'bye-id'),
 
182
            ('hello.c', 'hello-id'),
 
183
            ('sub', 'sub-id'),
 
184
            ('sub/a', 'a-id'),
 
185
            ('zz.c', 'zzc-id'),
 
186
            ], [(path, ie.file_id) for path, ie in inv.iter_entries(
 
187
            from_dir='src-id')])
 
188
 
 
189
        # Test not recursing at the root level
 
190
        self.assertEqual([
 
191
            ('', 'tree-root'),
 
192
            ('Makefile', 'makefile-id'),
 
193
            ('doc', 'doc-id'),
 
194
            ('src', 'src-id'),
 
195
            ('zz', 'zz-id'),
 
196
            ], [(path, ie.file_id) for path, ie in inv.iter_entries(
 
197
            recursive=False)])
 
198
 
 
199
        # Test not recursing at a subdirectory level
 
200
        self.assertEqual([
 
201
            ('bye.c', 'bye-id'),
 
202
            ('hello.c', 'hello-id'),
 
203
            ('sub', 'sub-id'),
 
204
            ('zz.c', 'zzc-id'),
 
205
            ], [(path, ie.file_id) for path, ie in inv.iter_entries(
 
206
            from_dir='src-id', recursive=False)])
 
207
 
 
208
    def test_iter_just_entries(self):
 
209
        inv = self.prepare_inv_with_nested_dirs()
 
210
        self.assertEqual([
 
211
            'a-id',
 
212
            'bye-id',
 
213
            'doc-id',
 
214
            'hello-id',
 
215
            'makefile-id',
 
216
            'src-id',
 
217
            'sub-id',
 
218
            'tree-root',
 
219
            'zz-id',
 
220
            'zzc-id',
 
221
            ], sorted([ie.file_id for ie in inv.iter_just_entries()]))
 
222
 
160
223
    def test_iter_entries_by_dir(self):
161
 
        inv = self.make_inventory('tree-root')
162
 
        for args in [('src', 'directory', 'src-id'),
163
 
                     ('doc', 'directory', 'doc-id'),
164
 
                     ('src/hello.c', 'file', 'hello-id'),
165
 
                     ('src/bye.c', 'file', 'bye-id'),
166
 
                     ('zz', 'file', 'zz-id'),
167
 
                     ('src/sub/', 'directory', 'sub-id'),
168
 
                     ('src/zz.c', 'file', 'zzc-id'),
169
 
                     ('src/sub/a', 'file', 'a-id'),
170
 
                     ('Makefile', 'file', 'makefile-id')]:
171
 
            inv.add_path(*args)
 
224
        inv = self. prepare_inv_with_nested_dirs()
172
225
        self.assertEqual([
173
226
            ('', 'tree-root'),
174
227
            ('Makefile', 'makefile-id'),
232
285
            ('src/bye.c', 'bye-id'),
233
286
            ], [(path, ie.file_id) for path, ie in inv.iter_entries_by_dir(
234
287
                specific_file_ids=('bye-id',), yield_parents=True)])
235
 
 
236
 
    def test_add_recursive(self):
237
 
        parent = InventoryDirectory('src-id', 'src', 'tree-root')
238
 
        child = InventoryFile('hello-id', 'hello.c', 'src-id')
239
 
        parent.children[child.file_id] = child
240
 
        inv = self.make_inventory('tree-root')
241
 
        inv.add(parent)
242
 
        self.assertEqual('src/hello.c', inv.id2path('hello-id'))
 
288
 
 
289
 
 
290
class TestInventoryFiltering(TestInventory):
 
291
 
 
292
    def test_inv_filter_empty(self):
 
293
        inv = self.prepare_inv_with_nested_dirs()
 
294
        new_inv = inv.filter([])
 
295
        self.assertEqual([
 
296
            ('', 'tree-root'),
 
297
            ], [(path, ie.file_id) for path, ie in new_inv.iter_entries()])
 
298
    
 
299
    def test_inv_filter_files(self):
 
300
        inv = self.prepare_inv_with_nested_dirs()
 
301
        new_inv = inv.filter(['zz-id', 'hello-id', 'a-id'])
 
302
        self.assertEqual([
 
303
            ('', 'tree-root'),
 
304
            ('src', 'src-id'),
 
305
            ('src/hello.c', 'hello-id'),
 
306
            ('src/sub', 'sub-id'),
 
307
            ('src/sub/a', 'a-id'),
 
308
            ('zz', 'zz-id'),
 
309
            ], [(path, ie.file_id) for path, ie in new_inv.iter_entries()])
 
310
    
 
311
    def test_inv_filter_dirs(self):
 
312
        inv = self.prepare_inv_with_nested_dirs()
 
313
        new_inv = inv.filter(['doc-id', 'sub-id'])
 
314
        self.assertEqual([
 
315
            ('', 'tree-root'),
 
316
            ('doc', 'doc-id'),
 
317
            ('src', 'src-id'),
 
318
            ('src/sub', 'sub-id'),
 
319
            ('src/sub/a', 'a-id'),
 
320
            ], [(path, ie.file_id) for path, ie in new_inv.iter_entries()])
 
321
 
 
322
    def test_inv_filter_files_and_dirs(self):
 
323
        inv = self.prepare_inv_with_nested_dirs()
 
324
        new_inv = inv.filter(['makefile-id', 'src-id'])
 
325
        self.assertEqual([
 
326
            ('', 'tree-root'),
 
327
            ('Makefile', 'makefile-id'),
 
328
            ('src', 'src-id'),
 
329
            ('src/bye.c', 'bye-id'),
 
330
            ('src/hello.c', 'hello-id'),
 
331
            ('src/sub', 'sub-id'),
 
332
            ('src/sub/a', 'a-id'),
 
333
            ('src/zz.c', 'zzc-id'),
 
334
            ], [(path, ie.file_id) for path, ie in new_inv.iter_entries()])
 
335
 
 
336
    def test_inv_filter_entry_not_present(self):
 
337
        inv = self.prepare_inv_with_nested_dirs()
 
338
        new_inv = inv.filter(['not-present-id'])
 
339
        self.assertEqual([
 
340
            ('', 'tree-root'),
 
341
            ], [(path, ie.file_id) for path, ie in new_inv.iter_entries()])