Skip to content

Commit 91c539e

Browse files
committed
add backoff to add custom module method
1 parent 8f309a8 commit 91c539e

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

securitycenter/snippets_management_api/security_health_analytics_custom_modules_test.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -132,6 +132,9 @@ def extract_custom_module_id(module_name):
132132
return ""
133133

134134

135+
@backoff.on_exception(
136+
backoff.expo, (InternalServerError, ServiceUnavailable, NotFound), max_tries=3
137+
)
135138
def add_custom_module(org_id: str):
136139
"""
137140
Adds a new SHA custom module.

0 commit comments

Comments
 (0)