mongration 0.0.3.pre.beta → 0.0.4.pre.beta

data/lib/mongration/migrate_all_up.rb

@@ -0,0 +1,20 @@
+module Mongration
+
+  # @private
+  class MigrateAllUp
+
+    def self.perform
+      new.perform
+    end
+
+    def perform
+      files_to_migrate.take_while(&:up)
+    end
+
+    private
+
+    def files_to_migrate
+      File.pending
+    end
+  end
+end

data/lib/mongration/migrate_down.rb

@@ -0,0 +1,20 @@
+module Mongration
+
+  # @private
+  class MigrateDown
+
+    def initialize(version)
+      @version = version
+    end
+
+    def perform
+      files_to_rollback.each(&:down)
+    end
+
+    private
+
+    def files_to_rollback
+      File.migrated.select { |f| f.version > @version }.reverse
+    end
+  end
+end

data/lib/mongration/migrate_up.rb

@@ -0,0 +1,20 @@
+module Mongration
+
+  # @private
+  class MigrateUp
+
+    def initialize(version)
+      @version = version
+    end
+
+    def perform
+      files_to_migrate.take_while(&:up)
+    end
+
+    private
+
+    def files_to_migrate
+      File.pending.select { |f| f.version <= @version }
+    end
+  end
+end

data/lib/mongration/migration.rb

@@ -0,0 +1,37 @@
+module Mongration
+
+  # @private
+  class Migration
+    include Mongoid::Document
+
+    store_in({ collection: 'mongration_migrations' })
+
+    field :file_name, type: String
+
+    field :created_at, type: Time, default: -> { Time.now }
+    field :deleted_at, type: Time
+
+    default_scope -> { where(deleted_at: nil) }
+
+    def self.create_by_file_name(file_name)
+      create(file_name: file_name)
+    end
+
+    def self.destroy_by_file_name(file_name)
+      where(file_name: file_name).first.destroy
+    end
+
+    def self.file_names
+      Migration.pluck(:file_name)
+    end
+
+    def self.last
+      all.to_a.max
+    end
+
+    def destroy(*)
+      self.deleted_at = Time.now
+      save!
+    end
+  end
+end

data/lib/mongration/null_output.rb

@@ -0,0 +1,6 @@
+module Mongration
+  class NullOutput
+    def puts(*); end
+    def write(*); end
+  end
+end

data/lib/mongration/rake_tasks.rb

@@ -0,0 +1,40 @@
+namespace :db do
+
+  desc 'Runs #up on any migration files found that have not already been run'
+  task migrate: :environment do
+    version = ENV['VERSION']
+    Mongration.migrate(version)
+  end
+
+  desc 'Runs #down on all migration files from the most recent migration'
+  task rollback: :environment do
+    step = ENV['STEP'] ? ENV['STEP'].to_i : 1
+    Mongration.rollback(step)
+  end
+
+  desc 'Returns the version for the most recent migration (i.e. the number of migrations that have been run, not the number of migration files)'
+  task version: :environment do
+    version = Mongration.version
+    Mongration.out.puts "Current version: #{version}"
+  end
+
+  namespace :migrate do
+
+    desc 'Creates a new migration file in the migration directory'
+    task :create, [:name] => [:environment] do |t, args|
+      name = args[:name] || ENV['NAME']
+      path = Mongration.create_migration(name)
+      Mongration.out.puts "Created #{path}"
+    end
+
+    desc 'Outputs a list of files and whether or not they have been migrated'
+    task status: :environment do
+      migrations = Mongration.status
+      Mongration.out.puts ['Status', 'Migration ID' 'Migration Name'].join("\t")
+      Mongration.out.puts '-' * 50
+      migrations.each do |migration|
+        Mongration.out.puts [migration.status, migration.migration_id, migration.migration_name].join("\t")
+      end
+    end
+  end
+end

data/lib/mongration/rollback.rb

@@ -3,25 +3,14 @@ module Mongration
   # @private
   class Rollback
 
-    def self.perform(migration)
-      new(migration).perform
-    end
-
-    def initialize(migration)
-      @migration = migration
+    def self.perform
+      new.perform
     end
 
     def perform
-      files_to_rollback.sort.reverse.each(&:down)
-      @migration.destroy
-    end
-
-    private
-
-    def files_to_rollback
-      @migration.file_names.map do |file_name|
-        Mongration::File.new(file_name)
-      end
+      return unless Migration.exists?
+      file = File.migrated.last
+      file.down
     end
   end
 end

data/lib/mongration/status.rb

@@ -0,0 +1,27 @@
+module Mongration
+
+  # @private
+  module Status
+    extend self
+
+    FileStatus = Struct.new(:status, :migration_id, :migration_name)
+
+    def migrations
+      performed_migrations + pending_migrations
+    end
+
+    private
+
+    def performed_migrations
+      File.migrated.map do |file|
+        FileStatus.new('up', file.version, file.name)
+      end
+    end
+
+    def pending_migrations
+      File.pending.map do |file|
+        FileStatus.new('down', file.version, file.name)
+      end
+    end
+  end
+end

data/lib/mongration/version.rb

@@ -1,3 +1,3 @@
 module Mongration
-  VERSION = '0.0.3-beta'
+  VERSION = '0.0.4-beta'
 end

data/lib/os_check.rb

@@ -0,0 +1,7 @@
+require 'rbconfig'
+
+module OsCheck
+  def self.mac_osx?
+    RbConfig::CONFIG['host_os'] =~ /darwin|mac os/
+  end
+end

data/spec/fixtures/integration.rb

@@ -1,39 +1,39 @@
 module IntegrationFixtures
-  def create(file_name, up, down)
-    Mongration::MigrationFileWriter.
-      write(file_name, up: up, down: down, dir: Mongration.configuration.dir)
+  def create_migration(file_name, options = {})
+    Mongration::CreateMigration::MigrationFileWriter.
+      write(file_name + '.rb', options.merge(dir: Mongration.configuration.dir))
   end
 
   def foo_create_migration
-    create(
-      '001_add_foo.rb',
-      'Foo.instances << Foo.new',
-      'Foo.instances.pop'
-      )
+    create_migration(
+      '001_add_foo',
+      up: 'Foo.instances << Foo.new',
+      down: 'Foo.instances.pop'
+    )
   end
 
   def bar_create_migration
-    create(
-      '002_add_bar.rb',
-      'Bar.instances << Bar.new',
-      'Bar.instances.pop'
-      )
+    create_migration(
+      '002_add_bar',
+      up: 'Bar.instances << Bar.new',
+      down: 'Bar.instances.pop'
+    )
   end
 
   def foo_update_migration
-    create(
-      '0003_add_name_to_foo.rb',
-      'foo = Foo.instances.first; foo.name = "Test"',
-      'foo = Foo.instances.first; foo.name = ""'
-      )
+    create_migration(
+      '0003_add_name_to_foo',
+      up: 'foo = Foo.instances.first; foo.name = "Test"',
+      down: 'foo = Foo.instances.first; foo.name = ""'
+    )
   end
 
   def bar_update_migration
-    create(
-      '004_add_name_to_bar.rb',
-      'bar = Bar.instances.first; bar.name = "Test"',
-      'bar = Bar.instances.first; bar.name = ""'
-      )
+    create_migration(
+      '004_add_name_to_bar',
+      up: 'bar = Bar.instances.first; bar.name = "Test"',
+      down: 'bar = Bar.instances.first; bar.name = ""'
+    )
   end
 end
 
@@ -64,7 +64,3 @@ class Bar
 
   attr_accessor :name
 end
-
-RSpec.configure do |config|
-  config.include(IntegrationFixtures)
-end

data/spec/integration/{template_spec.rb → create_migration_spec.rb}

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-describe Mongration do
+describe 'Mongration.create_migration' do
 
   it 'creates number-based template with CamelCase file name' do
     Mongration.create_migration('AddFoo')
@@ -67,4 +67,8 @@ end
 EOS
     )
   end
+
+  it 'returns file name of the migration created' do
+    expect(Mongration.create_migration('add_foo')).to eq(File.join('spec', 'db', 'migrate', '001_add_foo.rb'))
+  end
 end

data/spec/integration/migrate_spec.rb

@@ -0,0 +1,76 @@
+require 'spec_helper'
+
+describe 'Mongration.migrate' do
+
+  context 'when the migration is successful' do
+    before do
+      foo_create_migration
+    end
+
+    let(:migrate) { Mongration.migrate }
+
+    it 'runs a single migration' do
+      migrate
+      expect(Foo.count).to eq(1)
+    end
+
+    it 'includes success message' do
+      expect(Mongration.out).to receive(:puts).with('001 AddFoo: migrating')
+      expect(Mongration.out).to receive(:puts).with('001 AddFoo: migrated')
+      migrate
+    end
+  end
+
+  context 'when there is an error in the first migration' do
+    before do
+      create_migration(
+        '1_migration_with_error',
+        up: 'raise StandardError'
+      )
+      create_migration(
+        '2_migration_without_error',
+        up: 'Foo.instances << Foo.new'
+      )
+    end
+
+    let(:migrate) { Mongration.migrate }
+
+    it 'does not run later migrations' do
+      migrate
+      expect(Foo.instances).to be_empty
+    end
+
+    it 'includes failed message' do
+      expect(Mongration.out).to receive(:puts).with('1 MigrationWithError: migrating')
+      expect(Mongration.out).to receive(:puts).with(/StandardError: An error has occured, this and all later migrations cancelled/)
+      migrate
+    end
+  end
+
+  it 'skips the first migration and runs a second migration' do
+    foo_create_migration
+    Mongration.migrate
+
+    bar_create_migration
+    Mongration.migrate
+
+    expect(Foo.count).to eq(1)
+    expect(Bar.count).to eq(1)
+  end
+
+  it 'does not migrate when there are no migrations' do
+    foo_create_migration
+    Mongration.migrate
+    Mongration.migrate
+    expect(Foo.count).to eq(1)
+  end
+
+  it 'migrates files in order' do
+    foo_update_migration
+    foo_create_migration
+    Mongration.migrate
+
+    foo = Foo.instances.first
+    expect(foo.name).to eq('Test')
+  end
+end

data/spec/integration/migrate_with_version_spec.rb

@@ -0,0 +1,38 @@
+require 'spec_helper'
+
+describe 'Mongration.migrate(version)' do
+
+  context 'when the given version is above the current version' do
+    it 'migrates up to the given version' do
+      foo_create_migration
+      bar_create_migration
+      Mongration.migrate('001')
+
+      expect(Foo.count).to eq(1)
+      expect(Bar.count).to eq(0)
+    end
+  end
+
+  context 'when the given version is below the current version' do
+    it 'migrates down to the given version' do
+      foo_create_migration
+      bar_create_migration
+      Mongration.migrate
+
+      Mongration.migrate('001')
+
+      expect(Foo.count).to eq(1)
+      expect(Bar.count).to eq(0)
+    end
+  end
+
+  context 'when the version does not exist' do
+    it 'outputs message' do
+      foo_create_migration
+      bar_create_migration
+
+      expect(Mongration.out).to receive(:puts).with('Invalid Version: 003 does not exist.')
+      Mongration.migrate('003')
+    end
+  end
+end

data/spec/integration/{mongration_spec.rb → rollback_spec.rb}

@@ -1,25 +1,6 @@
 require 'spec_helper'
 
-describe Mongration do
-
-  it 'runs a single migration' do
-    foo_create_migration
-
-    Mongration.migrate
-
-    expect(Foo.count).to eq(1)
-  end
-
-  it 'skips the first migration and runs a second migration' do
-    foo_create_migration
-    Mongration.migrate
-
-    bar_create_migration
-    Mongration.migrate
-
-    expect(Foo.count).to eq(1)
-    expect(Bar.count).to eq(1)
-  end
+describe 'Mongration.rollback' do
 
   it 'rollsback a migration' do
     foo_create_migration
@@ -31,8 +12,6 @@ describe Mongration do
 
   it 'can rollback twice' do
     foo_create_migration
-    Mongration.migrate
-
     bar_create_migration
     Mongration.migrate
 
@@ -49,28 +28,24 @@ describe Mongration do
     expect(Foo.count).to eq(1)
   end
 
-  it 'does not migrate when there are no migrations' do
-    foo_create_migration
-    Mongration.migrate
+  it 'rolls back files in order' do
+    bar_create_migration
+    bar_update_migration
     Mongration.migrate
-    expect(Foo.count).to eq(1)
-  end
 
-  it 'migrates files in order' do
-    foo_update_migration
-    foo_create_migration
-    Mongration.migrate
+    Mongration.rollback
+    Mongration.rollback
 
-    foo = Foo.instances.first
-    expect(foo.name).to eq('Test')
+    expect(Bar.count).to eq(0)
   end
 
-  it 'rollbacks files in order' do
+  it 'rolls back multiple times when passed a step' do
+    foo_create_migration
     bar_create_migration
-    bar_update_migration
     Mongration.migrate
-    Mongration.rollback
 
+    Mongration.rollback(2)
+    expect(Foo.count).to eq(0)
     expect(Bar.count).to eq(0)
   end
 end