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 6f6bc4011221cf9d5977d76ad83da643014003ab..fa56b2354f3301db490f1001b593229125c9f890 100644 (file)
--- 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()