ensure solves are removed when averages are removed

justinj · Oct 24, 2013 · fa9a468 · fa9a468
1 parent 99d3f61
commit fa9a468
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/lib/model/average.rb b/lib/model/average.rb
@@ -8,6 +8,10 @@ class Average < Sequel::Model
     many_to_one :competition
     many_to_many :tags
 
+    def before_destroy
+      solves.each { |solve| solve.destroy }
+      super
+    end
 
     def initialize(args={})
       args[:visible] = false if args[:visible].nil?

diff --git a/routes/average.rb b/routes/average.rb
@@ -42,7 +42,6 @@
 get "/average/delete_confirm/:id" do
   authenticate!
   @average = RCDB::Average.first(id: params[:id])
-  @average.remove_all_solves
   @average.destroy
   redirect "/average"
 end