Update for QuickCheck 2.

[hath.git] / src / Maskbits.hs

diff --git a/src/Maskbits.hs b/src/Maskbits.hs
index cbd4dbe86fd0d76efa15ea349dfd353d06e62985..06314e395f2043c12da778f81405203692e5ded2 100644 (file)
--- a/src/Maskbits.hs
+++ b/src/Maskbits.hs
@@ -1,8 +1,11 @@
 module Maskbits
 ( Maskbits(..),
+  decrement,
   maskbits_from_string
 ) where
 
+import Test.QuickCheck
+
 -- A type representing the number of bits in a CIDR netmask.
 data Maskbits = None
               | Zero
@@ -38,7 +41,81 @@ data Maskbits = None
               | Thirty
               | ThirtyOne
               | ThirtyTwo
-              deriving (Eq, Ord, Show)
+              deriving (Eq, Ord)
+
+
+instance Show Maskbits where
+    show None        = "None"
+    show Zero        = "0"
+    show One         = "1"
+    show Two         = "2"
+    show Three       = "3"
+    show Four        = "4"
+    show Five        = "5"
+    show Six         = "6"
+    show Seven       = "7"
+    show Eight       = "8"
+    show Nine        = "9"
+    show Ten         = "10"
+    show Eleven      = "11"
+    show Twelve      = "12"
+    show Thirteen    = "13"
+    show Fourteen    = "14"
+    show Fifteen     = "15"
+    show Sixteen     = "16"
+    show Seventeen   = "17"
+    show Eighteen    = "18"
+    show Nineteen    = "19"
+    show Twenty      = "20"
+    show TwentyOne   = "21"
+    show TwentyTwo   = "22"
+    show TwentyThree = "23"
+    show TwentyFour  = "24"
+    show TwentyFive  = "25"
+    show TwentySix   = "26"
+    show TwentySeven = "27"
+    show TwentyEight = "28"
+    show TwentyNine  = "29"
+    show Thirty      = "30"
+    show ThirtyOne   = "31"
+    show ThirtyTwo   = "32"
+
+
+instance Arbitrary Maskbits where
+    arbitrary = elements [ Zero,
+                           One,
+                           Two,
+                           Three,
+                           Four,
+                           Five,
+                           Six,
+                           Seven,
+                           Eight,
+                           Nine,
+                           Ten,
+                           Eleven,
+                           Twelve,
+                           Thirteen,
+                           Fourteen,
+                           Fifteen,
+                           Sixteen,
+                           Seventeen,
+                           Eighteen,
+                           Nineteen,
+                           Twenty,
+                           TwentyOne,
+                           TwentyTwo,
+                           TwentyThree,
+                           TwentyFour,
+                           TwentyFive,
+                           TwentySix,
+                           TwentySeven,
+                           TwentyEight,
+                           TwentyNine,
+                           Thirty,
+                           ThirtyOne,
+                           ThirtyTwo ]
+
 
 
 -- There are only 32 bits in an IPv4 address, so there
@@ -85,3 +162,41 @@ maskbits_from_string s =
     case (reads s :: [(Int, String)]) of
       []   -> None
       x:_ -> maskbits_from_int (fst x)
+
+
+
+decrement :: Maskbits -> Maskbits
+decrement None = None
+decrement Zero = None
+decrement One = Zero
+decrement Two = One
+decrement Three = Two
+decrement Four = Three
+decrement Five = Four
+decrement Six = Five
+decrement Seven = Six
+decrement Eight = Seven
+decrement Nine = Eight
+decrement Ten = Nine
+decrement Eleven = Ten
+decrement Twelve = Eleven
+decrement Thirteen = Twelve
+decrement Fourteen = Thirteen
+decrement Fifteen = Fourteen
+decrement Sixteen = Fifteen
+decrement Seventeen = Sixteen
+decrement Eighteen = Seventeen
+decrement Nineteen = Eighteen
+decrement Twenty = Nineteen
+decrement TwentyOne = Twenty
+decrement TwentyTwo = TwentyOne
+decrement TwentyThree = TwentyTwo
+decrement TwentyFour = TwentyThree
+decrement TwentyFive = TwentyFour
+decrement TwentySix = TwentyFive
+decrement TwentySeven = TwentySix
+decrement TwentyEight = TwentySeven
+decrement TwentyNine = TwentyEight
+decrement Thirty = TwentyNine
+decrement ThirtyOne = Thirty
+decrement ThirtyTwo = ThirtyOne