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

@@ -1,25 +1,29 @@
 require 'helper'
-require 'swot'
 
 VALID = [  "foo.gov",
             "http://foo.mil",
             "[e-mail]",
             "foo.gov.au",
-            "http://www.foo.gouv.fr",
+            "https://www.foo.gouv.fr",
             "[e-mail]",
             "foo.bar.baz.gov.au",
-            "[e-mail]"
+            "[e-mail]",
+            ".gov",
+            "foo.fed.us",
         ]
 
 INVALID = [ "foo.bar.com",
             "[e-mail]",
             "http://www.foo.biz",
             "foo.uk",
             "gov",
             "[e-mail]",
-            "[e-mail]"
+            "[e-mail]",
+            "foo",
+            "",
+            nil,
           ]
 
 class TestGman < Test::Unit::TestCase
 
   should "recognize government email addresses and domains" do
@@ -37,6 +41,50 @@
   should "not contain any educational domains" do
     Gman.list.each do |entry|
       assert_equal false, Swot::is_academic?(entry.name), "#{entry.name} is an academic domain"
     end
   end
+
+  should "not contain any invalid domains" do
+    Gman.list.each do |entry|
+      assert_equal true, PublicSuffix.valid?("foo.#{entry.name}"), "#{entry.name} is not a valid domain"
+    end
+  end
+
+  should "not allow educational domains" do
+    assert_equal false, Gman::valid?("[e-mail]")
+  end
+
+  should "properly parse domains from strings" do
+    assert_equal "github.gov", Gman::get_domain("[e-mail]")
+    assert_equal "foo.github.gov", Gman::get_domain("foo.github.gov")
+    assert_equal "github.gov", Gman::get_domain("http://github.gov")
+    assert_equal "github.gov", Gman::get_domain("https://github.gov")
+    assert_equal ".gov", Gman::get_domain(".gov")
+    assert_equal nil, Gman.get_domain("foo")
+  end
+
+  should "validate mx records when asked" do
+    assert_equal true, Gman.valid?("[e-mail]", true)
+    assert_equal false, Gman.valid?("[e-mail]", true)
+    assert_equal true, Gman.valid?("[e-mail]", false)
+  end
+
+  should "pass any url on the list" do
+    Gman.list.each do |entry|
+      assert_equal true, Gman.valid?("http://foo.#{entry.name}/bar"), "http://foo.#{entry.name}/bar is not a valid"
+    end
+  end
+
+  should "pass any email on the list" do
+    Gman.list.each do |entry|
+      assert_equal true, Gman.valid?("foo@bar.#{entry.name}"), "foo@bar.#{entry.name} is not a valid"
+    end
+  end
+
+  should "pass any domain on the list" do
+    Gman.list.each do |entry|
+      assert_equal true, Gman.valid?("foo.#{entry.name}"), "foo.#{entry.name} is not a valid domain"
+    end
+  end
+
 end