Separate the Network.DNS.RBL.Weight module and fix the doctests.

[dead/harbl.git] / test / Doctests.hs

diff --git a/test/Doctests.hs b/test/Doctests.hs
index d183b4b664d289ea123da751f562a937c34ff1ae..558a1e270ff80154d88f57c3f54abf60757b54d3 100644 (file)
--- a/test/Doctests.hs
+++ b/test/Doctests.hs
@@ -4,10 +4,16 @@ where
 import Test.DocTest
 import System.FilePath.Find ((==?), always, extension, find)
 
-find_sources :: IO [FilePath]
-find_sources = find always (extension ==? ".hs") "src/"
+find_lib_sources :: IO [FilePath]
+find_lib_sources = find always (extension ==? ".hs") "harbl/src/"
+
+find_cli_sources :: IO [FilePath]
+find_cli_sources = find always (extension ==? ".hs") "harbl-cli/src/"
 
 main :: IO ()
 main = do
-  sources <- find_sources
-  doctest $ ["-isrc", "-idist/build/autogen"] ++ sources
+  lib_sources <- find_lib_sources
+  cli_sources <- find_cli_sources
+  let sources = lib_sources ++ cli_sources
+  let flags = ["-iharbl/src", "-iharbl-cli/src", "-idist/build/autogen"]
+  doctest $ flags ++ sources