Model CleanUp add end_date

Refs #1934

1 files changed, 7 insertions, 6 deletions

diff --git a/app/models/clean_up.rb b/app/models/clean_up.rb
index 607362682..a44bb46a4 100644
--- a/app/models/clean_up.rb
+++ b/app/models/clean_up.rb
@@ -3,7 +3,7 @@ class CleanUp < ActiveRecord::Base
   belongs_to :referential
   has_one :clean_up_result
 
-  validates :expected_date, presence: true
+  validates :begin_date, presence: true
   after_commit :perform_cleanup, :on => :create
 
   def perform_cleanup
@@ -12,15 +12,16 @@ class CleanUp < ActiveRecord::Base
 
   def clean
     result = {}
-    tms = Chouette::TimeTable.validity_out_from_on?(expected_date)
-    tms.each.map(&:delete)
-
-    result['time_table_count']      = tms.size
+    result['time_table_count']      = self.clean_time_tables
     result['vehicle_journey_count'] = self.clean_vehicle_journeys
     result['journey_pattern_count'] = self.clean_journey_patterns
     result
   end
 
+  def clean_time_tables
+    Chouette::TimeTable.validity_out_between?(begin_date, end_date).delete_all
+  end
+
   def clean_vehicle_journeys
     ids = Chouette::VehicleJourney.includes(:time_tables).where(:time_tables => {id: nil}).pluck(:id)
     Chouette::VehicleJourney.where(id: ids).delete_all
@@ -61,6 +62,6 @@ class CleanUp < ActiveRecord::Base
 
   def log_failed message_attributs
     update_attribute(:ended_at, Time.now)
-    # self.clean_up_result.create(message_key: :failed, message_attributs: message_attributs)
+    CleanUpResult.create(clean_up: self, message_key: :failed, message_attributs: message_attributs)
   end
 end