Add support for SHA512 (#1)

Author: ddalu5

README.md

@@ -15,7 +15,7 @@ Tested on:
 
 ## Documentation
 
-Latest version is 1.0.1
+Latest version is 1.0.2
 
 ## Installation
 
@@ -31,6 +31,7 @@ The supported checksum algorithms are:
 
 - md5
 - sha256
+- sha512
 
 #### Target files sizes
 

pyntegrity/config.py

@@ -17,13 +17,16 @@
 """
 # Hash settings
 
+SHA512_LENGTH = 128
 SHA256_LENGTH = 64
 MD5_LENGTH = 32
 
+SHA512_REX = r"^[A-Fa-f\d]{128}$"
 SHA256_REX = r"^[A-Fa-f\d]{64}$"
 MD5_REX = r"^[a-fA-F\d]{32}$"
 
 SUPPORTED_HASH_ALGOS = {
     "sha256": {"REX": SHA256_REX, "LENGTH": SHA256_LENGTH},
     "md5": {"REX": MD5_REX, "LENGTH": MD5_LENGTH},
+    "sha512": {"REX": SHA512_REX, "LENGTH": SHA512_LENGTH},
 }

pyntegrity/core.py

@@ -29,10 +29,6 @@
 
 from .config import SUPPORTED_HASH_ALGOS
 
-HashAlgoEnum = enum.Enum(
-    "HashAlgoEnum", {key: key for key in SUPPORTED_HASH_ALGOS.keys()}
-)
-
 
 def detect_hash_algo(checksum_str: str):
     """

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name="pyntegrity",
-    version="1.0.1",
+    version="1.0.2",
     description="Pyntegrity is a Python package that helps you check a file integrity.",
     long_description=long_description,
     long_description_content_type="text/markdown",

tests/test_core.py

@@ -50,6 +50,15 @@ def test_validate_checksum_str_sha256_nok(self):
         with self.assertRaises(HashStrNotValidException):
             validate_checksum_str(invalid_sha256, "sha256")
 
+    def test_validate_checksum_str_sha512_ok(self):
+        valid_sha512 = "d80f94038e28749518984f0a7827530fc272117f952db85d54baa05ea89b6fc6fc9a4059be18c9b8762139404bd43c59e7ad5814ddc0eb73bdc6a6355b16b718"
+        self.assertTrue(validate_checksum_str(valid_sha512, "sha512"))
+
+    def test_validate_checksum_str_sha512_nok(self):
+        invalid_sha512 = "d80f94038e28749518984f0a78275x0fc272117f952db85d54baa05ea89b6fc6fc9a4059be18c9b8762139404bd43c59e7ad5814ddc0eb73bdc6a6355b16b718"
+        with self.assertRaises(HashStrNotValidException):
+            validate_checksum_str(invalid_sha512, "sha512")
+
 
 class TestGetFileFromStr(TestCase):
     def test_get_file_path_from_str_ok(self):
@@ -66,7 +75,7 @@ def test_get_file_path_from_str_nok_not_found(self):
 
 
 class TestGetFileChecksum(TestCase):
-    def test_get_file_checksum_text(self):
+    def test_get_file_checksum_text_md5_true(self):
         obj = IntegrityValidator(
             str_path="tests/data/test_file.txt",
             checksum_str="bbe4f28b27120e0a9611d90d242bc656",
@@ -76,7 +85,7 @@ def test_get_file_checksum_text(self):
             "bbe4f28b27120e0a9611d90d242bc656",
         )
 
-    def test_get_file_checksum_bin(self):
+    def test_get_file_checksum_bin_md5_true(self):
         obj = IntegrityValidator(
             str_path="tests/data/test_file.dat",
             checksum_str="0cb988d042a7f28dd5fe2b55b3f5ac7a",
@@ -86,6 +95,56 @@ def test_get_file_checksum_bin(self):
             "0cb988d042a7f28dd5fe2b55b3f5ac7a",
         )
 
+    def test_get_file_checksum_text_md5_false(self):
+        obj = IntegrityValidator(
+            str_path="tests/data/test_file.txt",
+            checksum_str="bbe4f28b27120e0a9611d90d242bc657",
+        )
+        self.assertNotEqual(
+            obj.get_file_checksum("md5"),
+            "bbe4f28b27120e0a9611d90d242bc657",
+        )
+
+    def test_get_file_checksum_bin_sha256_true(self):
+        obj = IntegrityValidator(
+            str_path="tests/data/test_file.dat",
+            checksum_str="a12871fee210fb8619291eaea194581cbd2531e4b23759d225f6806923f63222",
+        )
+        self.assertEqual(
+            obj.get_file_checksum("sha256"),
+            "a12871fee210fb8619291eaea194581cbd2531e4b23759d225f6806923f63222",
+        )
+
+    def test_get_file_checksum_text_sha256_false(self):
+        obj = IntegrityValidator(
+            str_path="tests/data/test_file.txt",
+            checksum_str="a12871fee210fb8619291eaea194581cbd2531e4b23759d225f6806923f63223",
+        )
+        self.assertNotEqual(
+            obj.get_file_checksum("sha256"),
+            "a12871fee210fb8619291eaea194581cbd2531e4b23759d225f6806923f63223",
+        )
+
+    def test_get_file_checksum_bin_sha512_true(self):
+        obj = IntegrityValidator(
+            str_path="tests/data/test_file.dat",
+            checksum_str="bdd81ab233bceb6ad233cd1871509320a18d0335a891cf98730990e8923e1dda04f3358e9c7e1c3d16b16f408cfafb6af53254ef3023ed2436533808b6ca9933",
+        )
+        self.assertEqual(
+            obj.get_file_checksum("sha512"),
+            "bdd81ab233bceb6ad233cd1871509320a18d0335a891cf98730990e8923e1dda04f3358e9c7e1c3d16b16f408cfafb6af53254ef3023ed2436533808b6ca9933",
+        )
+
+    def test_get_file_checksum_text_sha512_false(self):
+        obj = IntegrityValidator(
+            str_path="tests/data/test_file.txt",
+            checksum_str="add81ab233bceb6ad233cd1871509320a18d0335a891cf98730990e8923e1dda04f3358e9c7e1c3d16b16f408cfafb6af53254ef3023ed2436533808b6ca9933",
+        )
+        self.assertNotEqual(
+            obj.get_file_checksum("sha512"),
+            "add81ab233bceb6ad233cd1871509320a18d0335a891cf98730990e8923e1dda04f3358e9c7e1c3d16b16f408cfafb6af53254ef3023ed2436533808b6ca9933",
+        )
+
 
 class TestValidateFileIntegrity(TestCase):
     def test_validate_file_integrity_true(self):