minitest-stackprof 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a5af0b46c8d4b91058e0f63bc62768f8811c9dee
-  data.tar.gz: 2ae770ad77a7001a88aeea12224f72c05646c487
+  metadata.gz: d37f637d5bc432deb02f2008a81953bf83ec2d87
+  data.tar.gz: 75f32dd7afd597f563f19bfb8e77e83f0164eb5c
 SHA512:
-  metadata.gz: a5030bed3b726adbca52f604dc970e315706ebf1b72002fbbf7c79d1d2ca7b325a3c94132602191b6380dac53ea0b532fc68f084d107fa4cb684761964f0bc2b
-  data.tar.gz: e6108ef0c873b2eea6934666fe5a728b523a08753a51808fe789a5ba8e7f696b08b9e1a03adf646411f8903a7fe399319a222a05f9d181d0f9e6f417648cf41e
+  metadata.gz: 4e74e10f559bf7193c71373ad2d3f96dd2352594a6da860bad1a7471da6bd83fd79cfc5395154c731d223dcfc9cd2ee74b471653511b5ca63ced5f41903ecd09
+  data.tar.gz: 8eb51fcbc0a9109a48c52bc08f380d1d818092eb6a43ba20f8751f762fd142e2e0d4e41139590897d02089cb49692cbcd54863a0686e63b9ea5e18a23e60aa77

data/lib/minitest/stackprof/version.rb

@@ -1,5 +1,5 @@
 module Minitest
     module Stackprof
-        VERSION = "0.1.0"
+        VERSION = "0.2.0"
     end
 end

data/lib/minitest/stackprof_plugin.rb

@@ -1,13 +1,41 @@
 module Minitest
     module StackProfPlugin
+        @include_setup_and_teardown = false
+
+        def self.include_setup_and_teardown?
+            @include_setup_and_teardown
+        end
+
+        def self.include_setup_and_teardown=(flag)
+            @include_setup_and_teardown = !!flag
+        end
+
         def before_setup
-            StackProf.start
+            if StackProfPlugin.include_setup_and_teardown?
+                StackProf.start(raw: true)
+            end
+            super
+        end
+
+        def after_setup
+            super
+            if !StackProfPlugin.include_setup_and_teardown?
+                StackProf.start(raw: true)
+            end
+        end
+
+        def before_teardown
             super
+            if !StackProfPlugin.include_setup_and_teardown?
+                StackProf.stop
+            end
         end
 
         def after_teardown
             super
-            StackProf.stop
+            if StackProfPlugin.include_setup_and_teardown?
+                StackProf.stop
+            end
         end
     end
 
@@ -24,14 +52,18 @@ module Minitest
     end
 
     def self.plugin_stackprof_options(opts, options)
-        opts.on '--stackprof[=FILE]', 'run under stackprof (optionally specifies the mode)' do |mode|
-            options[:stackprof] = mode || "stackprof-minitest-#{Process.pid}.dump"
+        opts.on '--profile-setup', 'profile the setup and teardown as well as the test' do
+            StackProfPlugin.include_setup_and_teardown = true
+        end
+        opts.on '--stackprof[=FILE]', String, 'run under stackprof (optionally specifies the output file)' do |file|
+            options[:stackprof] = file || "stackprof-minitest-#{Process.pid}.dump"
         end
     end
     def self.plugin_stackprof_init(options)
         if output_file = options[:stackprof]
             require 'stackprof'
             puts "Stackprof enabled - dumping results to #{output_file}"
+            puts "Stackprof: #{"NOT " if !StackProfPlugin.include_setup_and_teardown?} profiling the setup/teardown"
             Minitest::Test.prepend StackProfPlugin
             reporter << StackProfReporter.new(output_file)
         end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: minitest-stackprof
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Sylvain Joyeux
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2015-09-23 00:00:00.000000000 Z
+date: 2016-04-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: minitest