Merge pull request #83 from osyoyu/ensure-cleanup

Ensure @@session is cleaned up

Author: osyoyu

lib/pf2.rb

@@ -30,5 +30,10 @@ def self.profile(**kwargs, &block)
     result = stop
     @@session = nil # let GC clean up the session
     result
+  ensure
+    if defined?(@@session) && @@session != nil
+      stop
+      @@session = nil
+    end
   end
 end

test/pf2_test.rb

@@ -39,4 +39,16 @@ def test_profile_passes_custom_options_to_start
   def test_profile_raises_error_when_called_without_block
     assert_raises(ArgumentError, "block required") { Pf2.profile }
   end
+
+  def test_profile_stops_session_even_if_given_block_raises
+    begin
+      Pf2.profile do
+        raise
+      end
+    rescue
+      # nice catch, do nothing
+    end
+
+    assert_nil Pf2.class_variable_get(:@@session)
+  end
 end