Pop transaction for transactions stack in case of code failure (excep…

…tion)

lib/strokedb/transaction.rb

@@ -38,10 +38,15 @@ def execute
       Thread.current[:strokedb_transactions].push self
 
       @timestamp = LTS.new(store.timestamp.counter,uuid)
+
+      begin
+        result = yield(self)
+      rescue 
+        throw $!
+      ensure
+        Thread.current[:strokedb_transactions].pop
+      end
 
-      result = yield(self)
-
-      Thread.current[:strokedb_transactions].pop
 
       result
     end

spec/lib/strokedb/stores/transaction_spec.rb

@@ -65,6 +65,15 @@
     Document.find(@doc.uuid).should be_nil
   end
 
+  it "should pop transaction on code that raises some exception" do
+    lambda do
+      @txn.execute do |txn|
+        raise Exception
+      end
+    end.should raise_error(Exception)
+    Thread.current[:strokedb_transactions].should be_empty
+  end
+
 
 end