Add run tests

1 files changed, 5 insertions, 0 deletions

diff --git a/tests/dnsviz_probe_tests.py b/tests/dnsviz_probe_tests.py
index 54be61e..5373d71 100644
--- a/tests/dnsviz_probe_tests.py
+++ b/tests/dnsviz_probe_tests.py
@@ -5,6 +5,7 @@ import binascii
 import gzip
 import logging
 import os
+import subprocess
 import tempfile
 import unittest
 
@@ -987,5 +988,9 @@ ns1.example 0 IN A 192.0.2.1
             for tmpfile in (names_file, names_file_unicode, example_names_only):
                 os.remove(tmpfile.name)
 
+    def test_dnsviz_probe_run(self):
+        self.assertEqual(subprocess.call(['./bin/dnsviz', 'probe', '-d', '0', '-o', '/dev/null', 'example.com']), 0)
+        self.assertEqual(subprocess.call(['./bin/dnsviz', 'probe', '-d', '0', '-A', '-o', '/dev/null', 'example.com']), 0)
+
 if __name__ == '__main__':
     unittest.main()