Rev 3232: Minimally test stacking and unstacking a repository. in http://people.ubuntu.com/~robertc/baz2.0/shallow-branch

[Robert Collins]

At http://people.ubuntu.com/~robertc/baz2.0/shallow-branch

------------------------------------------------------------
revno: 3232
revision-id:robertc at robertcollins.net-20080222015839-hylnnuh74gw9mnpw
parent: robertc at robertcollins.net-20080222015318-hea8aug1bqiowzk1
committer: Robert Collins <robertc at robertcollins.net>
branch nick: StackableBranch
timestamp: Fri 2008-02-22 12:58:39 +1100
message:
  Minimally test stacking and unstacking a repository.
modified:
  bzrlib/tests/test_branch.py    test_branch.py-20060116013032-97819aa07b8ab3b5
=== modified file 'bzrlib/tests/test_branch.py'
--- a/bzrlib/tests/test_branch.py	2008-02-22 01:53:18 +0000
+++ b/bzrlib/tests/test_branch.py	2008-02-22 01:58:39 +0000
@@ -354,6 +354,14 @@
         branch = self.make_branch('a', format=self.get_format_name())
         self.assertFileEqual('\n', 'a/.bzr/branch/stacked-on')
 
+    def test_stack_and_unstack(self):
+        branch = self.make_branch('a', format=self.get_format_name())
+        target = self.make_branch('b')
+        branch.set_stacked_on(target.base)
+        self.assertEqual(target.base, branch.get_stacked_on())
+        branch.set_stacked_on(None)
+        self.assertRaises(errors.NotStacked, branch.get_stacked_on)
+
 
 class TestBranchReference(TestCaseWithTransport):
     """Tests for the branch reference facility."""