mongoid_migrations 1.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: e4c2f765c51f0bad595b39e311cbccc425c54aec
+  data.tar.gz: 7f3005b08f1ab8812f5ad861d268bc7df4bc908d
+SHA512:
+  metadata.gz: 3838008127e118580ecb48a8cc41af114a1e8fed42aa9163252aabfd40caffeb2a8d7eae16ab617d51e0431a7f87b36d612065885a8f4e6f4b2d6c8982bfe09b
+  data.tar.gz: 1f126fb2313d9859a1725918634972b32fdd7bf11238a92ea3c9153a0b6b9a1d6eeab21a7aeb631b537ee998fb1a9ad9ad9275df913bd3fdded6e5a05d6de16e

data/.gitignore

@@ -0,0 +1,6 @@
+.bundle
+.DS_Store
+.rvmrc
+.idea
+log/
+.ruby-version

data/Gemfile

@@ -0,0 +1,7 @@
+source "http://rubygems.org"
+gemspec :name => 'mongoid_migrations'
+
+group :test do
+  gem 'minitest'
+  gem 'pry'
+end

data/Gemfile.lock

@@ -0,0 +1,50 @@
+PATH
+  remote: .
+  specs:
+    mongoid_migrations (1.0.1)
+      activesupport (>= 4.2.0)
+      bundler (>= 1.0.0)
+      mongoid (>= 5.0.0)
+
+GEM
+  remote: http://rubygems.org/
+  specs:
+    activemodel (5.1.1)
+      activesupport (= 5.1.1)
+    activesupport (5.1.1)
+      concurrent-ruby (~> 1.0, >= 1.0.2)
+      i18n (~> 0.7)
+      minitest (~> 5.1)
+      tzinfo (~> 1.1)
+    bson (4.2.1)
+    coderay (1.1.1)
+    concurrent-ruby (1.0.5)
+    i18n (0.8.4)
+    method_source (0.8.2)
+    minitest (5.10.2)
+    mongo (2.4.2)
+      bson (>= 4.2.1, < 5.0.0)
+    mongoid (6.1.1)
+      activemodel (~> 5.0)
+      mongo (>= 2.4.1, < 3.0.0)
+    pry (0.10.4)
+      coderay (~> 1.1.0)
+      method_source (~> 0.8.1)
+      slop (~> 3.4)
+    rake (12.0.0)
+    slop (3.6.0)
+    thread_safe (0.3.6)
+    tzinfo (1.2.3)
+      thread_safe (~> 0.1)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  minitest
+  mongoid_migrations!
+  pry
+  rake
+
+BUNDLED WITH
+   1.14.6

data/Rakefile

@@ -0,0 +1,14 @@
+require "bundler/gem_tasks"
+
+$:.unshift(File.dirname(__FILE__))
+
+task :default => ['test:mongoid:migrations']
+
+namespace :test do
+  namespace :mongoid do
+    desc "Test mongoid migrations"
+    task :migrations do
+      load 'test/migration_test.rb'
+    end
+  end
+end

data/lib/mongoid_migrations.rb

@@ -0,0 +1,11 @@
+# encoding: utf-8
+require 'bundler/setup'
+Bundler.require(:mongoid_migrations)
+
+# Add base to path incase not included as a gem
+$:.unshift(File.dirname(__FILE__)) unless
+  $:.include?(File.dirname(__FILE__)) || $:.include?(File.expand_path(File.dirname(__FILE__)))
+
+require 'mongoid_migrations/version'
+require 'mongoid_migrations/models/data_migration'
+require 'mongoid_migrations/active_record_ext/migrations'

data/lib/mongoid_migrations/active_record_ext/migrations.rb

@@ -0,0 +1,422 @@
+# encoding: utf-8
+
+module Mongoid #:nodoc
+  # Exception that can be raised to stop migrations from going backwards.
+  class IrreversibleMigration < RuntimeError
+  end
+
+  class DuplicateMigrationVersionError < RuntimeError#:nodoc:
+    def initialize(version)
+      super("Multiple migrations have the version number #{version}")
+    end
+  end
+
+  class DuplicateMigrationNameError < RuntimeError#:nodoc:
+    def initialize(name)
+      super("Multiple migrations have the name #{name}")
+    end
+  end
+
+  class UnknownMigrationVersionError < RuntimeError#:nodoc:
+    def initialize(version)
+      super("No migration with version number #{version}")
+    end
+  end
+
+  class IllegalMigrationNameError < RuntimeError#:nodoc:
+    def initialize(name)
+      super("Illegal name for migration file: #{name}\n\t(only lower case letters, numbers, and '_' allowed)")
+    end
+  end
+
+  # Data migrations can manage the modification of data. It's a solution to the common problem of modifying
+  # data between code revisions within a document oriented database.
+  #
+  # Example of simple migration for a system dependency:
+  #
+  #   class AddBaselineSurveySchema < Mongoid::Migration
+  #     def self.up
+  #       SurveySchema.create(:label => 'Baseline Survey')
+  #     end
+  #
+  #     def self.down
+  #       SurveySchema.where(:label => 'Baseline Survey').first.destroy
+  #     end
+  #   end
+  #
+  # == Timestamped Migrations
+  #
+  # By default, generates migrations that look like:
+  #
+  #    20080717013526_your_migration_name.rb
+  #
+  # The prefix is a generation timestamp (in UTC).
+  #
+  # If you'd prefer to use numeric prefixes, you can turn timestamped migrations
+  # off by setting:
+  #
+  #    Mongoid.config.timestamped_migrations = false
+  #
+  # In environment.rb.
+  #
+  class Migration
+    @@verbose = true
+    cattr_accessor :verbose
+
+    class << self
+      def up_with_benchmarks #:nodoc:
+        migrate(:up)
+      end
+
+      def down_with_benchmarks #:nodoc:
+        migrate(:down)
+      end
+
+      # Execute this migration in the named direction
+      def migrate(direction)
+        return unless respond_to?(direction)
+
+        case direction
+          when :up   then announce "migrating"
+          when :down then announce "reverting"
+        end
+
+        result = nil
+        time = Benchmark.measure { result = send("#{direction}_without_benchmarks") }
+
+        case direction
+          when :up   then announce "migrated (%.4fs)" % time.real; write
+          when :down then announce "reverted (%.4fs)" % time.real; write
+        end
+
+        result
+      end
+
+      # Because the method added may do an alias_method, it can be invoked
+      # recursively. We use @ignore_new_methods as a guard to indicate whether
+      # it is safe for the call to proceed.
+      def singleton_method_added(sym) #:nodoc:
+        return if defined?(@ignore_new_methods) && @ignore_new_methods
+
+        begin
+          @ignore_new_methods = true
+
+          case sym
+            when :up, :down
+              # singleton_class.send(:alias_method_chain, sym, "benchmarks")
+              singleton_class.send(:alias_method, "#{sym}_without_benchmarks", sym)
+              singleton_class.send(:alias_method, sym, "#{sym}_with_benchmarks")
+          end
+        ensure
+          @ignore_new_methods = false
+        end
+      end
+
+      def write(text="")
+        puts(text) if verbose
+      end
+
+      def announce(message)
+        version = defined?(@version) ? @version : nil
+
+        text = "#{version} #{name}: #{message}"
+        length = [0, 75 - text.length].max
+        write "== %s %s" % [text, "=" * length]
+      end
+
+      def say(message, subitem=false)
+        write "#{subitem ? "   ->" : "--"} #{message}"
+      end
+
+      def say_with_time(message)
+        say(message)
+        result = nil
+        time = Benchmark.measure { result = yield }
+        say "%.4fs" % time.real, :subitem
+        say("#{result} rows", :subitem) if result.is_a?(Integer)
+        result
+      end
+
+      def suppress_messages
+        save, self.verbose = verbose, false
+        yield
+      ensure
+        self.verbose = save
+      end
+
+      def connection
+        # ActiveRecord::Base.connection
+        if ::Mongoid.respond_to?(:default_client)
+          ::Mongoid.default_client
+        else
+          ::Mongoid.default_session
+        end
+      end
+
+      def method_missing(method, *arguments, &block)
+        arg_list = arguments.map(&:inspect) * ', '
+
+        say_with_time "#{method}(#{arg_list})" do
+          # unless arguments.empty? || method == :execute
+          #   arguments[0] = Migrator.proper_table_name(arguments.first)
+          # end
+          connection.send(method, *arguments, &block)
+        end
+      end
+    end
+  end
+
+  # MigrationProxy is used to defer loading of the actual migration classes
+  # until they are needed
+  class MigrationProxy
+
+    attr_accessor :name, :version, :filename
+
+    delegate :migrate, :announce, :write, :to=>:migration
+
+    private
+
+      def migration
+        @migration ||= load_migration
+      end
+
+      def load_migration
+        require(File.expand_path(filename))
+        name.constantize
+      end
+
+  end
+
+  class Migrator#:nodoc:
+    class << self
+      def migrate(migrations_path, target_version = nil)
+        case
+          when target_version.nil?              then up(migrations_path, target_version)
+          when current_version > target_version then down(migrations_path, target_version)
+          else                                       up(migrations_path, target_version)
+        end
+      end
+
+      def rollback(migrations_path, steps=1)
+        move(:down, migrations_path, steps)
+      end
+
+      def forward(migrations_path, steps=1)
+        move(:up, migrations_path, steps)
+      end
+
+      def up(migrations_path, target_version = nil)
+        self.new(:up, migrations_path, target_version).migrate
+      end
+
+      def down(migrations_path, target_version = nil)
+        self.new(:down, migrations_path, target_version).migrate
+      end
+
+      def run(direction, migrations_path, target_version)
+        self.new(direction, migrations_path, target_version).run
+      end
+
+      def migrations_path
+        'db/migrate'
+      end
+
+      def generate(migration_name)
+        file_name = "#{Time.now.utc.strftime("%Y%m%d%H%M%S")}_#{migration_name.underscore}.rb"
+
+        File.open("#{migrations_path}/#{file_name}", 'w') do |f|
+          f << "class #{migration_name.camelize} < Mongoid::Migration\n"
+          f << "  def self.up\n  end\n\n"
+          f << "  def self.down\n  end\n"
+          f << "end\n"
+        end
+      end
+
+      # def schema_migrations_table_name
+      #   # Base.table_name_prefix + 'schema_migrations' + Base.table_name_suffix
+      #   'data_migrations'
+      # end
+
+      def get_all_versions
+        # table = Arel::Table.new(schema_migrations_table_name)
+        #         Base.connection.select_values(table.project(table['version']).to_sql).map(&:to_i).sort
+        DataMigration.all.map {|datamigration| datamigration.version.to_i }.sort
+      end
+
+      def current_version
+        # sm_table = schema_migrations_table_name
+        # if Base.connection.table_exists?(sm_table)
+        #   get_all_versions.max || 0
+        # else
+        #   0
+        # end
+        get_all_versions.max || 0
+      end
+
+      def proper_table_name(name)
+        # Use the Active Record objects own table_name, or pre/suffix from ActiveRecord::Base if name is a symbol/string
+        # name.table_name rescue "#{ActiveRecord::Base.table_name_prefix}#{name}#{ActiveRecord::Base.table_name_suffix}"
+        name
+      end
+
+      private
+
+      def move(direction, migrations_path, steps)
+        migrator = self.new(direction, migrations_path)
+        start_index = migrator.migrations.index(migrator.current_migration)
+
+        if start_index
+          finish = migrator.migrations[start_index + steps]
+          version = finish ? finish.version : 0
+          send(direction, migrations_path, version)
+        end
+      end
+    end
+
+    def initialize(direction, migrations_path, target_version = nil)
+      # raise StandardError.new("This database does not yet support migrations") unless Base.connection.supports_migrations?
+      # Base.connection.initialize_schema_migrations_table
+      @direction, @migrations_path, @target_version = direction, migrations_path, target_version
+    end
+
+    def current_version
+      migrated.last || 0
+    end
+
+    def current_migration
+      migrations.detect { |m| m.version == current_version }
+    end
+
+    def run
+      target = migrations.detect { |m| m.version == @target_version }
+      raise UnknownMigrationVersionError.new(@target_version) if target.nil?
+      unless (up? && migrated.include?(target.version.to_i)) || (down? && !migrated.include?(target.version.to_i))
+        target.migrate(@direction)
+        record_version_state_after_migrating(target.version)
+      end
+    end
+
+    def migrate
+      current = migrations.detect { |m| m.version == current_version }
+      target = migrations.detect { |m| m.version == @target_version }
+
+      if target.nil? && !@target_version.nil? && @target_version > 0
+        raise UnknownMigrationVersionError.new(@target_version)
+      end
+
+      start = up? ? 0 : (migrations.index(current) || 0)
+      finish = migrations.index(target) || migrations.size - 1
+      runnable = migrations[start..finish]
+
+      # skip the last migration if we're headed down, but not ALL the way down
+      runnable.pop if down? && !target.nil?
+
+      runnable.each do |migration|
+        # puts "Migrating to #{migration.name} (#{migration.version})"
+
+        # On our way up, we skip migrating the ones we've already migrated
+        next if up? && migrated.include?(migration.version.to_i)
+
+        # On our way down, we skip reverting the ones we've never migrated
+        if down? && !migrated.include?(migration.version.to_i)
+          migration.announce 'never migrated, skipping'; migration.write
+          next
+        end
+
+        # begin
+        #   ddl_transaction do
+        #     migration.migrate(@direction)
+        #     record_version_state_after_migrating(migration.version)
+        #   end
+        # rescue => e
+        #   canceled_msg = Base.connection.supports_ddl_transactions? ? "this and " : ""
+        #   raise StandardError, "An error has occurred, #{canceled_msg}all later migrations canceled:\n\n#{e}", e.backtrace
+        # end
+        begin
+          migration.migrate(@direction)
+          record_version_state_after_migrating(migration.version)
+        rescue => e
+          raise StandardError, "An error has occurred, #{migration.version} and all later migrations canceled:\n\n#{e}", e.backtrace
+        end
+      end
+    end
+
+    def migrations
+      @migrations ||= begin
+        files = Dir["#{@migrations_path}/[0-9]*_*.rb"]
+
+        migrations = files.inject([]) do |klasses, file|
+          version, name = file.scan(/([0-9]+)_([_a-z0-9]*).rb/).first
+
+          raise IllegalMigrationNameError.new(file) unless version
+          version = version.to_i
+
+          if klasses.detect { |m| m.version == version }
+            raise DuplicateMigrationVersionError.new(version)
+          end
+
+          if klasses.detect { |m| m.name == name.camelize }
+            raise DuplicateMigrationNameError.new(name.camelize)
+          end
+
+          migration = MigrationProxy.new
+          migration.name     = name.camelize
+          migration.version  = version
+          migration.filename = file
+          klasses << migration
+        end
+
+        migrations = migrations.sort_by(&:version)
+        down? ? migrations.reverse : migrations
+      end
+    end
+
+    def pending_migrations
+      already_migrated = migrated
+      migrations.reject { |m| already_migrated.include?(m.version.to_i) }
+    end
+
+    def migrated
+      @migrated_versions ||= self.class.get_all_versions
+    end
+
+    private
+      def record_version_state_after_migrating(version)
+        # table = Arel::Table.new(self.class.schema_migrations_table_name)
+
+        @migrated_versions ||= []
+        # if down?
+        #   @migrated_versions.delete(version)
+        #   table.where(table["version"].eq(version.to_s)).delete
+        # else
+        #   @migrated_versions.push(version).sort!
+        #   table.insert table["version"] => version.to_s
+        # end
+        if down?
+          @migrated_versions.delete(version)
+          DataMigration.where(:version => version.to_s).first.destroy
+        else
+          @migrated_versions.push(version).sort!
+          DataMigration.create(:version => version.to_s)
+        end
+      end
+
+      def up?
+        @direction == :up
+      end
+
+      def down?
+        @direction == :down
+      end
+
+      # Wrap the migration in a transaction only if supported by the adapter.
+      def ddl_transaction(&block)
+        # if Base.connection.supports_ddl_transactions?
+        #   Base.transaction { block.call }
+        # else
+        #   block.call
+        # end
+        block.call
+      end
+  end
+end

data/lib/mongoid_migrations/models/data_migration.rb

@@ -0,0 +1,5 @@
+class DataMigration
+  include Mongoid::Document
+
+  field :version
+end

data/lib/mongoid_migrations/tasks.rb

@@ -0,0 +1,102 @@
+# require 'mongoid_migrations/tasks'
+# will give you the resque tasks
+
+namespace :db do
+  unless Rake::Task.task_defined?("db:drop")
+    desc 'Drops all the collections for the database for the current env'
+    task :drop do
+      Mongoid.master.collections.each {|col| col.drop_indexes && col.drop unless ['system.indexes', 'system.users'].include?(col.name) }
+    end
+  end
+
+  unless Rake::Task.task_defined?("db:seed")
+    # if another ORM has defined db:seed, don't run it twice.
+    desc 'Load the seed data from db/seeds.rb'
+    task :seed do
+      seed_file = File.join('db', 'seeds.rb')
+      load(seed_file) if File.exist?(seed_file)
+    end
+  end
+
+  unless Rake::Task.task_defined?("db:setup")
+    desc 'Create the database, and initialize with the seed data'
+    task :setup => [ 'db:create', 'db:seed' ]
+  end
+
+  unless Rake::Task.task_defined?("db:reseed")
+    desc 'Delete data and seed'
+    task :reseed => [ 'db:drop', 'db:seed' ]
+  end
+
+  unless Rake::Task.task_defined?("db:create")
+    task :create do
+      # noop
+    end
+  end
+
+  desc 'Current database version'
+  task :version do
+    puts Mongoid::Migrator.current_version.to_s
+  end
+
+  desc "Migrate the database through scripts in db/migrate. Target specific version with VERSION=x. Turn off output with VERBOSE=false."
+  task :migrate do
+    Mongoid::Migration.verbose = ENV["VERBOSE"] ? ENV["VERBOSE"] == "true" : true
+    Mongoid::Migrator.migrate("db/migrate/", ENV["VERSION"] ? ENV["VERSION"].to_i : nil)
+  end
+
+  namespace :migrate do
+    desc  'Rollback the database one migration and re migrate up. If you want to rollback more than one step, define STEP=x. Target specific version with VERSION=x.'
+    task :redo do
+      if ENV["VERSION"]
+        Rake::Task["db:migrate:down"].invoke
+        Rake::Task["db:migrate:up"].invoke
+      else
+        Rake::Task["db:rollback"].invoke
+        Rake::Task["db:migrate"].invoke
+      end
+    end
+
+    desc 'Resets your database using your migrations for the current environment'
+    # should db:create be changed to db:setup? It makes more sense wanting to seed
+    task :reset => ["db:drop", "db:create", "db:migrate"]
+
+    desc 'Runs the "up" for a given migration VERSION.'
+    task :up do
+      version = ENV["VERSION"] ? ENV["VERSION"].to_i : nil
+      raise "VERSION is required" unless version
+      Mongoid::Migrator.run(:up, "db/migrate/", version)
+    end
+
+    desc 'Runs the "down" for a given migration VERSION.'
+    task :down do
+      version = ENV["VERSION"] ? ENV["VERSION"].to_i : nil
+      raise "VERSION is required" unless version
+      Mongoid::Migrator.run(:down, "db/migrate/", version)
+    end
+
+    desc 'Create a migration file for given migration NAME'
+    task :create do
+      migration_name = ENV["NAME"] || 'migration'
+      Mongoid::Migrator.generate(migration_name)
+    end
+  end
+
+  desc 'Rolls the database back to the previous migration. Specify the number of steps with STEP=n'
+  task :rollback do
+    step = ENV['STEP'] ? ENV['STEP'].to_i : 1
+    Mongoid::Migrator.rollback('db/migrate/', step)
+  end
+
+  namespace :schema do
+    task :load do
+      # noop
+    end
+  end
+
+  namespace :test do
+    task :prepare do
+      # Stub out for MongoDB
+    end
+  end
+end

data/lib/mongoid_migrations/version.rb

@@ -0,0 +1,3 @@
+module MongoidMigrations #:nodoc:
+  VERSION = '1.0.1'
+end

data/mongoid_migrations.gemspec

@@ -0,0 +1,28 @@
+require File.join(File.dirname(__FILE__), 'lib', 'mongoid_migrations', 'version')
+
+Gem::Specification.new do |s|
+  s.platform    = Gem::Platform::RUBY
+  s.name        = 'mongoid_migrations'
+  s.version     = MongoidMigrations::VERSION
+  s.summary     = 'Data migrations for Rack applications with Mongoid'
+  s.license     = 'MIT'
+  s.description = 'Migrations for Mongoid'
+
+  s.required_ruby_version     = '>= 1.8.6'
+  s.required_rubygems_version = '>= 1.3.6'
+
+  s.author            = 'Nimrod Popper'
+  s.email             = 'nimrod@yotpo.com'
+  s.date              = %q{2017-06-06}
+  s.homepage          = 'https://github.com/YotpoLtd/mongoid_migrations'
+
+  s.require_paths = ['lib']
+  s.files         = Dir['.gitignore', 'Gemfile', 'Gemfile.lock', 'Rakefile', 'README.rdoc', 'mongoid_migrations.gemspec', 'lib/**/*']
+  s.test_files    = Dir['test/**/*']
+  s.has_rdoc      = false
+
+  s.add_runtime_dependency('bundler', '>= 1.0.0')
+  s.add_runtime_dependency('mongoid', '>= 5.0.0')
+  s.add_runtime_dependency('activesupport',  '>= 4.2.0')
+  s.add_development_dependency 'rake'
+end

data/test/config.rb

@@ -0,0 +1,8 @@
+require File.join(File.dirname(__FILE__), '..', 'lib', 'mongoid_migrations')
+
+Mongoid.configure.connect_to('mongoid_test')
+
+# require all models
+Dir[File.join(File.dirname(__FILE__), 'models', '*.rb')].each { |file| require file }
+
+MIGRATIONS_ROOT = File.join(File.dirname(__FILE__), 'migrations')

data/test/helper.rb

@@ -0,0 +1,11 @@
+$:.unshift(File.dirname(__FILE__))
+
+require 'bundler/setup'
+Bundler.require(:test)
+
+require 'mongoid'
+require 'config'
+require 'minitest/autorun'
+require 'rake'
+require 'rake/testtask'
+require 'rdoc/task'

data/test/migration_test.rb

@@ -0,0 +1,134 @@
+require File.dirname(__FILE__) + '/helper'
+
+class Mongoid::Migration
+  class <<self
+    attr_accessor :message_count
+
+    def puts(text="")
+      self.message_count ||= 0
+      self.message_count += 1
+    end
+  end
+end
+
+module Mongoid
+  class TestCase < ActiveSupport::TestCase #:nodoc:
+
+    def setup
+      Mongoid::Migration.verbose = true
+      # same as db:drop command in lib/mongoid_migrations/mongoid_ext/railties/database.rake
+      if Mongoid.respond_to?(:default_client)
+        Mongoid.default_client.database.drop
+      else
+        Mongoid.default_session.drop
+      end
+    end
+
+    def teardown; end
+
+    def test_drop_works
+      assert_equal 0, Mongoid::Migrator.current_version, "db:drop should take us down to version 0"
+    end
+
+    def test_finds_migrations
+      assert Mongoid::Migrator.new(:up, MIGRATIONS_ROOT + "/valid").migrations.size == 2
+      assert_equal 2, Mongoid::Migrator.new(:up, MIGRATIONS_ROOT + "/valid").pending_migrations.size
+    end
+
+    def test_migrator_current_version
+      Mongoid::Migrator.migrate(MIGRATIONS_ROOT + "/valid", 20100513054656)
+      assert_equal(20100513054656, Mongoid::Migrator.current_version)
+    end
+
+    def test_migrator
+      assert SurveySchema.first.nil?, "All SurveySchemas should be clear before migration run"
+
+      Mongoid::Migrator.up(MIGRATIONS_ROOT + "/valid")
+
+      assert_equal 20100513063902, Mongoid::Migrator.current_version
+      assert !SurveySchema.first.nil?
+
+      Mongoid::Migrator.down(MIGRATIONS_ROOT + "/valid")
+      assert_equal 0, Mongoid::Migrator.current_version
+
+      assert SurveySchema.create(:label => 'Questionable Survey')
+      assert_equal 1, SurveySchema.all.size
+    end
+
+    def test_migrator_two_up_and_one_down
+      assert SurveySchema.where(:label => 'Baseline Survey').first.nil?
+      assert_equal 0, SurveySchema.all.size
+
+      Mongoid::Migrator.up(MIGRATIONS_ROOT + "/valid", 20100513054656)
+
+      assert !SurveySchema.where(:label => 'Baseline Survey').first.nil?
+      assert_equal 1, SurveySchema.all.size
+
+      assert SurveySchema.where(:label => 'Improvement Plan Survey').first.nil?
+
+      Mongoid::Migrator.up(MIGRATIONS_ROOT + "/valid", 20100513063902)
+      assert_equal 20100513063902, Mongoid::Migrator.current_version
+
+      assert !SurveySchema.where(:label => 'Improvement Plan Survey').first.nil?
+      assert_equal 2, SurveySchema.all.size
+
+      Mongoid::Migrator.down(MIGRATIONS_ROOT + "/valid", 20100513054656)
+      assert_equal 20100513054656, Mongoid::Migrator.current_version
+
+      assert SurveySchema.where(:label => 'Improvement Plan Survey').first.nil?
+      assert !SurveySchema.where(:label => 'Baseline Survey').first.nil?
+      assert_equal 1, SurveySchema.all.size
+    end
+
+    def test_finds_pending_migrations
+      Mongoid::Migrator.up(MIGRATIONS_ROOT + "/valid", 20100513054656)
+      pending_migrations = Mongoid::Migrator.new(:up, MIGRATIONS_ROOT + "/valid").pending_migrations
+
+      assert_equal 1, pending_migrations.size
+      assert_equal pending_migrations[0].version, 20100513063902
+      assert_equal pending_migrations[0].name, 'AddImprovementPlanSurveySchema'
+    end
+
+    def test_migrator_rollback
+      Mongoid::Migrator.migrate(MIGRATIONS_ROOT + "/valid")
+      assert_equal(20100513063902, Mongoid::Migrator.current_version)
+
+      Mongoid::Migrator.rollback(MIGRATIONS_ROOT + "/valid")
+      assert_equal(20100513054656, Mongoid::Migrator.current_version)
+
+      Mongoid::Migrator.rollback(MIGRATIONS_ROOT + "/valid")
+      assert_equal(0, Mongoid::Migrator.current_version)
+    end
+
+    def test_migrator_forward
+      Mongoid::Migrator.migrate(MIGRATIONS_ROOT + "/valid", 20100513054656)
+      assert_equal(20100513054656, Mongoid::Migrator.current_version)
+
+      Mongoid::Migrator.forward(MIGRATIONS_ROOT + "/valid", 20100513063902)
+      assert_equal(20100513063902, Mongoid::Migrator.current_version)
+    end
+
+    def test_migrator_with_duplicate_names
+      assert_raise(Mongoid::DuplicateMigrationNameError) do
+        Mongoid::Migrator.migrate(MIGRATIONS_ROOT + "/duplicate/names", nil)
+      end
+    end
+
+    def test_migrator_with_duplicate_versions
+      assert_raise(Mongoid::DuplicateMigrationVersionError) do
+        Mongoid::Migrator.migrate(MIGRATIONS_ROOT + "/duplicate/versions", nil)
+      end
+    end
+
+    def test_migrator_with_missing_version_numbers
+      assert_raise(Mongoid::UnknownMigrationVersionError) do
+        Mongoid::Migrator.migrate(MIGRATIONS_ROOT + "/valid", 500)
+      end
+    end
+
+    def test_migration_returns_connection_without_error
+      Mongoid::Migration.connection
+    end
+
+  end
+end

data/test/migrations/duplicate/names/20100513073457_add_duplicate_survey_schema.rb

@@ -0,0 +1,9 @@
+class AddDuplicateSurveySchema < Mongoid::Migration
+  def self.up
+    SurveySchema.create(:label => 'Duplicate Survey')
+  end
+
+  def self.down
+    SurveySchema.where(:label => 'Duplicate Survey').first.destroy
+  end
+end

data/test/migrations/duplicate/names/20100513073724_add_duplicate_survey_schema.rb

@@ -0,0 +1,9 @@
+class AddDuplicateSurveySchema < Mongoid::Migration
+  def self.up
+    SurveySchema.create(:label => 'Duplicate Survey')
+  end
+
+  def self.down
+    SurveySchema.where(:label => 'Duplicate Survey').first.destroy
+  end
+end

data/test/migrations/duplicate/versions/20100513073457_add_another_duplicate_survey_schema.rb

@@ -0,0 +1,9 @@
+class AddDuplicateSurveySchema < Mongoid::Migration
+  def self.up
+    SurveySchema.create(:label => 'Duplicate Survey')
+  end
+
+  def self.down
+    SurveySchema.where(:label => 'Duplicate Survey').first.destroy
+  end
+end

data/test/migrations/duplicate/versions/20100513073457_add_duplicate_survey_schema.rb

@@ -0,0 +1,9 @@
+class AddDuplicateSurveySchema < Mongoid::Migration
+  def self.up
+    SurveySchema.create(:label => 'Duplicate Survey')
+  end
+
+  def self.down
+    SurveySchema.where(:label => 'Duplicate Survey').first.destroy
+  end
+end

data/test/migrations/valid/20100513054656_add_baseline_survey_schema.rb

@@ -0,0 +1,10 @@
+class AddBaselineSurveySchema < Mongoid::Migration
+  def self.up
+    SurveySchema.create(:id => '4c47bf87f3395c339c000001',
+                        :label => 'Baseline Survey')
+  end
+
+  def self.down
+    SurveySchema.where(:label => 'Baseline Survey').first.destroy
+  end
+end

data/test/migrations/valid/20100513063902_add_improvement_plan_survey_schema.rb

@@ -0,0 +1,9 @@
+class AddImprovementPlanSurveySchema < Mongoid::Migration
+  def self.up
+    SurveySchema.create(:label => 'Improvement Plan Survey')
+  end
+
+  def self.down
+    SurveySchema.where(:label => 'Improvement Plan Survey').first.destroy
+  end
+end

data/test/models/survey_schema.rb

@@ -0,0 +1,6 @@
+class SurveySchema
+  include Mongoid::Document
+	include Mongoid::Timestamps
+	
+	field :label
+end

metadata

@@ -0,0 +1,129 @@
+--- !ruby/object:Gem::Specification
+name: mongoid_migrations
+version: !ruby/object:Gem::Version
+  version: 1.0.1
+platform: ruby
+authors:
+- Nimrod Popper
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-06-06 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+- !ruby/object:Gem::Dependency
+  name: mongoid
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 5.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 5.0.0
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 4.2.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 4.2.0
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Migrations for Mongoid
+email: nimrod@yotpo.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- Gemfile.lock
+- Rakefile
+- lib/mongoid_migrations.rb
+- lib/mongoid_migrations/active_record_ext/migrations.rb
+- lib/mongoid_migrations/models/data_migration.rb
+- lib/mongoid_migrations/tasks.rb
+- lib/mongoid_migrations/version.rb
+- mongoid_migrations.gemspec
+- test/config.rb
+- test/helper.rb
+- test/migration_test.rb
+- test/migrations/duplicate/names/20100513073457_add_duplicate_survey_schema.rb
+- test/migrations/duplicate/names/20100513073724_add_duplicate_survey_schema.rb
+- test/migrations/duplicate/versions/20100513073457_add_another_duplicate_survey_schema.rb
+- test/migrations/duplicate/versions/20100513073457_add_duplicate_survey_schema.rb
+- test/migrations/valid/20100513054656_add_baseline_survey_schema.rb
+- test/migrations/valid/20100513063902_add_improvement_plan_survey_schema.rb
+- test/models/survey_schema.rb
+homepage: https://github.com/YotpoLtd/mongoid_migrations
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 1.8.6
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 1.3.6
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.8
+signing_key: 
+specification_version: 4
+summary: Data migrations for Rack applications with Mongoid
+test_files:
+- test/config.rb
+- test/helper.rb
+- test/migration_test.rb
+- test/migrations/duplicate/names/20100513073457_add_duplicate_survey_schema.rb
+- test/migrations/duplicate/names/20100513073724_add_duplicate_survey_schema.rb
+- test/migrations/duplicate/versions/20100513073457_add_another_duplicate_survey_schema.rb
+- test/migrations/duplicate/versions/20100513073457_add_duplicate_survey_schema.rb
+- test/migrations/valid/20100513054656_add_baseline_survey_schema.rb
+- test/migrations/valid/20100513063902_add_improvement_plan_survey_schema.rb
+- test/models/survey_schema.rb