fix: base str traversal

Author: JGSweets

dataprofiler/tests/labelers/test_character_level_cnn_model.py

@@ -16,7 +16,8 @@
 )
 
 _file_dir = os.path.dirname(os.path.abspath(__file__))
-_resource_labeler_dir = Path(importlib.resources.files("resources")) / "labelers"
+with importlib.resources.as_file(importlib.resources.files("resources")) as base:
+    _resource_labeler_dir = Path(base) / "labelers"
 
 
 mock_model_parameters = {

dataprofiler/tests/labelers/test_column_name_model.py

@@ -13,7 +13,8 @@
 from dataprofiler.labelers.column_name_model import ColumnNameModel
 
 _file_dir = os.path.dirname(os.path.abspath(__file__))
-_resource_labeler_dir = Path(importlib.resources.files("resources")) / "labelers"
+with importlib.resources.as_file(importlib.resources.files("resources")) as base:
+    _resource_labeler_dir = Path(base) / "labelers"
 
 
 mock_model_parameters = {

dataprofiler/tests/labelers/test_data_labelers.py

@@ -152,9 +152,10 @@ def test_load_from_disk(self, *mocks):
         import importlib
         from pathlib import Path
 
-        default_labeler_dir = (
-            Path(importlib.resources.files("resources")) / "labelers/structured_model"
-        )
+        with importlib.resources.as_file(
+            importlib.resources.files("resources")
+        ) as base:
+            default_labeler_dir = Path(base) / "labelers/structured_model"
 
         data_labeler = dp.DataLabeler.load_from_disk(default_labeler_dir)
         self.assertIsInstance(data_labeler, BaseDataLabeler)

dataprofiler/tests/labelers/test_data_processing.py

@@ -225,7 +225,10 @@ def test_load_from_library(self, mocked_load, *mocks):
         BaseDataProcessor.load_from_library("default")
 
         # assert called with proper load_processor dirpath
-        default_labeler_dir = Path(importlib.resources.files("resources")) / "labelers"
+        with importlib.resources.as_file(
+            importlib.resources.files("resources")
+        ) as base:
+            default_labeler_dir = Path(base) / "labelers"
 
         mocked_load.assert_called_with(os.path.join(default_labeler_dir, "default"))
 

dataprofiler/tests/labelers/test_integration_column_name_data_labeler.py

@@ -12,7 +12,8 @@
     DirectPassPreprocessor,
 )
 
-default_labeler_dir = Path(importlib.resources.files("resources")) / "labelers"
+with importlib.resources.as_file(importlib.resources.files("resources")) as base:
+    default_labeler_dir = Path(base) / "labelers"
 
 
 class TestColumnNameDataLabeler(unittest.TestCase):

dataprofiler/tests/labelers/test_integration_regex_data_labeler.py

@@ -7,7 +7,8 @@
 
 from dataprofiler.labelers.data_labelers import BaseDataLabeler
 
-default_labeler_dir = Path(importlib.resources.files("resources")) / "labelers"
+with importlib.resources.as_file(importlib.resources.files("resources")) as base:
+    default_labeler_dir = Path(base) / "labelers"
 
 
 class TestRegexDataLabeler(unittest.TestCase):

dataprofiler/tests/labelers/test_regex_model.py

@@ -11,7 +11,8 @@
 from dataprofiler.labelers.regex_model import RegexModel
 
 _file_dir = os.path.dirname(os.path.abspath(__file__))
-_resource_labeler_dir = Path(importlib.resources.files("resources")) / "labelers"
+with importlib.resources.as_file(importlib.resources.files("resources")) as base:
+    _resource_labeler_dir = Path(base) / "labelers"
 
 
 mock_model_parameters = {