perf_check 0.3.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: d80f52906352baede601135b143828f3b1983985
-  data.tar.gz: 3b1300b56aebd3733cd2d9bb15addcfea6f6b9ce
+  metadata.gz: f8cdfcd06c215cfa05e4267afa28abf9f8d8efba
+  data.tar.gz: a895cac67e5622ea31c997239f7371aac08fce8a
 SHA512:
-  metadata.gz: 75e17dca3e1e25fae73c741a3263b5247859ea9276d356d2f73f9511e82c3dd301bd8247f56eff59e4c4d6fbf1132115712073b4da66902985b87438df29f856
-  data.tar.gz: 4c04b113dab60e0a3b634b7c5cf7020e34369bc93bd47f4648eb3546e97b48b45deb8b43c7f32adbc2415ed59d3973d92ad94b6788bcbc032a0007c0d804af9b
+  metadata.gz: a9c4c040e1d5e27d2f21b59f527efe287ebdfdacd794c447ada57c08c85c725406695f6759c2a305474875688fb7f6c6f13c4f0e10db4fc2094cc9e984564ea6
+  data.tar.gz: eba0557a4d74c324892cf01a9a4c2a5abbf6f6b0c7cb78df3be9289f1d7fa0342900c43569b101d55d8b9e09b31ca661856275817794ac0daa06588377bb6dbc

data/lib/perf_check/config.rb

@@ -44,6 +44,10 @@ class PerfCheck
       config.caching = false
     end
 
+    opts.on('--run-migrations', 'Run migrations on the branch and unmigrate at the end') do
+      config[:run_migrations?] = true
+    end
+
     opts.on('--fail-fast', '-f', 'Bail immediately on non-200 HTTP response') do
       config[:fail_fast?] = true
     end

data/lib/perf_check/git.rb

@@ -67,5 +67,18 @@ class PerfCheck
         logger.fatal("Problem with git stash! Bailing...") && abort
       end
     end
+
+    def self.migrations_to_run_down
+      current_migrations_not_on_master.map { |filename| File.basename(filename, '.rb').split('_').first }
+    end
+
+    def self.current_migrations_not_on_master
+      %x{git diff origin/master --name-only --diff-filter=A db/migrate/}.split.reverse
+    end
+    private_class_method :current_migrations_not_on_master
+
+    def self.clean_db
+      `git checkout db`
+    end
   end
 end

data/lib/perf_check.rb

@@ -49,6 +49,8 @@ class PerfCheck
   private
 
   def profile_requests
+    run_migrations_up if options.run_migrations?
+
     server.restart
     test_cases.each_with_index do |test, i|
       trigger_before_start_callbacks(test)
@@ -65,6 +67,20 @@ class PerfCheck
 
       test.run(server, options)
     end
+  ensure
+    run_migrations_down if options.run_migrations?
+  end
+
+  def run_migrations_up
+    Bundler.with_clean_env{ puts `bundle exec rake db:migrate` }
+    Git.clean_db
+  end
+
+  def run_migrations_down
+    Git.migrations_to_run_down.each do |version|
+      Bundler.with_clean_env{ puts `bundle exec rake db:migrate:down VERSION=#{version}` }
+    end
+    Git.clean_db
   end
 end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: perf_check
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - rubytune
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-07-02 00:00:00.000000000 Z
+date: 2016-01-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: colorize
@@ -76,7 +76,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.2.2
+rubygems_version: 2.4.5
 signing_key: 
 specification_version: 4
 summary: PERF CHECKKK!