From ae1d95d4fa9c22e8e1da6aeaefb5ff8c7f101ea1 Mon Sep 17 00:00:00 2001
From: LeoHsiao <leohsiao@foxmail.com>
Date: Wed, 16 Sep 2020 22:22:08 +0800
Subject: [PATCH] Refactor tiff-test.sh to tiff_test()

---
 tests/bash_tests/testcases.py | 80 +++++++++++++++++++++++++++++++++++
 tests/bash_tests/utils.py     | 25 ++++++-----
 2 files changed, 95 insertions(+), 10 deletions(-)

diff --git a/tests/bash_tests/testcases.py b/tests/bash_tests/testcases.py
index a3e2abc3..8c4240e4 100644
--- a/tests/bash_tests/testcases.py
+++ b/tests/bash_tests/testcases.py
@@ -1016,3 +1016,83 @@ set Exif.Photo.DateTimeDigitized 2020:05:26 07:31:42
         out     += BT.Executer('stringto-test')
         BT.reportTest('stringto-test', out)
 
+
+    def tiff_test(self):
+        # TIFF parser test driver
+
+        def exifprobe(img):
+            out  = BT.Output()
+
+            # Convert each line from that format:
+            #   Exif.Image.NewSubfileType       Long    1  Primary image
+            # to that format:
+            #   NewSubfileType 	 Long 	 1 	 "Primary image"
+            tags = []
+            typs = []
+            lens = []
+            vals = []
+            e    = BT.Executer('exiv2 -pa {img}', vars(), redirect_stderr_to_stdout=False)
+            for line in e.stdout.split('\n'):
+                fields = re.findall(r'^(\S*)\s*(\S*)\s*(\d*)\s*(.*)$', line)[0]
+                tags  += [fields[0].split('.')[2]]      # tag
+                typs  += [fields[1]]                    # type
+                lens  += [fields[2]]                    # length
+                vals  += [fields[3]]                    # value
+
+            out += 'exiv2 -pa output'
+            for i in range(len(tags)):
+                out += ' \t '.join([tags[i], typs[i], lens[i], '"{}"'.format(vals[i])])
+
+            # Parse -pS output:
+            # address |    tag                           |      type |    count |   offset | value
+            #     254 | 0x00fe NewSubfileType            |      LONG |        1 |        0 | 0
+            TAGS = []
+            TYPS = []
+            LENS = []
+            VALS = []
+            e    = BT.Executer('exiv2 -pS {img}', vars(), redirect_stderr_to_stdout=False)
+            for line in e.stdout.split('\n')[2:-1]:
+                fields = [i.strip(' ') for i in line.split('|')]
+                TAGS  += [fields[1].split(' ')[1]]
+                TYPS  += [fields[2]]
+                LENS  += [fields[3]]
+                VALS  += [fields[5]]
+
+            out += '\nexiv2 -pS output'
+            for i in range(len(TAGS)):
+                out += ' \t '.join([TAGS[i], TYPS[i], LENS[i], '"{}"'.format(VALS[i])])
+
+            out += '\nAnalysis'
+            out += 'count =  {} COUNT =  {}'.format(len(tags), len(TAGS))
+            
+            # Make them have the same number of lines
+            max_lines  = max(len(TAGS), len(tags))
+            for _list in [tags, typs, lens, vals, TAGS, TYPS, LENS, VALS]:
+                _list += [''] * (max_lines - len(_list))
+
+            # Compare the main fields of each line
+            for i in range(max_lines):
+                if TAGS[i]  != tags[i]:
+                    out     += 'TAG {} {} mismatch'.format(TAGS[i], tags[i]).replace('  ', ' ')
+
+                TYPS[i]      = TYPS[i].upper()
+                typs[i]      = typs[i].upper()
+                if TYPS[i]  != typs[i]:
+                    out     += 'TYPE {} {} mismatch'.format(TYPS[i], typs[i]).replace('  ', ' ')
+
+                if LENS[i]  != lens[i]:
+                    out     += 'Length {} {} mismatch'.format(LENS[i], lens[i]).replace('  ', ' ')
+
+                if typs[i]  == 'ASCII' and VALS[i] != vals[i]:
+                    out     += 'Value {} {} mismatch'.format(VALS[i], vals[i]).replace('  ', ' ')
+
+            return str(out)
+
+        test_file = 'mini9.tif'
+        BT.copyTestFile(test_file)
+        out      = BT.Output()
+        out     += exifprobe(test_file)
+        out     += BT.Executer('tiff-test {test_file}', vars())
+        out     += exifprobe(test_file)
+        BT.reportTest('tiff-test', out)
+
diff --git a/tests/bash_tests/utils.py b/tests/bash_tests/utils.py
index 4a5e213d..ea7d3738 100644
--- a/tests/bash_tests/utils.py
+++ b/tests/bash_tests/utils.py
@@ -343,17 +343,22 @@ def diffCheck(file1, file2, in_bytes=False, encoding=None):
 
 def simply_diff(file1, file2, encoding=None):
     """ Find the first different line of the two text files """
-    encoding         = encoding or Config.encoding
-    list1            = cat(file1, encoding=encoding).split('\n')
-    list2            = cat(file2, encoding=encoding).split('\n')
-    if list1        == list2:
+    encoding    = encoding or Config.encoding
+    list1       = cat(file1, encoding=encoding).split('\n')
+    list2       = cat(file2, encoding=encoding).split('\n')
+    if list1   == list2:
         return
-    report           = []
-    report          += ['{}: {} lines'.format(file1, len(list1))]
-    report          += ['{}: {} lines'.format(file2, len(list2))]
-    max_lines        = max(len(list1), len(list2))
-    for i in [list1, list2]:
-        i           += [''] * (max_lines - len(i))    # Make them have the same number of lines
+
+    report      = []
+    report     += ['{}: {} lines'.format(file1, len(list1))]
+    report     += ['{}: {} lines'.format(file2, len(list2))]
+
+    # Make them have the same number of lines
+    max_lines   = max(len(list1), len(list2))
+    for _list in [list1, list2]:
+        _list  += [''] * (max_lines - len(_list))
+
+    # Compare each line
     for i in range(max_lines):
         if list1[i] != list2[i]:
             report  += ['The first mismatch is in line {}:'.format(i + 1)]