1 files changed, 58 insertions, 0 deletions
diff --git a/share/extensions/tests/test_guillotine.py b/share/extensions/tests/test_guillotine.py
new file mode 100644
index 0000000..d478491
--- /dev/null
+++ b/share/extensions/tests/test_guillotine.py
@@ -0,0 +1,58 @@
+# coding=utf-8
+import os
+import tarfile
+from guillotine import Guillotine
+from inkex.tester import ComparisonMixin, TestCase
+
+class TestGuillotineBasic(ComparisonMixin, TestCase):
+    """Test the Guillotine extension"""
+    stderr_protect = False
+    effect_class = Guillotine
+    compare_file = 'svg/guides.svg'
+    comparisons = [
+        ('--image=f{}oo',),
+        ('--ignore=true',),
+    ]
+
+    def test_all_comparisons(self):
+        """Images are extracted to a file directory"""
+        for args in self.comparisons:
+            # Create a landing directory for generated images
+            outdir = os.path.join(self.tempdir, 'img')
+            args += ('--directory={}/'.format(outdir),)
+
+            # But also set this directory into the compare file
+            compare_file = os.path.join(self.tempdir, 'compare_file.svg')
+            with open(self.data_file(self.compare_file), 'rb') as rhl:
+                with open(compare_file, 'wb') as whl:
+                    whl.write(rhl.read().replace(b'{tempdir}', outdir.encode('utf8')))
+
+            self.assertEffect(compare_file, args=args)
+            self.assertTrue(os.path.isdir(outdir))
+
+            infile = self.get_compare_outfile(args)
+            if os.environ.get('EXPORT_COMPARE', False):
+                self.export_comparison(outdir, infile)
+
+            with tarfile.open(infile) as tar_handle:
+                for item in tar_handle:
+                    fileobj = tar_handle.extractfile(item)
+                    with open(os.path.join(outdir, item.name), 'rb') as fhl:
+                        self.assertEqual(fileobj.read(), fhl.read(), "File '{}'".format(item.name))
+
+    @staticmethod
+    def export_comparison(outdir, outfile):
+        """Export the files as a tar file for manual comparison"""
+        tarname = outfile + '.export'
+        tar = tarfile.open(tarname, 'w|')
+
+        # We make a tar archive so we can test it.
+        for name in sorted(os.listdir(outdir)):
+            with open(os.path.join(outdir, name), 'rb') as fhl:
+                fhl.seek(0, 2)
+                info = tarfile.TarInfo(name)
+                info.size = fhl.tell()
+                fhl.seek(0)
+                tar.addfile(info, fhl)
+        tar.close()
+        print("Written output: {}.export".format(outfile))