From 56daa7e4f347fabc7d6989172e6b167181a5949d Mon Sep 17 00:00:00 2001
From: John Goerzen <jgoerzen@complete.org>
Date: Wed, 28 May 2008 03:49:28 -0500
Subject: [PATCH] Test infrastructure improvements

---
 OfflineIMAP.cabal   | 3 +--
 testsrc/runtests.hs | 6 ++++++
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/OfflineIMAP.cabal b/OfflineIMAP.cabal
index 92ac69d..710864a 100644
--- a/OfflineIMAP.cabal
+++ b/OfflineIMAP.cabal
@@ -73,5 +73,4 @@ Executable runtests
   Hs-Source-Dirs: src, testsrc
   GHC-Options: -O2
   Build-Depends: QuickCheck
-  Extensions: ExistentialQuantification, OverlappingInstances,
-      UndecidableInstances
+  Extensions: ExistentialQuantification, FlexibleInstances
diff --git a/testsrc/runtests.hs b/testsrc/runtests.hs
index 079e482..99d9a1c 100644
--- a/testsrc/runtests.hs
+++ b/testsrc/runtests.hs
@@ -44,6 +44,12 @@ testv = runVerbTestText (HU.putTextToHandle stderr True) $ HU.TestList alltHU
 
 testq = runTests "Test Stuff" defOpt (map (run . snd) allt)
 
+instance (Arbitrary k, Eq k, Ord k) => Arbitrary (Map.Map k ()) where
+    arbitrary = 
+        do items <- arbitrary
+           return $ keysToMap items
+    coarbitrary = coarbitrary . Map.keys
+
 -- Modified from HUnit
 runVerbTestText :: HU.PutText st -> HU.Test -> IO (HU.Counts, st)
 runVerbTestText (HU.PutText put us) t = do
-- 
2.39.2