From: Peter van Dijk <peter.van.dijk@powerdns.com>
Date: Tue, 5 Jul 2022 10:21:22 +0000 (+0200)
Subject: auth py tests: also test TCP LUA shared/non-shared mode
X-Git-Tag: auth-4.8.0-alpha0~1^2~2
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=1d76a10342afc6499aaf5dc350e43c840abf1222;p=thirdparty%2Fpdns.git

auth py tests: also test TCP LUA shared/non-shared mode
---

diff --git a/regression-tests.auth-py/test_LuaRecords.py b/regression-tests.auth-py/test_LuaRecords.py
index 6f6bc40112..fa56b2354f 100644
--- a/regression-tests.auth-py/test_LuaRecords.py
+++ b/regression-tests.auth-py/test_LuaRecords.py
@@ -937,7 +937,7 @@ createforward6.example.org.                 3600 IN NS   ns2.example.org.
                 self.assertRcodeEqual(res, dns.rcode.NOERROR)
                 self.assertEqual(res.answer, response.answer)
 
-    def _getCounter(self):
+    def _getCounter(self, tcp=False):
         """
         Helper function for shared/non-shared testing
         """
@@ -946,8 +946,10 @@ createforward6.example.org.                 3600 IN NS   ns2.example.org.
         query = dns.message.make_query(name, 'TXT')
         responses = []
 
+        sender = self.sendTCPQuery if tcp else self.sendUDPQuery
+
         for i in range(50):
-            res = self.sendUDPQuery(query)
+            res = sender(query)
             responses.append(res.answer[0][0])
 
         return(responses)
@@ -957,9 +959,11 @@ createforward6.example.org.                 3600 IN NS   ns2.example.org.
         Test non-shared behaviour
         """
 
-        res = set(self._getCounter())
+        resUDP = set(self._getCounter(tcp=False))
+        resTCP = set(self._getCounter(tcp=True))
 
-        self.assertEqual(len(res), 1)
+        self.assertEqual(len(resUDP), 1)
+        self.assertEqual(len(resTCP), 1)
 
 class TestLuaRecordsShared(TestLuaRecords):
     _config_template = """
@@ -976,9 +980,11 @@ lua-health-checks-interval=1
         Test shared behaviour
         """
 
-        res = set(self._getCounter())
+        resUDP = set(self._getCounter(tcp=False))
+        resTCP = set(self._getCounter(tcp=True))
 
-        self.assertEqual(len(res), 50)
+        self.assertEqual(len(resUDP), 50)
+        self.assertEqual(len(resTCP), 50)
 
 if __name__ == '__main__':
     unittest.main()