Return Repo from porcelain.clone().

dulwich/porcelain.py

@@ -113,6 +113,7 @@ def clone(source, target=None, bare=False, outstream=sys.stdout):
     :param target: Path to target repository (optional)
     :param bare: Whether or not to create a bare repository
     :param outstream: Optional stream to write progress to
+    :return: The new repository
     """
     client, host_path = get_transport_and_path(source)
 
@@ -129,6 +130,7 @@ def clone(source, target=None, bare=False, outstream=sys.stdout):
         determine_wants=r.object_store.determine_wants_all,
         progress=outstream.write)
     r["HEAD"] = remote_refs["HEAD"]
+    return r
 
 
 def add(repo=".", paths=None):

dulwich/repo.py

@@ -864,6 +864,7 @@ class BaseRepo(object):
         :param determine_wants: Optional function to determine what refs to
             fetch.
         :param progress: Optional progress function
+        :return: The local refs
         """
         if determine_wants is None:
             determine_wants = lambda heads: heads.values()

dulwich/tests/test_porcelain.py

@@ -99,7 +99,8 @@ class CloneTests(PorcelainTestCase):
         target_path = tempfile.mkdtemp()
         outstream = StringIO()
         self.addCleanup(shutil.rmtree, target_path)
-        porcelain.clone(self.repo.path, target_path, outstream=outstream)
+        r = porcelain.clone(self.repo.path, target_path, outstream=outstream)
+        self.assertEquals(r.path, target_path)
         self.assertEquals(Repo(target_path).head(), c3.id)