Merge lp:~julian-edwards/maas/backport-r1303 into lp:maas/1.2

[Julian Edwards]

Add a command to write the full DNS config.  This will be called by the postinst script of maas-dns. (backport r1303)

[Julian Edwards (julian-edwards)]

review: Approve

=== modified file 'src/maasserver/management/commands/set_up_dns.py'
--- src/maasserver/management/commands/set_up_dns.py	2012-08-13 11:04:44 +0000
+++ src/maasserver/management/commands/set_up_dns.py	2012-12-03 07:43:19 +0000
@@ -6,7 +6,7 @@
 This creates a basic, blank DNS configuration which will allow MAAS to
 reload its configuration once zone files will be written.
 
-The main purpose of this command is for it to be run when MAAS is
+The main purpose of this command is for it to be run when 'maas-dns' is
 installed.
 """
 
 
=== added file 'src/maasserver/management/commands/write_dns_config.py'
--- src/maasserver/management/commands/write_dns_config.py	1970-01-01 00:00:00 +0000
+++ src/maasserver/management/commands/write_dns_config.py	2012-12-03 07:43:19 +0000
@@ -0,0 +1,34 @@
+# Copyright 2012 Canonical Ltd.  This software is licensed under the
+# GNU Affero General Public License version 3 (see the file LICENSE).
+
+"""Django command: write the MAAS named zone files.
+
+If any of the cluster controllers connected to this MAAS region controller
+is configured to manage DNS, write the DNS configuration.
+
+The main purpose of this command is for it to be run when 'maas-dns' is
+installed.
+"""
+
+from __future__ import (
+    absolute_import,
+    print_function,
+    unicode_literals,
+    )
+
+__metaclass__ = type
+__all__ = [
+    'Command',
+    ]
+
+from django.core.management.base import BaseCommand
+from maasserver.dns import write_full_dns_config
+
+
+class Command(BaseCommand):
+    help = (
+        "Write the DNS configuration files and reload the DNS server if "
+        "this region has cluster controllers configured to manage DNS.")
+
+    def handle(self, *args, **options):
+        write_full_dns_config()
 
=== added file 'src/maasserver/tests/test_commands_write_dns_config.py'
--- src/maasserver/tests/test_commands_write_dns_config.py	1970-01-01 00:00:00 +0000
+++ src/maasserver/tests/test_commands_write_dns_config.py	2012-12-03 07:43:19 +0000
@@ -0,0 +1,52 @@
+# Copyright 2012 Canonical Ltd.  This software is licensed under the
+# GNU Affero General Public License version 3 (see the file LICENSE).
+
+"""Tests for the write_dns_config command."""
+
+from __future__ import (
+    absolute_import,
+    print_function,
+    unicode_literals,
+    )
+
+__metaclass__ = type
+__all__ = []
+
+
+import os
+
+from celery.conf import conf
+from django.core.management import call_command
+from maasserver.testing.testcase import TestCase
+from maasserver.testing.factory import factory
+from django.conf import settings
+from netaddr import (
+    IPNetwork,
+    )
+from maasserver.enum import (
+    NODEGROUP_STATUS,
+    NODEGROUPINTERFACE_MANAGEMENT,
+    )
+from provisioningserver import tasks
+from testtools.matchers import (
+    FileExists,
+    )
+
+
+class TestWriteDNSConfigCommand(TestCase):
+
+    def test_write_dns_config_writes_zone_file(self):
+        dns_conf_dir = self.make_dir()
+        self.patch(conf, 'DNS_CONFIG_DIR', dns_conf_dir)
+        self.patch(settings, 'DNS_CONNECT', True)
+        # Prevent rndc task dispatch.
+        self.patch(tasks, "rndc_command")
+        domain = factory.getRandomString()
+        factory.make_node_group(
+            name=domain,
+            network=IPNetwork('192.168.0.1/24'),
+            status=NODEGROUP_STATUS.ACCEPTED,
+            management=NODEGROUPINTERFACE_MANAGEMENT.DHCP_AND_DNS)
+        call_command('write_dns_config')
+        zone_file = os.path.join(dns_conf_dir, 'zone.%s' % domain)
+        self.assertThat(zone_file, FileExists())