Implement formatting of symlink targets. With unit test.

2 files changed, 27 insertions, 2 deletions

diff --git a/summainlib.py b/summainlib.py
index ad6fee0..1d0dfb4 100644
--- a/summainlib.py
+++ b/summainlib.py
@@ -35,10 +35,12 @@ class FilesystemObject(object):
     '''
     
     def __init__(self, filename, stat_result=None, sha1=None,
-                 open_file=None):
+                 open_file=None, readlink=None):
         stat_result = stat_result or os.lstat(filename)
         sha1 = sha1 or hashlib.sha1()
         self.open_file = open_file or self.open_file
+        readlink = readlink or os.readlink
+        
         self.values = dict()
         self['Name'] = filename
         self['Mtime'] = self.format_time(stat_result.st_mtime)
@@ -53,6 +55,8 @@ class FilesystemObject(object):
         self['Group'] = self.lookup_group(stat_result.st_gid)
         if stat.S_ISREG(stat_result.st_mode):
             self['Sha-1'] = self.compute_sha1(filename, sha1)
+        if stat.S_ISLNK(stat_result.st_mode):
+            self['Target'] = readlink(filename)
 
     def format_time(self, timestamp):
         return time.strftime('%Y-%m-%d %H:%M:%S +0000', 
diff --git a/summainlib_tests.py b/summainlib_tests.py
index cbfb11c..c206155 100644
--- a/summainlib_tests.py
+++ b/summainlib_tests.py
@@ -51,6 +51,19 @@ class FakeOpenFile(object):
         pass
 
 
+class FakeReadlink(object):
+
+    def __init__(self, parent):
+        self.parent = parent
+
+    def __call__(self, filename):
+        if stat.S_ISLNK(self.parent.st.st_mode):
+            self.target = 'symlink'
+        else:
+            self.target = ''
+        return self.target
+
+
 class FilesystemObjectTests(unittest.TestCase):
 
     def setUp(self):
@@ -66,7 +79,8 @@ class FilesystemObjectTests(unittest.TestCase):
     def new(self, name):
         return summainlib.FilesystemObject(name, stat_result=self.st,
                                            sha1=FakeSha1(),
-                                           open_file=FakeOpenFile())
+                                           open_file=FakeOpenFile(),
+                                           readlink=FakeReadlink(self))
         
     def test_formats_simple_name_identically(self):
         self.assertEqual(self.new('foo')['Name'], 'foo')
@@ -112,3 +126,10 @@ class FilesystemObjectTests(unittest.TestCase):
         self.st.st_mode = stat.S_IFDIR | 0755
         self.assertEqual(self.new('foo')['Sha-1'], '')
 
+    def test_formats_target_correctly_for_symlink(self):
+        self.st.st_mode = stat.S_IFLNK | 0777
+        self.assertEqual(self.new('foo')['Target'], 'symlink')
+
+    def test_formats_target_correctly_for_regular_file(self):
+        self.assertEqual(self.new('foo')['Target'], '')
+