X-Git-Url: http://gitweb.michael.orlitzky.com/?p=haeredes.git;a=blobdiff_plain;f=src%2FDNS.hs;h=cda8b96274414d6bd90b6dd71bf8d5f1f08e2309;hp=a1ed2bdf718790d5fce2aaa77cda275da77eb3f1;hb=4f6bb3da01e7136f1e17a7d2d28518eefe18fc36;hpb=e8fe3d4c76b719d256a3c04d462819cd98218111

diff --git a/src/DNS.hs b/src/DNS.hs
index a1ed2bd..cda8b96 100644
--- a/src/DNS.hs
+++ b/src/DNS.hs
@@ -1,8 +1,10 @@
 module DNS (
   LookupResult,
+  dns_tests,
   lookupMX',
   lookupNS',
   normalize,
+  normalize_case,
   resolve_address )
 where
 
@@ -30,6 +32,9 @@ import Network.DNS (
   rdata,
   rrtype,
   withResolver )
+import Test.Framework (Test, testGroup)
+import Test.Framework.Providers.HUnit (testCase)
+import Test.HUnit (assertEqual)
 import Text.Read (readMaybe)
 
 type LookupResult = (Domain, Maybe [Domain])
@@ -111,3 +116,18 @@ normalize_root d
 
 normalize_case :: Domain -> Domain
 normalize_case = BS.map toLower
+
+
+test_normalize_case :: Test
+test_normalize_case =
+  testCase desc $
+    assertEqual desc expected actual
+  where
+    desc = "normalize_case lowercases DNS names"
+    expected = BS.pack "example.com"
+    actual = normalize_case $ BS.pack "ExAmPlE.COM"
+
+dns_tests :: Test
+dns_tests =
+  testGroup "DNS Tests" [
+    test_normalize_case ]