X-Git-Url: https://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=test%2FTestSuite.hs;h=7cd8d937938b9e245156823ed5e3b393c5b5d3cb;hb=eee156f562f9c1c1194a67cef12f146304d88ce9;hp=897807f9b6216b2f91a3a35ce525648393fcd051;hpb=1d34c6adebfc203d38d7051584de7f1733a34d6f;p=hath.git

diff --git a/test/TestSuite.hs b/test/TestSuite.hs
index 897807f..7cd8d93 100644
--- a/test/TestSuite.hs
+++ b/test/TestSuite.hs
@@ -1,15 +1,21 @@
-import Test.HUnit
+import Test.Tasty ( TestTree, defaultMain, localOption, testGroup )
+import Test.Tasty.QuickCheck ( QuickCheckTests(..), QuickCheckMaxRatio(..) )
+import Cidr ( cidr_properties, cidr_tests )
+import IPv4Address ( ipv4address_properties, ipv4address_tests )
+import Octet ( octet_properties, octet_tests )
 
-import Cidr (cidr_tests)
-import IPv4Address (ipv4address_tests)
-import Maskable (maskable_tests)
-import Octet (octet_tests)
+tests :: TestTree
+tests = testGroup "All Tests" [
+          cidr_properties,
+          cidr_tests,
+          ipv4address_properties,
+          ipv4address_tests,
+          octet_properties,
+          octet_tests ]
 
-test_suite = TestList (concat [cidr_tests,
-                               ipv4address_tests,
-                               maskable_tests,
-                               octet_tests])
-
-main :: IO Counts
-main = do
-  runTestTT test_suite
+-- TODO: Run 5000 generated tests, we have a large space.
+main :: IO ()
+main =
+  defaultMain $
+    localOption (QuickCheckTests 5000) $
+    localOption (QuickCheckMaxRatio 50) tests