some test cases from the field
[pdns-pipe-nmc.git] / test.hs
diff --git a/test.hs b/test.hs
index 3bcd4bc12187fffcbe08faff6f915540d50ebb46..117994d76e6e706503b81982f0976af42d3d0b80 100644 (file)
--- a/test.hs
+++ b/test.hs
@@ -3,6 +3,7 @@
 module Main where
 
 import Prelude hiding (readFile)
+import System.Environment
 import Data.ByteString.Lazy (ByteString)
 import Data.ByteString.Lazy.Char8 (readFile)
 import System.IO.Error
@@ -15,6 +16,7 @@ queryOp key = catch (readFile key >>= return . Right)
                     (\e -> return (Left (show (e :: IOException))))
 
 main = do
-        descendNmcDom queryOp []               (seedNmcDom "root") >>= print
-        descendNmcDom queryOp ["_tcp","_smtp"] (seedNmcDom "root") >>= print
+        (d:_) <- getArgs
+        descendNmcDom queryOp []               (seedNmcDom d) >>= print
+        descendNmcDom queryOp ["_tcp","_smtp"] (seedNmcDom d) >>= print