Move NPM configuration and client behind plugin boundary
This commit is contained in:
+74
-27
@@ -5,48 +5,49 @@ from zipfile import ZipFile
|
||||
from unittest.mock import Mock, patch
|
||||
|
||||
from dns_sync.adapters import Endpoint
|
||||
from dns_sync.config import AppConfig
|
||||
from dns_sync.certificates.material import CertificateMaterial
|
||||
from dns_sync.npm_client import NpmClient
|
||||
from dns_sync.plugins.adapters.npm.client import NpmClient
|
||||
|
||||
|
||||
class NpmClientEndpointTests(unittest.TestCase):
|
||||
def config(self):
|
||||
return AppConfig.from_env()
|
||||
|
||||
def test_default_client_uses_first_configured_endpoint(self):
|
||||
def test_client_uses_explicit_endpoint(self):
|
||||
with patch.dict(
|
||||
os.environ,
|
||||
{
|
||||
"NPM_ENDPOINTS": "192.168.1.3,192.168.1.4:8081",
|
||||
"DNS_TARGET_IP": "192.168.1.101",
|
||||
"NPM_IDENTITY": "test@example.com",
|
||||
"NPM_SECRET": "not-a-real-secret",
|
||||
},
|
||||
clear=True,
|
||||
):
|
||||
config = self.config()
|
||||
endpoint = Endpoint(
|
||||
"npm",
|
||||
"192.168.1.3:81",
|
||||
)
|
||||
with patch(
|
||||
"dns_sync.npm_client.requests.Session.request"
|
||||
"dns_sync.plugins.adapters.npm.client.requests.Session.request"
|
||||
) as request:
|
||||
client = NpmClient(config)
|
||||
client = NpmClient(endpoint)
|
||||
|
||||
request.assert_not_called()
|
||||
self.assertEqual(client.api_url, "http://192.168.1.3:81/api")
|
||||
self.assertEqual(
|
||||
client.api_url,
|
||||
"http://192.168.1.3:81/api",
|
||||
)
|
||||
|
||||
def test_client_can_select_specific_configured_endpoint(self):
|
||||
def test_client_can_select_specific_endpoint(self):
|
||||
with patch.dict(
|
||||
os.environ,
|
||||
{
|
||||
"NPM_ENDPOINTS": "192.168.1.3,192.168.1.4:8081",
|
||||
"DNS_TARGET_IP": "192.168.1.101",
|
||||
"NPM_IDENTITY": "test@example.com",
|
||||
"NPM_SECRET": "not-a-real-secret",
|
||||
},
|
||||
clear=True,
|
||||
):
|
||||
config = self.config()
|
||||
client = NpmClient(config, config.npm_endpoints[1])
|
||||
endpoint = Endpoint(
|
||||
"npm",
|
||||
"192.168.1.4:8081",
|
||||
)
|
||||
client = NpmClient(endpoint)
|
||||
|
||||
self.assertEqual(
|
||||
client.api_url,
|
||||
@@ -61,19 +62,66 @@ class NpmClientEndpointTests(unittest.TestCase):
|
||||
with patch.dict(
|
||||
os.environ,
|
||||
{
|
||||
"PROXY_IP": "192.168.1.101",
|
||||
"NPM_IDENTITY": "test@example.com",
|
||||
"NPM_SECRET": "not-a-real-secret",
|
||||
},
|
||||
clear=True,
|
||||
):
|
||||
config = self.config()
|
||||
endpoint = Endpoint("verizon", "192.168.1.1")
|
||||
endpoint = Endpoint(
|
||||
"verizon",
|
||||
"192.168.1.1:443",
|
||||
)
|
||||
|
||||
with self.assertRaisesRegex(RuntimeError, "npm endpoint"):
|
||||
NpmClient(config, endpoint)
|
||||
with self.assertRaisesRegex(
|
||||
RuntimeError,
|
||||
"npm endpoint",
|
||||
):
|
||||
NpmClient(endpoint)
|
||||
|
||||
def test_client_rejects_invalid_api_scheme(self):
|
||||
with patch.dict(
|
||||
os.environ,
|
||||
{
|
||||
"NPM_IDENTITY": "test@example.com",
|
||||
"NPM_SECRET": "not-a-real-secret",
|
||||
"PROXY_API_SCHEME": "ftp",
|
||||
},
|
||||
clear=True,
|
||||
):
|
||||
with self.assertRaisesRegex(
|
||||
RuntimeError,
|
||||
"PROXY_API_SCHEME",
|
||||
):
|
||||
NpmClient(
|
||||
Endpoint(
|
||||
"npm",
|
||||
"192.168.1.3:81",
|
||||
)
|
||||
)
|
||||
|
||||
def test_client_owns_tls_verification_setting(self):
|
||||
with patch.dict(
|
||||
os.environ,
|
||||
{
|
||||
"NPM_IDENTITY": "test@example.com",
|
||||
"NPM_SECRET": "not-a-real-secret",
|
||||
"PROXY_API_SCHEME": "https",
|
||||
"PROXY_VERIFY_TLS": "false",
|
||||
},
|
||||
clear=True,
|
||||
):
|
||||
client = NpmClient(
|
||||
Endpoint(
|
||||
"npm",
|
||||
"192.168.1.3:81",
|
||||
)
|
||||
)
|
||||
|
||||
self.assertEqual(
|
||||
client.api_url,
|
||||
"https://192.168.1.3:81/api",
|
||||
)
|
||||
self.assertFalse(client.verify_tls)
|
||||
|
||||
|
||||
class NpmClientMutationTests(unittest.TestCase):
|
||||
@@ -81,17 +129,16 @@ class NpmClientMutationTests(unittest.TestCase):
|
||||
with patch.dict(
|
||||
os.environ,
|
||||
{
|
||||
"NPM_ENDPOINTS": "192.168.1.3",
|
||||
"DNS_TARGET_IP": "192.168.1.101",
|
||||
"NPM_IDENTITY": "test@example.com",
|
||||
"NPM_SECRET": "not-a-real-secret",
|
||||
},
|
||||
clear=True,
|
||||
):
|
||||
config = AppConfig.from_env()
|
||||
return NpmClient(
|
||||
config,
|
||||
config.npm_endpoints[0],
|
||||
Endpoint(
|
||||
"npm",
|
||||
"192.168.1.3:81",
|
||||
)
|
||||
)
|
||||
|
||||
@staticmethod
|
||||
|
||||
Reference in New Issue
Block a user