~bzr-pqm/bzr/bzr.dev

601 by Martin Pool
- whitebox tests for branch path handling
1
#! /usr/bin/python
2
3
from bzrlib.branch import ScratchBranch
4
from bzrlib.errors import NotBranchError
5
from unittest import TestCase
6
import os, unittest
7
8
class BranchPathTestCase(TestCase):
9
    """test for branch path lookups
10
11
    Branch.relpath and bzrlib.branch._relpath do a simple but subtle
12
    job: given a path (either relative to cwd or absolute), work out
13
    if it is inside a branch and return the path relative to the base.
14
    """
15
    
16
    def runTest(self):
17
        from bzrlib.branch import _relpath
18
        import tempfile, shutil
19
        
20
        savedir = os.getcwdu()
21
        dtmp = tempfile.mkdtemp()
22
23
        def rp(p):
24
            return _relpath(dtmp, p)
25
        
26
        try:
27
            # check paths inside dtmp while standing outside it
28
            self.assertEqual(rp(os.path.join(dtmp, 'foo')), 'foo')
29
30
            # root = nothing
31
            self.assertEqual(rp(dtmp), '')
32
33
            self.assertRaises(NotBranchError,
34
                              rp,
35
                              '/etc')
36
37
            # now some near-miss operations -- note that
38
            # os.path.commonprefix gets these wrong!
39
            self.assertRaises(NotBranchError,
40
                              rp,
41
                              dtmp.rstrip('\\/') + '2')
42
43
            self.assertRaises(NotBranchError,
44
                              rp,
45
                              dtmp.rstrip('\\/') + '2/foo')
46
47
            # now operations based on relpath of files in current
48
            # directory, or nearby
49
            os.chdir(dtmp)
50
51
            self.assertEqual(rp('foo/bar/quux'), 'foo/bar/quux')
52
53
            self.assertEqual(rp('foo'), 'foo')
54
55
            self.assertEqual(rp('./foo'), 'foo')
56
57
            self.assertEqual(rp(os.path.abspath('foo')), 'foo')
58
59
            self.assertRaises(NotBranchError,
60
                              rp, '../foo')
61
62
        finally:
63
            os.chdir(savedir)
64
            shutil.rmtree(dtmp)
65
66
                              
67
if __name__ == '__main__':
68
    unittest.main()
69