squasher 0.1.7 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 2bf39c604130d794f42f29a6a71328154c55c60e
-  data.tar.gz: 448ea347111fe23d91f500c43a9f1ecc86047f17
+  metadata.gz: 4e3e107d6711573cf3fc5acd9b237fde3186c5b0
+  data.tar.gz: 0a710e069d472696efd9802c3427b6b2f463af83
 SHA512:
-  metadata.gz: c8960b288cfc369f217f51705a7c8827f3d223343f3c055eafb030a48887657a2dbd90d9bbe50e7b5f2a8a915cd585bb8365e99f6ad2f4b49f6126c0a9b47967
-  data.tar.gz: 05e4a577fb8b698713b1c181f65e8ae3316608acb1e6dbe604eafce00030f423a02960bc632e497365623ad0128baf4366be81706fe013d40c6053be00fb43ac
+  metadata.gz: d4ea404426d7555d6c0564578990be8d76140b1c27b5a96bf11c737252359b45904844df57c4f1ca5bfb8a56816e353ad7702b73b68b213d2fe89f20fd3383ce
+  data.tar.gz: 4eea8c07ec83255821bea140e655fddebc85d03c6c10cff2d51f7a16a44e10031fc6a2487799d81f03869b5788a707dec1e48217e968d0c8b3c659c329ed5714

data/.gitignore

@@ -3,6 +3,7 @@
 /.bundle
 /.config
 /.rspec
+Gemfile.custom
 Gemfile.lock
 InstalledFiles
 _yardoc

data/.travis.yml

@@ -1,9 +1,8 @@
-before_install:
-  - gem update bundler
+cache: bundler
 language: ruby
 rvm:
-  - 2.1.0
-  - 2.0.0
-  - 1.9.3
-  - jruby-19mode
-  - rbx
+  - ruby-head
+  - 2.2
+  - 2.1
+  - 2.0
+  - jruby

data/Gemfile

@@ -3,9 +3,4 @@ source 'https://rubygems.org'
 # Specify your gem's dependencies in squasher.gemspec
 gemspec
 
-platforms :rbx do
-  gem 'json'
-  gem 'racc'
-  gem 'rubysl', '~> 2.0'
-  gem 'psych'
-end
+eval(File.read('Gemfile.custom')) if File.exist?('Gemfile.custom')

data/README.md

@@ -36,12 +36,28 @@ You can tell `squasher` a more detailed date, for example:
     $ squasher 2013/12    #prior to December 2013
     $ squasher 2013/12/19 #prior to 19 December 2013
 
+### Options
+
+`-d` - execute in **dry** mode - test a squashing process without deleting old migrations. The final output will be
+printed in the console.
+
+`-r` - reuse a database from previous squashing process. The option can be used in the next cases:
+  - you've run squasher in **dry** mode before and there were no errors
+  - you're squashing migrations gradually. For example, you want to squash migrations from 2013 till 2015, but the
+process breaks in a migration from 2014. In this situation you squash till 2014, save the squasher's
+database at the end, make a patch in the broken migration and run again the suite with `-r` option. As the result
+squasher will not need to create the db schema and all data from the previous migrations will be there.
+
 ## Requirements
 
-It works and was tested on Ruby 1.9.3+ and Rails 3.1+. It also requires a valid configuration in `config/database.yml` and using Ruby format in `db/schema.rb` (default Rails use-case).
+It works and was tested on Ruby 2.0+ and Rails 3.1+. It also requires a valid configuration in `config/database.yml` and using Ruby format in `db/schema.rb` (default Rails use-case).
 If an old migration inserted data (created ActiveRecord model records) you will lose this code in the squashed migration, **BUT** `squasher` will ask you to leave a tmp database which will have all data that was inserted while migrating. Using this database you could add that data as another migration, or into `config/seed.rb` (the expected place for this stuff).
 
 ## Changelog
+- 0.2.0
+  - add **dry** mode and ability to reuse the previous squasher database
+  - improve database config processing
+  - raise the minimum supported version of Ruby
 - 0.1.7
   - a regression fix of the log output ([@lime](https://github.com/lime))
   - improve a multi-platform support ([@johncarney](https://github.com/johncarney))

data/bin/squasher

@@ -1,9 +1,10 @@
 #!/usr/bin/env ruby
 require 'squasher'
 
-case ARGV[0]
-when 'clean'   then Squasher.clean
-when /\A\d{4}/ then Squasher.squash(ARGV[0])
+if ARGV[0] == 'clean'
+  Squasher.clean
+elsif ARGV[-1] =~ /\A\d{4}/
+  Squasher.squash(ARGV[-1], ARGV[0...-1])
 else
   Squasher.tell(:usage)
 end

data/lib/squasher.rb

@@ -6,10 +6,19 @@ module Squasher
   autoload :Render,  'squasher/render'
   autoload :Worker,  'squasher/worker'
 
-  def squash(raw_date)
+  def squash(raw_date, raw_options)
     parts = raw_date.to_s.split('/').map(&:to_i)
     date = Time.new(*parts)
-    Worker.process(date)
+
+    options = raw_options.map do |o|
+      o = o.gsub('-', '').to_sym
+      unless Worker::OPTIONS.include?(o)
+        tell(:wrong_option, o: o)
+        error(:usage)
+      end
+      o
+    end
+    Worker.process(date, options)
   end
 
   def clean
@@ -28,9 +37,9 @@ module Squasher
 
   def tell(key, options = {})
     message = messages.fetch(key.to_s)
+    message = message.join("\n") if message.is_a?(Array)
     message = colorize(message)
-    message = message % options
-    puts message
+    puts message % options
   end
 
   def error(*args)
@@ -48,10 +57,9 @@ module Squasher
     @messages = JSON.load(File.open(path))
   end
 
+  COLORS = ['red', 'green', 'yellow', 'blue'].each_with_index.inject({}) { |r, (k, i)| r.merge!(k => "03#{ i + 1 }") }
+
   def colorize(message)
-    message.gsub(/\:(\w+)\<([^>]+)\>/) do |match|
-      color_code = { "red" => "031", "green" => "032", "yellow" => "033" }[$1]
-      "\033[#{ color_code }m#{ $2 }\033[039m"
-    end
+    message.gsub(/\:(\w+)\<([^>]+)\>/) { |_| "\033[#{ COLORS[$1] }m#{ $2 }\033[039m" }
   end
 end

data/lib/squasher/config.rb

@@ -4,12 +4,14 @@ require 'erb'
 
 module Squasher
   class Config
+    attr_reader :schema_file
+
     def initialize
-      @root_path = Dir.pwd
-    end
+      root_path = Dir.pwd
 
-    def schema_file
-      @schema_file ||= from_root('db', 'schema.rb')
+      @schema_file = File.join(root_path, 'db', 'schema.rb')
+      @migrations_folder = File.join(root_path, 'db', 'migrate')
+      @dbconfig_file = File.join(root_path, 'config', 'database.yml')
     end
 
     def migration_files
@@ -34,52 +36,38 @@ module Squasher
       list = [dbconfig_file, schema_file]
       list.each do |file|
         next unless File.exists?(file)
-        FileUtils.mv file, "#{ file }.sbackup"
+        FileUtils.mv file, "#{ file }.sq"
       end
-      update_dbconfig_file
+
+      File.open(dbconfig_file, 'wb') { |stream| stream.write dbconfig.to_yaml }
 
       yield
 
     ensure
       list.each do |file|
-        next unless File.exists?("#{ file }.sbackup")
-        FileUtils.mv "#{ file }.sbackup", file
+        next unless File.exists?("#{ file }.sq")
+        FileUtils.mv "#{ file }.sq", file
       end
     end
 
     private
 
-    attr_reader :root_path
-
-    def from_root(*subfolders)
-      File.join(root_path, *subfolders)
-    end
-
-    def migrations_folder
-      @migrations_folder ||= from_root('db', 'migrate')
-    end
-
-    def dbconfig_file
-      @dbconfig_file ||= from_root('config', 'database.yml')
-    end
+    attr_reader :migrations_folder, :dbconfig_file
 
     def dbconfig
       return @dbconfig if defined?(@dbconfig)
       return @dbconfig = nil unless File.exists?(dbconfig_file)
 
+      @dbconfig = nil
+
       begin
-        content = File.read(dbconfig_file).gsub(/database: (.+)/, 'database: squasher')
-        parsed_content = ERB.new(content).result(binding)
-        @dbconfig = YAML.load(parsed_content)
-        @dbconfig = nil unless @dbconfig.keys.include?('development')
+        content = YAML.load(ERB.new(File.read(dbconfig_file)).result(binding))
+        if content.has_key?('development')
+          @dbconfig = { 'development' => content['development'].merge('database' => 'squasher') }
+        end
       rescue
-        @dbconfig = nil
       end
       @dbconfig
     end
-
-    def update_dbconfig_file
-      File.open(dbconfig_file, 'wb') { |stream| stream.write dbconfig.to_yaml }
-    end
   end
 end

data/lib/squasher/messages.json

@@ -1,9 +1,17 @@
 {
-  "keep_database" : "Squasher's created the `:green<squasher>` database for its needs.\nIt might be useful if to keep it if any of your deleted migrations inserted data into database.\nKeep it (:green<yes> / :red<no>)?",
+  "keep_database" : [
+    "Squasher's created the `:green<squasher>` database for its needs.",
+    "It might be useful to keep it if any of your deleted migrations inserts data or",
+    "you squash migrations in few steps (look at -r option).",
+    "Keep it (:green<yes> / :red<no>)?"
+  ],
 
   "apply_clean": "Do you want to clean your database from the old schema migration records(:red<yes>/:green<no>)?",
 
-  "migration_folder_missing" : "The folder with migrations is missing.\nAre you sure that you're in the :red<root of a rails> application?",
+  "migration_folder_missing" : [
+    "The folder with migrations is missing.",
+    "Are you sure that you're in the :red<root of a rails> application?"
+  ],
 
   "no_migrations" : "There are no migrations in the folder prior to :red<%{date}>",
 
@@ -15,5 +23,20 @@
 
   "db_cleaning": "Squasher is applying the clean migration",
 
-  "usage": "Example usage:\n  :green<squasher> :yellow<year[/month][/day]> #squash migrations prior to a specified date\n  :green<squasher> :yellow<clean>              #generate or update a cleaning migration and apply it\n  :green<squasher> :yellow<info>               #show this message"
+  "db_reuse": "Squasher is reusing the database from the previous squashing",
+
+  "usage": [
+    "Example usage:",
+    "  :green<squasher> [options] :yellow<year[/month][/day]> => squash migrations prior to a specified date",
+    "  :green<squasher> :yellow<clean>                        => generate or update a cleaning migration and apply it",
+    "  :green<squasher> :yellow<info>                         => show the message",
+    "",
+    "Options:",
+    "  :yellow<-d>       => execute in :blue<`dry`> mode - test a squashing process without removing old migrations",
+    "  :yellow<-r>       => reuse a database from the previous squashing"
+  ],
+
+  "wrong_option": "You provided a wrong option :red<%{o}>",
+
+  "dry_mode_finished": "The `dry` mode is finished. Below the init schema's content:"
 }

data/lib/squasher/templates/init_schema.rb.erb

@@ -6,6 +6,6 @@ class InitSchema < ActiveRecord::Migration
   end
 
   def down
-    raise "Can not revert initial migration"
+    raise ActiveRecord::IrreversibleMigration, "The initial migration is not revertable"
   end
 end

data/lib/squasher/worker.rb

@@ -2,28 +2,35 @@ require 'fileutils'
 
 module Squasher
   class Worker
-    attr_reader :date
+    OPTIONS = [:d, :r]
+
+    attr_reader :date, :options
 
     def self.process(*args)
       new(*args).process
     end
 
-    def initialize(date)
+    def initialize(date, options = [])
       @date = date
+      @options = options
     end
 
     def process
       check!
 
       result = under_squash_env do
-        File.open(config.migration_file(finish_timestamp, :init_schema), 'wb') do |stream|
-          stream << ::Squasher::Render.render(:init_schema, config)
+        if options.include?(:d)
+          Squasher.tell(:dry_mode_finished)
+          puts Render.render(:init_schema, config)
+        else
+          path = config.migration_file(finish_timestamp, :init_schema)
+          File.open(path, 'wb') { |io| io << Render.render(:init_schema, config) }
+          migrations.each { |file| FileUtils.rm(file) }
         end
 
-        migrations.each { |file| FileUtils.rm(file) }
-
         Squasher.rake("db:drop") unless Squasher.ask(:keep_database)
       end
+
       Squasher.clean if result && Squasher.ask(:apply_clean)
     end
 
@@ -64,10 +71,17 @@ module Squasher
 
     def under_squash_env
       config.stub_dbconfig do
-        if Squasher.rake("db:drop db:create", :db_create) &&
-          Squasher.rake("db:migrate VERSION=#{ finish_timestamp }", :db_migrate)
-          yield
+        if options.include?(:r)
+          Squasher.tell(:db_reuse)
+        else
+          return unless Squasher.rake("db:drop db:create", :db_create)
         end
+
+        return unless Squasher.rake("db:migrate VERSION=#{ finish_timestamp }", :db_migrate)
+
+        yield
+
+        true
       end
     end
   end

data/spec/lib/{squasher/cleaner_spec.rb → cleaner_spec.rb}

@@ -5,8 +5,8 @@ describe Squasher::Cleaner do
   let(:expected_file) { File.join(fake_root, 'db', 'migrate', '20140102030405_squasher_clean.rb') }
 
   before do
-    Time.stub(:now => Time.new(2014, 1, 2, 3, 4, 5))
-    Squasher.stub(:rake).with("db:migrate", :db_cleaning)
+    allow(Time).to receive(:now).and_return(Time.new(2014, 1, 2, 3, 4, 5))
+    allow(Squasher).to receive(:rake).with("db:migrate", :db_cleaning)
   end
 
   after do
@@ -14,7 +14,7 @@ describe Squasher::Cleaner do
   end
 
   it 'create a new migration' do
-    Squasher::Cleaner.any_instance.stub(:prev_migration => nil)
+    allow_any_instance_of(Squasher::Cleaner).to receive(:prev_migration).and_return(nil)
     Squasher.clean
 
     expect(clean_migrations).to include(expected_file)

data/spec/lib/{squasher/config_spec.rb → config_spec.rb}

@@ -7,19 +7,19 @@ describe Squasher::Config do
     subject(:result) { config.dbconfig? }
 
     it 'a file is exists and it has a valid content' do
-      expect(result).to be_true
+      expect(result).to be_truthy
     end
 
     it 'a file is exists but doesnt have a valid content' do
-      config.stub(:dbconfig_file => File.join(fake_root, 'config', 'invalid_database.yml'))
+      allow(config).to receive(:dbconfig_file).and_return(File.join(fake_root, 'config', 'invalid_database.yml'))
 
-      expect(result).to be_false
+      expect(result).to be_falsey
     end
 
     it 'a file is not exists' do
-      config.stub(:dbconfig_file => File.join(fake_root, 'config', 'not_existed.yml'))
+      allow(config).to receive(:dbconfig_file).and_return(File.join(fake_root, 'config', 'not_existed.yml'))
 
-      expect(result).to be_false
+      expect(result).to be_falsey
     end
   end
 
@@ -30,8 +30,7 @@ describe Squasher::Config do
           content = YAML.load(stream.read)
           expect(content["development"]["database"]).to eq("squasher")
           expect(content["development"]["encoding"]).to eq("utf-8")
-          expect(content["another_development"]["database"]).to eq("squasher")
-          expect(content["another_development"]["encoding"]).to eq("utf-8")
+          expect(content).not_to have_key("another_development")
         end
       end
     end
@@ -39,14 +38,14 @@ describe Squasher::Config do
     it 'recover original schema and db config files if some error raised' do
       begin
         config.stub_dbconfig do
-          expect(file_exists?('config', 'database.yml')).to be_true
-          expect(file_exists?('config', 'database.yml.sbackup')).to be_true
+          expect(file_exists?('config', 'database.yml')).to be_truthy
+          expect(file_exists?('config', 'database.yml.sq')).to be_truthy
 
           raise RuntimeError, "Unexpected system error"
         end
       rescue RuntimeError
-        expect(file_exists?('config', 'database.yml')).to be_true
-        expect(file_exists?('config', 'database.yml.sbackup')).to be_false
+        expect(file_exists?('config', 'database.yml')).to be_truthy
+        expect(file_exists?('config', 'database.yml.sq')).to be_falsey
       end
     end
 

data/spec/lib/squasher_spec.rb

@@ -6,8 +6,8 @@ describe Squasher do
     specify { expected_covert("2013", Time.new(2013, 1, 1)) }
 
     def expected_covert(input, expected)
-      expect(Squasher::Worker).to receive(:process).with(expected)
-      Squasher.squash(input)
+      expect(Squasher::Worker).to receive(:process).with(expected, [])
+      Squasher.squash(input, [])
     end
   end
 end

data/spec/lib/worker_spec.rb

@@ -0,0 +1,70 @@
+require 'spec_helper'
+require 'tempfile'
+
+describe Squasher::Worker do
+  context 'failed on #check!' do
+    let(:worker) { described_class.new(Time.new(2012, 6, 20)) }
+
+    specify 'command was run not in application root' do
+      allow_any_instance_of(Squasher::Config).to receive(:migrations_folder?).and_return(false)
+
+      expect_exit_with(:migration_folder_missing)
+    end
+
+    specify 'db configuration is invalid' do
+      allow_any_instance_of(Squasher::Config).to receive(:dbconfig?).and_return(false)
+
+      expect_exit_with(:dbconfig_invalid)
+    end
+
+    specify 'matched migrations was not found' do
+      expect_exit_with(:no_migrations, :date => "2012/06/20")
+    end
+
+    def expect_exit_with(*args)
+      expect(Squasher).to receive(:error).with(*args).and_call_original
+      expect { worker.process }.to raise_error(SystemExit)
+    end
+  end
+
+  it 'creates a new squashed migration & remove selected migrations' do
+    worker = described_class.new(Time.new(2014))
+    allow(worker).to receive(:under_squash_env).and_yield.and_return(true)
+    new_migration_path = File.join(Dir.tmpdir, 'init_schema.rb')
+    allow_any_instance_of(Squasher::Config).to receive(:migration_file).with('20131213090719', :init_schema).and_return(new_migration_path)
+
+    expect(FileUtils).to receive(:rm).with(File.join(fake_root, 'db', 'migrate', '20131205160936_first_migration.rb'))
+    expect(FileUtils).to receive(:rm).with(File.join(fake_root, 'db', 'migrate', '20131213090719_second_migration.rb'))
+    expect(Squasher).to receive(:ask).with(:keep_database).and_return(false)
+    expect(Squasher).to receive(:rake).with("db:drop")
+    expect(Squasher).to receive(:ask).with(:apply_clean).and_return(true)
+    expect(Squasher).to receive(:clean)
+
+    worker.process
+
+    expect(File.exists?(new_migration_path)).to be_truthy
+    File.open(new_migration_path) do |stream|
+      content = stream.read
+      expect(content).to include("InitSchema")
+      expect(content).to include('create_table "managers", :force => true do |t|')
+    end
+  end
+
+  specify 'the dry mode' do
+    worker = described_class.new(Time.new(2014), [:d])
+    allow(worker).to receive(:under_squash_env).and_yield.and_return(true)
+    expect(Squasher).to receive(:tell).with(:dry_mode_finished).and_call_original
+    expect(Squasher).to receive(:ask).with(:keep_database).and_return(false)
+    expect(Squasher).to receive(:rake).with("db:drop")
+    expect(Squasher).to receive(:ask).with(:apply_clean).and_return(false)
+    worker.process
+  end
+
+  specify 'reuse of an existing database' do
+    worker = described_class.new(Time.new(2014), [:r])
+    expect(Squasher).to receive(:tell).with(:db_reuse).and_call_original
+    expect(Squasher).not_to receive(:rake).with("db:drop db:create", :db_create)
+    allow(Squasher).to receive(:rake).with("db:migrate VERSION=20131213090719", :db_migrate).and_return(false)
+    worker.process
+  end
+end

data/spec/spec_helper.rb

@@ -1,13 +1,22 @@
 require 'bundler/setup'
 Bundler.require
 
-def fake_root
-  @fake_root ||= File.join(File.dirname(__FILE__), 'fake_app')
+
+module SpecHelpers
+  def fake_root
+    File.join(File.dirname(__FILE__), 'fake_app')
+  end
+end
+
+module Squasher
+  class Dir < ::Dir
+    def self.pwd
+      File.join(File.dirname(__FILE__), 'fake_app')
+    end
+  end
 end
 
 RSpec.configure do |config|
   config.order = 'random'
-  config.before do
-    Squasher::Config.any_instance.stub(:root_path => fake_root)
-  end
+  config.include SpecHelpers
 end

data/squasher.gemspec

@@ -4,7 +4,7 @@ $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 
 Gem::Specification.new do |spec|
   spec.name          = "squasher"
-  spec.version       = "0.1.7"
+  spec.version       = "0.2.0"
   spec.authors       = ["Sergey Pchelintsev"]
   spec.email         = ["mail@sergeyp.me"]
   spec.description   = %q{Squash your old migrations}
@@ -19,5 +19,5 @@ Gem::Specification.new do |spec|
 
   spec.add_development_dependency "bundler", ">= 1.3"
   spec.add_development_dependency "rake"
-  spec.add_development_dependency "rspec", "~> 2.14.0"
+  spec.add_development_dependency "rspec", ">= 3.3.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: squasher
 version: !ruby/object:Gem::Version
-  version: 0.1.7
+  version: 0.2.0
 platform: ruby
 authors:
 - Sergey Pchelintsev
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-02-16 00:00:00.000000000 Z
+date: 2016-07-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -42,16 +42,16 @@ dependencies:
   name: rspec
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 2.14.0
+        version: 3.3.0
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 2.14.0
+        version: 3.3.0
 description: Squash your old migrations
 email:
 - mail@sergeyp.me
@@ -82,10 +82,10 @@ files:
 - spec/fake_app/db/migrate/20140101010101_squasher_clean.rb
 - spec/fake_app/db/migrate/20140103124257_third_migration.rb
 - spec/fake_app/db/schema.rb
-- spec/lib/squasher/cleaner_spec.rb
-- spec/lib/squasher/config_spec.rb
-- spec/lib/squasher/worker_spec.rb
+- spec/lib/cleaner_spec.rb
+- spec/lib/config_spec.rb
 - spec/lib/squasher_spec.rb
+- spec/lib/worker_spec.rb
 - spec/spec_helper.rb
 - squasher.gemspec
 homepage: https://github.com/jalkoby/squasher
@@ -120,8 +120,9 @@ test_files:
 - spec/fake_app/db/migrate/20140101010101_squasher_clean.rb
 - spec/fake_app/db/migrate/20140103124257_third_migration.rb
 - spec/fake_app/db/schema.rb
-- spec/lib/squasher/cleaner_spec.rb
-- spec/lib/squasher/config_spec.rb
-- spec/lib/squasher/worker_spec.rb
+- spec/lib/cleaner_spec.rb
+- spec/lib/config_spec.rb
 - spec/lib/squasher_spec.rb
+- spec/lib/worker_spec.rb
 - spec/spec_helper.rb
+has_rdoc: 

data/spec/lib/squasher/worker_spec.rb

@@ -1,52 +0,0 @@
-require 'spec_helper'
-require 'tempfile'
-
-describe Squasher::Worker do
-  context 'failed on #check!' do
-    let(:worker) { described_class.new(Time.new(2012, 6, 20)) }
-
-    it 'command was run not in application root' do
-      Squasher::Config.any_instance.stub(:migrations_folder? => false)
-
-      expect_exit_with(:migration_folder_missing)
-    end
-
-    it 'db configuration is invalid' do
-      Squasher::Config.any_instance.stub(:dbconfig? => false)
-
-      expect_exit_with(:dbconfig_invalid)
-    end
-
-    it 'matched migrations was not found' do
-      expect_exit_with(:no_migrations, :date => "2012/06/20")
-    end
-
-    def expect_exit_with(*args)
-      expect(Squasher).to receive(:error).with(*args).and_call_original
-      expect { worker.process }.to raise_error(SystemExit)
-    end
-  end
-
-  it 'create a new squashed migration & remove selected migrations' do
-    worker = described_class.new(Time.new(2014))
-    worker.stub(:under_squash_env).and_yield.and_return(true)
-    new_migration_path = File.join(Dir.tmpdir, 'init_schema.rb')
-    Squasher::Config.any_instance.stub(:migration_file).with('20131213090719', :init_schema).and_return(new_migration_path)
-
-    FileUtils.should_receive(:rm).with(File.join(fake_root, 'db', 'migrate', '20131205160936_first_migration.rb'))
-    FileUtils.should_receive(:rm).with(File.join(fake_root, 'db', 'migrate', '20131213090719_second_migration.rb'))
-    Squasher.should_receive(:ask).with(:keep_database).and_return(false)
-    Squasher.should_receive(:rake).with("db:drop")
-    Squasher.should_receive(:ask).with(:apply_clean).and_return(true)
-    Squasher.should_receive(:clean)
-
-    worker.process
-
-    expect(File.exists?(new_migration_path)).to be_true
-    File.open(new_migration_path) do |stream|
-      content = stream.read
-      expect(content).to include("InitSchema")
-      expect(content).to include('create_table "managers", :force => true do |t|')
-    end
-  end
-end