From 24cf0a6a4c3380af5289a762d285d459086a9991 Mon Sep 17 00:00:00 2001
From: Mike McLean <mikem@redhat.com>
Date: Feb 06 2023 12:17:33 +0000
Subject: simple unit tests for splicing code


---

diff --git a/tests/test_lib/data/rpms/test-deps-1-1.fc24.x86_64.rpm.signed b/tests/test_lib/data/rpms/test-deps-1-1.fc24.x86_64.rpm.signed
new file mode 100644
index 0000000..eaf322e
Binary files /dev/null and b/tests/test_lib/data/rpms/test-deps-1-1.fc24.x86_64.rpm.signed differ
diff --git a/tests/test_lib/test_spliced_sig.py b/tests/test_lib/test_spliced_sig.py
new file mode 100644
index 0000000..107bcd8
--- /dev/null
+++ b/tests/test_lib/test_spliced_sig.py
@@ -0,0 +1,40 @@
+# coding=utf-8
+from __future__ import absolute_import
+import os.path
+import shutil
+import tempfile
+import unittest
+
+import koji
+
+
+
+class TestCheckSigMD5(unittest.TestCase):
+
+    def setUp(self):
+        self.tempdir = tempfile.mkdtemp()
+        orig_path = os.path.join(os.path.dirname(__file__), 'data/rpms/test-deps-1-1.fc24.x86_64.rpm')
+        orig_signed = os.path.join(os.path.dirname(__file__), 'data/rpms/test-deps-1-1.fc24.x86_64.rpm.signed')
+        self.path = '%s/test.rpm' % self.tempdir
+        self.signed = '%s/test.rpm.signed' % self.tempdir
+        shutil.copyfile(orig_path, self.path)
+        shutil.copyfile(orig_signed, self.signed)
+
+    def tearDown(self):
+        shutil.rmtree(self.tempdir)
+
+    def test_spliced_sig_reader(self):
+        contents_orig = open(self.path, 'rb').read()
+        contents_signed = open(self.signed, 'rb').read()
+        sighdr = koji.rip_rpm_sighdr(self.signed)
+        contents_spliced = koji.spliced_sig_reader(self.path, sighdr).read()
+        self.assertEqual(contents_signed, contents_spliced)
+        self.assertNotEqual(contents_signed, contents_orig)
+
+    def test_splice_rpm_sighdr(self):
+        contents_signed = open(self.signed, 'rb').read()
+        sighdr = koji.rip_rpm_sighdr(self.signed)
+        dst = '%s/signed-copy.rpm' % self.tempdir
+        koji.splice_rpm_sighdr(sighdr, self.path, dst=dst)
+        contents_spliced = open(dst, 'rb').read()
+        self.assertEqual(contents_signed, contents_spliced)