~bzr-pqm/bzr/bzr.dev

601 by Martin Pool
- whitebox tests for branch path handling
1
#! /usr/bin/python
2
768 by Martin Pool
- start some tests for directory renames
3
import os
4
import unittest
5
6
from bzrlib.selftest import InTempDir, TestBase
7
from bzrlib.branch import ScratchBranch, Branch
778 by Martin Pool
- simple revert of text files
8
from bzrlib.errors import NotBranchError, NotVersionedError
9
10
781 by Martin Pool
- start of simple test for unknown-file reporting
11
class Unknowns(InTempDir):
12
    def runTest(self):
13
        b = Branch('.', init=True)
14
15
        self.build_tree(['hello.txt',
16
                         'hello.txt~'])
17
18
        self.assertEquals(list(b.unknowns()),
19
                          ['hello.txt'])
20
21
778 by Martin Pool
- simple revert of text files
22
class Revert(InTempDir):
23
    """Test selected-file revert"""
24
    def runTest(self):
25
        b = Branch('.', init=True)
26
27
        self.build_tree(['hello.txt'])
28
        file('hello.txt', 'w').write('initial hello')
29
30
        self.assertRaises(NotVersionedError,
31
                          b.revert, ['hello.txt'])
32
        
33
        b.add(['hello.txt'])
34
        b.commit('create initial hello.txt')
35
36
        self.check_file_contents('hello.txt', 'initial hello')
37
        file('hello.txt', 'w').write('new hello')
38
        self.check_file_contents('hello.txt', 'new hello')
39
40
        # revert file modified since last revision
41
        b.revert(['hello.txt'])
42
        self.check_file_contents('hello.txt', 'initial hello')
782 by Martin Pool
- Branch.revert copies files to backups before reverting them
43
        self.check_file_contents('hello.txt~', 'new hello')
778 by Martin Pool
- simple revert of text files
44
782 by Martin Pool
- Branch.revert copies files to backups before reverting them
45
        # reverting again clobbers the backup
778 by Martin Pool
- simple revert of text files
46
        b.revert(['hello.txt'])
47
        self.check_file_contents('hello.txt', 'initial hello')
782 by Martin Pool
- Branch.revert copies files to backups before reverting them
48
        self.check_file_contents('hello.txt~', 'initial hello')
778 by Martin Pool
- simple revert of text files
49
768 by Martin Pool
- start some tests for directory renames
50
51
52
class RenameDirs(InTempDir):
53
    """Test renaming directories and the files within them."""
54
    def runTest(self):
55
        b = Branch('.', init=True)
56
        self.build_tree(['dir/', 'dir/sub/', 'dir/sub/file'])
57
        b.add(['dir', 'dir/sub', 'dir/sub/file'])
58
59
        b.commit('create initial state')
60
61
        # TODO: lift out to a test helper that checks the shape of
62
        # an inventory
63
        
64
        revid = b.revision_history()[0]
65
        self.log('first revision_id is {%s}' % revid)
66
        
67
        inv = b.get_revision_inventory(revid)
68
        self.log('contents of inventory: %r' % inv.entries())
69
70
        self.check_inventory_shape(inv,
71
                                   ['dir', 'dir/sub', 'dir/sub/file'])
72
771 by Martin Pool
- more tests of directory renames
73
        b.rename_one('dir', 'newdir')
74
75
        self.check_inventory_shape(b.inventory,
76
                                   ['newdir', 'newdir/sub', 'newdir/sub/file'])
77
78
        b.rename_one('newdir/sub', 'newdir/newsub')
79
        self.check_inventory_shape(b.inventory,
80
                                   ['newdir', 'newdir/newsub',
81
                                    'newdir/newsub/file'])
82
768 by Martin Pool
- start some tests for directory renames
83
        
84
85
86
class BranchPathTestCase(TestBase):
601 by Martin Pool
- whitebox tests for branch path handling
87
    """test for branch path lookups
88
89
    Branch.relpath and bzrlib.branch._relpath do a simple but subtle
90
    job: given a path (either relative to cwd or absolute), work out
91
    if it is inside a branch and return the path relative to the base.
92
    """
768 by Martin Pool
- start some tests for directory renames
93
601 by Martin Pool
- whitebox tests for branch path handling
94
    def runTest(self):
95
        from bzrlib.branch import _relpath
96
        import tempfile, shutil
97
        
98
        savedir = os.getcwdu()
99
        dtmp = tempfile.mkdtemp()
100
101
        def rp(p):
102
            return _relpath(dtmp, p)
103
        
104
        try:
105
            # check paths inside dtmp while standing outside it
106
            self.assertEqual(rp(os.path.join(dtmp, 'foo')), 'foo')
107
108
            # root = nothing
109
            self.assertEqual(rp(dtmp), '')
110
111
            self.assertRaises(NotBranchError,
112
                              rp,
113
                              '/etc')
114
115
            # now some near-miss operations -- note that
116
            # os.path.commonprefix gets these wrong!
117
            self.assertRaises(NotBranchError,
118
                              rp,
119
                              dtmp.rstrip('\\/') + '2')
120
121
            self.assertRaises(NotBranchError,
122
                              rp,
123
                              dtmp.rstrip('\\/') + '2/foo')
124
125
            # now operations based on relpath of files in current
126
            # directory, or nearby
127
            os.chdir(dtmp)
128
129
            self.assertEqual(rp('foo/bar/quux'), 'foo/bar/quux')
130
131
            self.assertEqual(rp('foo'), 'foo')
132
133
            self.assertEqual(rp('./foo'), 'foo')
134
135
            self.assertEqual(rp(os.path.abspath('foo')), 'foo')
136
137
            self.assertRaises(NotBranchError,
138
                              rp, '../foo')
139
140
        finally:
141
            os.chdir(savedir)
142
            shutil.rmtree(dtmp)