spec/stache/mustache/handler_spec.rb in stache-1.0.0.rc vs spec/stache/mustache/handler_spec.rb in stache-1.0.0

@@ -36,6 +36,28 @@
       }.should raise_error(NameError, "uninitialized constant Profiles::Index::Foo")
 
       Object.send(:remove_const, :Profiles)
     end
   end
+
+  describe "#mustache_class_from_template with config module wrapper set" do
+    before do
+      Stache.wrapper_module_name = "Wrapper"
+    end
+
+    it "returns the appropriate mustache class" do
+      module Wrapper; class HelloWorld < ::Stache::Mustache::View; end; end
+      @handler.mustache_class_from_template(@template).should == Wrapper::HelloWorld
+      Object.send(:remove_const, :Wrapper)
+    end
+    it "is clever about folders and such" do
+      @template.stub!(:virtual_path).and_return("profiles/index")
+      module Wrapper; module Profiles; class Index < ::Stache::Mustache::View; end; end; end
+      @handler.mustache_class_from_template(@template).should == Wrapper::Profiles::Index
+      Object.send(:remove_const, :Wrapper)
+    end
+
+    after do
+      Stache.wrapper_module_name = nil
+    end
+  end
 end