[PATCH] posix: fix split() for the case where the path is at the root of the filesystem

Remy Blank remy.blank at pobox.com
Sat Jan 5 19:50:09 CST 2013


# HG changeset patch
# User Remy Blank <remy.blank at pobox.com>
# Date 1357401181 -3600
# Node ID c068597242626dbb517e750a7ef0bed664131517
# Parent  83aa4359c49f67bcb98fb9c7d885ed4ac7443239
posix: fix split() for the case where the path is at the root of the filesystem

posixpath.split() strips '/' from the dirname *unless it is the root*. This
patch reproduces this behavior in posix.split(). The old behavior causes a
crash when creating a file at the root of the repo with localrepo.wfile()
when the repo is at the root of the filesystem.

diff --git a/mercurial/posix.py b/mercurial/posix.py
--- a/mercurial/posix.py
+++ b/mercurial/posix.py
@@ -21,14 +21,32 @@ umask = os.umask(0)
 os.umask(umask)
 
 def split(p):
-    '''Same as os.path.split, but faster'''
+    '''Same as os.path.split, but faster
+
+    >>> split('/absolute/path/to/file')
+    ('/absolute/path/to', 'file')
+    >>> split('relative/path/to/file')
+    ('relative/path/to', 'file')
+    >>> split('file_alone')
+    ('', 'file_alone')
+    >>> split('path/to/directory/')
+    ('path/to/directory', '')
+    >>> split('/multiple/path//separators')
+    ('/multiple/path', 'separators')
+    >>> split('/file_at_root')
+    ('/', 'file_at_root')
+    >>> split('///multiple_leading_separators_at_root')
+    ('///', 'multiple_leading_separators_at_root')
+    >>> split('')
+    ('', '')
+    '''
     ht = p.rsplit('/', 1)
     if len(ht) == 1:
         return '', p
     nh = ht[0].rstrip('/')
     if nh:
         return nh, ht[1]
-    return ht
+    return ht[0] + '/', ht[1]
 
 def openhardlinks():
     '''return true if it is safe to hold open file handles to hardlinks'''
diff --git a/tests/test-doctest.py b/tests/test-doctest.py
--- a/tests/test-doctest.py
+++ b/tests/test-doctest.py
@@ -6,6 +6,7 @@ import doctest
 
 import mercurial.util
 doctest.testmod(mercurial.util)
+doctest.testmod(mercurial.util.platform)
 
 import mercurial.changelog
 doctest.testmod(mercurial.changelog)


More information about the Mercurial-devel mailing list