add mock get_private_key for unit-test

Signed-off-by: HH <[email protected]>
flyteorg · Sep 10, 2023 · 70e6b22 · 70e6b22
1 parent c16591a
commit 70e6b22
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/flytekit/unit/types/structured_dataset/test_snowflake.py b/tests/flytekit/unit/types/structured_dataset/test_snowflake.py
@@ -33,9 +33,10 @@ def wf() -> pd.DataFrame:
     return t2(sd=sd)
 
 
+@mock.patch("flytekit.types.structured.snowflake.get_private_key", return_value="pb")
 @mock.patch("snowflake.connector.connect")
 @pytest.mark.asyncio
-async def test_sf_wf(mock_connect):
+async def test_sf_wf(mock_connect, mock_get_private_key):
     class mock_dataframe:
         def to_dataframe(self):
             return pd.DataFrame({"Name": ["Tom", "Joseph"], "Age": [20, 22]})