test/unit/test_helpers.rb in spontaneous-0.2.0.alpha2 vs test/unit/test_helpers.rb in spontaneous-0.2.0.alpha3

@@ -2,13 +2,84 @@
 
 require File.expand_path('../../test_helper', __FILE__)
 
 class HelpersTest < MiniTest::Spec
 
-  context "HTML helper" do
+  def setup
+    @site = setup_site
+    @site.paths.add :templates, File.expand_path("../../fixtures/helpers/templates", __FILE__)
+    @renderer = S::Output::Template::Renderer.new(false)
+    S::Output.renderer = @renderer
+
+  end
+
+  def teardown
+    teardown_site
+  end
+
+  context "Helpers" do
+    should "be assignable to a particular format" do
+      CustomHelper1 = Site.helper :html do
+        def here_is_my_custom_helper1; end
+      end
+
+      included_helpers = [
+        CustomHelper1,
+        Spontaneous::Output::Helpers::ConditionalCommentHelper,
+        Spontaneous::Output::Helpers::ClassesHelper,
+        Spontaneous::Output::Helpers::ScriptHelper,
+        Spontaneous::Output::Helpers::StylesheetHelper
+      ]
+      helper_module = Site.context :html
+      join = included_helpers & helper_module.ancestors
+      Set.new(join).should == Set.new(included_helpers)
+      helper_module.respond_to?(:here_is_my_custom_helper1).should be_true
+    end
+
+    should "be assigned to all formats if none given" do
+      CustomHelper2 = Site.helper do
+        extend self
+        def here_is_my_custom_helper2; end
+      end
+
+      assert CustomHelper2.respond_to?(:here_is_my_custom_helper2)
+
+      helper_module = Site.context :html
+      helper_module.ancestors.include?(CustomHelper2).should be_true
+
+      helper_module = Site.context :pdf
+      helper_module.ancestors.include?(CustomHelper2).should be_true
+    end
+
+    should "be available during the render step" do
+      class Page < Spontaneous::Page
+        add_output :mobile
+      end
+
+      Site.helper :html do
+        def here_is_my_custom_helper3
+          "here_is_my_custom_helper3"
+        end
+        extend self
+      end
+
+      Site.helper :mobile do
+        def here_is_my_custom_helper4
+          "here_is_my_custom_helper4"
+        end
+        extend self
+      end
+
+      page = Page.new
+      page.render(:html).should == "<here_is_my_custom_helper3>\n"
+      page.render(:mobile).should == "<here_is_my_custom_helper4>\n"
+    end
+  end
+
+  context "Classes helper" do
     setup do
-      @helper = Spontaneous::Render::Helpers::HTMLHelper
+      @helper = Spontaneous::Output::Helpers::ClassesHelper
     end
     should "enable easy addition of classes" do
       classes = @helper.classes("a", "b", "c", :active => false, :invisible => true)
       classes.should == %(class="a b c invisible")
 
@@ -25,9 +96,15 @@
       classes = @helper.classes("a b c")
       classes.should == %(class="a b c")
 
       classes = @helper.classes("a b c", :active => false, :invisible => true)
       classes.should == %(class="a b c invisible")
+    end
+  end
+
+  context "ConditionalComment helper" do
+    setup do
+      @helper = Spontaneous::Output::Helpers::ConditionalCommentHelper
     end
 
     should "provide a wrapper around IE conditional comments" do
       @helper.ie_only.should == "<!--[if IE]>"
       @helper.ie_only(6).should == "<!--[if IE 6]>"