From 1d76a10342afc6499aaf5dc350e43c840abf1222 Mon Sep 17 00:00:00 2001 From: Peter van Dijk Date: Tue, 5 Jul 2022 12:21:22 +0200 Subject: [PATCH] auth py tests: also test TCP LUA shared/non-shared mode --- regression-tests.auth-py/test_LuaRecords.py | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) 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() -- 2.47.2