rabl-rails/test/deep_nesting_test.rb

require 'test_helper'

class DeepNestingTest < ActiveSupport::TestCase

  class Post
    attr_accessor :id, :title

    def initialize(id, title)
      @id, @title = id, title
    end

    def comments
      [Struct.new(:id, :content).new(1, 'first'), Struct.new(:id, :content).new(2, 'second')]
    end
  end

  setup do
    RablRails::Library.reset_instance
    @post = Post.new(42, 'I rock !')
    @user = User.new(1, 'foobar', 'male')
    @user.stub(:posts).and_return([@post])
    @user.stub(:respond_to?).with(:each).and_return(false)

    @context = Context.new
    @context.stub(:instance_variable_get).with(:@user).and_return(@user)
    @context.stub(:instance_variable_get).with(:@virtual_path).and_return('users/show')
    @context.stub(:instance_variable_get).with(:@_assigns).and_return({})
    @context.stub(:lookup_context).and_return(mock(:find_template => mock(:source => %{ object :@comment\n attribute :content })))
  end

  test "compile and render deep nesting template" do
    source = %{
      object :@user
      attributes :id, :name
      child :posts do
        attribute :title
        child :comments do
          extends 'comments/show'
        end
      end
    }

    assert_equal(MultiJson.encode(:user => {
      :id => 1,
      :name => 'foobar',
      :posts => [{
        :title => 'I rock !',
        :comments => [
          { :content => 'first' },
          { :content => 'second' }
        ]
      }]
    }), RablRails::Library.instance.get_rendered_template(source, @context))
  end
end