1 files changed, 12 insertions, 0 deletions

diff --git a/clients/netbox/config_file_client_test.go b/clients/netbox/config_file_client_test.go
index 117db00..7af4647 100644
--- a/clients/netbox/config_file_client_test.go
+++ b/clients/netbox/config_file_client_test.go
@@ -13,11 +13,15 @@ import (
 var (
         ipNet1 *net.IPNet
         ipNet2 *net.IPNet
+        ip1    net.IP
+        ip2    net.IP
 )
 
 func init() {
         _, ipNet1, _ = net.ParseCIDR("127.0.0.1/8")
         _, ipNet2, _ = net.ParseCIDR("10.0.10.0/24")
+        ip1 = net.ParseIP("127.0.0.1")
+        ip2 = net.ParseIP("127.0.0.2")
 }
 
 var testFs fstest.MapFS
@@ -57,6 +61,7 @@ netbox:
   sites:
     site1:
       tag1: [ "127.0.0.1/8", "10.0.10.0/24" ]
+      dns-servers: [ "127.0.0.1", "127.0.0.2" ]
   tags:
     tag1: [ "127.0.0.1/8", "10.0.10.0/24" ]
   services:
@@ -186,6 +191,13 @@ func (s *ConfigFileNetboxClientSuite) TestGetServicesForVMMissingVM() {
         assert.ErrorContains(s.T(), err, "VM named foo not in config")
 }
 
+func (s *ConfigFileNetboxClientSuite) TestGetDnsServersForSite() {
+        ips, err := s.c.GetDnsServersForSite(context.TODO(), "site1")
+        assert.NoError(s.T(), err)
+        assert.True(s.T(), ip1.Equal(ips[0]))
+        assert.True(s.T(), ip2.Equal(ips[1]))
+}
+
 func TestConfigFileNetboxClientSuite(t *testing.T) {
         suite.Run(t, &ConfigFileNetboxClientSuite{})
 }