Custom dump path

Gemfile.lock

@@ -6,6 +6,9 @@ PATH
 GEM
   remote: https://rubygems.org/
   specs:
+    metaclass (0.0.1)
+    mocha (0.14.0)
+      metaclass (~> 0.0.1)
     rake (10.1.0)
     rake-compiler (0.9.1)
       rake
@@ -14,5 +17,6 @@ PLATFORMS
   ruby
 
 DEPENDENCIES
+  mocha
   rake-compiler
   stackprof!

lib/stackprof/middleware.rb

@@ -3,12 +3,14 @@
 module StackProf
   class Middleware
     def initialize(app, options = {})
-      @app = app
-      @options = options
-      @num_reqs = options[:save_every] || nil
-      Middleware.mode = options[:mode] || :cpu
+      @app       = app
+      @options   = options
+      @num_reqs  = options[:save_every] || nil
+
+      Middleware.mode     = options[:mode] || :cpu
       Middleware.interval = options[:interval] || 1000
-      Middleware.enabled = options[:enabled]
+      Middleware.enabled  = options[:enabled]
+      Middleware.path     = options[:path] || 'tmp'
       at_exit{ Middleware.save? } if options[:save_at_exit]
     end
 
@@ -26,17 +28,19 @@ def call(env)
     end
 
     class << self
-      attr_accessor :enabled, :mode, :interval
+      attr_accessor :enabled, :mode, :interval, :path
       alias enabled? enabled
 
       def save
         if results = StackProf.results
-          FileUtils.mkdir_p('tmp')
-          File.open("tmp/stackprof-#{results[:mode]}-#{Process.pid}-#{Time.now.to_i}.dump", 'wb') do |f|
+          FileUtils.mkdir_p(Middleware.path)
+          filename = "stackprof-#{results[:mode]}-#{Process.pid}-#{Time.now.to_i}.dump"
+          File.open(File.join(Middleware.path, filename), 'wb') do |f| 
             f.write Marshal.dump(results)
           end
         end
       end
+
     end
   end
 end

stackprof.gemspec

@@ -18,4 +18,5 @@ Gem::Specification.new do |s|
   s.license = 'MIT'
 
   s.add_development_dependency 'rake-compiler'
+  s.add_development_dependency 'mocha'
 end

test/test_middleware.rb

@@ -0,0 +1,41 @@
+$:.unshift File.expand_path('../../lib', __FILE__)
+require 'stackprof'
+require 'stackprof/middleware'
+require 'test/unit'
+require 'mocha/setup'
+
+class StackProf::MiddlewareTest < Test::Unit::TestCase
+
+  def test_path_default
+    StackProf::Middleware.new(Object.new)
+
+    assert_equal 'tmp', StackProf::Middleware.path
+  end
+
+  def test_path_custom
+    StackProf::Middleware.new(Object.new, { path: '/foo' })
+
+    assert_equal '/foo', StackProf::Middleware.path
+  end
+
+  def test_save_default
+    StackProf::Middleware.new(Object.new)
+
+    StackProf.stubs(:results).returns({ mode: 'foo' })
+    FileUtils.expects(:mkdir_p).with('tmp')
+    File.expects(:open).with(regexp_matches(/^tmp\/stackprof-foo/), 'wb')
+
+    StackProf::Middleware.save
+  end
+
+  def test_save_custom
+    StackProf::Middleware.new(Object.new, { path: '/foo' })
+
+    StackProf.stubs(:results).returns({ mode: 'foo' })
+    FileUtils.expects(:mkdir_p).with('/foo')
+    File.expects(:open).with(regexp_matches(/^\/foo\/stackprof-foo/), 'wb')
+
+    StackProf::Middleware.save
+  end
+
+end