test/test_email.rb in messenger-0.0.3 vs test/test_email.rb in messenger-0.1.0

@@ -18,9 +18,38 @@
         assert_equal ["[e-mail]"], Mail::TestMailer.deliveries.first.to
         assert_equal ["[e-mail]"], Mail::TestMailer.deliveries.first.from
         assert_equal "Test", Mail::TestMailer.deliveries.first.subject
         assert_equal "Test message", Mail::TestMailer.deliveries.first.body.to_s
       end
+
+      should "raise if trying to send to an invalid URL" do
+        assert_raises URLError do
+          Email.send("mailto:test", :body => "whatever", :email_from => "[e-mail]", :email_subject => "Test")
+        end
+      end
+
+      should "obfuscate the URL" do
+        assert_equal "mailto:[e-mail]", Email.obfuscate("mailto:[e-mail]")
+      end
+
+      should "raise if trying obfuscate an invalid URL" do
+        assert_raises URLError do
+          Email.obfuscate("mailto:test")
+        end
+      end
+    end
+
+    context "Email notificaiton URL validation" do
+      should "return true for good URLs" do
+        assert true, Email.valid_url?("mailto:[e-mail]")
+      end
+
+      should "return false for bad URLs" do
+        assert_equal false, Email.valid_url?("mailto:")
+        assert_equal false, Email.valid_url?("mailto:test")
+        assert_equal false, Email.valid_url?("mailto:@example.com")
+        assert_equal false, Email.valid_url?("mailto:example.com")
+      end
     end
 
   end
 
 end