1
# Copyright (C) 2005 by Canonical Ltd
1
# Copyright (C) 2005 Canonical Ltd
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
11
# GNU General Public License for more details.
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18
18
"""Tests of simple versioning operations"""
20
# TODO: test adding a file whose directory is not versioned
21
20
# TODO: test trying to commit within a directory that is not yet added
25
from bzrlib.selftest import InTempDir, BzrTestBase
25
from bzrlib.tests import TestCaseInTempDir
26
26
from bzrlib.branch import Branch
29
class Mkdir(InTempDir):
27
from bzrlib.trace import mutter
28
from bzrlib.osutils import pathjoin
29
from bzrlib.workingtree import WorkingTree
32
class TestVersioning(TestCaseInTempDir):
31
35
"""Basic 'bzr mkdir' operation"""
32
from bzrlib.commands import run_bzr
35
run_bzr(['mkdir', 'foo'])
38
self.run_bzr('mkdir foo')
36
39
self.assert_(os.path.isdir('foo'))
38
self.assertRaises(OSError, run_bzr, ['mkdir', 'foo'])
40
from bzrlib.diff import compare_trees
41
from bzrlib.branch import Branch
44
delta = compare_trees(b.basis_tree(), b.working_tree())
41
self.run_bzr('mkdir foo', retcode=3)
43
wt = WorkingTree.open('.')
45
delta = wt.changes_from(wt.basis_tree())
46
47
self.log('delta.added = %r' % delta.added)
49
50
self.assertEquals(delta.added[0][0], 'foo')
50
51
self.failIf(delta.modified)
54
class AddInUnversioned(InTempDir):
56
"""Try to add a file in an unversioned directory.
58
smart_add may eventually add the parent as necessary, but simple
59
branch add doesn't do that.
61
from bzrlib.branch import Branch
62
from bzrlib.errors import NotVersionedError
64
b = Branch('.', init=True)
66
self.build_tree(['foo/',
69
self.assertRaises(NotVersionedError,
74
class SubdirCommit(BzrTestBase):
53
def test_mkdir_in_subdir(self):
54
"""'bzr mkdir' operation in subdirectory"""
57
self.run_bzr('mkdir dir')
58
self.assert_(os.path.isdir('dir'))
61
self.log('Run mkdir in subdir')
62
self.run_bzr('mkdir subdir')
63
self.assert_(os.path.isdir('subdir'))
66
wt = WorkingTree.open('.')
68
delta = wt.changes_from(wt.basis_tree())
70
self.log('delta.added = %r' % delta.added)
72
self.assertEquals(len(delta.added), 2)
73
self.assertEquals(delta.added[0][0], 'dir')
74
self.assertEquals(delta.added[1][0], pathjoin('dir','subdir'))
75
self.failIf(delta.modified)
77
def test_mkdir_w_nested_trees(self):
78
"""'bzr mkdir' with nested trees"""
89
self.run_bzr('mkdir dir a/dir a/b/dir')
90
self.failUnless(os.path.isdir('dir'))
91
self.failUnless(os.path.isdir('a/dir'))
92
self.failUnless(os.path.isdir('a/b/dir'))
94
wt = WorkingTree.open('.')
95
wt_a = WorkingTree.open('a')
96
wt_b = WorkingTree.open('a/b')
98
delta = wt.changes_from(wt.basis_tree())
99
self.assertEquals(len(delta.added), 1)
100
self.assertEquals(delta.added[0][0], 'dir')
101
self.failIf(delta.modified)
103
delta = wt_a.changes_from(wt_a.basis_tree())
104
self.assertEquals(len(delta.added), 1)
105
self.assertEquals(delta.added[0][0], 'dir')
106
self.failIf(delta.modified)
108
delta = wt_b.changes_from(wt_b.basis_tree())
109
self.assertEquals(len(delta.added), 1)
110
self.assertEquals(delta.added[0][0], 'dir')
111
self.failIf(delta.modified)
113
def check_branch(self):
114
"""After all the above changes, run the check and upgrade commands.
116
The upgrade should be a no-op."""
117
b = Branch.open(u'.')
118
mutter('branch has %d revisions', b.revno())
120
mutter('check branch...')
121
from bzrlib.check import check
125
class SubdirCommit(TestCaseInTempDir):
127
def test_subdir_commit(self):
76
128
"""Test committing a subdirectory, and committing within a directory."""
77
129
run_bzr = self.run_bzr
78
130
eq = self.assertEqual
80
132
self.build_tree(['a/', 'b/'])
135
b = Branch.open(u'.')
85
137
for fn in ('a/one', 'b/two', 'top'):
86
138
file(fn, 'w').write('old contents')
89
run_bzr('commit', '-m', 'first revision')
141
run_bzr(['commit', '-m', 'first revision'])
91
143
for fn in ('a/one', 'b/two', 'top'):
92
144
file(fn, 'w').write('new contents')
94
run_bzr('commit', 'a', '-m', 'commit a only')
96
old = b.revision_tree(b.lookup_revision(1))
97
new = b.revision_tree(b.lookup_revision(2))
146
mutter('start selective subdir commit')
147
run_bzr(['commit', 'a', '-m', 'commit a only'])
149
old = b.repository.revision_tree(b.get_rev_id(1))
150
new = b.repository.revision_tree(b.get_rev_id(2))
99
153
eq(new.get_file_by_path('b/two').read(), 'old contents')
100
154
eq(new.get_file_by_path('top').read(), 'old contents')
101
155
eq(new.get_file_by_path('a/one').read(), 'new contents')
104
159
# commit from here should do nothing
105
run_bzr('commit', '.', '-m', 'commit subdir only', '--unchanged')
106
v3 = b.revision_tree(b.lookup_revision(3))
160
run_bzr(['commit', '.', '-m', 'commit subdir only', '--unchanged'])
161
v3 = b.repository.revision_tree(b.get_rev_id(3))
107
163
eq(v3.get_file_by_path('b/two').read(), 'old contents')
108
164
eq(v3.get_file_by_path('top').read(), 'old contents')
109
165
eq(v3.get_file_by_path('a/one').read(), 'new contents')
111
168
# commit in subdirectory commits whole tree
112
run_bzr('commit', '-m', 'commit whole tree from subdir')
113
v4 = b.revision_tree(b.lookup_revision(4))
114
eq(v4.get_file_by_path('b/two').read(), 'new contents')
169
run_bzr(['commit', '-m', 'commit whole tree from subdir'])
170
v4 = b.repository.revision_tree(b.get_rev_id(4))
172
eq(v4.get_file_by_path('b/two').read(), 'new contents')
115
173
eq(v4.get_file_by_path('top').read(), 'new contents')
117
176
# TODO: factor out some kind of assert_tree_state() method
121
class SubdirAdd(InTempDir):
123
"""Add in subdirectory should add only things from there down"""
125
from bzrlib.branch import Branch
126
from bzrlib.commands import run_bzr
128
eq = self.assertEqual
132
b = Branch('.', init=True)
133
self.build_tree(['src/', 'README'])
135
eq(sorted(b.unknowns()),
138
eq(run_bzr(['add', 'src']), 0)
140
self.build_tree(['src/foo.c'])
143
eq(run_bzr(['add']), 0)
145
eq(sorted(b.unknowns()),
147
eq(len(b.inventory), 3)
150
eq(run_bzr(['add']), 0)
151
eq(list(b.unknowns()), [])
179
if __name__ == '__main__':