1 files changed, 29 insertions, 3 deletions
diff --git a/spec/models/vehicle_translation_spec.rb b/spec/models/vehicle_translation_spec.rb
index 47be9ff3a..bc16f2260 100644
--- a/spec/models/vehicle_translation_spec.rb
+++ b/spec/models/vehicle_translation_spec.rb
@@ -13,7 +13,32 @@ describe VehicleTranslation do
                                   )}
   subject {Factory.build(:vehicle_translation,
                          :vehicle_journey_id => vehicle_journey.id,
-                         :first_stop_departure_time => "12:00")}
+                         :first_stop_time => "12:00",
+                         :departure_or_arrival => "departure",
+                         :duration => 9,
+                         :count => 2)}
+
+  describe "#first_stop_time=" do
+    context "when setting invalid value" do
+      it "should have an error on first_stop_departure_time" do
+        subject.first_stop_time = "dummy"
+        subject.valid?
+        subject.errors[ :first_stop_time].should_not be_nil
+      end
+    end
+
+  end
+  describe "#evaluate_delta" do
+    it "should return 3600 seconds" do
+      subject.evaluate_delta( Time.parse( "11:00")).should == 3600.0
+    end
+  end
+  describe "#first_delta" do
+    it "should return 3600 seconds" do
+      subject.should_receive( :first_vjas_time).and_return( Time.parse( "11:00"))
+      subject.first_delta
+    end
+  end
 
   describe "#translate" do
     it "should add new vehicle" do
@@ -51,10 +76,11 @@ describe VehicleTranslation do
     end
     it "should add vehicle where vehicle_journey_at_stops are translated with #duration" do
       read_vehicle = Chouette::VehicleJourney.find(vehicle_journey.id)  # read from bd, change time values
+      delta = subject.first_delta
       subject.translate
       last_created_vehicle.vehicle_journey_at_stops.each_with_index do |vjas, index|
-        vjas.departure_time.should == (read_vehicle.vehicle_journey_at_stops[index].departure_time + subject.duration.minutes)
-        vjas.arrival_time.should == (read_vehicle.vehicle_journey_at_stops[index].arrival_time + subject.duration.minutes)
+        vjas.departure_time.should == (read_vehicle.vehicle_journey_at_stops[index].departure_time + delta + subject.duration.minutes)
+        vjas.arrival_time.should == (read_vehicle.vehicle_journey_at_stops[index].arrival_time + delta + subject.duration.minutes)
       end
     end
   end