diff options
Diffstat (limited to 'tests/unit/test_file_utils.py')
| -rw-r--r-- | tests/unit/test_file_utils.py | 14 |
1 files changed, 12 insertions, 2 deletions
diff --git a/tests/unit/test_file_utils.py b/tests/unit/test_file_utils.py index 75be4f6..2c6f0b6 100644 --- a/tests/unit/test_file_utils.py +++ b/tests/unit/test_file_utils.py @@ -5,6 +5,7 @@ import unittest from dataclasses import dataclass from typing import Callable from unittest import mock +import pytest from gn3.fs_helpers import extract_uploaded_file from gn3.fs_helpers import get_dir_hash from gn3.fs_helpers import jsonfile_to_dict @@ -21,17 +22,20 @@ class MockFile: class TestFileUtils(unittest.TestCase): """Test cases for procedures defined in fs_helpers.py""" + @pytest.mark.unit_test def test_get_dir_hash(self): """Test that a directory is hashed correctly""" test_dir = os.path.join(os.path.dirname(__file__), "test_data") self.assertEqual("3aeafab7d53b4f76d223366ae7ee9738", get_dir_hash(test_dir)) + @pytest.mark.unit_test def test_get_dir_hash_non_existent_dir(self): """Test thata an error is raised when the dir does not exist""" self.assertRaises(FileNotFoundError, get_dir_hash, "/non-existent-file") + @pytest.mark.unit_test def test_jsonfile_to_dict(self): """Test that a json file is parsed correctly""" "" json_file = os.path.join(os.path.dirname(__file__), "test_data", @@ -39,12 +43,14 @@ class TestFileUtils(unittest.TestCase): self.assertEqual("Longer description", jsonfile_to_dict(json_file).get("description")) + @pytest.mark.unit_test def test_jsonfile_to_dict_nonexistent_file(self): """Test that a ValueError is raised when the json file is non-existent""" self.assertRaises(FileNotFoundError, jsonfile_to_dict, "/non-existent-dir") + @pytest.mark.unit_test @mock.patch("gn3.fs_helpers.tarfile") @mock.patch("gn3.fs_helpers.secure_filename") def test_extract_uploaded_file(self, mock_file, mock_tarfile): @@ -60,11 +66,12 @@ non-existent""" "upload-data.tar.gz") mock_tarfile.open.assert_called_once_with("/tmp/abcdef-abcdef/" "upload-data.tar.gz") - mock_tarfile.open.return_value.extractall.assert_called_once_with( + mock_tarfile.open.return_value.__enter__.return_value.extractall.assert_called_once_with( path='/tmp/abcdef-abcdef') mock_file.assert_called_once_with("upload-data.tar.gz") self.assertEqual(result, {"status": 0, "token": "abcdef-abcdef"}) + @pytest.mark.unit_test @mock.patch("gn3.fs_helpers.secure_filename") def test_extract_uploaded_file_non_existent_gzip(self, mock_file): """Test that the right error message is returned when there is a problem @@ -78,13 +85,15 @@ extracting the file""" "error": "gzip failed to unpack file" }) + @pytest.mark.unit_test def test_cache_ipfs_file_cache_hit(self): """Test that the correct file location is returned if there's a cache hit""" # Create empty file test_dir = "/tmp/QmQPeNsJPyVWPFDVHb77w8G42Fvo15z4bG2X8D2GhfbSXc-test" if not os.path.exists(test_dir): os.mkdir(test_dir) - open(f"{test_dir}/genotype.txt", "a").close() + with open(f"{test_dir}/genotype.txt", "a", encoding="utf8"): + pass file_loc = cache_ipfs_file( ipfs_file=("/ipfs/" "QmQPeNsJPyVWPFDVHb" @@ -96,6 +105,7 @@ extracting the file""" os.rmdir(test_dir) self.assertEqual(file_loc, f"{test_dir}/genotype.txt") + @pytest.mark.unit_test @mock.patch("gn3.fs_helpers.ipfshttpclient") def test_cache_ipfs_file_cache_miss(self, mock_ipfs): |