X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=test%2FTestSuite.hs;h=172bcfcb42e9dd649fa64977f35261b8e0aa8d1c;hb=18a81546e7c4ad10574918efb08b2e104e911d73;hp=897807f9b6216b2f91a3a35ce525648393fcd051;hpb=1d34c6adebfc203d38d7051584de7f1733a34d6f;p=hath.git

diff --git a/test/TestSuite.hs b/test/TestSuite.hs
index 897807f..172bcfc 100644
--- a/test/TestSuite.hs
+++ b/test/TestSuite.hs
@@ -1,15 +1,43 @@
-import Test.HUnit
+{-# LANGUAGE NoMonomorphismRestriction #-}
+import Data.Monoid (mempty)
+import Test.Framework (
+  Test,
+  defaultMainWithOpts )
+import Test.Framework.Options
+import Test.Framework.Runners.Options
 
-import Cidr (cidr_tests)
-import IPv4Address (ipv4address_tests)
-import Maskable (maskable_tests)
-import Octet (octet_tests)
+import Cidr (
+  cidr_properties,
+  cidr_tests )
+import IPv4Address (
+  ipv4address_properties,
+  ipv4address_tests )
+import Octet (
+  octet_properties,
+  octet_tests )
 
-test_suite = TestList (concat [cidr_tests,
-                               ipv4address_tests,
-                               maskable_tests,
-                               octet_tests])
+tests :: [Test.Framework.Test]
+tests = [ cidr_properties,
+          cidr_tests,
+          ipv4address_properties,
+          ipv4address_tests,
+          octet_properties,
+          octet_tests ]
 
-main :: IO Counts
+main :: IO ()
 main = do
-  runTestTT test_suite
+  let empty_test_opts = mempty :: TestOptions
+  let my_test_opts = empty_test_opts {
+    --
+    -- Increase to 5000 when,
+    --   https://github.com/batterseapower/test-framework/issues/34
+    -- is fixed.
+    --
+    topt_maximum_generated_tests = Just 1000
+  }
+  let empty_runner_opts = mempty :: RunnerOptions
+  let my_runner_opts = empty_runner_opts {
+    ropt_test_options = Just my_test_opts
+  }
+
+  defaultMainWithOpts tests my_runner_opts