Skip to content

Commit 2762610

Browse files
committed
fix: mock update polling and network manager post method for unit testcases
1 parent e3ba9da commit 2762610

File tree

1 file changed

+14
-4
lines changed

1 file changed

+14
-4
lines changed

tests/e2e/get_flag_test.py

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,13 @@ def _helper_function_to_run_test(
5656
with patch(
5757
"vwo.vwo_builder.VWOBuilder.get_settings",
5858
return_value=settings_files.get(data.get("settings")),
59-
) as mock_get_settings:
59+
) as mock_get_settings, patch(
60+
"vwo.vwo_builder.VWOBuilder.update_poll_interval_and_check_and_poll",
61+
return_value=None, # Mocking the method to return nothing
62+
), patch(
63+
"vwo.packages.network_layer.manager.network_manager.NetworkManager.post",
64+
return_value=None, # Mocking the post method to return nothing
65+
):
6066
storage = user_storage()
6167
options = {"sdk_key": self.sdk_key, "account_id": self.account_id}
6268

@@ -132,13 +138,17 @@ def _helper_function_to_run_salt_test(self, salt_test_data: List[Dict[str, Any]]
132138
for test_data in salt_test_data:
133139
# Debug print for the test data
134140
storage.clear()
135-
print("DEBUG: Test Data:")
136-
print(test_data)
137141

138142
with patch(
139143
"vwo.vwo_builder.VWOBuilder.get_settings",
140144
return_value=settings_files.get(test_data.get("settings")),
141-
) as mock_get_settings:
145+
) as mock_get_settings, patch(
146+
"vwo.vwo_builder.VWOBuilder.update_poll_interval_and_check_and_poll",
147+
return_value=None, # Mocking the method to return nothing
148+
), patch(
149+
"vwo.packages.network_layer.manager.network_manager.NetworkManager.post",
150+
return_value=None, # Mocking the post method to return nothing
151+
):
142152
# Debug print for the settings file being used
143153
settings_file = settings_files.get(test_data.get("settings"))
144154
options = {

0 commit comments

Comments
 (0)