def get_dns_class(self, data):
if data['dnsapi'] == 'dns_ali':
import panelDnsapi
public.mod_reload(panelDnsapi)
dns_class = panelDnsapi.AliyunDns(key=data['dns_param'][0],
secret=data['dns_param'][1])
return dns_class
elif data['dnsapi'] == 'dns_dp':
dns_class = sewer.DNSPodDns(DNSPOD_ID=data['dns_param'][0],
DNSPOD_API_KEY=data['dns_param'][1])
return dns_class
elif data['dnsapi'] == 'dns_cx':
import panelDnsapi
public.mod_reload(panelDnsapi)
dns_class = panelDnsapi.CloudxnsDns(key=data['dns_param'][0],
secret=data['dns_param'][1])
result = dns_class.get_domain_list()
if result['code'] == 1:
return dns_class
elif data['dnsapi'] == 'dns_bt':
import panelDnsapi
public.mod_reload(panelDnsapi)
dns_class = panelDnsapi.Dns_com()
return dns_class
return False
def setUp(self):
self.domain_name = "example.com"
self.domain_dns_value = "mock-domain_dns_value"
self.DNSPOD_ID = "0123456"
self.DNSPOD_API_KEY = "mock-api-key"
self.DNSPOD_API_BASE_URL = "https://some-mock-url.com"
self.test_datas = [
{
"domain_name": "example.com",
"domain_dns_value": "mock-domain_dns_value",
"expected_domain_name": "example.com",
"expected_sub_domain_name": "_acme-challenge",
},
{
"domain_name": "sub1.example.com",
"domain_dns_value": "mock-domain_dns_value",
"expected_domain_name": "example.com",
"expected_sub_domain_name": "_acme-challenge.sub1",
},
{
"domain_name": "sub1.sub2.example.com",
"domain_dns_value": "mock-domain_dns_value",
"expected_domain_name": "example.com",
"expected_sub_domain_name": "_acme-challenge.sub1.sub2",
},
]
with mock.patch("requests.post") as mock_requests_post, mock.patch(
"requests.get"
) as mock_requests_get:
mock_requests_post.return_value = test_utils.MockResponse()
mock_requests_get.return_value = test_utils.MockResponse()
self.dns_class = sewer.DNSPodDns(
DNSPOD_ID=self.DNSPOD_ID,
DNSPOD_API_KEY=self.DNSPOD_API_KEY,
DNSPOD_API_BASE_URL=self.DNSPOD_API_BASE_URL,
)
