[#4177] Added generators for Contract.

2010-06-17 15:16:11 -07:00
parent 309a6193d6
commit a5ef5f5999
1 changed files with 14 additions and 0 deletions
--- a/app/models/contract.rb
+++ b/app/models/contract.rb
@@ -39,4 +39,18 @@ class Contract < ActiveRecord::Base
      errors.add :end_date, :greater_than_start_date
    end
  end
+
+  if Rails.env.test?
+    generator_for :name, :method => :next_name
+    generator_for :executed => true
+    generator_for(:start_date) { Date.yesterday }
+    generator_for(:end_date) { Date.tomorrow }
+
+    def self.next_name
+      @last_name ||= 'Contract 0000'
+      @last_name.succ!
+    end
+
+  end
+  
 end