custom rollback references

Author: borisnadion

Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    deb (0.0.8)
+    deb (0.0.10)
       docile
       rails
 

app/models/deb/transaction.rb

@@ -18,12 +18,12 @@ def self.start(&block)
       Docile.dsl_eval(Deb::Builder.new, &block).build
     end
 
-    def rollback!
+    def rollback!(ref = nil)
       tran = self
       self.class.start! do
         tran.debit_items.each { |di| credit(di.account, di.amount) }
         tran.credit_items.each { |ci| debit(ci.account, ci.amount) }
-        reference(tran)
+        reference(ref || tran)
         kind(tran.kind)
         description("Rollback of #{tran.description}")
       end

lib/deb/version.rb

@@ -1,3 +1,3 @@
 module Deb
-  VERSION = "0.0.9"
+  VERSION = "0.0.10"
 end

spec/models/transaction_spec.rb

@@ -131,12 +131,30 @@ module Deb
         @liability.reload.current_balance.should == 7
       end
 
-      describe :rollback do
+      it "should receive custom rollback references" do
+        @transaction.save!
+        transaction2 = Transaction.start do
+          debit @asset, 12
+          credit @revenue, 5
+          credit @liability, 7
+          description "foobar"
+          kind "baz"
+        end
+        transaction2.save!
+        @rollback = @transaction.rollback!(transaction2)
+        @rollback.transactionable.should == transaction2
+      end
+
+      describe :default_rollback do
         before(:each) do
           @transaction.save!
           @rollback = @transaction.rollback!
         end
 
+        it "should keep the reference" do
+          @rollback.transactionable.should == @transaction
+        end
+
         it "should be okay" do
           @rollback.should_not be_new_record
         end