sts-sponsors team mailing list archive

[Christian Grabowski <mp+436300@xxxxxxxxxxxxxxxxxx>]

Christian Grabowski has proposed merging ~cgrabowski/maas:fix_missing_reverse_records into maas:master.

Commit message:
use zone_info.zone_name instead of domain to freeze/thaw reverse updates

fix default ttls



Requested reviews:
  MAAS Maintainers (maas-maintainers)

For more details, see:
https://code.launchpad.net/~cgrabowski/maas/+git/maas/+merge/436300
-- 
Your team MAAS Committers is subscribed to branch maas:master.

diff --git a/src/maasserver/triggers/system.py b/src/maasserver/triggers/system.py
index bf4c51f..c6313f9 100644
--- a/src/maasserver/triggers/system.py
+++ b/src/maasserver/triggers/system.py
@@ -2143,7 +2143,7 @@ def render_dns_dynamic_update_node(op):
           domain text;
           address_ttl int;
         BEGIN
-          IF ((TG_OP = 'INSERT' OR  TG_OP = 'UPDATE') AND TG_LEVEL = 'ROW') THEN
+          IF ((TG_OP = 'INSERT' OR TG_OP = 'UPDATE') AND TG_LEVEL = 'ROW') THEN
               IF NEW.node_type <> {NODE_TYPE.DEVICE} AND NEW.node_type <> {NODE_TYPE.MACHINE} THEN
                   PERFORM pg_notify('sys_dns_updates', 'RELOAD');
               END IF;
diff --git a/src/provisioningserver/dns/config.py b/src/provisioningserver/dns/config.py
index 729b570..6e11f73 100644
--- a/src/provisioningserver/dns/config.py
+++ b/src/provisioningserver/dns/config.py
@@ -62,6 +62,8 @@ class DynamicDNSUpdate:
         else:
             if ip.version == 6:
                 rectype = "AAAA"
+        if kwargs.get("ttl") == 0:  # default ttl
+            kwargs["ttl"] = 30
         return cls(answer=answer, rectype=rectype, **kwargs)
 
     @classmethod
diff --git a/src/provisioningserver/dns/tests/test_zoneconfig.py b/src/provisioningserver/dns/tests/test_zoneconfig.py
index 10ee0f7..1db577e 100644
--- a/src/provisioningserver/dns/tests/test_zoneconfig.py
+++ b/src/provisioningserver/dns/tests/test_zoneconfig.py
@@ -8,6 +8,7 @@ from itertools import chain
 import os.path
 import random
 from tempfile import mktemp
+from unittest.mock import call
 
 from netaddr import IPAddress, IPNetwork, IPRange
 from testtools.matchers import (
@@ -24,6 +25,7 @@ from maastesting.factory import factory
 from maastesting.matchers import MockNotCalled
 from maastesting.testcase import MAASTestCase
 from provisioningserver.dns import actions
+import provisioningserver.dns.actions
 from provisioningserver.dns.config import (
     DynamicDNSUpdate,
     get_nsupdate_key_path,
@@ -488,6 +490,44 @@ class TestDNSForwardZoneConfig(MAASTestCase):
             stdin=expected_stdin.encode("ascii"),
         )
 
+    def test_full_reload_calls_freeze_thaw(self):
+        patch_zone_file_config_path(self)
+        execute_rndc_command = self.patch(
+            provisioningserver.dns.actions, "execute_rndc_command"
+        )
+        domain = factory.make_string()
+        network = factory.make_ipv4_network()
+        ipv4_hostname = factory.make_name("host")
+        ipv4_ip = factory.pick_ip_in_network(network)
+        ipv6_hostname = factory.make_name("host")
+        ipv6_ip = factory.make_ipv6_address()
+        ipv6_network = factory.make_ipv6_network()
+        dynamic_range = IPRange(ipv6_network.first, ipv6_network.last)
+        ttl = random.randint(10, 300)
+        mapping = {
+            ipv4_hostname: HostnameIPMapping(None, ttl, {ipv4_ip}),
+            ipv6_hostname: HostnameIPMapping(None, ttl, {ipv6_ip}),
+        }
+        dns_zone_config = DNSForwardZoneConfig(
+            domain,
+            serial=random.randint(1, 100),
+            mapping=mapping,
+            default_ttl=ttl,
+            dynamic_ranges=[dynamic_range],
+        )
+        self.patch(dns_zone_config, "get_GENERATE_directives")
+        run_command = self.patch(actions, "run_command")
+        dns_zone_config.write_config()
+        dns_zone_config.force_config_write = True
+        dns_zone_config.write_config()
+        self.assertCountEqual(
+            execute_rndc_command.call_args_list,
+            [
+                call(("freeze", domain), timeout=2),
+                call(("thaw", domain), timeout=2),
+            ],
+        )
+
 
 class TestDNSReverseZoneConfig(MAASTestCase):
     """Tests for DNSReverseZoneConfig."""
@@ -1077,6 +1117,34 @@ class TestDNSReverseZoneConfig(MAASTestCase):
             stdin=expected_stdin.encode("ascii"),
         )
 
+    def test_full_reload_calls_freeze_thaw(self):
+        patch_zone_file_config_path(self)
+        execute_rndc_command = self.patch(
+            provisioningserver.dns.actions, "execute_rndc_command"
+        )
+        domain = factory.make_string()
+        network = IPNetwork("10.0.0.0/24")
+        ip1 = factory.pick_ip_in_network(network)
+        ip2 = factory.pick_ip_in_network(network)
+        hostname1 = f"{factory.make_string()}.{domain}"
+        hostname2 = f"{factory.make_string()}.{domain}"
+        zone = DNSReverseZoneConfig(
+            domain,
+            serial=random.randint(1, 100),
+            network=network,
+        )
+        run_command = self.patch(actions, "run_command")
+        zone.write_config()
+        zone.force_config_write = True
+        zone.write_config()
+        self.assertCountEqual(
+            execute_rndc_command.call_args_list,
+            [
+                call(("freeze", "0.0.10.in-addr.arpa"), timeout=2),
+                call(("thaw", "0.0.10.in-addr.arpa"), timeout=2),
+            ],
+        )
+
 
 class TestDNSReverseZoneConfig_GetGenerateDirectives(MAASTestCase):
     """Tests for `DNSReverseZoneConfig.get_GENERATE_directives()`."""
diff --git a/src/provisioningserver/dns/zoneconfig.py b/src/provisioningserver/dns/zoneconfig.py
index 3171bae..bc02651 100644
--- a/src/provisioningserver/dns/zoneconfig.py
+++ b/src/provisioningserver/dns/zoneconfig.py
@@ -650,26 +650,33 @@ class DNSReverseZoneConfig(DomainConfigBase):
             else:
                 Path(f"{zi.target_path}.jnl").unlink(missing_ok=True)
                 self.requires_reload = True
-                self.write_zone_file(
-                    zi.target_path,
-                    self.make_parameters(),
-                    {
-                        "mappings": {
-                            "PTR": self.get_PTR_mapping(
-                                self._mapping, zi.subnetwork
-                            )
-                        },
-                        "other_mapping": [],
-                        "generate_directives": {
-                            "PTR": generate_directives,
-                            "CNAME": self.get_rfc2317_GENERATE_directives(
-                                zi.subnetwork,
-                                self._rfc2317_ranges,
-                                self.domain,
-                            ),
+                needs_freeze_thaw = self.zone_file_exists(zi)
+                if needs_freeze_thaw:
+                    bind_freeze_zone(zone=zi.zone_name)
+                try:
+                    self.write_zone_file(
+                        zi.target_path,
+                        self.make_parameters(),
+                        {
+                            "mappings": {
+                                "PTR": self.get_PTR_mapping(
+                                    self._mapping, zi.subnetwork
+                                )
+                            },
+                            "other_mapping": [],
+                            "generate_directives": {
+                                "PTR": generate_directives,
+                                "CNAME": self.get_rfc2317_GENERATE_directives(
+                                    zi.subnetwork,
+                                    self._rfc2317_ranges,
+                                    self.domain,
+                                ),
+                            },
                         },
-                    },
-                )
+                    )
+                finally:
+                    if needs_freeze_thaw:
+                        bind_thaw_zone(zone=zi.zone_name)
                 PROMETHEUS_METRICS.update(
                     "maas_dns_full_zonefile_write_count",
                     "inc",