def test_hash_directory(self): """ Ensure HashDirectory will correctly hash all files in a directory """ hash_directory = HashDirectory('tests/fixtures/test-data') results = hash_directory.hash_files() self.assertEqual(results['documents/Important Document 1.odt'], 'd460a36805fb460c038d96723f206b20') self.assertEqual(results['documents/Important Presentation.odp'], '1911ec839cedcbf00739a7d3447ec3a3') self.assertEqual(results['pictures/Picture #1.jpg'], '6eec850e32622c0e33bdae08ced29e24') self.assertEqual(results['documents/exclude_me.txt'], '2e7d8cb32bb82e838506aff5600182d1') self.assertEqual(len(results), 4)
def test_hash_directory_keyboard_interupt(self): """ Ensure HashDirectory gracefully handles a KeyboardInterrupt """ with patch('furtive.hasher.multiprocessing.Pool') as mock_pool: pool = MagicMock() pool.map.side_effect = KeyboardInterrupt mock_pool.return_value = pool hash_directory = HashDirectory('tests/fixtures/test-data') results = hash_directory.hash_files() pool.terminate.assert_called_once_with() self.assertEqual(results, {})
def test_hash_directory_errors_when_given_file(self): """ Ensure HashDirectory raises an error when a file is provided instead of a directory """ with self.assertRaises(OSError): hash_directory = HashDirectory('tests/fixtures/test-data/documents/exclude_me.txt') hash_directory.hash_files()