VehicleJourneyAtStop implementation of checksum concerns

2 files changed, 17 insertions, 0 deletions

diff --git a/app/models/chouette/time_table_date.rb b/app/models/chouette/time_table_date.rb
index b881c9a5d..2aa86a14e 100644
--- a/app/models/chouette/time_table_date.rb
+++ b/app/models/chouette/time_table_date.rb
@@ -1,4 +1,6 @@
 class Chouette::TimeTableDate < Chouette::ActiveRecord
+  include ChecksumSupport
+
   self.primary_key = "id"
   belongs_to :time_table, inverse_of: :dates
   acts_as_list :scope => 'time_table_id = #{time_table_id}',:top_of_list => 0
@@ -12,5 +14,11 @@ class Chouette::TimeTableDate < Chouette::ActiveRecord
     ActiveModel::Name.new Chouette::TimeTableDate, Chouette, "TimeTableDate"
   end
 
+  def checksum_attributes
+    [].tap do |attrs|
+      attrs << self.date.to_s
+      attrs << self.in_out.to_s
+    end
+  end
 end
 
diff --git a/app/models/chouette/vehicle_journey_at_stop.rb b/app/models/chouette/vehicle_journey_at_stop.rb
index 4518d09cf..090a3251e 100644
--- a/app/models/chouette/vehicle_journey_at_stop.rb
+++ b/app/models/chouette/vehicle_journey_at_stop.rb
@@ -25,6 +25,15 @@ module Chouette
       end
     end
 
+    def checksum_attributes
+      [].tap do |attrs|
+        attrs << self.departure_time.try(:to_s, :time)
+        attrs << self.arrival_time.try(:to_s, :time)
+        attrs << self.departure_day_offset.to_s
+        attrs << self.arrival_day_offset.to_s
+      end
+    end
+
     after_initialize :set_virtual_attributes
     def set_virtual_attributes
       @_destroy = false