ensure host ends in a slash

lib/sitemap_generator/sitemap_location.rb

@@ -9,16 +9,27 @@ class SitemapLocation < Hash
 
     PATH_OUTPUT_WIDTH = 47 # Character width of the path in the summary lines
 
-    [:host, :adapter].each do |method|
-      define_method(method) do
-        raise SitemapGenerator::SitemapError, "No value set for #{method}" unless self[method]
-        self[method]
-      end
+    def assert_value(key)
+      raise SitemapGenerator::SitemapError, "No value set for #{key}" unless self[key]
+    end
+
+    def append_slash(key)
+      SitemapGenerator::Utilities.append_slash(self[key])
+    end
+
+    def adapter
+      assert_value(:adapter)
+      self[:adapter]
+    end
+
+    def host
+      assert_value(:host)
+      append_slash(:host)
     end
 
     [:public_path, :sitemaps_path].each do |method|
       define_method(method) do
-        Pathname.new(SitemapGenerator::Utilities.append_slash(self[method]))
+        Pathname.new(append_slash(method))
       end
     end
 

spec/sitemap_generator/sitemap_location_spec.rb

@@ -35,6 +35,11 @@
     }.should raise_error
   end
 
+  it "should append slash to host" do
+    location = SitemapGenerator::SitemapLocation.new(:filename => nil, :namer => nil, :host => "http://www.slash.com")
+    location.host.should == "http://www.slash.com/"
+  end
+
   it "should accept a Namer option" do
     @namer = SitemapGenerator::SimpleNamer.new(:xxx)
     location = SitemapGenerator::SitemapLocation.new(:namer => @namer)