X-Git-Url: http://gitweb.michael.orlitzky.com/?p=hath.git;a=blobdiff_plain;f=src%2FCidr.hs;h=007e94b9425ae819e3c624ae4765bdf3017d3a42;hp=af9c9e8ba55a6c9aaaa5190baf4800599a1623cc;hb=8430f78a834a85ca58cf1b445380679da6b9fc4e;hpb=a8c2d85d8611f7f56eaa0c3406ec423d68f81f96

diff --git a/src/Cidr.hs b/src/Cidr.hs
index af9c9e8..007e94b 100644
--- a/src/Cidr.hs
+++ b/src/Cidr.hs
@@ -307,7 +307,8 @@ cidr_properties =
       prop_normalize_preserves_equality,
       prop_ord_instance_antisymmetric,
       prop_ord_instance_reflexive,
-      prop_ord_instance_transitive ]
+      prop_ord_instance_transitive,
+      prop_ord_uses_addr_when_masks_equal ]
 
 
 -- HUnit Tests
@@ -593,3 +594,18 @@ prop_ord_instance_antisymmetric =
     prop :: Cidr -> Cidr -> Property
     prop cidr1 cidr2 =
       (cidr1 <= cidr2 && cidr2 <= cidr1) ==> cidr1 == cidr2
+
+
+-- When comparing two CIDRs with the same mask, the comparison
+-- should be numeric (i.e. whatever the IPv4Address does).
+-- Of course, we have to normalize first.
+prop_ord_uses_addr_when_masks_equal :: TestTree
+prop_ord_uses_addr_when_masks_equal =
+  testProperty "The CIDR order is the IPv4Address order for equal masks" prop
+  where
+    prop :: Cidr -> Cidr -> Property
+    prop cidr1 cidr2 =
+      (mask1 == mask2) ==> (cidr1 <= cidr2) == (addr1 <= addr2)
+      where
+        (Cidr addr1 mask1) = normalize cidr1
+        (Cidr addr2 mask2) = normalize cidr2