fix checksum test

SilenZcience · Dec 13, 2024 · 721d6ca · 721d6ca
1 parent 9e11a72
commit 721d6ca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cat_win/tests/src/service/test_checksum.py b/cat_win/tests/src/service/test_checksum.py
@@ -39,7 +39,7 @@ def test_checksum_colored(self):
         self.assertEqual(get_checksum_from_file(test_file_path, ['X', 'Y']), expected_output)
 
     def test_checksum_invalid_file(self):
-        self.assertEqual(get_checksum_from_file('randomFileThatHopefullyDoesNotExistWithWeirdCharsForSafety*!?\\/:<>|'), 'OSError')
+        self.assertIn(get_checksum_from_file('randomFileThatHopefullyDoesNotExistWithWeirdCharsForSafety*!?\\/:<>|'), ['FileNotFoundError', 'OSError'])
 
     def test_print_checksum(self):
         expected_output = f"XChecksum of '{test_file_path}':Y\n"