~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/blackbox/test_versioning.py

  • Committer: NamNguyen
  • Date: 2007-08-01 06:14:14 UTC
  • mto: (2789.1.1 ianc-integration)
  • mto: This revision was merged to the branch mainline in revision 2790.
  • Revision ID: namnguyen-20070801061414-u0tzrfgcz6z604lz
``Branch.hooks`` now supports ``pre_commit`` hook.

The hook's signature is

::

  hook(local, master, old_revno, old_revid, new_revno, new_revid,
       deleted_paths, added_paths, future_revision_tree)

``deleted_paths`` and ``added_paths`` are lists of paths. Renamed paths are
recorded in both ``deleted_paths`` and ``added_paths`` (i.e. deleted then
added).

``future_revision_tree`` is obtained from ``CommitBuilder.revision_tree``
to save hooks from getting it from the branch again.

For example, hooks can get a file:

::

  for path in added_paths:
      id = future_revision_tree.path2id(path)
      if future_revision_tree.kind(id) == 'file':
          file = future_revision_tree.get_file(id)
          ...

or export a tree and do ``make check`` or similar

::

  import bzrlib.export
  bzrlib.export.export(future_revision_tree, 'tmp_space')


If the commit is to be rejected, hooks should raise an ``Exception``.

Show diffs side-by-side

added added

removed removed

Lines of Context:
12
12
#
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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
 
18
18
"""Tests of simple versioning operations"""
22
22
 
23
23
import os
24
24
 
 
25
from bzrlib.tests import TestCaseInTempDir
25
26
from bzrlib.branch import Branch
 
27
from bzrlib.trace import mutter
26
28
from bzrlib.osutils import pathjoin
27
 
from bzrlib.tests import TestCaseInTempDir, TestCaseWithTransport
28
 
from bzrlib.trace import mutter
29
29
from bzrlib.workingtree import WorkingTree
30
30
 
31
31
 
122
122
        check(b, False)
123
123
 
124
124
 
125
 
class SubdirCommit(TestCaseWithTransport):
 
125
class SubdirCommit(TestCaseInTempDir):
126
126
 
127
127
    def test_subdir_commit(self):
128
 
        """Test committing a subdirectory, and committing a directory."""
129
 
        tree = self.make_branch_and_tree('.')
130
 
        b = tree.branch
 
128
        """Test committing a subdirectory, and committing within a directory."""
 
129
        run_bzr = self.run_bzr
 
130
        eq = self.assertEqual
 
131
 
131
132
        self.build_tree(['a/', 'b/'])
132
 
        def set_contents(contents):
133
 
            self.build_tree_contents([
134
 
                ('a/one', contents),
135
 
                ('b/two', contents),
136
 
                ('top', contents),
137
 
                ])
138
 
        set_contents('old contents')
139
 
        tree.smart_add(['.'])
140
 
        tree.commit('first revision')
141
 
        set_contents('new contents')
 
133
 
 
134
        run_bzr('init')
 
135
        b = Branch.open(u'.')
 
136
 
 
137
        for fn in ('a/one', 'b/two', 'top'):
 
138
            file(fn, 'w').write('old contents')
 
139
 
 
140
        run_bzr('add')
 
141
        run_bzr(['commit', '-m', 'first revision'])
 
142
 
 
143
        for fn in ('a/one', 'b/two', 'top'):
 
144
            file(fn, 'w').write('new contents')
142
145
 
143
146
        mutter('start selective subdir commit')
144
 
        self.run_bzr(['commit', 'a', '-m', 'commit a only'])
 
147
        run_bzr(['commit', 'a', '-m', 'commit a only'])
145
148
 
 
149
        old = b.repository.revision_tree(b.get_rev_id(1))
146
150
        new = b.repository.revision_tree(b.get_rev_id(2))
147
 
        new.lock_read()
148
151
 
149
 
        self.assertEqual(new.get_file_by_path('b/two').read(), 'old contents')
150
 
        self.assertEqual(new.get_file_by_path('top').read(), 'old contents')
151
 
        self.assertEqual(new.get_file_by_path('a/one').read(), 'new contents')
152
 
        new.unlock()
 
152
        eq(new.get_file_by_path('b/two').read(), 'old contents')
 
153
        eq(new.get_file_by_path('top').read(), 'old contents')
 
154
        eq(new.get_file_by_path('a/one').read(), 'new contents')
153
155
 
154
156
        os.chdir('a')
155
157
        # commit from here should do nothing
156
 
        self.run_bzr(['commit', '.', '-m', 'commit subdir only', '--unchanged'])
 
158
        run_bzr(['commit', '.', '-m', 'commit subdir only', '--unchanged'])
157
159
        v3 = b.repository.revision_tree(b.get_rev_id(3))
158
 
        v3.lock_read()
159
 
        self.assertEqual(v3.get_file_by_path('b/two').read(), 'old contents')
160
 
        self.assertEqual(v3.get_file_by_path('top').read(), 'old contents')
161
 
        self.assertEqual(v3.get_file_by_path('a/one').read(), 'new contents')
162
 
        v3.unlock()
 
160
        eq(v3.get_file_by_path('b/two').read(), 'old contents')
 
161
        eq(v3.get_file_by_path('top').read(), 'old contents')
 
162
        eq(v3.get_file_by_path('a/one').read(), 'new contents')
163
163
 
164
164
        # commit in subdirectory commits whole tree
165
 
        self.run_bzr(['commit', '-m', 'commit whole tree from subdir'])
 
165
        run_bzr(['commit', '-m', 'commit whole tree from subdir'])
166
166
        v4 = b.repository.revision_tree(b.get_rev_id(4))
167
 
        v4.lock_read()
168
 
        self.assertEqual(v4.get_file_by_path('b/two').read(), 'new contents')
169
 
        self.assertEqual(v4.get_file_by_path('top').read(), 'new contents')
170
 
        v4.unlock()
 
167
        eq(v4.get_file_by_path('b/two').read(), 'new contents')
 
168
        eq(v4.get_file_by_path('top').read(), 'new contents')
171
169
 
172
170
        # TODO: factor out some kind of assert_tree_state() method
 
171
 
 
172
 
 
173
if __name__ == '__main__':
 
174
    import unittest
 
175
    unittest.main()