require 'spec_helper'

describe "NestedAttribute behavior" do
  before do
    class Bar < ActiveFedora::Base
      belongs_to :car, :property=>:has_member
      has_metadata :type=>ActiveFedora::SimpleDatastream, :name=>"someData" do |m|
        m.field "uno", :string
        m.field "dos", :string
      end
      delegate :uno, to: 'someData', multiple: false
      delegate :dos, to: 'someData', multiple: false
    end

    # base Car class, used in test for association updates and :allow_destroy flag
    class Car < ActiveFedora::Base
      has_many :bars, :property=>:has_member
      accepts_nested_attributes_for :bars, :allow_destroy=>true
    end

    # class used in test for :reject_if=>:all_blank
    class CarAllBlank < Car
      accepts_nested_attributes_for :bars, :reject_if=>:all_blank
    end

    # class used in test for :reject_if with proc object
    class CarProc < Car
      accepts_nested_attributes_for :bars, :reject_if=>proc { |attributes| attributes['uno'].blank? }
    end

    # class used in test for :reject_if with method name as symbol
    class CarSymbol < Car
      accepts_nested_attributes_for :bars, :reject_if=>:uno_blank

      def uno_blank(attributes)
        attributes['uno'].blank?
      end
    end

    # class used in test for :limit
    class CarWithLimit < Car
      accepts_nested_attributes_for :bars, :limit => 1
    end
  end

  it "should update the child objects" do
    @car, @bar1, @bar2 = create_car_with_bars

    @car.attributes = {:bars_attributes=>[{:id=>@bar1.pid, :uno=>"bar1 uno"}, {:uno=>"newbar uno"}, {:id=>@bar2.pid, :_destroy=>'1', :uno=>'bar2 uno'}]}
    Bar.find(@bar1.pid).uno.should == 'bar1 uno'
    Bar.where(:id => @bar2.pid).first.should be_nil
    Bar.where(:uno => "newbar uno").first.should_not be_nil

    bars = @car.bars(true)
    bars.should include(@bar1)
    bars.should_not include(@bar2)
  end

  it "should reject attributes when all blank" do
    @car, @bar1, @bar2 = create_car_with_bars(CarAllBlank)

    @car.bars.count.should == 2
    @car.attributes = {:bars_attributes=>[{}, {:id=>@bar1.pid, :uno=>"bar1 uno"}]}
    @car.bars(true).count.should == 2

    @bar1.reload
    @bar1.uno.should == "bar1 uno"
  end

  it "should reject attributes based on proc" do
    @car, @bar1, @bar2 = create_car_with_bars(CarProc)

    @car.attributes = {:bars_attributes=>[{}, {:id=>@bar1.pid, :uno=>"bar1 uno"}, {:id=>@bar2.pid, :dos=>"bar2 dos"}]}
    @bar1.reload
    @bar2.reload
    @bar1.uno.should == "bar1 uno"
    @bar2.dos.should be_nil
  end

  it "should reject attributes base on method name" do
    @car, @bar1, @bar2 = create_car_with_bars(CarSymbol)

    @car.attributes = {:bars_attributes=>[{}, {:id=>@bar1.pid, :uno=>"bar1 uno"}, {:id=>@bar2.pid, :dos=>"bar2 dos"}]}
    @bar1.reload
    @bar2.reload
    @bar1.uno.should == "bar1 uno"
    @bar2.dos.should be_nil
  end

  it "should throw TooManyRecords" do
    @car, @bar1, @bar2 = create_car_with_bars(CarWithLimit)

    lambda {
      @car.attributes = {:bars_attributes=>[{}]}
    }.should_not raise_exception

    lambda {
      @car.attributes = {:bars_attributes=>[{}, {}]}
    }.should raise_exception(ActiveFedora::NestedAttributes::TooManyRecords)
  end

  private

  # Helper method used to create 1 Car and 2 Bars (with option to provide classes for both models)
  #
  # @param car_class [class] class for new `car` object, default Car
  # @param bar_class [class] class for new `bar` object, default Bar
  #
  # @return [car,bar,bar] returns 1 Car and 2 Bars
  def create_car_with_bars(car_class = Car, bar_class = Bar)
    car = car_class.new; car.save

    bar1 = bar_class.new(:car=>car); bar1.save
    bar2 = bar_class.new(:car=>car); bar2.save
    [car, bar1, bar2]
  end

end