Add DNSUnitTest.hs

[haskell-dns.git] / Setup.lhs

diff --git a/Setup.lhs b/Setup.lhs
index 5bde0de962274e26328ecffabb4e9f6430a10cc9..335aacc692f0b63163d762914dc1a6bc60797a9a 100644 (file)
--- a/Setup.lhs
+++ b/Setup.lhs
@@ -1,3 +1,8 @@
 #!/usr/bin/env runhaskell
 > import Distribution.Simple
-> main = defaultMain
+> import System.Cmd
+> import System.Exit
+> main = defaultMainWithHooks (defaultUserHooks { runTests = runTestUnit })
+>     where
+>       runTestUnit _ _ _ _
+>           = system "./dist/build/DNSUnitTest/DNSUnitTest" >> return ()