add unit tests for get_check_sum

Author: ddalu5

pyntegrity/core.py

@@ -108,5 +108,5 @@ def validate_file_integrity(self):
     def get_file_checksum(self):
         with self.file.open() as file_to_validate:
             file_content = file_to_validate.read()
-            self.hashlib_obj.update(file_content)
+            self.hashlib_obj.update(file_content.encode())
             return self.hashlib_obj.hexdigest()

tests/test_core.py

@@ -2,8 +2,9 @@
 from unittest import TestCase
 
 from pyntegrity.core import detect_hash_algo
-from pyntegrity.core import get_file_path_from_str
+from pyntegrity.core import IntegrityValidator
 from pyntegrity.core import validate_checksum_str
+from pyntegrity.core import get_file_path_from_str
 
 from pyntegrity.exceptions import FileNotFoundException
 from pyntegrity.exceptions import ObjectNotAFileException
@@ -62,3 +63,19 @@ def test_get_file_path_from_str_nok_is_not_file(self):
     def test_get_file_path_from_str_nok_not_found(self):
         with self.assertRaises(FileNotFoundException):
             get_file_path_from_str("tests/data/doesnt_exists.csv")
+
+
+class TestGetFileChecksum(TestCase):
+    def test_get_file_checksum_text(self):
+        obj = IntegrityValidator(
+            str_path="tests/data/test_file.txt",
+            checksum_str="bbe4f28b27120e0a9611d90d242bc656",
+        )
+        self.assertEqual(obj.get_file_checksum(), "bbe4f28b27120e0a9611d90d242bc656")
+
+    def test_get_file_checksum_bin(self):
+        obj = IntegrityValidator(
+            str_path="tests/data/test_file.dat",
+            checksum_str="0cb988d042a7f28dd5fe2b55b3f5ac7a",
+        )
+        self.assertEqual(obj.get_file_checksum(), "0cb988d042a7f28dd5fe2b55b3f5ac7a")