spec/lib/flapjack/gateways/api/entity_methods_spec.rb in flapjack-0.7.17 vs spec/lib/flapjack/gateways/api/entity_methods_spec.rb in flapjack-0.7.18

@@ -411,11 +411,11 @@
       last_response.status.should == 403
     end
 
     it "deletes a scheduled maintenance period for an entity check" do
       start_time = Time.now + (60 * 60) # an hour from now
-      entity_check.should_receive(:delete_scheduled_maintenance).with(:start_time => start_time.to_i)
+      entity_check.should_receive(:end_scheduled_maintenance).with(start_time.to_i)
 
       Flapjack::Data::EntityCheck.should_receive(:for_entity).
         with(entity, check, :redis => redis).and_return(entity_check)
 
       Flapjack::Data::Entity.should_receive(:find_by_name).
@@ -424,22 +424,22 @@
       delete "/scheduled_maintenances", :check => {entity_name => check}, :start_time => start_time.iso8601
       last_response.status.should == 204
     end
 
     it "doesn't delete a scheduled maintenance period if the start time isn't passed" do
-      entity_check.should_not_receive(:delete_scheduled_maintenance)
+      entity_check.should_not_receive(:end_scheduled_maintenance)
 
       delete "/scheduled_maintenances", :check => {entity_name => check}
       last_response.status.should == 403
     end
 
     it "deletes scheduled maintenance periods for multiple entity checks" do
       start_time = Time.now + (60 * 60) # an hour from now
 
       entity_check_2 = mock(Flapjack::Data::EntityCheck)
 
-      entity_check.should_receive(:delete_scheduled_maintenance).with(:start_time => start_time.to_i)
-      entity_check_2.should_receive(:delete_scheduled_maintenance).with(:start_time => start_time.to_i)
+      entity_check.should_receive(:end_scheduled_maintenance).with(start_time.to_i)
+      entity_check_2.should_receive(:end_scheduled_maintenance).with(start_time.to_i)
 
       Flapjack::Data::EntityCheck.should_receive(:for_entity).
         with(entity, check, :redis => redis).and_return(entity_check)
       Flapjack::Data::EntityCheck.should_receive(:for_entity).
         with(entity, 'foo', :redis => redis).and_return(entity_check_2)