test/test_gman.rb in gman-0.1.0 vs test/test_gman.rb in gman-0.2.0

@@ -20,10 +20,11 @@
             "[e-mail]",
             "[e-mail]",
             "foo",
             "",
             nil,
+            " ",
           ]
 
 class TestGman < Test::Unit::TestCase
 
   should "recognize government email addresses and domains" do
@@ -85,6 +86,23 @@
     Gman.list.each do |entry|
       assert_equal true, Gman.valid?("foo.#{entry.name}"), "foo.#{entry.name} is not a valid domain"
     end
   end
 
+  should "not err out on invalid domains" do
+    assert_equal false, Gman.valid?("[e-mail]")
+    assert_equal "act.gov.au", Gman.get_domain("[e-mail]")
+    assert_equal nil, Gman.domain_parts("[e-mail]")
+  end
+
+  should "return public suffix domain" do
+    assert_equal PublicSuffix::Domain, Gman.domain_parts("whitehouse.gov").class
+    assert_equal NilClass, Gman.domain_parts("foo.bar").class
+  end
+
+  should "parse domain parts" do
+    assert_equal "gov", Gman.domain_parts("[e-mail]").tld
+    assert_equal "bar", Gman.domain_parts("foo.bar.gov").sld
+    assert_equal "bar", Gman.domain_parts("https://foo.bar.gov").sld
+    assert_equal "bar.gov", Gman.domain_parts("[e-mail]").domain
+  end
 end