Fix key format propagation to children in template.

Author: rolftimmermans

lib/jbuilder_template.rb

@@ -3,9 +3,9 @@ def self.encode(context)
     new(context)._tap { |jbuilder| yield jbuilder }.target!
   end
 
-  def initialize(context)
+  def initialize(context, *args)
     @context = context
-    super()
+    super(*args)
   end
 
   def partial!(options, locals = {})
@@ -21,7 +21,7 @@ def partial!(options, locals = {})
 
   private
     def _new_instance
-      __class__.new(@context)
+      __class__.new(@context, @key_formatter)
     end
 end
 

test/jbuilder_template_test.rb

@@ -15,4 +15,25 @@ class JbuilderTemplateTest < ActiveSupport::TestCase
 
     assert_equal "hello", JSON.parse(json)["content"]
   end
+
+  test "key_format! with parameter" do
+    json = JbuilderTemplate.new(binding)
+    json.key_format! :camelize => [:lower]
+    json.camel_style "for JS"
+
+    assert_equal ['camelStyle'], json.attributes!.keys
+  end
+
+  test "key_format! propagates to child elements" do
+    json = JbuilderTemplate.new(binding)
+    json.key_format! :upcase
+    json.level1 "one"
+    json.level2 do |json|
+      json.value "two"
+    end
+
+    result = json.attributes!
+    assert_equal "one", result["LEVEL1"]
+    assert_equal "two", result["LEVEL2"]["VALUE"]
+  end
 end