Fix testcases and also fix errors the testcase brought up

[selectricity-live] / app / models / election.rb

diff --git a/app/models/election.rb b/app/models/election.rb
index 59a10bfbe601387eac1b72de2c9fcec03c99560c..9c6337fdcf56ff78d7ee1d63a04ba1bfedf23bf0 100644 (file)
--- a/app/models/election.rb
+++ b/app/models/election.rb
@@ -15,6 +15,12 @@ class Election < ActiveRecord::Base
   
   require 'date'
   
+  def initialize(params={})
+    super
+    self.enddate = read_attribute( :enddate ) || \
+                   Time.now + 30.days - 1.second
+  end
+
   def other_methods
     if election_method
       @other_methods = ELECTION_TYPES.reject {|i| i == election_method}
@@ -27,17 +33,6 @@ class Election < ActiveRecord::Base
   def startdate
     read_attribute( :startdate ) || Time.now
   end
-  
-  def enddate
-    date = read_attribute( :enddate ) || Time.now + 14
-    date - 1.second
-  end
-
-  def enddate=(date)
-    date += 1.day
-    date = Time.gm(*date)
-    super(date)
-  end
 
   def votes
     votes = Array.new
@@ -56,7 +51,6 @@ class Election < ActiveRecord::Base
 
   def start_blockers
     reasons = []
-    
     if self.candidates.length <= 1
       reasons << "You must have at least two candidates."
     end