Add tests for probing for zone files

1 files changed, 27 insertions, 0 deletions

diff --git a/tests/local_probe_tests.py b/tests/local_probe_tests.py
new file mode 100644
index 0000000..fcd4724
--- /dev/null
+++ b/tests/local_probe_tests.py
@@ -0,0 +1,27 @@
+import io
+import os
+import subprocess
+import unittest
+
+DATA_DIR = os.path.dirname(__file__)
+EXAMPLE_COM_SIGNED = os.path.join(DATA_DIR, 'zone', 'example.com.zone.signed')
+EXAMPLE_COM_ZONE = os.path.join(DATA_DIR, 'zone', 'example.com.zone')
+EXAMPLE_COM_DELEGATION = os.path.join(DATA_DIR, 'zone', 'example.com.zone-delegation')
+
+class DNSProbeTestCase(unittest.TestCase):
+    def setUp(self):
+        self.devnull = io.open('/dev/null', 'wb')
+
+    def tearDown(self):
+        self.devnull.close()
+
+    def test_authoritative_probe_root(self):
+        self.assertEqual(subprocess.call(
+            ['dnsviz', 'probe', '-d', '0', '-A', 
+                '-x' 'example.com:%s' % EXAMPLE_COM_SIGNED,
+                '-N' 'example.com:%s' % EXAMPLE_COM_DELEGATION,
+                '-D' 'example.com:%s' % EXAMPLE_COM_DELEGATION,
+                'example.com'], stdout=self.devnull), 0)
+
+if __name__ == '__main__':
+    unittest.main()