add a unit test

tests/test_lib/test_check_sigmd5.py

@@ -0,0 +1,35 @@
+# coding=utf-8
+from __future__ import absolute_import
+import os.path
+import shutil
+import tempfile
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
+
+from koji.util import check_sigmd5
+
+
+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')
+        self.path = '%s/test.rpm' % self.tempdir
+        shutil.copyfile(orig_path, self.path)
+
+    def tearDown(self):
+        shutil.rmtree(self.tempdir)
+
+    def test_check_sigmd5_good(self):
+        value = check_sigmd5(self.path)
+        self.assertEqual(value, True)
+
+    def test_check_sigmd5_bad(self):
+        # corrupt the file by truncating it
+        with open(self.path, 'a+b') as f:
+            f.seek(5, 2)
+            f.truncate()
+        value = check_sigmd5(self.path)
+        self.assertEqual(value, False)