mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-05-28 16:29:54 +02:00
Test fix for security scanner
This commit is contained in:
parent
dfec8c200e
commit
fa6ed6f318
3 changed files with 7 additions and 1 deletions
|
@ -45,6 +45,9 @@ except NameError:
|
|||
# Attn: these imports should NOT be at the top of the file
|
||||
try:
|
||||
from .client import CLIENT, commands
|
||||
except ImportError:
|
||||
pass
|
||||
try:
|
||||
from .errors import RegistryError, ErrorCode
|
||||
from epplib.models import common, info
|
||||
from epplib.responses import extensions
|
||||
|
|
|
@ -74,6 +74,8 @@ class EPPLibWrapper:
|
|||
}
|
||||
|
||||
self._pool = None
|
||||
# Since we reuse the same creds for each pool, we can test on
|
||||
# one socket, and if successful, then we know we can connect.
|
||||
if not settings.DEBUG or self._test_registry_connection_success():
|
||||
self._pool = EppConnectionPool(
|
||||
client=self._client, login=self._login, options=options
|
||||
|
|
|
@ -42,7 +42,8 @@ class Socket:
|
|||
return code >= 2000
|
||||
|
||||
def test_connection_success(self):
|
||||
"""Tests if a successful connection can be made with the registry"""
|
||||
"""Tests if a successful connection can be made with the registry.
|
||||
Tries 3 times"""
|
||||
# Something went wrong if this doesn't exist
|
||||
if not hasattr(self.client, "connect"):
|
||||
return False
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue