Add tests about compiled templates

lib/rabl-fast-json/compiler.rb

@@ -37,7 +37,6 @@ module RablFastJson
       return unless block_given?
       name = :"_glue#{@glue_count}"
       @glue_count += 1
-      Rails.logger.warn "[RABL] glue called with data = #{data.inspect}"
       _compile_sub_template(name, data, &block)
     end
 

lib/rabl-fast-json/template.rb

@@ -19,12 +19,11 @@ module RablFastJson
     end
 
     def render_resource(data = nil, source = nil)
-      output = {}
-
       data ||= @object
       source ||= @source
 
-      source.each_pair { |key, value|
+      source.inject({}) { |output, current|
+        key, value = current
         out = case value
         when Symbol
           data.send(value) # attributes
@@ -37,22 +36,19 @@ module RablFastJson
             value.each_pair { |k, v|
               output[k] = object.send(v)
             }
-
+            next output
-            next
           else # child
             object.respond_to?(:each) ? render_collection(object, value) : render_resource(object, value)
           end
         end
         output[key] = out
-      }
         output
+      }
     end
 
     def render_collection(collection = nil, source = nil)
-      output = []
       collection ||= @object
-      collection.each { |o| output << render_resource(o, source) }
+      collection.inject([]) { |output, o| output << render_resource(o, source) }
-      output
     end
   end
 end

test/compiled_template_test.rb

@@ -4,9 +4,36 @@ class TestCompiledTemplate < ActiveSupport::TestCase
 
   setup do
     @context = Context.new
-    @user = User.new
+    @data = User.new(1, 'foobar', 'male')
-    @context.set_assign('user', @user)
+    @data.stub(:respond_to?).with(:each).and_return(false)
+    @context.stub(:instance_variable_get).with(:@data).and_return(@data)
     @template = RablFastJson::CompiledTemplate.new
     @template.context = @context
+    @template.data = :@data
+  end
+
+  test "render single object attributes" do
+    @template.source = { :id => :id, :name => :name }
+    assert_equal({ :id => 1, :name => 'foobar'}, @template.render)
+  end
+
+  test "render object as a child" do
+    @template.source = { :author => { :_data => :@data, :name => :name } }
+    assert_equal({ :author => { :name => 'foobar' } }, @template.render)
+  end
+
+  test "render glued attributes from single object" do
+    @template.source = { :_glue0 => { :_data => :@data, :name => :name } }
+    assert_equal({ :name => 'foobar' }, @template.render)
+  end
+
+  test "render collection with attributes" do
+    @data = [User.new(1, 'foo', 'male'), User.new(2, 'bar', 'female')]
+    @context.stub(:instance_variable_get).with(:@data).and_return(@data)
+    @template.source = { :uid => :id, :name => :name, :gender => :sex }
+    assert_equal([
+      { :uid => 1, :name => 'foo', :gender => 'male'},
+      { :uid => 2, :name => 'bar', :gender => 'female'}
+    ], @template.render)
   end
 end

test/compiler_test.rb

@@ -49,7 +49,7 @@ class CompilerTest < ActiveSupport::TestCase
   end
 
   test "glue is compiled as a child but with anonymous name" do
-    t = @compiler.compile_source(%{ glue :@user do attribute :name end })
+    t = @compiler.compile_source(%{ glue(:@user) do attribute :name end })
     assert_equal({ :_glue0 => { :_data => :@user, :name => :name } }, t.source)
   end
 

test/test_helper.rb

@@ -31,6 +31,4 @@ class Context
   end
 end
 
-class User
+User = Struct.new(:id, :name, :sex)
-  attr_accessor :name
-end