~bzr-pqm/bzr/bzr.dev

6478.2.1 by Jelmer Vernooij
Accept path element list as argument to Tree.path2id.
1
# Copyright (C) 2012 Canonical Ltd
2
#
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
7
#
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
# GNU General Public License for more details.
12
#
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
17
from bzrlib import (
18
    errors,
19
    )
20
from bzrlib.tests.per_tree import TestCaseWithTree
21
22
class IdTests(TestCaseWithTree):
23
24
    def setUp(self):
25
        super(IdTests, self).setUp()
26
        work_a = self.make_branch_and_tree('wta')
27
        self.build_tree(['wta/bla', 'wta/dir/', 'wta/dir/file'])
28
        work_a.add(['bla', 'dir', 'dir/file'], ['bla-id', 'dir-id', 'file-id'])
29
        work_a.commit('add files')
30
        self.tree_a = self.workingtree_to_test_tree(work_a)
31
32
    def test_path2id(self):
33
        self.assertEquals('bla-id', self.tree_a.path2id('bla'))
34
        self.assertEquals('dir-id', self.tree_a.path2id('dir'))
35
        self.assertIs(None, self.tree_a.path2id('idontexist'))
36
37
    def test_path2id_list(self):
38
        self.assertEquals('bla-id', self.tree_a.path2id(['bla']))
39
        self.assertEquals('dir-id', self.tree_a.path2id(['dir']))
40
        self.assertEquals('file-id', self.tree_a.path2id(['dir', 'file']))
41
        self.assertEquals(self.tree_a.get_root_id(),
42
            self.tree_a.path2id([]))
43
        self.assertIs(None, self.tree_a.path2id(['idontexist']))
44
        self.assertIs(None, self.tree_a.path2id(['dir', 'idontexist']))
45
46
    def test_id2path(self):
47
        self.addCleanup(self.tree_a.lock_read().unlock)
48
        self.assertEquals('bla', self.tree_a.id2path('bla-id'))
49
        self.assertEquals('dir', self.tree_a.id2path('dir-id'))
50
        self.assertEquals('dir/file', self.tree_a.id2path('file-id'))
51
        self.assertRaises(errors.NoSuchId, self.tree_a.id2path, 'nonexistant')