2424import socket
2525
2626class TestCloudStackDNSFramework (cloudstackTestCase ):
27-
27+
2828 @classmethod
2929 def setUpClass (cls ):
3030 """
@@ -80,12 +80,12 @@ def setUpClass(cls):
8080 time .sleep (15 )
8181
8282 cls .logger .info ("PDNS is up and running" )
83-
83+
8484 # Construct PDNS URL once
8585 cls .pdns_url = f"http://{ cls .marvin_vm_ip } "
8686 cls .logger .info (f"PDNS endpoint: { cls .pdns_url } " )
87-
88-
87+
88+
8989 def test_01_list_dns_providers (self ):
9090 """
9191 List DNS providers, expect PowerDNS provider to be present
@@ -95,7 +95,7 @@ def test_01_list_dns_providers(self):
9595 response = self .api_client .listDnsProviders (list_providers_cmd )
9696 self .assertIsNotNone (response , "Failed to list DNS providers" )
9797 self .logger .info (f"DNS Providers found: { [provider .name for provider in response ]} " )
98-
98+
9999 def test_02_add_dns_server (self ):
100100 """
101101 Register PDNS as DNS provider in CloudStack
@@ -107,7 +107,7 @@ def test_02_add_dns_server(self):
107107 self .__class__ .dns_server_id = response .id
108108 self .logger .info (f"DNS Provider added: { response .id } " )
109109 self .assertIsNotNone (response .id , "DNS server ID should not be None" )
110-
110+
111111
112112 def test_03_list_dns_servers (self ):
113113 """
@@ -121,7 +121,7 @@ def test_03_list_dns_servers(self):
121121 self .assertEqual (len (response ), 1 , "Expected exactly one DNS server" )
122122 self .assertEqual (response [0 ].id , self .dns_server_id , "DNS server ID mismatch" )
123123
124-
124+
125125 def test_04_create_dns_zone (self ):
126126 """
127127 Create a DNS zone in the added PDNS provider
@@ -133,7 +133,7 @@ def test_04_create_dns_zone(self):
133133 self .__class__ .dns_zone_id = response .id
134134 self .logger .info (f"DNS Zone created: { response .id } " )
135135
136-
136+
137137 def test_05_list_dns_zones (self ):
138138 """
139139 List DNS zones and verify the newly created zone is present
@@ -160,7 +160,7 @@ def test_06_create_a_dns_record(self):
160160 )
161161 self .assertIsNotNone (response , "Failed to create DNS record" )
162162 self .assertEqual (response .name , "www.example.com" , "DNS record name mismatch" )
163-
163+
164164 def test_07_create_aaaa_dns_records (self ):
165165 """
166166 Create AAAA DNS records in the previously created zone
@@ -180,7 +180,7 @@ def test_08_create_mx_dns_record(self):
180180 """
181181 Create an MX DNS record in the previously created zone
182182 """
183- self .logger .info ("Creating an MX DNS record" )
183+ self .logger .info ("Creating an MX DNS record" )
184184 response = self ._create_record (
185185 self .dns_zone_id ,
186186 "example.com" ,
@@ -263,13 +263,13 @@ def test_12_delete_dns_server(self):
263263 response = self .api_client .listDnsServers (list_cmd )
264264 dns_servers = response or []
265265 self .assertEqual (len (dns_servers ), 0 , "Expected no DNS servers after deletion" )
266-
266+
267267 @classmethod
268268 def tearDownClass (cls ):
269269 """
270270 Stop PDNS after tests
271271 """
272-
272+
273273 try :
274274 cls .logger .info ("Stopping PDNS stack..." )
275275
@@ -293,7 +293,7 @@ def _create_record(self, zone_id, name, rtype, contents):
293293 cmd .contents = contents
294294
295295 return self .api_client .createDnsRecord (cmd )
296-
296+
297297
298298 def _add_dns_server (self ):
299299 cmd = addDnsServer .addDnsServerCmd ()
@@ -308,11 +308,12 @@ def _add_dns_server(self):
308308 cmd .publicdomainsuffix = "pdns-public.example.com"
309309
310310 return self .api_client .addDnsServer (cmd )
311-
311+
312312 def _create_zone (self , server_id ):
313313 cmd = createDnsZone .createDnsZoneCmd ()
314314 cmd .dnsserverid = server_id
315315 cmd .name = "example.com"
316316 cmd .description = "Test DNS Zone for PDNS"
317317
318- return self .api_client .createDnsZone (cmd )
318+ return self .api_client .createDnsZone (cmd )
319+
0 commit comments