ardm-migrations 1.2.0

data/lib/dm-migrations/sql/table_modifier.rb

@@ -0,0 +1,57 @@
+module SQL
+  class TableModifier
+    extend DataMapper::Property::Lookup
+
+    attr_accessor :table_name, :opts, :statements, :adapter
+
+    def initialize(adapter, table_name, opts = {}, &block)
+      @adapter = adapter
+      @table_name = table_name.to_s
+      @opts = (opts)
+
+      @statements = []
+
+      self.instance_eval &block
+    end
+
+    def add_column(name, type, opts = {})
+      column = SQL::TableCreator::Column.new(@adapter, name, type, opts)
+      @statements << "ALTER TABLE #{quoted_table_name} ADD COLUMN #{column.to_sql}"
+    end
+
+    def drop_column(name)
+      # raise NotImplemented for SQLite3. Can't ALTER TABLE, need to copy table.
+      # We'd have to inspect it, and we can't, since we aren't executing any queries yet.
+      # TODO instead of building the SQL queries when executing the block, create AddColumn,
+      # AlterColumn and DropColumn objects that get #to_sql'd
+      if name.is_a?(Array)
+        name.each{ |n| drop_column(n) }
+      else
+        @statements << "ALTER TABLE #{quoted_table_name} DROP COLUMN #{quote_column_name(name)}"
+      end
+    end
+    alias_method :drop_columns, :drop_column
+
+    def rename_column(name, new_name, opts = {})
+      # raise NotImplemented for SQLite3
+      @statements << "ALTER TABLE #{quoted_table_name} RENAME COLUMN #{quote_column_name(name)} TO #{quote_column_name(new_name)}"
+    end
+
+    def change_column(name, type, opts = {})
+      column = SQL::TableCreator::Column.new(@adapter, name, type, opts)
+      @statements << @adapter.change_column_type_statement(table_name, column)
+    end
+
+    def quote_column_name(name)
+      @adapter.send(:quote_name, name.to_s)
+    end
+
+    def quoted_table_name
+      @adapter.send(:quote_name, table_name)
+    end
+
+    def to_sql
+      @statements.join(';')
+    end
+  end
+end

data/lib/dm-migrations/sql.rb

@@ -0,0 +1,5 @@
+require 'dm-migrations/sql/table_creator'
+require 'dm-migrations/sql/table_modifier'
+require 'dm-migrations/sql/sqlite'
+require 'dm-migrations/sql/mysql'
+require 'dm-migrations/sql/postgres'

data/lib/dm-migrations/version.rb

@@ -0,0 +1,5 @@
+module DataMapper
+  module Migrations
+    VERSION = '1.2.0'
+  end
+end

data/lib/dm-migrations.rb

@@ -0,0 +1,3 @@
+require 'dm-core'
+require 'dm-migrations/migration'
+require 'dm-migrations/auto_migration'

data/lib/spec/example/migration_example_group.rb

@@ -0,0 +1,73 @@
+require 'spec/matchers/migration_matchers'
+
+require 'spec'
+
+module Spec
+  module Example
+    class MigrationExampleGroup < Spec::Example::ExampleGroup
+      include Spec::Matchers::Migration
+
+      before(:all) do
+        if this_migration.adapter.supports_schema_transactions?
+          run_prereq_migrations
+        end
+      end
+
+      before(:each) do
+        if ! this_migration.adapter.supports_schema_transactions?
+          run_prereq_migrations
+        else
+          this_migration.adapter.begin_transaction
+        end
+      end
+
+      after(:each) do
+        if this_migration.adapter.supports_schema_transactions?
+          this_migration.adapter.rollback_transaction
+        end
+      end
+
+      after(:all) do
+        this_migration.adapter.recreate_database
+      end
+
+      def run_prereq_migrations
+        "running n-1 migrations"
+        all_databases.each do |db|
+          db.adapter.recreate_database
+        end
+        @@migrations.sort.each do |migration|
+          break if migration.name.to_s == migration_name.to_s
+          migration.perform_up
+        end
+      end
+
+      def run_migration
+        this_migration.perform_up
+      end
+
+      def migration_name
+        @migration_name ||= self.class.instance_variable_get("@description_text").to_s
+      end
+
+      def all_databases
+        @@migrations.map { |m| m.database }.uniq
+      end
+
+      def this_migration
+        @@migrations.select { |m| m.name.to_s == migration_name }.first
+      end
+
+      def select(sql)
+        this_migration.adapter.select(sql)
+      end
+
+      def table(table_name)
+        this_migration.adapter.table(table_name)
+      end
+
+      Spec::Example::ExampleGroupFactory.register(:migration, self)
+
+    end
+  end
+end

data/lib/spec/matchers/migration_matchers.rb

@@ -0,0 +1,106 @@
+module Spec
+  module Matchers
+    module Migration
+
+      def have_table(table_name)
+        HaveTableMatcher.new(table_name)
+      end
+
+      def have_column(column_name)
+        HaveColumnMatcher.new(column_name)
+      end
+
+      def permit_null
+        NullableColumnMatcher.new
+      end
+
+      def be_primary_key
+        PrimaryKeyMatcher.new
+      end
+
+      class HaveTableMatcher
+
+        attr_accessor :table_name, :repository
+
+        def initialize(table_name)
+          @table_name = table_name
+        end
+
+        def matches?(repository)
+          repository.adapter.storage_exists?(table_name)
+        end
+
+        def failure_message
+          %(expected #{repository} to have table '#{table_name}')
+        end
+
+        def negative_failure_message
+          %(expected #{repository} to not have table '#{table_name}')
+        end
+
+      end
+
+      class HaveColumnMatcher
+
+        attr_accessor :table, :column_name
+
+        def initialize(column_name)
+          @column_name = column_name
+        end
+
+        def matches?(table)
+          @table = table
+          table.columns.map { |c| c.name }.include?(column_name.to_s)
+        end
+
+        def failure_message
+          %(expected #{table} to have column '#{column_name}')
+        end
+
+        def negative_failure_message
+          %(expected #{table} to not have column '#{column_name}')
+        end
+
+      end
+
+      class NullableColumnMatcher
+
+        attr_accessor :column
+
+        def matches?(column)
+          @column = column
+          ! column.not_null
+        end
+
+        def failure_message
+          %(expected #{column.name} to permit NULL)
+        end
+
+        def negative_failure_message
+          %(expected #{column.name} to be NOT NULL)
+        end
+
+      end
+
+      class PrimaryKeyMatcher
+
+        attr_accessor :column
+
+        def matches?(column)
+          @column = column
+          column.primary_key
+        end
+
+        def failure_message
+          %(expected #{column.name} to be PRIMARY KEY)
+        end
+
+        def negative_failure_message
+          %(expected #{column.name} to not be PRIMARY KEY)
+        end
+
+      end
+
+    end
+  end
+end