add simple unit test

logicalclocks · Dec 19, 2024 · 8f031dc · 8f031dc
1 parent 2fcd32d
commit 8f031dc
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/python/tests/test_storage_connector.py b/python/tests/test_storage_connector.py
@@ -113,7 +113,19 @@ def test_default_path(self, mocker):
         )
         sc.read(data_format="csv")
         # assert
-        assert "s3://test-bucket" in mock_engine_read.call_args[0][3]
+        assert "" == mock_engine_read.call_args[0][3]
+
+    def test_get_path(self, mocker):
+        mocker.patch("hsfs.engine.get_instance", return_value=spark.Engine())
+        sc = storage_connector.S3Connector(
+            id=1, name="test_connector", featurestore_id=1, bucket="test-bucket", path="abc/def"
+        )
+
+        # act
+        result = sc._get_path("some/location")
+
+        # assert
+        assert "s3://test-bucket/abc/def/some/location" == result
 
 
 class TestRedshiftConnector: