From 2c836040573628f96010f8983a244e6fb15e8679 Mon Sep 17 00:00:00 2001
From: netkas <netkas@n64.cc>
Date: Fri, 10 Jan 2025 13:34:18 -0500
Subject: [PATCH] Refactor mockRecord method to support string input.

---
 src/Socialbox/Classes/ServerResolver.php | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/Socialbox/Classes/ServerResolver.php b/src/Socialbox/Classes/ServerResolver.php
index c34138c..8d166ac 100644
--- a/src/Socialbox/Classes/ServerResolver.php
+++ b/src/Socialbox/Classes/ServerResolver.php
@@ -127,8 +127,13 @@
          * @param DnsRecord $record The DNS record to be associated with the specified domain.
          * @return void
          */
-        public static function mockRecord(string $domain, DnsRecord $record): void
+        public static function addMock(string $domain, DnsRecord|string $record): void
         {
+            if(is_string($record))
+            {
+                $record = DnsHelper::parseTxt($record);
+            }
+
             self::$mockedRecords[$domain] = $record;
         }