From f3b0df19379dfc19a73caaaf4a3f1281ed4c1650 Mon Sep 17 00:00:00 2001
From: Jonas Braathen <jonas@ponas.no>
Date: Wed, 20 May 2026 23:19:56 +0200
Subject: [PATCH] Add tests and migration for previous commit

---
 mdb/migrations/0013_host_allow_email.py | 18 +++++++++++++
 tests/test_zone_sync.py                 | 36 +++++++++++++++++++++++++
 2 files changed, 54 insertions(+)
 create mode 100644 mdb/migrations/0013_host_allow_email.py

diff --git a/mdb/migrations/0013_host_allow_email.py b/mdb/migrations/0013_host_allow_email.py
new file mode 100644
index 0000000..44ee6e2
--- /dev/null
+++ b/mdb/migrations/0013_host_allow_email.py
@@ -0,0 +1,18 @@
+# Generated by Django 6.0.3 on 2026-05-20 21:09
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('mdb', '0012_alter_domaintxtrecord_target'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='host',
+            name='allow_email',
+            field=models.BooleanField(default=False),
+        ),
+    ]
diff --git a/tests/test_zone_sync.py b/tests/test_zone_sync.py
index efe2c34..ef0a024 100644
--- a/tests/test_zone_sync.py
+++ b/tests/test_zone_sync.py
@@ -4,6 +4,7 @@ from unittest.mock import MagicMock
 import pytest
 from freezegun import freeze_time
 
+from mdb.models import Host, HostType, Interface, OperatingSystem
 from mdb.services.zone_sync import ZoneSyncService
 
 EXPECTED_ZONE_FILE_CONTENTS = """\
@@ -183,3 +184,38 @@ def test_sync_zone_records_success_and_diff(db_domain, tmp_path):
     assert change["errors"] is None
     assert change["old_serial"] == 0
     assert change["new_serial"] == 2024010101
+
+
+def add_host(domain, hostname, allow_email):
+    host_type = HostType.objects.create(host_type="server", description="")
+    operating_system = OperatingSystem.objects.create(
+        name="Debian", version="12", arch="x86_64"
+    )
+    host = Host.objects.create(
+        hostname=hostname,
+        host_type=host_type,
+        operating_system=operating_system,
+        allow_email=allow_email,
+    )
+    Interface.objects.create(
+        name="eth0",
+        macaddr="00:11:22:33:44:55",
+        host=host,
+        domain=domain,
+    )
+    # Creating the interface bumps the domain serial via a signal; reload so the
+    # in-memory domain stays consistent with the DB.
+    domain.refresh_from_db()
+    return host
+
+
+def test_zone_file_contents_includes_spf_record_when_email_allowed(db_domain):
+    add_host(db_domain, "mailhost", allow_email=True)
+    content = db_domain.zone_file_contents()
+    assert 'mailhost\tTXT\t"v=spf1 mx -all"\n' in content
+
+
+def test_zone_file_contents_omits_spf_record_when_email_not_allowed(db_domain):
+    add_host(db_domain, "plainhost", allow_email=False)
+    content = db_domain.zone_file_contents()
+    assert "v=spf1 mx -all" not in content