dnsdist doc typo fix

[thirdparty/pdns.git] / regression-tests.dnsdist / test_CheckConfig.py

diff --git a/regression-tests.dnsdist/test_CheckConfig.py b/regression-tests.dnsdist/test_CheckConfig.py
index 2034fc8511d89b1e189b27e65d41dd2d5b2cb9e6..ce9345c05ed68d684a370432a925b3033617c0a1 100644 (file)
--- a/regression-tests.dnsdist/test_CheckConfig.py
+++ b/regression-tests.dnsdist/test_CheckConfig.py
@@ -2,11 +2,11 @@
 import unittest
 import os
 import subprocess
-import sys
+import time
 
 class TestCheckConfig(unittest.TestCase):
 
-    def tryDNSDist(self, configTemplate, shouldBeSuccessful=True):
+    def tryDNSDist(self, configTemplate, shouldBeSuccessful=True, delay=1):
         conffile = 'dnsdist_test.conf'
         with open(conffile, 'w') as conf:
             conf.write("-- Autogenerated by dnsdisttests.py\n")
@@ -17,10 +17,14 @@ class TestCheckConfig(unittest.TestCase):
         with open(os.devnull, 'w') as fdDevNull:
             dnsdist = subprocess.Popen(dnsdistcmd, close_fds=True, stdout=fdDevNull)
 
-        if dnsdist.poll() is not None:
-            if dnsdist.returncode != 0 and not shouldBeSuccessful:
-                sys.exit(1)
-            sys.exit(0)
+        if dnsdist.poll() is None:
+            time.sleep(delay)
+            self.assertNotEqual(dnsdist.poll(), None)
+
+        if shouldBeSuccessful:
+            self.assertEquals(dnsdist.returncode, 0)
+        else:
+            self.assertNotEqual(dnsdist.returncode, 0)
 
     def testWorkingConfig(self):
         """
@@ -29,21 +33,12 @@ class TestCheckConfig(unittest.TestCase):
         configTemplate = """
             newServer{address="127.0.0.1:53"}
             truncateTC(true)
-            addAnyTCRule()
-            addAction(RegexRule("evil[0-9]{4,}\\\\.regex\\\\.tests\\\\.powerdns\\\\.com$"), RCodeAction(5))
+            addAction(AndRule{QTypeRule(DNSQType.ANY), TCPRule(false)}, TCAction())
+            addAction(RegexRule("evil[0-9]{4,}\\\\.regex\\\\.tests\\\\.powerdns\\\\.com$"), RCodeAction(DNSRCode.REFUSED))
             mySMN = newSuffixMatchNode()
             mySMN:add(newDNSName("nameAndQtype.tests.powerdns.com."))
-            addAction(AndRule{SuffixMatchNodeRule(mySMN), QTypeRule("TXT")}, RCodeAction(4))
+            addAction(AndRule{SuffixMatchNodeRule(mySMN), QTypeRule("TXT")}, RCodeAction(DNSRCode.NOTIMP))
             addAction(makeRule("drop.test.powerdns.com."), DropAction())
-            block=newDNSName("powerdns.org.")
-            function blockFilter(dq)
-                if(dq.qname:isPartOf(block))
-                then
-                    print("Blocking *.powerdns.org")
-                    return true
-                end
-                return false
-            end
         """
 
         self.tryDNSDist(configTemplate)
@@ -69,7 +64,7 @@ class TestCheckConfig(unittest.TestCase):
         CheckConfig: Invalid parameter
         """
         configTemplate = """
-          newServer { address="127.0.0.355" }
+          addACL("127.0.0.355")
         """
         self.tryDNSDist(configTemplate, False)