From a8f399ab7fce7fd767a48be22c9084e5b707e102 Mon Sep 17 00:00:00 2001
From: Kameron Kenny <kkenny379@gmail.com>
Date: Thu, 19 Dec 2024 11:03:18 -0500
Subject: [PATCH] bats

---
 test/integration/dns.bat | 32 ++++++++++++++++++--------------
 1 file changed, 18 insertions(+), 14 deletions(-)

diff --git a/test/integration/dns.bat b/test/integration/dns.bat
index 1604f1e..6139a5d 100644
--- a/test/integration/dns.bat
+++ b/test/integration/dns.bat
@@ -1,15 +1,20 @@
 #!/usr/bin/env bats
 
-domain="kameronkenny.com"
-serial="24070101"
-@test "Test A Record: ${domain}" {
-  r=$(dig -t A ${domain} +noall +answer | awk '{ print $5 }')
-  [ "$r" == "204.8.15.81" ]
+setup() {
+  load 'bats-support'
+  load 'bats-assert'
 }
 
-@test "Test SOA Serial: ${domain}" {
-  r=$(dig -t SOA ${domain} +noall +answer +multi | grep serial | awk '{ print $1 }')
-  [ "$r" == "${serial}" ]
+export domain="kameronkenny.com"
+export serial="24070101"
+@test "Test A Record: kameronkenny.com" {
+  run dig -t A kameronkenny.com +noall +answer
+  assert_output --partial '204.8.15.81'
+}
+
+@test "Test SOA Serial: kameronkenny.com" {
+  run dig -t SOA kameronkenny.com +noall +answer +multi
+  assert_output --partial "${serial}"
 }
 
 # in.thelinuxpro.net
@@ -17,12 +22,11 @@ serial="24121801"
 ## nagios.in.thelinuxpro.net
 domain="nagios.in.thelinuxpro.net"
 @test "Test A Record: ${domain}" {
-  r=$(dig -t A ${domain} +noall +answer | awk '{ print $5 }')
-  [ "$r" == "10.99.23.36" ]
+  dig -t A ${domain} +noall +answer
+  assert_output --partial '10.99.23.36'
 }
 
-@test "Test SOA Serial: ${domain}" {
-  skip
-  r=$(dig -t SOA nagios.in.thelinuxpro.net +noall +answer +multi | grep serial | awk '{ print $1 }')
-  [ "$r" == "${serial}" ]
+@test "Test SOA Serial: nagios.in.thelinuxpro.net" {
+  dig -t SOA nagios.in.thelinuxpro.net +noall +answer +multi
+  assert_output --partial "${serial}"
 }