Add the date fields in Entity.

Author: hainesr

lib/cff/entity.rb

@@ -25,6 +25,8 @@ class Entity < ModelPart
       'city',
       'country',
       'email',
+      'date-end',
+      'date-start',
       'fax',
       'location',
       'name',
@@ -48,5 +50,31 @@ def initialize(param)
       end
     end
 
+    # :call-seq:
+    #   date_end = date
+    #
+    # Set the `date-end` field. If a non-Date object is passed in it will
+    # be parsed into a Date.
+    def date_end=(date)
+      unless Date === date
+        date = Date.parse(date)
+      end
+
+      @fields['date-end'] = date
+    end
+
+    # :call-seq:
+    #   date_start = date
+    #
+    # Set the `date-start` field. If a non-Date object is passed in it will
+    # be parsed into a Date.
+    def date_start=(date)
+      unless Date === date
+        date = Date.parse(date)
+      end
+
+      @fields['date-start'] = date
+    end
+
   end
 end

test/cff_entity_test.rb

@@ -57,6 +57,20 @@ def test_simple_fields_set_and_output_correctly
     end
   end
 
+  def test_date_fields_set_and_output_correctly
+    date = Date.today
+    @entity.date_end = date
+    @entity.date_start = date
+
+    assert_equal @entity.date_end, date
+    assert_equal @entity.date_start, date
+
+    y = @entity.fields.to_yaml
+
+    assert y.include? "date-end: #{date.to_s}\n"
+    assert y.include? "date-start: #{date.to_s}\n"
+  end
+
   def test_tel_fax_fields_set_and_output_correctly
     number = "+44 (0) 161-234-5678"
     @entity.fax = number