Use locking and write groups properly, fixes compatibility with packs.
[jelmer/subvertpy.git] / tests / test_branch.py
index 93dbe0fb2fff8c6f6da0e6d4d62b40077e803ffb..564354e64ca41cadc8f6f3f8f77d5fc1706d06de 100644 (file)
@@ -1,4 +1,4 @@
-# Copyright (C) 2006 Jelmer Vernooij <jelmer@samba.org>
+# Copyright (C) 2006-2007 Jelmer Vernooij <jelmer@samba.org>
 
 # This program is free software; you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
 # along with this program; if not, write to the Free Software
 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
+"""Branch tests."""
+
 from bzrlib.branch import Branch
-from bzrlib.bzrdir import BzrDir, BzrDirTestProviderAdapter, BzrDirFormat
+from bzrlib.bzrdir import BzrDir
+from bzrlib.errors import NoSuchFile, NoSuchRevision, NotBranchError
 from bzrlib.repository import Repository
 from bzrlib.trace import mutter
 
 import os
+from unittest import TestCase
 
-import svn.core, svn.client
-
-import format
+from branch import FakeControlFiles, SvnBranchFormat
+from convert import load_dumpfile
+from fileids import generate_svn_file_id
+from repository import MAPPING_VERSION, generate_svn_revision_id, SVN_PROP_BZR_REVISION_ID
 from tests import TestCaseWithSubversionRepository
 
 class WorkingSubversionBranch(TestCaseWithSubversionRepository):
+    def test_last_rev_rev_hist(self):
+        repos_url = self.make_client("a", "dc")
+        branch = Branch.open(repos_url)
+        branch.revision_history()
+        self.assertEqual(branch.generate_revision_id(0), branch.last_revision())
+
+    def test_get_branch_path_root(self):
+        repos_url = self.make_client("a", "dc")
+        branch = Branch.open(repos_url)
+        self.assertEqual("", branch.get_branch_path())
+
+    def test_get_branch_path_subdir(self):
+        repos_url = self.make_client("a", "dc")
+        self.build_tree({"dc/trunk": None})
+        self.client_add("dc/trunk")
+        self.client_commit("dc", "Add branch")
+        branch = Branch.open(repos_url+"/trunk")
+        self.assertEqual("trunk", branch.get_branch_path())
+
+    def test_open_nonexistant(self):
+        repos_url = self.make_client("a", "dc")
+        self.assertRaises(NotBranchError, Branch.open, repos_url + "/trunk")
+
+    def test_last_rev_rev_info(self):
+        repos_url = self.make_client("a", "dc")
+        branch = Branch.open(repos_url)
+        self.assertEqual((1, branch.generate_revision_id(0)),
+                branch.last_revision_info())
+        branch.revision_history()
+        self.assertEqual((1, branch.generate_revision_id(0)),
+                branch.last_revision_info())
+
+    def test_lookup_revision_id_unknown(self):
+        repos_url = self.make_client("a", "dc")
+        branch = Branch.open(repos_url)
+        self.assertRaises(NoSuchRevision, 
+                lambda: branch.lookup_revision_id("bla"))
+
+    def test_lookup_revision_id(self):
+        repos_url = self.make_client("a", "dc")
+        branch = Branch.open(repos_url)
+        self.assertEquals(0, 
+                branch.lookup_revision_id(branch.last_revision()))
+
+    def test_set_parent(self):
+        repos_url = self.make_client('a', 'dc')
+        branch = Branch.open(repos_url)
+        branch.set_parent("foobar")
+
     def test_num_revnums(self):
         repos_url = self.make_client('a', 'dc')
         bzrdir = BzrDir.open("svn+"+repos_url)
         branch = bzrdir.open_branch()
-        self.assertEqual(None, branch.last_revision())
+        self.assertEqual(branch.generate_revision_id(0),
+                         branch.last_revision())
 
         self.build_tree({'dc/foo': "data"})
         self.client_add("dc/foo")
         self.client_commit("dc", "My Message")
+        self.client_update("dc")
         
         bzrdir = BzrDir.open("svn+"+repos_url)
         branch = bzrdir.open_branch()
-        repos = bzrdir.open_repository()
+        repos = bzrdir.find_repository()
 
-        self.assertEqual("svn-v1:1@%s-" % repos.uuid, branch.last_revision())
+        self.assertEqual(repos.generate_revision_id(1, "", "none"), 
+                branch.last_revision())
 
         self.build_tree({'dc/foo': "data2"})
         self.client_commit("dc", "My Message")
@@ -49,22 +106,69 @@ class WorkingSubversionBranch(TestCaseWithSubversionRepository):
         branch = Branch.open("svn+"+repos_url)
         repos = Repository.open("svn+"+repos_url)
 
-        self.assertEqual("svn-v1:2@%s-" % repos.uuid, branch.last_revision())
+        self.assertEqual(repos.generate_revision_id(2, "", "none"),
+                branch.last_revision())
+
+    def test_set_revision_history(self):
+        repos_url = self.make_client('a', 'dc')
+        branch = Branch.open("svn+"+repos_url)
+        self.assertRaises(NotImplementedError, branch.set_revision_history, [])
+
+    def test_break_lock(self):
+        repos_url = self.make_client('a', 'dc')
+        branch = Branch.open("svn+"+repos_url)
+        branch.control_files.break_lock()
+
+    def test_repr(self):
+        repos_url = self.make_client('a', 'dc')
+        branch = Branch.open("svn+"+repos_url)
+        self.assertEqual("SvnBranch('svn+%s')" % repos_url, branch.__repr__())
+
+    def test_get_physical_lock_status(self):
+        repos_url = self.make_client('a', 'dc')
+        branch = Branch.open("svn+"+repos_url)
+        self.assertFalse(branch.get_physical_lock_status())
+
+    def test_set_push_location(self):
+        repos_url = self.make_client('a', 'dc')
+        branch = Branch.open("svn+"+repos_url)
+        self.assertRaises(NotImplementedError, branch.set_push_location, [])
+
+    def test_get_parent(self):
+        repos_url = self.make_client('a', 'dc')
+        branch = Branch.open("svn+"+repos_url)
+        self.assertEqual("svn+"+repos_url, branch.get_parent())
+
+    def test_append_revision(self):
+        repos_url = self.make_client('a', 'dc')
+        branch = Branch.open("svn+"+repos_url)
+        branch.append_revision([])
+
+    def test_get_push_location(self):
+        repos_url = self.make_client('a', 'dc')
+        branch = Branch.open("svn+"+repos_url)
+        self.assertIs(None, branch.get_push_location())
 
     def test_revision_history(self):
         repos_url = self.make_client('a', 'dc')
 
         branch = Branch.open("svn+"+repos_url)
-        self.assertEqual([], branch.revision_history())
+        self.assertEqual([branch.generate_revision_id(0)], 
+                branch.revision_history())
 
         self.build_tree({'dc/foo': "data"})
         self.client_add("dc/foo")
+        self.client_set_prop("dc", SVN_PROP_BZR_REVISION_ID+"none", 
+                "42 mycommit\n")
         self.client_commit("dc", "My Message")
+        self.client_update("dc")
         
         branch = Branch.open("svn+"+repos_url)
         repos = Repository.open("svn+"+repos_url)
 
-        self.assertEqual(["svn-v1:1@%s-" % repos.uuid], branch.revision_history())
+        self.assertEqual([repos.generate_revision_id(0, "", "none"), 
+                    repos.generate_revision_id(1, "", "none")], 
+                branch.revision_history())
 
         self.build_tree({'dc/foo': "data34"})
         self.client_commit("dc", "My Message")
@@ -73,11 +177,50 @@ class WorkingSubversionBranch(TestCaseWithSubversionRepository):
         repos = Repository.open("svn+"+repos_url)
 
         self.assertEqual([
-            "svn-v1:1@%s-" % repos.uuid, 
-            "svn-v1:2@%s-" % repos.uuid],
+            repos.generate_revision_id(0, "", "none"),
+            "mycommit",
+            repos.generate_revision_id(2, "", "none")],
             branch.revision_history())
 
-    def test_get_nick(self):
+    def test_revision_id_to_revno_none(self):
+        """The None revid should map to revno 0."""
+        repos_url = self.make_client('a', 'dc')
+        branch = Branch.open(repos_url)
+        self.assertEquals(0, branch.revision_id_to_revno(None))
+
+    def test_revision_id_to_revno_nonexistant(self):
+        """revision_id_to_revno() should raise NoSuchRevision if
+        the specified revision did not exist in the branch history."""
+        repos_url = self.make_client('a', 'dc')
+        branch = Branch.open(repos_url)
+        self.assertRaises(NoSuchRevision, branch.revision_id_to_revno, "bla")
+    
+    def test_revision_id_to_revno_simple(self):
+        repos_url = self.make_client('a', 'dc')
+        self.build_tree({'dc/foo': "data"})
+        self.client_add("dc/foo")
+        self.client_set_prop("dc", "bzr:revision-id:v%d-none" % MAPPING_VERSION, 
+                            "2 myrevid\n")
+        self.client_commit("dc", "My Message")
+        branch = Branch.open(repos_url)
+        self.assertEquals(2, branch.revision_id_to_revno("myrevid"))
+
+    def test_revision_id_to_revno_older(self):
+        repos_url = self.make_client('a', 'dc')
+        self.build_tree({'dc/foo': "data"})
+        self.client_add("dc/foo")
+        self.client_set_prop("dc", "bzr:revision-id:v%d-none" % MAPPING_VERSION, 
+                            "2 myrevid\n")
+        self.client_commit("dc", "My Message")
+        self.build_tree({'dc/foo': "someotherdata"})
+        self.client_set_prop("dc", "bzr:revision-id:v%d-none" % MAPPING_VERSION, 
+                            "2 myrevid\n3 mysecondrevid\n")
+        self.client_commit("dc", "My Message")
+        branch = Branch.open(repos_url)
+        self.assertEquals(3, branch.revision_id_to_revno("mysecondrevid"))
+        self.assertEquals(2, branch.revision_id_to_revno("myrevid"))
+
+    def test_get_nick_none(self):
         repos_url = self.make_client('a', 'dc')
 
         self.build_tree({'dc/foo': "data"})
@@ -88,6 +231,32 @@ class WorkingSubversionBranch(TestCaseWithSubversionRepository):
 
         self.assertIs(None, branch.nick)
 
+    def test_get_nick_path(self):
+        repos_url = self.make_client('a', 'dc')
+
+        self.build_tree({'dc/trunk': None})
+        self.client_add("dc/trunk")
+        self.client_commit("dc", "My Message")
+
+        branch = Branch.open("svn+"+repos_url+"/trunk")
+
+        self.assertEqual("trunk", branch.nick)
+
+    def test_get_revprops(self):
+        repos_url = self.make_client('a', 'dc')
+
+        self.build_tree({'dc/foo': "data"})
+        self.client_add("dc/foo")
+        self.client_set_prop("dc", "bzr:revision-info", 
+                "properties: \n\tbranch-nick: mybranch\n")
+        self.client_commit("dc", "My Message")
+
+        branch = Branch.open("svn+"+repos_url)
+
+        rev = branch.repository.get_revision(branch.last_revision())
+
+        self.assertEqual("mybranch", rev.properties["branch-nick"])
+
     def test_fetch_replace(self):
         filename = os.path.join(self.test_dir, "dumpfile")
         open(filename, 'w').write("""SVN-fs-dump-format-version: 2
@@ -306,8 +475,11 @@ Content-length: 8
 
 foohosts""")
         os.mkdir("new")
+        os.mkdir("old")
+
+        load_dumpfile("dumpfile", "old")
 
-        url = "dumpfile/branches/foobranch"
+        url = "old/branches/foobranch"
         mutter('open %r' % url)
         olddir = BzrDir.open(url)
 
@@ -316,13 +488,19 @@ foohosts""")
         newbranch = newdir.open_branch()
 
         uuid = "6f95bc5c-e18d-4021-aca8-49ed51dbcb75"
+        newbranch.lock_read()
         tree = newbranch.repository.revision_tree(
-                "svn-v1:7@%s-branches%%2ffoobranch" % uuid)
+                generate_svn_revision_id(uuid, 7, "branches/foobranch", 
+                "trunk0"))
 
-        weave = tree.get_weave(tree.inventory.path2id("hosts"))
-        self.assertEqual(['svn-v1:6@%s-branches%%2ffoobranch' % uuid, 
-                          'svn-v1:7@%s-branches%%2ffoobranch' % uuid],
+        weave = newbranch.repository.weave_store.get_weave(
+            tree.inventory.path2id("hosts"),
+            newbranch.repository.get_transaction())
+        self.assertEqual([
+            generate_svn_revision_id(uuid, 6, "branches/foobranch", "trunk0"),
+            generate_svn_revision_id(uuid, 7, "branches/foobranch", "trunk0")],
                           weave.versions())
+        newbranch.unlock()
  
 
     def test_fetch_odd(self):
@@ -359,25 +537,99 @@ foohosts""")
 
         newbranch = newdir.open_branch()
 
-        uuid = olddir.open_repository().uuid
+        uuid = olddir.find_repository().uuid
         tree = newbranch.repository.revision_tree(
-                "svn-v1:6@%s-branches%%2ffoobranch" % uuid)
-
-        weave = tree.get_weave(tree.inventory.path2id("hosts"))
-        self.assertEqual(['svn-v1:1@%s-trunk' % uuid, 
-                          'svn-v1:2@%s-trunk' % uuid, 
-                          'svn-v1:3@%s-trunk' % uuid, 
-                          'svn-v1:6@%s-branches%%2ffoobranch' % uuid],
-                          weave.versions())
+             generate_svn_revision_id(uuid, 6, "branches/foobranch", "trunk0"))
+        transaction = newbranch.repository.get_transaction()
+        newbranch.repository.lock_read()
+        weave = newbranch.repository.weave_store.get_weave(
+                tree.inventory.path2id("hosts"), transaction)
+        self.assertEqual(set([
+            generate_svn_revision_id(uuid, 1, "trunk", "trunk0"),
+            generate_svn_revision_id(uuid, 2, "trunk", "trunk0"),
+            generate_svn_revision_id(uuid, 3, "trunk", "trunk0"),
+            generate_svn_revision_id(uuid, 6, "branches/foobranch", "trunk0")]),
+                          set(weave.versions()))
+        newbranch.repository.unlock()
+
+    def test_check(self):
+        self.make_client('d', 'dc')
+        branch = Branch.open('d')
+        result = branch.check()
+        self.assertEqual(branch, result.branch) 
  
+    def test_generate_revision_id(self):
+        self.make_client('d', 'dc')
+        self.build_tree({'dc/bla/bloe': None})
+        self.client_add("dc/bla")
+        self.client_commit("dc", "bla")
+        branch = Branch.open('d')
+        self.assertEqual("svn-v%d-none:%s::1" % (MAPPING_VERSION, branch.repository.uuid),  branch.generate_revision_id(1))
+
+    def test_create_checkout(self):
+        repos_url = self.make_client('d', 'dc')
+
+        self.build_tree({'dc/trunk': None, 'dc/trunk/hosts': 'hej1'})
+        self.client_add("dc/trunk")
+        self.client_commit("dc", "created trunk and added hosts") #1
+
+        url = "svn+"+repos_url+"/trunk"
+        oldbranch = Branch.open(url)
+
+        newtree = self.create_checkout(oldbranch, "e")
+        self.assertTrue(newtree.branch.repository.has_revision(
+           oldbranch.generate_revision_id(1)))
+
+        self.assertTrue(os.path.exists("e/.bzr"))
+        self.assertFalse(os.path.exists("e/.svn"))
+
+    def test_create_checkout_lightweight(self):
+        repos_url = self.make_client('d', 'dc')
+
+        self.build_tree({'dc/trunk': None, 
+                         'dc/trunk/hosts': 'hej1'})
+        self.client_add("dc/trunk")
+        self.client_commit("dc", "created trunk and added hosts") #1
+
+        url = "svn+"+repos_url+"/trunk"
+        oldbranch = Branch.open(url)
+
+        newtree = self.create_checkout(oldbranch, "e", lightweight=True)
+        self.assertEqual(oldbranch.generate_revision_id(1), newtree.base_revid)
+        self.assertTrue(os.path.exists("e/.svn"))
+        self.assertFalse(os.path.exists("e/.bzr"))
+
+    def test_create_checkout_lightweight_stop_rev(self):
+        repos_url = self.make_client('d', 'dc')
+
+        self.build_tree({'dc/trunk': None, 
+                         'dc/trunk/hosts': 'hej1'})
+        self.client_add("dc/trunk")
+        self.client_commit("dc", "created trunk and added hosts") #1
+        self.client_update("dc")
+        
+        self.build_tree({'dc/trunk/hosts': 'bloe'})
+        self.client_commit("dc", "added another revision")
+        self.client_update("dc")
+
+        url = "svn+"+repos_url+"/trunk"
+        oldbranch = Branch.open(url)
+
+        newtree = self.create_checkout(oldbranch, "e", revision_id=
+           oldbranch.generate_revision_id(1), lightweight=True)
+        self.assertEqual(oldbranch.generate_revision_id(1),
+           newtree.base_revid)
+        self.assertTrue(os.path.exists("e/.svn"))
+        self.assertFalse(os.path.exists("e/.bzr"))
+
     def test_fetch_branch(self):
-        repos_url = self.make_client('d', 'sc')
+        self.make_client('d', 'sc')
 
         self.build_tree({'sc/foo/bla': "data"})
         self.client_add("sc/foo")
         self.client_commit("sc", "foo")
 
-        olddir = BzrDir.open("sc")
+        olddir = self.open_checkout_bzrdir("sc")
 
         os.mkdir("dc")
         
@@ -387,22 +639,91 @@ foohosts""")
                 olddir.open_branch().last_revision(),
                 newdir.open_branch().last_revision())
 
+    def test_fetch_dir_upgrade(self):
+        repos_url = self.make_client('d', 'sc')
+
+        self.build_tree({'sc/trunk/mylib/bla': "data", "sc/branches": None})
+        self.client_add("sc/trunk")
+        self.client_add("sc/branches")
+        self.client_commit("sc", "foo")
+
+        self.client_copy("sc/trunk/mylib", "sc/branches/abranch")
+        self.client_commit("sc", "Promote mylib")
+
+        olddir = self.open_checkout_bzrdir("sc/branches/abranch")
+
+        os.mkdir("dc")
+        
+        newdir = olddir.sprout('dc')
+
+        self.assertEqual(
+                olddir.open_branch().last_revision(),
+                newdir.open_branch().last_revision())
+
+    def test_fetch_branch_downgrade(self):
+        repos_url = self.make_client('d', 'sc')
+
+        self.build_tree({'sc/trunk': None, "sc/branches/abranch/bla": 'foo'})
+        self.client_add("sc/trunk")
+        self.client_add("sc/branches")
+        self.client_commit("sc", "foo")
+
+        self.client_copy("sc/branches/abranch", "sc/trunk/mylib")
+        self.client_commit("sc", "Demote mylib")
+
+        olddir = self.open_checkout_bzrdir("sc/trunk")
+
+        os.mkdir("dc")
+        
+        newdir = olddir.sprout('dc')
+
+        self.assertEqual(
+                olddir.open_branch().last_revision(),
+                newdir.open_branch().last_revision())
+
+
+
     def test_ghost_workingtree(self):
         # Looks like bazaar has trouble creating a working tree of a 
         # revision that has ghost parents
-        repos_url = self.make_client('d', 'sc')
+        self.make_client('d', 'sc')
 
         self.build_tree({'sc/foo/bla': "data"})
         self.client_add("sc/foo")
-        self.client_set_prop("sc", "bzr:merge", "some-ghost\n")
+        self.client_set_prop("sc", "bzr:ancestry:v3-none", "some-ghost\n")
         self.client_commit("sc", "foo")
 
-        olddir = BzrDir.open("sc")
+        olddir = self.open_checkout_bzrdir("sc")
 
         os.mkdir("dc")
         
         newdir = olddir.sprout('dc')
-        newdir.open_repository().get_revision(
+        newdir.find_repository().get_revision(
                 newdir.open_branch().last_revision())
-        newdir.open_repository().get_revision_inventory(
+        newdir.find_repository().get_revision_inventory(
                 newdir.open_branch().last_revision())
+
+class TestFakeControlFiles(TestCase):
+    def test_get_utf8(self):
+        f = FakeControlFiles()
+        self.assertRaises(NoSuchFile, f.get_utf8, "foo")
+
+
+    def test_get(self):
+        f = FakeControlFiles()
+        self.assertRaises(NoSuchFile, f.get, "foobla")
+
+class BranchFormatTests(TestCase):
+    def setUp(self):
+        self.format = SvnBranchFormat()
+
+    def test_initialize(self):
+        self.assertRaises(NotImplementedError, self.format.initialize, None)
+
+    def test_get_format_string(self):
+        self.assertEqual("Subversion Smart Server", 
+                         self.format.get_format_string())
+
+    def test_get_format_description(self):
+        self.assertEqual("Subversion Smart Server", 
+                         self.format.get_format_description())