ApplicationHelper.get_host does not break on invalid url

Marcin Ciunelis
1 parent 4e67af18
Showing 2 changed files with 33 additions and 2 deletions Show diff stats
app/helpers/application_helper.rb
spec/helpers/application_helper_spec.rb
@@ -23,8 +23,12 @@ module ApplicationHelper
   end
   def get_host(url)
-    uri = url && URI.parse(url)
-    uri.blank? ? "N/A" : uri.host
+    begin
+      uri = url && URI.parse(url)
+      uri.blank? ? "N/A" : uri.host
+    rescue URI::InvalidURIError
+      "N/A"
+    end
   end
@@ -0,0 +1,27 @@
+require 'spec_helper'
+
+# Specs in this file have access to a helper object that includes
+# the AbcHelper. For example:
+#
+# describe AbcHelper do
+#   describe "string concat" do
+#     it "concats two strings with spaces" do
+#       helper.concat_strings("this","that").should == "this that"
+#     end
+#   end
+# end
+describe ApplicationHelper do
+  describe "get_host" do
+    it "returns host if url is valid" do
+      helper.get_host("http://example.com/resource/12").should == 'example.com'
+    end
+    
+    it "returns 'N/A' when url is not valid" do
+      helper.get_host("some string").should == 'N/A'
+    end
+
+    it "returns 'N/A' when url is empty" do
+      helper.get_host({}).should == 'N/A'
+    end
+  end
+end
@@ -0,0 +1,27 @@		@@ -0,0 +1,27 @@
	1	+require 'spec_helper'
	2	+
	3	+# Specs in this file have access to a helper object that includes
	4	+# the AbcHelper. For example:
	5	+#
	6	+# describe AbcHelper do
	7	+# describe "string concat" do
	8	+# it "concats two strings with spaces" do
	9	+# helper.concat_strings("this","that").should == "this that"
	10	+# end
	11	+# end
	12	+# end
	13	+describe ApplicationHelper do
	14	+ describe "get_host" do
	15	+ it "returns host if url is valid" do
	16	+ helper.get_host("http://example.com/resource/12").should == 'example.com'
	17	+ end
	18	+
	19	+ it "returns 'N/A' when url is not valid" do
	20	+ helper.get_host("some string").should == 'N/A'
	21	+ end
	22	+
	23	+ it "returns 'N/A' when url is empty" do
	24	+ helper.get_host({}).should == 'N/A'
	25	+ end
	26	+ end
	27	+end