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, )