From 7c8f2d14652800521a8611010db6adcac7bdbcc1 Mon Sep 17 00:00:00 2001
From: Ian Norton <inorton@gmail.com>
Date: Wed, 2 Nov 2016 17:22:01 +0000
Subject: [PATCH] Very simple fast test of hashing performance for moderate
 sized files

---
 hashingtests.py | 67 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 67 insertions(+)
 create mode 100644 hashingtests.py

diff --git a/hashingtests.py b/hashingtests.py
new file mode 100644
index 00000000..ae55d742
--- /dev/null
+++ b/hashingtests.py
@@ -0,0 +1,67 @@
+#!/usr/bin/env python3
+#
+# This file is part of the clcache project.
+#
+# The contents of this file are subject to the BSD 3-Clause License, the
+# full text of which is available in the accompanying LICENSE file at the
+# root directory of this project.
+#
+
+import os
+import unittest
+import shutil
+import clcache
+import uuid
+
+THIS_DIR = os.path.dirname(__file__)
+ASSETS_DIR = os.path.join(THIS_DIR, "tests", "hashtests")
+
+
+class TestFileHashing(unittest.TestCase):
+    """
+    Test class to create lots of test files and test performance in hashing them.
+    """
+    NUM_TEST_FILES = 1000
+    TEST_FILES = list()
+
+    @classmethod
+    def setUpClass(cls):
+        """
+        Create many files in folders so that we can hash them in the test
+        :return:
+        """
+        if os.path.isdir(ASSETS_DIR):
+            shutil.rmtree(ASSETS_DIR)
+        os.makedirs(ASSETS_DIR)
+
+        for _ in range(cls.NUM_TEST_FILES):
+            fstr = str(uuid.uuid4())
+            fdir = fstr[:3]
+            fname = fstr[3:]
+            if not os.path.isdir(os.path.join(ASSETS_DIR, fdir)):
+                os.makedirs(os.path.join(ASSETS_DIR, fdir))
+            filepath = os.path.join(ASSETS_DIR, fdir, fname)
+            with open(filepath, "w") as fhandle:
+                fhandle.write("{}".format(fname * 8000))
+            cls.TEST_FILES.append(filepath)
+
+    def testExistsManyFiles(self):
+        """
+        Simply check all the files exist
+        :return:
+        """
+        for filepath in self.TEST_FILES:
+            assert os.path.exists(filepath)
+
+    def testHashManyFiles(self):
+        """
+        Hash all our files
+        """
+        for filepath in self.TEST_FILES:
+            filehash = clcache.getFileHash(filepath)
+            assert filehash
+
+
+if __name__ == "__main__":
+    unittest.TestCase.longMessage = True
+    unittest.main()