Handle absolute paths in zips.join_path

Author: mwilliamson

mammoth/zips.py

@@ -65,4 +65,13 @@ def split_path(path):
 
 
 def join_path(*args):
-    return "/".join(list(filter(None, args)))
+    non_empty_paths = list(filter(None, args))
+    
+    relevant_paths = []
+    for path in non_empty_paths:
+        if path.startswith("/"):
+            relevant_paths = [path]
+        else:
+            relevant_paths.append(path)
+    
+    return "/".join(relevant_paths)

tests/zips_tests.py

@@ -27,4 +27,11 @@ def empty_parts_are_ignored_when_joining_paths():
     assert_equal("a", zips.join_path("a", ""))
     assert_equal("b", zips.join_path("", "b"))
     assert_equal("a/b", zips.join_path("a", "", "b"))
-    
+
+
+@istest
+def when_joining_paths_then_absolute_paths_ignore_earlier_paths():
+    assert_equal("/b", zips.join_path("a", "/b"))
+    assert_equal("/b/c", zips.join_path("a", "/b", "c"))
+    assert_equal("/b", zips.join_path("/a", "/b"))
+    assert_equal("/a", zips.join_path("/a"))