Update vehicles_controller.rb

app/controllers/vehicles_controller.rb

@@ -41,17 +41,21 @@ class VehiclesController < ApplicationController
   
   # display a specific vehicle
   def show
-    @vehicle = Vehicle.find_by_id(params[:id])
+    begin
-    if not @vehicle
+      @vehicle = Vehicle.find_by_id(params[:id])
-      flash[:error] = "Vehicle Not Found"
+    rescue ActiveRecord::RecordNotFound
-      redirect_to :index
+      render_404
     end
   end
   
   # return an HTML form for editing a vehicle
   def edit
-    @vehicle = Vehicle.find_by_id(params[:id])
+    begin
-    @customers = Customer.all.order(:name)
+      @vehicle = Vehicle.find_by_id(params[:id])
+      @customers = Customer.all.order(:name)
+    rescue ActiveRecord::RecordNotFound
+      render_404
+    end
   end
   
   # update a specific vehicle
@@ -72,10 +76,10 @@ class VehiclesController < ApplicationController
     begin
       Vehicle.find_by_id(params[:id]).destroy
       flash[:notice] = "Vehicle deleted successfully"
-    rescue
+      redirect_to action: :index
-      flash[:error] = "No Vehicle Found"
+    rescue ActiveRecord::RecordNotFound
+      render_404
     end
-    redirect_to action: :index
   end
 
 end