← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/selftest/test_smart_add.py

  • Committer: Robert Collins
  • Date: 2005-10-17 11:56:54 UTC
  • mfrom: (1185.16.59)
  • Revision ID: robertc@robertcollins.net-20051017115654-662239e1587524a8
merge from martin.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/selftest/test_smart_add.py
1
1
import os
2
2
import unittest
3
3
 
4
 
from bzrlib.tests import TestCaseInTempDir, TestCase
 
4
from bzrlib.selftest import TestCaseInTempDir, TestCase
5
5
from bzrlib.branch import Branch
6
 
from bzrlib.errors import NotBranchError, NoSuchFile
7
 
from bzrlib.inventory import InventoryFile, Inventory
8
 
from bzrlib.workingtree import WorkingTree
9
 
from bzrlib.add import smart_add
 
6
from bzrlib.errors import NotBranchError, NotVersionedError
 
7
from bzrlib.inventory import InventoryFile
10
8
 
11
9
class TestSmartAdd(TestCaseInTempDir):
12
10
 
15
13
        from bzrlib.add import smart_add
16
14
        paths = ("original/", "original/file1", "original/file2")
17
15
        self.build_tree(paths)
18
 
        branch = Branch.initialize(u".")
19
 
        smart_add((u".",))
 
16
        branch = Branch.initialize(".")
 
17
        smart_add((".",), recurse=True)
20
18
        for path in paths:
21
 
            self.assertNotEqual(branch.working_tree().path2id(path), None)
 
19
            self.assertNotEqual(branch.inventory.path2id(path), None)
22
20
 
23
21
    def test_add_dot_from_subdir(self):
24
22
        """Test adding . from a subdir of the tree.""" 
25
23
        from bzrlib.add import smart_add
26
24
        paths = ("original/", "original/file1", "original/file2")
27
25
        self.build_tree(paths)
28
 
        branch = Branch.initialize(u".")
 
26
        branch = Branch.initialize(".")
29
27
        os.chdir("original")
30
 
        smart_add((u".",))
 
28
        smart_add((".",), recurse=True)
31
29
        for path in paths:
32
 
            self.assertNotEqual(branch.working_tree().path2id(path), None)
 
30
            self.assertNotEqual(branch.inventory.path2id(path), None)
33
31
 
34
32
    def test_add_tree_from_above_tree(self):
35
33
        """Test adding a tree from above the tree.""" 
41
39
        branch = Branch.initialize("branch")
42
40
        smart_add(("branch",))
43
41
        for path in paths:
44
 
            self.assertNotEqual(branch.working_tree().path2id(path), None)
 
42
            self.assertNotEqual(branch.inventory.path2id(path), None)
45
43
 
46
44
    def test_add_above_tree_preserves_tree(self):
47
45
        """Test nested trees are not affect by an add above them."""
48
 
        from bzrlib.add import smart_add
 
46
        from bzrlib.add import smart_add, add_reporter_null
 
47
        
49
48
        paths = ("original/", "original/file1", "original/file2")
50
49
        child_paths = ("path",)
51
50
        full_child_paths = ("original/child", "original/child/path")
53
52
                       "original/child/", "original/child/path")
54
53
        
55
54
        self.build_tree(build_paths)
56
 
        branch = Branch.initialize(u".")
 
55
        branch = Branch.initialize(".")
57
56
        child_branch = Branch.initialize("original/child")
58
 
        smart_add((u".",))
 
57
        smart_add((".",), True, add_reporter_null)
59
58
        for path in paths:
60
 
            self.assertNotEqual((path, branch.working_tree().path2id(path)),
 
59
            self.assertNotEqual((path, branch.inventory.path2id(path)),
61
60
                                (path, None))
62
61
        for path in full_child_paths:
63
 
            self.assertEqual((path, branch.working_tree().path2id(path)),
 
62
            self.assertEqual((path, branch.inventory.path2id(path)),
64
63
                             (path, None))
65
64
        for path in child_paths:
66
 
            self.assertEqual(child_branch.working_tree().path2id(path), None)
 
65
            self.assertEqual(child_branch.inventory.path2id(path), None)
67
66
 
68
67
    def test_add_paths(self):
69
68
        """Test smart-adding a list of paths."""
70
69
        from bzrlib.add import smart_add
71
70
        paths = ("file1", "file2")
72
71
        self.build_tree(paths)
73
 
        branch = Branch.initialize(u".")
 
72
        branch = Branch.initialize(".")
74
73
        smart_add(paths)
75
74
        for path in paths:
76
 
            self.assertNotEqual(branch.working_tree().path2id(path), None)
77
 
 
78
 
    def test_add_dry_run(self):
79
 
        """Test a dry run add, make sure nothing is added."""
80
 
        from bzrlib.commands import run_bzr
81
 
        eq = self.assertEqual
82
 
        b = Branch.initialize(u'.')
83
 
        t = b.working_tree()
84
 
        self.build_tree(['inertiatic/', 'inertiatic/esp'])
85
 
        eq(list(t.unknowns()), ['inertiatic'])
86
 
        self.capture('add --dry-run .')
87
 
        eq(list(t.unknowns()), ['inertiatic'])
88
 
 
89
 
    def test_add_non_existant(self):
90
 
        """Test smart-adding a file that does not exist."""
91
 
        from bzrlib.add import smart_add
92
 
        branch = Branch.initialize(u".")
93
 
        self.assertRaises(NoSuchFile, smart_add, 'non-existant-file')
94
 
 
95
 
    def test_returns(self):
96
 
        """Correctly returns added/ignored files"""
97
 
        from bzrlib.commands import run_bzr
98
 
        b = Branch.initialize(u'.')
99
 
        t = b.working_tree()
100
 
        self.build_tree(['inertiatic/', 'inertiatic/esp', 'inertiatic/CVS', 
101
 
                        'inertiatic/foo.pyc'])
102
 
        added, ignored = smart_add(u'.')
103
 
        self.AssertSubset(('inertiatic', 'inertiatic/esp'), added)
104
 
        self.AssertSubset(('CVS', '*.py[oc]'), ignored)
105
 
        self.AssertSubset(('inertiatic/CVS',), ignored['CVS'])
106
 
        self.AssertSubset(('inertiatic/foo.pyc',), ignored['*.py[oc]'])
107
 
 
108
 
 
 
75
            self.assertNotEqual(branch.inventory.path2id(path), None)
 
76
            
109
77
class TestSmartAddBranch(TestCaseInTempDir):
110
78
    """Test smart adds with a specified branch."""
111
79
 
112
80
    def test_add_dot_from_root(self):
113
81
        """Test adding . from the root of the tree.""" 
114
 
        from bzrlib.add import smart_add_tree
 
82
        from bzrlib.add import smart_add_branch
115
83
        paths = ("original/", "original/file1", "original/file2")
116
84
        self.build_tree(paths)
117
 
        Branch.initialize(u".")
118
 
        tree = WorkingTree()
119
 
        smart_add_tree(tree, (u".",))
 
85
        branch = Branch.initialize(".")
 
86
        smart_add_branch(branch, (".",))
120
87
        for path in paths:
121
 
            self.assertNotEqual(tree.path2id(path), None)
 
88
            self.assertNotEqual(branch.inventory.path2id(path), None)
122
89
 
123
90
    def test_add_dot_from_subdir(self):
124
91
        """Test adding . from a subdir of the tree.""" 
125
 
        from bzrlib.add import smart_add_tree
 
92
        from bzrlib.add import smart_add_branch
126
93
        paths = ("original/", "original/file1", "original/file2")
127
94
        self.build_tree(paths)
128
 
        Branch.initialize(u".")
129
 
        tree = WorkingTree()
 
95
        branch = Branch.initialize(".")
130
96
        os.chdir("original")
131
 
        smart_add_tree(tree, (u".",))
 
97
        smart_add_branch(branch, (".",))
132
98
        for path in paths:
133
 
            self.assertNotEqual(tree.path2id(path), None)
 
99
            self.assertNotEqual(branch.inventory.path2id(path), None)
134
100
 
135
101
    def test_add_tree_from_above_tree(self):
136
102
        """Test adding a tree from above the tree.""" 
137
 
        from bzrlib.add import smart_add_tree
 
103
        from bzrlib.add import smart_add_branch
138
104
        paths = ("original/", "original/file1", "original/file2")
139
105
        branch_paths = ("branch/", "branch/original/", "branch/original/file1",
140
106
                        "branch/original/file2")
141
107
        self.build_tree(branch_paths)
142
 
        Branch.initialize("branch")
143
 
        tree = WorkingTree("branch")
144
 
        smart_add_tree(tree, ("branch",))
 
108
        branch = Branch.initialize("branch")
 
109
        smart_add_branch(branch, ("branch",))
145
110
        for path in paths:
146
 
            self.assertNotEqual(tree.path2id(path), None)
 
111
            self.assertNotEqual(branch.inventory.path2id(path), None)
147
112
 
148
113
    def test_add_above_tree_preserves_tree(self):
149
114
        """Test nested trees are not affect by an add above them."""
150
 
        from bzrlib.add import smart_add_tree
 
115
        from bzrlib.add import smart_add_branch
151
116
        paths = ("original/", "original/file1", "original/file2")
152
117
        child_paths = ("path")
153
118
        full_child_paths = ("original/child", "original/child/path")
154
119
        build_paths = ("original/", "original/file1", "original/file2", 
155
120
                       "original/child/", "original/child/path")
156
121
        self.build_tree(build_paths)
157
 
        Branch.initialize(u".")
158
 
        tree = WorkingTree()
 
122
        branch = Branch.initialize(".")
159
123
        child_branch = Branch.initialize("original/child")
160
 
        smart_add_tree(tree, (u".",))
 
124
        smart_add_branch(branch, (".",))
161
125
        for path in paths:
162
 
            self.assertNotEqual((path, tree.path2id(path)),
 
126
            self.assertNotEqual((path, branch.inventory.path2id(path)),
163
127
                                (path, None))
164
128
        for path in full_child_paths:
165
 
            self.assertEqual((path, tree.path2id(path)),
 
129
            self.assertEqual((path, branch.inventory.path2id(path)),
166
130
                             (path, None))
167
131
        for path in child_paths:
168
 
            self.assertEqual(child_branch.working_tree().path2id(path), None)
 
132
            self.assertEqual(child_branch.inventory.path2id(path), None)
169
133
 
170
134
    def test_add_paths(self):
171
135
        """Test smart-adding a list of paths."""
172
 
        from bzrlib.add import smart_add_tree
 
136
        from bzrlib.add import smart_add_branch
173
137
        paths = ("file1", "file2")
174
138
        self.build_tree(paths)
175
 
        Branch.initialize(u".")
176
 
        tree = WorkingTree()
177
 
        smart_add_tree(tree, paths)
 
139
        branch = Branch.initialize(".")
 
140
        smart_add_branch(branch, paths)
178
141
        for path in paths:
179
 
            self.assertNotEqual(tree.path2id(path), None)
180
 
 
181
 
 
182
 
class TestAddActions(TestCaseInTempDir):
183
 
 
184
 
    def test_null(self):
185
 
        from bzrlib.add import add_action_null
186
 
        self.run_action(add_action_null, "", False)
187
 
 
188
 
    def test_add(self):
 
142
            self.assertNotEqual(branch.inventory.path2id(path), None)
 
143
 
 
144
class TestAddCallbacks(TestCaseInTempDir):
 
145
 
 
146
    def setUp(self):
 
147
        super(TestAddCallbacks, self).setUp()
189
148
        self.entry = InventoryFile("id", "name", None)
190
 
        from bzrlib.add import add_action_add
191
 
        self.run_action(add_action_add, "", True)
192
 
 
193
 
    def test_add_and_print(self):
194
 
        from bzrlib.add import add_action_add_and_print
195
 
        self.run_action(add_action_add_and_print, "added path\n", True)
196
 
 
197
 
    def test_print(self):
198
 
        from bzrlib.add import add_action_print
199
 
        self.run_action(add_action_print, "added path\n", False)
200
 
 
201
 
    def run_action(self, action, output, should_add):
 
149
 
 
150
    def test_null_callback(self):
 
151
        from bzrlib.add import add_reporter_null
 
152
        add_reporter_null('path', 'file', self.entry)
 
153
 
 
154
    def test_print_callback(self):
 
155
        from bzrlib.add import add_reporter_print
202
156
        from StringIO import StringIO
203
 
        inv = Inventory()
204
157
        stdout = StringIO()
205
 
 
206
 
        self.apply_redirected(None, stdout, None, action, inv, 'path', 'file')
207
 
        self.assertEqual(stdout.getvalue(), output)
208
 
 
209
 
        if should_add:
210
 
            self.assertNotEqual(inv.path2id('path'), None)
211
 
        else:
212
 
            self.assertEqual(inv.path2id('path'), None)
 
158
        self.apply_redirected(None, stdout, None, add_reporter_print,
 
159
                              'path', 'file', self.entry)
 
160
        self.assertEqual(stdout.getvalue(), "added path\n")