From 3bdd9282ce67d1094c36204e77e3f32356a2c059 Mon Sep 17 00:00:00 2001
From: David Waern <david.waern@gmail.com>
Date: Fri, 25 Nov 2011 02:16:20 +0100
Subject: Some more changes to test scripts.

---
 tests/html-tests/copy.hs     | 10 ++++++++--
 tests/html-tests/runtests.hs |  2 +-
 2 files changed, 9 insertions(+), 3 deletions(-)

(limited to 'tests')

diff --git a/tests/html-tests/copy.hs b/tests/html-tests/copy.hs
index fa18fe9c..fe1e5422 100644
--- a/tests/html-tests/copy.hs
+++ b/tests/html-tests/copy.hs
@@ -5,7 +5,6 @@ import System.Exit
 import System.Directory
 import Data.List
 import Control.Monad
-import Text.Regex
 
 
 main = do
@@ -27,4 +26,11 @@ copy file = do
   writeFile new (stripLinks contents)
 
 
-stripLinks f = subRegex (mkRegexWithOpts "<A HREF=[^>]*>" False False) f "<A HREF=\"\">"
+stripLinks str =
+  let prefix = "<a href=\"" in
+  case stripPrefix prefix str of
+    Just str' -> prefix ++ stripLinks (dropWhile (/= '"') str')
+    Nothing ->
+      case str of
+        [] -> []
+        x : xs -> x : stripLinks xs
diff --git a/tests/html-tests/runtests.hs b/tests/html-tests/runtests.hs
index 35de8c0b..25e53d1a 100644
--- a/tests/html-tests/runtests.hs
+++ b/tests/html-tests/runtests.hs
@@ -32,7 +32,7 @@ main = do
 
 test = do
   x <- doesFileExist haddockPath
-  when (not x) $ die "you need to run 'cabal build' successfully first"
+  unless x $ die "you need to run 'cabal build' successfully first"
 
   contents <- getDirectoryContents testDir
   args <- getArgs
-- 
cgit v1.2.3