X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FCidr.hs;h=77f1ae41fe42569d5801c50158aef48f1a595ce7;hb=a767881c16a119a90ff80d9c14106348d8bc310d;hp=08eebcba23fd75a533e687cd742fb6db4a7e6af8;hpb=f74d9ccfb98a2ac25b4eebbfadf9a16b7a5202e1;p=hath.git

diff --git a/src/Cidr.hs b/src/Cidr.hs
index 08eebcb..77f1ae4 100644
--- a/src/Cidr.hs
+++ b/src/Cidr.hs
@@ -1,10 +1,12 @@
 module Cidr
 ( Cidr(..),
   cidr_from_string,
+  cidr_tests,
   combine_all
 ) where
 
 import Data.List (nubBy)
+import Test.HUnit
 
 import IPv4Address
 import ListUtils
@@ -170,3 +172,20 @@ adjacent cidr1 cidr2
     addr2 = ipv4address cidr2
     mbits1 = maskbits cidr1
     mbits2 = maskbits cidr2
+
+
+
+
+
+-- HUnit Tests
+
+test_contains :: Test
+test_contains =
+    TestCase $ assertEqual "10.1.1.0/23 contains 10.1.1.0/24" (cidr1 `contains` cidr2) True
+    where
+      cidr1 = cidr_from_string "10.1.1.0/23"
+      cidr2 = cidr_from_string "10.1.1.0/24"
+
+
+cidr_tests :: [Test]
+cidr_tests = [ test_contains ]