1# Copyright (C) 2005, 2006, 2007, 2009, 2010, 2011, 2016 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
18import sys
19
20import breezy.errors
21from breezy.osutils import getcwd
22from breezy.tests import (
23    TestCaseWithTransport,
24    TestNotApplicable,
25    TestSkipped,
26    )
27from breezy import urlutils
28
29
30"""Tests for Branch parent URL"""
31
32
33class TestParent(TestCaseWithTransport):
34
35    def test_no_default_parent(self):
36        """Branches should have no parent by default"""
37        b = self.make_branch('.')
38        self.assertEqual(None, b.get_parent())
39
40    def test_set_get_parent(self):
41        """Set, re-get and reset the parent"""
42        b = self.make_branch('subdir')
43        url = 'http://example.com/bzr/bzr.dev'
44        b.set_parent(url)
45        self.assertEqual(url, b.get_parent())
46        self.assertEqual(url, b._get_parent_location())
47
48        b.set_parent(None)
49        self.assertEqual(None, b.get_parent())
50
51        b.set_parent('../other_branch')
52
53        expected_parent = urlutils.join(self.get_url('subdir'),
54                                        '../other_branch')
55        self.assertEqual(expected_parent, b.get_parent())
56        path = urlutils.join(self.get_url('subdir'), '../yanb')
57        b.set_parent(path)
58        self.assertEqual('../yanb', b._get_parent_location())
59        self.assertEqual(path, b.get_parent())
60
61        self.assertRaises(urlutils.InvalidURL, b.set_parent, u'\xb5')
62        b.set_parent(urlutils.escape(u'\xb5'))
63        self.assertEqual('%C2%B5', b._get_parent_location())
64
65        self.assertEqual(b.base + '%C2%B5', b.get_parent())
66
67        # Handle the case for older style absolute local paths
68        if sys.platform == 'win32':
69            # TODO: jam 20060515 Do we want to special case Windows local
70            #       paths as well? Nobody has complained about it.
71            pass
72        else:
73            b.lock_write()
74            b._set_parent_location('/local/abs/path')
75            b.unlock()
76            self.assertEqual('file:///local/abs/path', b.get_parent())
77
78    def test_get_invalid_parent(self):
79        b = self.make_branch('.')
80
81        cwd = getcwd()
82        n_dirs = len(cwd.split('/'))
83
84        # Force the relative path to be something invalid
85        # This should attempt to go outside the filesystem
86        path = ('../' * (n_dirs + 5)) + 'foo'
87        b.lock_write()
88        b._set_parent_location(path)
89        b.unlock()
90
91        # With an invalid branch parent, just return None
92        self.assertRaises(breezy.errors.InaccessibleParent, b.get_parent)
93
94    def test_win32_set_parent_on_another_drive(self):
95        if sys.platform != 'win32':
96            raise TestSkipped('windows-specific test')
97        b = self.make_branch('.')
98        base_url = b.controldir.transport.abspath('.')
99        if not base_url.startswith('file:///'):
100            raise TestNotApplicable('this test should be run with local base')
101        base = urlutils.local_path_from_url(base_url)
102        other = 'file:///D:/path'
103        if base[0] != 'C':
104            other = 'file:///C:/path'
105        b.set_parent(other)
106        self.assertEqual(other, b._get_parent_location())
107