def test_building_new_retry():
OCSP_SERVER = OCSPServer()
OCSPCache.ACTIVATE_SSD = False
OCSP_SERVER.OCSP_RETRY_URL = None
hname = "a1.us-east-1.snowflakecomputing.com"
os.environ["SF_OCSP_ACTIVATE_NEW_ENDPOINT"] = "true"
OCSP_SERVER.reset_ocsp_endpoint(hname)
assert (OCSP_SERVER.CACHE_SERVER_URL ==
"https://ocspssd.us-east-1.snowflakecomputing.com/ocsp/fetch")
assert (OCSP_SERVER.OCSP_RETRY_URL ==
"https://ocspssd.us-east-1.snowflakecomputing.com/ocsp/retry")
hname = "a1-12345.global.snowflakecomputing.com"
OCSP_SERVER.reset_ocsp_endpoint(hname)
assert (OCSP_SERVER.CACHE_SERVER_URL ==
"https://ocspssd-12345.global.snowflakecomputing.com/ocsp/fetch")
assert (OCSP_SERVER.OCSP_RETRY_URL ==
"https://ocspssd-12345.global.snowflakecomputing.com/ocsp/retry")
hname = "snowflake.okta.com"
OCSP_SERVER.reset_ocsp_endpoint(hname)
assert (OCSP_SERVER.CACHE_SERVER_URL ==
"https://ocspssd.snowflakecomputing.com/ocsp/fetch")
assert (OCSP_SERVER.OCSP_RETRY_URL ==
"https://ocspssd.snowflakecomputing.com/ocsp/retry")
del os.environ["SF_OCSP_ACTIVATE_NEW_ENDPOINT"]
def test_building_retry_url():
# privatelink retry url
OCSP_SERVER = OCSPServer()
OCSPCache.ACTIVATE_SSD = False
OCSP_SERVER.OCSP_RETRY_URL = None
OCSP_SERVER.CACHE_SERVER_URL = (
"http://ocsp.us-east-1.snowflakecomputing.com/ocsp_response_cache.json"
)
OCSP_SERVER.reset_ocsp_dynamic_cache_server_url(None)
assert (OCSP_SERVER.OCSP_RETRY_URL ==
"http://ocsp.us-east-1.snowflakecomputing.com/retry/{0}/{1}")
# privatelink retry url with port
OCSPCache.ACTIVATE_SSD = False
OCSP_SERVER.OCSP_RETRY_URL = None
OCSP_SERVER.CACHE_SERVER_URL = (
"http://ocsp.us-east-1.snowflakecomputing.com:80/ocsp_response_cache"
".json")
OCSP_SERVER.reset_ocsp_dynamic_cache_server_url(None)
assert (OCSP_SERVER.OCSP_RETRY_URL ==
"http://ocsp.us-east-1.snowflakecomputing.com:80/retry/{0}/{1}")
# non-privatelink retry url
OCSPCache.ACTIVATE_SSD = False
OCSP_SERVER.OCSP_RETRY_URL = None
OCSP_SERVER.CACHE_SERVER_URL = (
"http://ocsp.snowflakecomputing.com/ocsp_response_cache.json")
OCSP_SERVER.reset_ocsp_dynamic_cache_server_url(None)
assert OCSP_SERVER.OCSP_RETRY_URL is None
# non-privatelink retry url with port
OCSPCache.ACTIVATE_SSD = False
OCSP_SERVER.OCSP_RETRY_URL = None
OCSP_SERVER.CACHE_SERVER_URL = (
"http://ocsp.snowflakecomputing.com:80/ocsp_response_cache.json")
OCSP_SERVER.reset_ocsp_dynamic_cache_server_url(None)
assert OCSP_SERVER.OCSP_RETRY_URL is None
# ssd enabled for privatelink retry url
OCSPCache.ACTIVATE_SSD = True
OCSP_SERVER.OCSP_RETRY_URL = None
OCSP_SERVER.CACHE_SERVER_URL = (
"http://ocsp.us-east-1.snowflakecomputing.com/ocsp_response_cache.json"
)
OCSP_SERVER.reset_ocsp_dynamic_cache_server_url(None)
assert (OCSP_SERVER.OCSP_RETRY_URL ==
"http://ocsp.us-east-1.snowflakecomputing.com/retry")
# ssd enabled for privatelink retry url with port
OCSPCache.ACTIVATE_SSD = True
OCSP_SERVER.OCSP_RETRY_URL = None
OCSP_SERVER.CACHE_SERVER_URL = (
"http://ocsp.us-east-1.snowflakecomputing.com:80/ocsp_response_cache"
".json")
OCSP_SERVER.reset_ocsp_dynamic_cache_server_url(None)
assert (OCSP_SERVER.OCSP_RETRY_URL ==
"http://ocsp.us-east-1.snowflakecomputing.com:80/retry")
# ssd enabled for non-privatelink
OCSPCache.ACTIVATE_SSD = True
OCSP_SERVER.OCSP_RETRY_URL = None
OCSP_SERVER.CACHE_SERVER_URL = (
"http://ocsp.snowflakecomputing.com/ocsp_response_cache.json")
OCSP_SERVER.reset_ocsp_dynamic_cache_server_url(None)
assert OCSP_SERVER.OCSP_RETRY_URL is None
# ssd enabled for non-privatelink with port
OCSPCache.ACTIVATE_SSD = True
OCSP_SERVER.OCSP_RETRY_URL = None
OCSP_SERVER.CACHE_SERVER_URL = (
"http://ocsp.snowflakecomputing.com:80/ocsp_response_cache.json")
OCSP_SERVER.reset_ocsp_dynamic_cache_server_url(None)
assert OCSP_SERVER.OCSP_RETRY_URL is None