From a69ac3e509eed60f1801aca605531b6bc8f3e506 Mon Sep 17 00:00:00 2001 From: erinhmclark Date: Tue, 11 Feb 2025 09:46:22 +0000 Subject: [PATCH] Fix file hash reference in S3 tests --- tests/storages/test_S3_storage.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/tests/storages/test_S3_storage.py b/tests/storages/test_S3_storage.py index 2594e73..e532a18 100644 --- a/tests/storages/test_S3_storage.py +++ b/tests/storages/test_S3_storage.py @@ -2,13 +2,12 @@ from typing import Type import pytest from unittest.mock import MagicMock, patch from auto_archiver.core import Media -from auto_archiver.modules.hash_enricher import HashEnricher from auto_archiver.modules.s3_storage import s3_storage -class TestGDriveStorage: +class TestS3Storage: """ - Test suite for GDriveStorage. + Test suite for S3Storage. """ module_name: str = "s3_storage" storage: Type[s3_storage] @@ -66,7 +65,7 @@ class TestGDriveStorage: # Set duplicate checking config to true: self.storage.random_no_duplicate = True - with patch('auto_archiver.modules.hash_enricher.HashEnricher.calculate_hash') as mock_calc_hash, \ + with patch('auto_archiver.modules.s3_storage.s3_storage.calculate_file_hash') as mock_calc_hash, \ patch.object(self.storage, 'file_in_folder') as mock_file_in_folder: mock_calc_hash.return_value = 'beepboop123beepboop123beepboop123' mock_file_in_folder.return_value = 'existing_key.txt' @@ -87,8 +86,7 @@ class TestGDriveStorage: # Create test media with calculated hash media = Media("test.txt") media.key = "original_path.txt" - - with patch('auto_archiver.modules.hash_enricher.HashEnricher.calculate_hash') as mock_calculate_hash: + with patch('auto_archiver.modules.s3_storage.s3_storage.calculate_file_hash') as mock_calculate_hash: mock_calculate_hash.return_value = "beepboop123beepboop123beepboop123" # Verify upload assert self.storage.is_upload_needed(media) is False