$:.unshift File.join(File.dirname(__FILE__), '..', '..', 'lib')

require 'test/unit'

require 'buildmaster/tree_to_object'

module BuildMaster
class TreeToObjectTest < Test::Unit::TestCase
  def test_should_populate_string_fields
    content = <<CONTENT
field_one: value_one
field_two: value_two
CONTENT
    tree = YAML.load(content)
    object = SampleObject.new
    TreeToObject.new(tree, object).convert
    assert_equal('value_one', object.field_one)
    assert_equal('value_two', object.field_two)
  end

  def test_should_populate_array_fields
    content = <<CONTENT
array_field:
- field: valueone
  index: 1
- field: valuetwo
  index: 2
CONTENT
    object = TreeToObject.from_yaml(content, SampleObject.new)
    array = object.array_field
    assert_equal(2, array.size)
    assert_equal('valueone', array[0].field)
    assert_equal('1', array[0].index.to_s)
    assert_equal('valuetwo', array[1].field)
    assert_equal('2', array[1].index.to_s)
  end

  def test_should_populate_instance_fields
    content = <<CONTENT
object_two:
  field: my_field
  index: 5
CONTENT
    object = TreeToObject.from_yaml(content, SampleObject.new)
    actual = object.object_two
    assert_equal('my_field', actual.field)
    assert_equal('5', actual.index.to_s)
  end

  def test_should_raise_error_if_property_not_found
    content = <<CONTENT
not_field: value
CONTENT
    begin
      object = TreeToObject.from_yaml(content, SampleObject.new)
      fail('exception should have been thrown')
    rescue PropertyMatchError => error
      assert_equal(true, error.message.include?('not_field'))
      assert_equal(true, error.message.include?('string'))
    end
  end
  
  def test_should_raise_error_if_sub_property_not_found
    content = <<CONTENT
not_sub_property:
  field: value
CONTENT
    begin
      object = TreeToObject.from_yaml(content, SampleObject.new)
      fail('exception should have been thrown')
    rescue PropertyMatchError => error
      assert_equal(true, error.message.include?('not_sub_property'))    
      assert_equal(true, error.message.include?('sub property'))
    end
  end
  
  def test_should_raise_error_if_array_property_not_found
    content = <<CONTENT
not_array_property:
- name: title
- name: title
CONTENT
    begin
      object = TreeToObject.from_yaml(content, SampleObject.new)
      fail('exception should have been thrown')
    rescue PropertyMatchError => error
      assert_equal(true, error.message.include?('not_array_property'))
      assert_equal(true, error.message.include?('array'))
    end
  end
  
  def test_empty_content
    object = TreeToObject.from_yaml('', SampleObject.new)
    assert_equal(nil, object.field_one)
  end

end


class SampleObject
  attr_reader :field_one, :field_two, :object_two, :array_field
  attr_writer :field_one, :field_two
  
  def initialize
    @object_two = SampleTwo.new
    @array_field = Array.new
  end

  def add_to_array_field
    item = SampleTwo.new
    @array_field.push(item)
    return item
  end
end

class SampleTwo
  attr_reader :field, :index
  attr_writer :field, :index
end
end