Add a skeleton test suite.

[haeredes.git] / src / DNS.hs

diff --git a/src/DNS.hs b/src/DNS.hs
index 02e5a5524b866eb646102a544bcc0719a154080c..cda8b96274414d6bd90b6dd71bf8d5f1f08e2309 100644 (file)
--- a/src/DNS.hs
+++ b/src/DNS.hs
@@ -1,5 +1,6 @@
 module DNS (
   LookupResult,
+  dns_tests,
   lookupMX',
   lookupNS',
   normalize,
@@ -31,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])
@@ -112,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 ]