From 6b1de5158b7aa7b8b7e866c0731e82d40cc786a6 Mon Sep 17 00:00:00 2001
From: Viktor Liu <viktor@netbird.io>
Date: Sun, 19 Jan 2025 21:27:02 +0100
Subject: [PATCH] Fix engine test

---
 client/internal/engine.go      | 4 +++-
 client/internal/engine_test.go | 6 ++++++
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/client/internal/engine.go b/client/internal/engine.go
index dceb08ca583..a936a844c16 100644
--- a/client/internal/engine.go
+++ b/client/internal/engine.go
@@ -1082,7 +1082,9 @@ func toDNSConfig(protoDNSConfig *mgmProto.DNSConfig, network *net.IPNet) nbdns.C
 		dnsUpdate.NameServerGroups = append(dnsUpdate.NameServerGroups, dnsNSGroup)
 	}
 
-	addReverseZone(&dnsUpdate, network)
+	if len(dnsUpdate.CustomZones) > 0 {
+		addReverseZone(&dnsUpdate, network)
+	}
 
 	return dnsUpdate
 }
diff --git a/client/internal/engine_test.go b/client/internal/engine_test.go
index a92b5b77228..f19abdf7083 100644
--- a/client/internal/engine_test.go
+++ b/client/internal/engine_test.go
@@ -701,6 +701,9 @@ func TestEngine_UpdateNetworkMapWithDNSUpdate(t *testing.T) {
 								},
 							},
 						},
+						{
+							Domain: "0.66.100.in-addr.arpa.",
+						},
 					},
 					NameServerGroups: []*mgmtProto.NameServerGroup{
 						{
@@ -730,6 +733,9 @@ func TestEngine_UpdateNetworkMapWithDNSUpdate(t *testing.T) {
 						},
 					},
 				},
+				{
+					Domain: "0.66.100.in-addr.arpa.",
+				},
 			},
 			expectedNSGroupsLen: 1,
 			expectedNSGroups: []*nbdns.NameServerGroup{