Update VehicleJourney checksum in factory when stops are created. Refs #5417

2 files changed, 8 insertions, 0 deletions

diff --git a/app/models/concerns/checksum_support.rb b/app/models/concerns/checksum_support.rb
index c95e23bcf..b700ef286 100644
--- a/app/models/concerns/checksum_support.rb
+++ b/app/models/concerns/checksum_support.rb
@@ -26,4 +26,11 @@ module ChecksumSupport
       self.checksum = Digest::SHA256.new.hexdigest(self.checksum_source)
     end
   end
+
+  def update_checksum!
+    set_current_checksum_source
+    if checksum_source_changed?
+      update checksum: Digest::SHA256.new.hexdigest(checksum_source)
+    end
+  end
 end
diff --git a/spec/factories/chouette_vehicle_journey.rb b/spec/factories/chouette_vehicle_journey.rb
index 5f64bd502..7d63a2e58 100644
--- a/spec/factories/chouette_vehicle_journey.rb
+++ b/spec/factories/chouette_vehicle_journey.rb
@@ -30,6 +30,7 @@ FactoryGirl.define do
                    :arrival_time    => "2000-01-01 #{arrival_time} UTC",
                    :departure_time  => "2000-01-01 #{departure_time} UTC")
           end
+          vehicle_journey.update_checksum!
         end
 
         factory :vehicle_journey_odd do