Adding specs for deletion/destroy on Digital and DigitalLink

spec/models/digital_spec.rb

@@ -1,4 +1,4 @@
-require File.dirname(__FILE__) + '/../spec_helper'
+require 'spec_helper'
 
 describe Spree::Digital do
 
@@ -12,13 +12,13 @@ describe Spree::Digital do
   end
   
   context "#destroy" do
-    #it "should destroy associated digital_links" do
-    #  digital = FactoryGirl.create(:digital)
-    #  3.times { digital.digital_links.create! :order => FactoryGirl.create(:order) }
-    #  DigitalLink.count.should == 3      
-    #  digital.destroy
-    #  DigitalLink.count.should == 0      
-    #end
+    it "should destroy associated digital_links" do
+     digital = FactoryGirl.create(:digital)
+     3.times { digital.digital_links.create!({ :line_item => FactoryGirl.create(:line_item) }, :without_protection => true) }
+     Spree::DigitalLink.count.should == 3      
+     digital.destroy
+     Spree::DigitalLink.count.should == 0      
+    end
   end
   
 end

spec/models/variant_spec.rb

@@ -0,0 +1,18 @@
+require 'spec_helper'
+
+describe Spree::Variant do
+  before do
+    @product = FactoryGirl.create :product
+    @digital = FactoryGirl.create :digital, :variant => @product.master
+  end
+
+  let(:variant) { @product.master }
+
+  it "should delete all digitals on variant#destroy" do
+    digital_id = variant.digitals.first.id
+    Spree::Digital.find(digital_id).should_not be_nil
+    variant.digitals.count.should == 1
+    variant.destroy
+    expect { Spree::Digital.find(digital_id) }.to raise_error(ActiveRecord::RecordNotFound)
+  end
+end