Add relative_path method, with tests.

1 files changed, 24 insertions, 0 deletions

diff --git a/summainlib_tests.py b/summainlib_tests.py
index 313b3a4..283c22b 100644
--- a/summainlib_tests.py
+++ b/summainlib_tests.py
@@ -144,6 +144,30 @@ class FilesystemObjectTests(unittest.TestCase):
         fso = self.new('foo', mode=stat.S_IFDIR | 0777)
         output = fso.format()
         self.assert_('Size:' not in output)
+        
+    def test_relative_path_returns_path_if_not_starting_with_root(self):
+        fso = self.new('/foo/bar')
+        self.assertEqual(fso.relative_path('/yo'), '/foo/bar')
+        
+    def test_relative_path_returns_partial_path_if_starting_with_root(self):
+        fso = self.new('/foo/bar')
+        self.assertEqual(fso.relative_path('/foo'), 'bar')
+        
+    def test_relative_path_returns_dot_if_same_as_root_and_dir(self):
+        fso = self.new('/foo/bar', mode=stat.S_IFDIR)
+        self.assertEqual(fso.relative_path('/foo/bar'), '.')
+        
+    def test_relative_path_returns_path_if_same_as_root_and_not_dir(self):
+        fso = self.new('/foo/bar', mode=stat.S_IFREG)
+        self.assertEqual(fso.relative_path('/foo/bar'), '/foo/bar')
+        
+    def test_relative_path_returns_path_if_root_is_slashless_prefix(self):
+        fso = self.new('/foobar', mode=stat.S_IFREG)
+        self.assertEqual(fso.relative_path('/foo'), '/foobar')
+        
+    def test_relative_path_returns_partial_path_if_root_ends_in_slash(self):
+        fso = self.new('/foo/bar', mode=stat.S_IFREG)
+        self.assertEqual(fso.relative_path('/foo/'), 'bar')
 
 
 class FilesystemObjectNormalizedNumbersTests(unittest.TestCase):