spec/seeder_spec.rb in seeder-1.0 vs spec/seeder_spec.rb in seeder-1.1.0

@@ -74,14 +74,16 @@
       expect(seeder).to receive(:create_new_records).ordered
       seeder.create
     end
 
     it 'aborts when an exception is raised' do
-      expect(seeder).to receive(:create_new_records).and_raise
+      allow(seeder)
+        .to receive(:create_new_records)
+        .and_raise(ActiveRecord::RecordInvalid.new(Grade.new))
       initial_attributes = Grade.all.map(&:attributes)
 
-      expect { seeder.create }.to raise_error
+      expect { seeder.create }.to raise_error(ActiveRecord::RecordInvalid)
 
       expect(Grade.all.map(&:attributes)).to eq(initial_attributes)
     end
 
     it 'produces the appropriate results' do
@@ -89,10 +91,10 @@
       grade2 = Grade.create!(student_id: 1, course_id: 1)
 
       seeder.create
 
       expect(Grade.count).to eq(2)
-      expect(Grade.exists?(grade1)).to eq(false)
+      expect(Grade.exists?(grade1.id)).to eq(false)
 
       grade2.reload
       expect(grade2.student_id).to eq(1)
       expect(grade2.course_id).to eq(1)
       expect(grade2.grade).to eq(90)