X-Git-Url: http://gitweb.michael.orlitzky.com/?p=dead%2Fharbl.git;a=blobdiff_plain;f=test%2FDoctests.hs;h=558a1e270ff80154d88f57c3f54abf60757b54d3;hp=d183b4b664d289ea123da751f562a937c34ff1ae;hb=67a990c746599a0f64cffa8dd92157626fc4654e;hpb=d124148390284252851e2973e60d1fe85380330d

diff --git a/test/Doctests.hs b/test/Doctests.hs
index d183b4b..558a1e2 100644
--- 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