← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/selftest/whitebox.py

  • Committer: Martin Pool
  • Date: 2005-06-27 06:25:30 UTC
  • Revision ID: mbp@sourcefrog.net-20050627062530-0002d5c96b0ed3c1
- note bzr log changes

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/selftest/whitebox.py
 
1
#! /usr/bin/python
 
2
 
 
3
import os
 
4
import unittest
 
5
 
 
6
from bzrlib.selftest import InTempDir, TestBase
 
7
from bzrlib.branch import ScratchBranch, Branch
 
8
from bzrlib.errors import NotBranchError, NotVersionedError
 
9
 
 
10
 
 
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
 
 
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')
 
43
        self.check_file_contents('hello.txt~', 'new hello')
 
44
 
 
45
        # reverting again clobbers the backup
 
46
        b.revert(['hello.txt'])
 
47
        self.check_file_contents('hello.txt', 'initial hello')
 
48
        self.check_file_contents('hello.txt~', 'initial hello')
 
49
 
 
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
 
 
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
 
 
83
        
 
84
 
 
85
 
 
86
class BranchPathTestCase(TestBase):
 
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
    """
 
93
 
 
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)