X-Git-Url: http://gitweb.michael.orlitzky.com/?p=haeredes.git;a=blobdiff_plain;f=src%2FDNS.hs;fp=src%2FDNS.hs;h=cda8b96274414d6bd90b6dd71bf8d5f1f08e2309;hp=02e5a5524b866eb646102a544bcc0719a154080c;hb=4f6bb3da01e7136f1e17a7d2d28518eefe18fc36;hpb=66b80d96a2552c2c1e58a771346ce8e9f38acb34 diff --git a/src/DNS.hs b/src/DNS.hs index 02e5a55..cda8b96 100644 --- 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 ]