birdbath 1.3.4

data/.bundle/config

@@ -0,0 +1,2 @@
+--- 
+BUNDLE_DISABLE_SHARED_GEMS: "1"

data/.gitignore

@@ -0,0 +1,6 @@
+*.swp
+*.swo
+Gemfile.lock
+pkg/*
+*.gem
+.bundle

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in birdbath.gemspec
+gemspec

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2007 Micah Alles, Patrick Bacon, David Crosby
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,125 @@
+= Birdbath
+
+Birdbath provides methods which let you assert the current state of the schema and run your migrations against the _test_ database.
+
+== Install
+
+  gem install birdbath
+
+If you're using it outside of a Rails environment (for whatever reason) include the Birdbath module in your tests:
+
+  require 'test/unit'
+  require 'birdbath'
+
+  class MyTest < Test::Unit::TestCase
+    include Birdbath
+    
+    def test_something
+      ...
+    end
+  end
+
+== Use
+
+*assert_schema*: verifies the schema of the database exactly matches the one specified.
+
+  def test_the_schema
+    assert_schema do |s|
+      s.table :books do |t|
+        t.column :id,     :integer
+        t.column :title,  :string, :limit => 5
+        t.column :author, :string
+      end
+
+      s.table :reviews do |t|
+        t.column :id,      :integer
+        t.column :book_id, :integer
+        t.column :body,    :text
+        t.column :rating,  :integer, :default => 0
+        t.index  :book_id, :name => 'index_book_id_on_reviews'
+      end
+    end
+  end
+
+This would verify there are only two tables defined in the test database: _books_ and _reviews_ (schema_info is ignored).  It will also verify that the _book_ table has the three columns, _id_, _title_ and _author_, each with their respective types. Indexes are verified too.
+
+
+*assert_table*: verify a table is found exactly as specified:
+
+  assert_table :books do |t|
+    t.column :id,     :integer
+    t.column :title,  :string, :limit => 5
+    t.column :author, :string
+    t.index  :author, :name => 'index_author_on_books'
+  end
+
+
+*drop_all_tables*: does just what it says to your _test_ database.
+
+
+*migrate*: executes the migrations against the test database using the same mechanism as rake db:migrate.
+
+  def test_the_migrations
+    migrate
+    migrate :version => 0
+    migrate :version => 10
+    migrate
+  end
+  
+
+This would do the same thing as running the following rake commands, but within a test case:
+
+  rake db:migrate
+  rake db:migrate VERSION=0
+  rake db:migrate VERSION=10
+  rake db:migrate
+  
+
+By combining the two helpers you can write a test that shows you can run all your migrations and get the final schema:
+
+  def test_should_be_able_to_migrate_from_an_empty_schema
+    drop_all_tables
+
+    # we shouldn't have any tables
+    assert_schema do |s|
+    end
+
+    migrate  
+
+    assert_schema do |s|
+      s.table :books do |t|
+        t.column :id,     :integer
+        t.column :title,  :string
+        t.column :author, :string
+      end
+
+      s.table :reviews do |t|
+        t.column :id,      :integer
+        t.column :book_id, :integer
+        t.column :body,    :text
+        t.column :rating,  :integer
+        t.index  :book_id, :name => 'index_book_id_on_reviews'
+      end
+    end
+  end
+
+
+The *migrate* helper can also be useful for testing data tranformation migrations:
+
+  def test_should_get_rid_of_bad_data
+    drop_all_tables
+    migrate :version => 7
+    Book.reset_column_information
+    book = Book.create! :title => "bad title\nwith\todd    spacing"
+    migrate :version => 8 # should cleanse spacing in book titles
+    book.reload
+    assert_equal "bad title with odd spacing", book.title
+  end 
+
+== Authors
+
+* Micah Alles (alles@atomicobject.com)
+* David Crosby (crosby@atomicobject.com)
+* Patrick Bacon (bacon@atomicobject.com)
+* © 2007-2011 {Atomic Object}[http://www.atomicobject.com]
+* More Atomic Object {open source}[http://www.atomicobject.com/pages/Software+Commons] projects

data/Rakefile

@@ -0,0 +1,19 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks
+
+# GEM_VERSION = '1.3.3'
+# 
+# Hoe.new('birdbath', GEM_VERSION) do |p|
+#   p.author = "Micah Alles" 
+#   p.email = "micah@atomicobject.com" 
+#   p.url = "http://migrationtest.rubyforge.org" 
+#   p.summary = "A Rails plugin for testing migrations" 
+#   p.description = <<-EOS
+# birdbath makes testing your migrations easier by 
+# adding helper methods to Test::Unit::TestCase for asserting the 
+# current state of the schema and executing migrations against the 
+# test database.
+#   EOS
+# 
+# end
+# 

data/birdbath.gemspec

@@ -0,0 +1,24 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "birdbath/version"
+
+Gem::Specification.new do |s|
+  s.name        = "birdbath"
+  s.version     = Birdbath::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Micah Alles"]
+  s.email       = ["micah@atomicobject.com"]
+  s.homepage    = "http://github.com/atomicobject/birdbath"
+  s.summary     = %q{A Rails gem for testing migrations}
+  s.description = <<-EOS
+birdbath makes testing your migrations easier by 
+adding helper methods to Test::Unit::TestCase for asserting the 
+current state of the schema and executing migrations against the 
+test database.
+  EOS
+
+  s.rubyforge_project = "birdbath"
+
+  s.files         = `git ls-files`.split("\n")
+  s.require_paths = ["lib"]
+end

data/generators/full_migration_test/USAGE

@@ -0,0 +1,30 @@
+Description:
+    The controller generator creates stubs for a new controller and its views.
+
+    The generator takes a controller name and a list of views as arguments.
+    The controller name may be given in CamelCase or under_score and should
+    not be suffixed with 'Controller'.  To create a controller within a
+    module, specify the controller name as 'module/controller'.
+
+    The generator creates a controller class in app/controllers with view
+    templates in app/views/controller_name, a helper class in app/helpers,
+    and a functional test suite in test/functional.
+
+Example:
+    ./script/generate controller CreditCard open debit credit close
+
+    Credit card controller with URLs like /credit_card/debit.
+        Controller: app/controllers/credit_card_controller.rb
+        Views:      app/views/credit_card/debit.rhtml [...]
+        Helper:     app/helpers/credit_card_helper.rb
+        Test:       test/functional/credit_card_controller_test.rb
+
+Modules Example:
+    ./script/generate controller 'admin/credit_card' suspend late_fee
+
+    Credit card admin controller with URLs /admin/credit_card/suspend.
+        Controller: app/controllers/admin/credit_card_controller.rb
+        Views:      app/views/admin/credit_card/debit.rhtml [...]
+        Helper:     app/helpers/admin/credit_card_helper.rb
+        Test:       test/functional/admin/credit_card_controller_test.rb
+

data/generators/full_migration_test/full_migration_test_generator.rb

@@ -0,0 +1,10 @@
+class FullMigrationTestGenerator < Rails::Generator::Base
+  def manifest
+    record do |m|
+
+      m.directory 'test/migration'
+      m.template 'full_migration_test.rb',
+                 'test/migration/full_migration_test.rb'
+    end
+  end
+end

data/generators/full_migration_test/templates/full_migration_test.rb

@@ -0,0 +1,89 @@
+require File.expand_path(File.dirname(__FILE__) + '/../test_helper') 
+
+#
+# This excercises the full set of migrations for your Rails app.
+# It proves:
+#   - After full migration, the database is in the expected state, including:
+#     - All table structure
+#     - Default data (if any)
+#   - Full downward (version 0) migration functions correctly.
+#
+# YOU NEED TO:
+#   - Update "see_full_schema"
+#   - Update "see_data"
+# 
+class FullMigrationTest < ActionController::IntegrationTest
+
+  # 
+  # Transactional fixtures can, on occasion, cause migration tests to hang.
+  # Applying this setting here will turn transactional fixtures off for THIS
+  # SUITE ONLY
+  #
+  # self.use_transactional_fixtures = false
+
+  def conn
+    ActiveRecord::Base.connection
+  end
+
+  def see_empty_schema
+    assert_schema do |s|
+      # is nothing 
+    end
+  end
+
+  #
+  # Structure and Content assertions
+  #
+
+  # Fully assert db structure after full migration
+  def see_full_schema
+    # TODO: add assertions here to verify your schema was built
+    flunk "implement me"
+    
+    #
+    # Something like this can be used to see the entire schema
+    # is as expeted.
+    #
+    # assert_schema do |s|
+    #   s.table :cat_tails do |t|
+    #     t.column :id,    :integer
+    #     t.column :name,  :string
+    #   end
+    #
+    #   s.table :dogs do |t|
+    #     t.column :id,    :integer
+    #     t.column :name,  :string
+    #   end
+    # end
+
+    #
+    # Alternatively individual tables can be checked.
+    #
+    # assert_table :cats_tails do |s|
+    #   t.column :id,    :integer
+    #   t.column :name,  :string
+    # end
+  end
+
+  # Make sure data you expect your migrations to load are in there:
+  def see_default_data
+    # TODO: add assertions here to verify any default data was loaded
+  end
+
+  #
+  # TESTS
+  #
+
+  def test_full_migration
+    drop_all_tables
+    
+    see_empty_schema
+
+    migrate
+
+    see_full_schema
+
+    see_default_data
+  end
+
+end

data/generators/migration_test/migration_test_generator.rb

@@ -0,0 +1,33 @@
+class MigrationTestGenerator < Rails::Generator::NamedBase
+  def manifest
+    schema_version = class_name.to_i
+    raise "Invalid schema version '#{class_name}'" unless schema_version > 0
+
+    schema_version_string = schema_version.to_s.rjust(3,'0')
+
+    migration = Dir["db/migrate/#{schema_version_string}*.rb"].first
+    raise "No migration found for schema version #{schema_version_string}" unless migration
+
+    migration_name = File.basename(migration,'.rb').sub(/^\d+_/,'')
+    test_class_name = migration_name.camelize + "Test"
+    test_file = "test/migration/#{schema_version_string}_#{migration_name}_test.rb"
+
+    record do |m|
+      m.directory 'test/migration'
+      m.template 'migration_test.rb', test_file, :assigns => {
+        :test_class_name => test_class_name,
+        :migration_name => migration_name,
+        :schema_version => schema_version,
+        :previous_schema_version => previous_schema_version(schema_version_string)
+      }
+    end
+  end
+
+  def previous_schema_version(current_version_string)
+    versions = Dir["db/migrate/*.rb"].map do |migration_file| 
+      migration_file.match(/db\/migrate\/(\d+)_.*\.rb/)[1]
+    end.sort
+    prev_index = versions.index(current_version_string) - 1
+    prev_index < 0 ? -1 : versions[prev_index].to_i
+  end
+end

data/generators/migration_test/templates/migration_test.rb

@@ -0,0 +1,25 @@
+require File.expand_path(File.dirname(__FILE__) + '/../test_helper') 
+
+class <%= test_class_name %> < Test::Unit::TestCase
+ 
+  def teardown
+    migrate
+  end
+
+  def test_migration_should_<%= migration_name %>
+    drop_all_tables
+
+    migrate :version => <%= previous_schema_version %>
+
+    flunk "TODO: setup test data"
+
+    migrate :version => <%= schema_version %>
+
+    flunk "TODO: examine results"
+
+    migrate :version => <%= previous_schema_version %>
+
+    flunk "TODO: examine results"
+  end
+
+end

data/init.rb

@@ -0,0 +1,7 @@
+if RAILS_ENV == 'test'
+  require 'birdbath'
+  require 'test/unit'
+  Test::Unit::TestCase.class_eval do
+    include Birdbath
+  end
+end

data/install.rb

@@ -0,0 +1 @@
+# Install hook code here

data/lib/birdbath/version.rb

@@ -0,0 +1,3 @@
+module Birdbath
+  VERSION = "1.3.4"
+end

data/lib/birdbath.rb

@@ -0,0 +1,159 @@
+#--
+# Copyright (c) 2007 Micah Alles, Patrick Bacon, David Crosby
+# 
+# Permission is hereby granted, free of charge, to any person obtaining
+# a copy of this software and associated documentation files (the
+# "Software"), to deal in the Software without restriction, including
+# without limitation the rights to use, copy, modify, merge, publish,
+# distribute, sublicense, and/or sell copies of the Software, and to
+# permit persons to whom the Software is furnished to do so, subject to
+# the following conditions:
+# 
+# The above copyright notice and this permission notice shall be
+# included in all copies or substantial portions of the Software.
+# 
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+#++
+
+require 'test/unit/assertions'
+
+module Birdbath 
+  def self.migration_dir
+    @migration_dir || File.expand_path(RAILS_ROOT + '/db/migrate')
+  end
+
+  #
+  # sets the directory in which the migrations reside to be run with migrate
+  #
+  def self.migration_dir=(new_dir)
+    @migration_dir = new_dir
+  end
+
+  #
+  # verifies the schema exactly matches the one specified (schema_info does not have to be specified)
+  # 
+  #   assert_schema do |s|
+  #     s.table :dogs do |t|
+  #       t.column :id,   :integer, :default => 2
+  #       t.column :name, :string
+  #     end
+  #   end
+  #
+  def assert_schema
+    schema = Schema.new
+    yield schema
+    schema.verify
+  end
+
+  #
+  # verifies a single table exactly matches the one specified
+  # 
+  #   assert_table :dogs do |t|
+  #     t.column :id,   :integer, :default => 2
+  #     t.column :name, :string
+  #   end
+  #
+  def assert_table(name)
+    table = Table.new(name)
+    yield table
+    table.verify
+  end
+
+  #
+  # drops all tables in the database
+  # 
+  def drop_all_tables
+    ActiveRecord::Base.connection.tables.each do |table|
+      ActiveRecord::Base.connection.drop_table(table)
+    end
+  end
+
+  #
+  # executes your migrations
+  #
+  #   migrate                # same as rake db:migrate
+  #
+  # Options are:
+  # * :version - version to migrate to (same as VERSION=.. option to rake db:migrate)
+  # * :verbose - print migration status messages to STDOUT, defaults to false
+  # 
+  def migrate(opts={})
+    old_verbose = ActiveRecord::Migration.verbose
+    ActiveRecord::Migration.verbose = opts[:verbose].nil? ? false : opts[:verbose]
+    version = opts[:version] ? opts[:version].to_i : nil
+    ActiveRecord::Migrator.migrate(Birdbath.migration_dir, version)
+  ensure
+    ActiveRecord::Migration.verbose = old_verbose
+  end
+
+  module Connection #:nodoc:
+    def conn
+      ActiveRecord::Base.connection
+    end
+  end
+
+  class Schema
+    include Connection
+    include Test::Unit::Assertions
+
+    def initialize #:nodoc:
+      @tables = []
+    end
+
+    def table(name)
+      table = Table.new(name)
+      yield table
+      table.verify
+      @tables << table
+    end
+
+    def verify #:nodoc:
+      actual_tables = conn.tables.reject {|t| t == 'schema_info' || t == 'schema_migrations' }
+      expected_tables = @tables.map {|t| t.name }
+      assert_equal expected_tables.sort, actual_tables.sort, 'wrong tables in schema'
+    end
+  end
+
+  class Table
+    include Connection
+    include Test::Unit::Assertions
+    attr_reader :name
+
+    def initialize(name) #:nodoc:
+      @name = name.to_s
+      @columns = []
+      @indexes = []
+      assert conn.tables.include?(@name), "table <#{@name}> not found in schema"
+    end
+
+    def column(colname,type,options={})
+      colname = colname.to_s
+      @columns << colname
+      col = conn.columns(name).find {|c| c.name == colname }
+      assert_not_nil col, "column <#{colname}> not found in table <#{self.name}>"
+      assert_equal type, col.type, "wrong type for column <#{colname}> in table <#{name}>"
+      options.each do |k,v|
+        k = k.to_sym; actual = col.send(k); actual = actual.is_a?(String) ? actual.sub(/'$/,'').sub(/^'/,'') : actual
+        assert_equal v, actual, "column <#{colname}> in table <#{name}> has wrong :#{k}"
+      end
+    end
+
+    def index(column_name, options = {})
+      @indexes << "name <#{options[:name]}> columns <#{Array(column_name).join(",")}> unique <#{options[:unique] == true}>" 
+    end
+
+    def verify #:nodoc:
+      actual_columns = conn.columns(name).map {|c| c.name }
+      assert_equal @columns.sort, actual_columns.sort, "wrong columns for table: <#{name}>"
+
+      actual_indexes = conn.indexes(@name).collect { |i| "name <#{i.name}> columns <#{i.columns.join(",")}> unique <#{i.unique}>" }
+      assert_equal @indexes.sort, actual_indexes.sort, "wrong indexes for table: <#{name}>"
+    end
+  end
+end

data/tasks/birdbath_tasks.rake

@@ -0,0 +1,17 @@
+
+namespace :test do
+
+  Rake::TestTask.new(:migrations => "db:test:prepare") do |t|
+    t.libs << "test"
+    t.pattern = 'test/migration/**/*_test.rb'
+    t.verbose = true
+  end
+  Rake::Task['test:migrations'].comment = "Run the migration tests in test/migration"
+
+  task :migration => 'test:migrations'
+end
+
+task :test do
+  Rake::Task['test:migrations'].invoke rescue got_error = true
+  raise "Test failures" if got_error
+end

data/test/birdbath_test.rb

@@ -0,0 +1,263 @@
+require File.expand_path(File.dirname(__FILE__) + '/helper') 
+
+class BirdbathTest < Test::Unit::TestCase
+
+  def setup
+    load_default_schema
+    Birdbath.migration_dir = plugin_path('test/db/migrate_good')
+  end
+
+  #
+  # HELPERS
+  #
+  def see_failure(pattern='')
+    err = assert_raise(Test::Unit::AssertionFailedError) do
+      yield
+    end
+    assert_match(/#{pattern}/mi, err.message) 
+  end
+
+  def see_no_failure
+    assert_nothing_raised do
+      yield
+    end
+  end
+
+  def declare_columns_on_table(t)
+    t.column :id,   :integer
+    t.column :tail, :string, :default => 'top dog', :limit => 187
+  end
+  
+  #
+  # TESTS
+  #
+  def test_assert_schema_should_not_fail_if_schema_is_matched
+    see_no_failure do
+      assert_schema do |s|
+        s.table :dogs do |t|
+	  declare_columns_on_table(t)
+	  t.index :tail, :name => 'index_tail_on_dogs'
+        end
+
+      end
+    end
+  end
+
+  def test_assert_schema_should_fail_if_a_table_is_not_specified
+    see_failure 'wrong tables in schema.*dogs' do
+      assert_schema do |s|
+      end
+    end
+  end
+
+  def test_assert_schema_should_fail_if_a_table_is_not_found
+    see_failure 'table <things> not found in schema' do
+      assert_schema do |s|
+        s.table :things do |t|
+          t.column :id, :integer
+        end
+      end
+    end
+  end
+
+  def test_assert_schema_should_fail_if_a_column_is_not_specified
+    see_failure 'wrong columns for table.*dogs.*tail' do
+      assert_schema do |s|
+        s.table :dogs do |t|
+          t.column :id,   :integer
+        end
+      end
+    end
+  end
+
+  def test_assert_schema_should_fail_if_a_column_is_not_found
+    see_failure 'column <legs> not found in table <dogs>' do
+      assert_schema do |s|
+        s.table :dogs do |t|
+          t.column :id,   :integer
+          t.column :tail, :string
+          t.column :legs, :integer
+        end
+      end
+    end
+  end
+
+  def test_assert_schema_should_fail_if_wrong_options_not_specified
+    see_failure 'column <tail> in table <dogs> has wrong :default' do
+      assert_table :dogs do |t|
+        t.column :id,   :integer
+        t.column :tail, :string, :default => "blah"
+      end
+    end
+  end
+
+  def test_assert_table_should_not_fail_if_table_is_matched
+    see_no_failure do
+      assert_table :dogs do |t|
+        declare_columns_on_table(t)
+	t.index :tail, :name => 'index_tail_on_dogs'
+      end
+    end
+  end
+
+  def test_assert_table_should_fail_if_a_table_is_not_found
+    see_failure 'table <things> not found in schema' do
+      assert_table :things do |t|
+        t.column :id, :integer
+      end
+    end
+  end
+
+  def test_assert_table_should_fail_if_a_column_is_not_specified
+    see_failure 'wrong columns for table.*dogs.*tail' do
+      assert_table :dogs do |t|
+        t.column :id,   :integer
+      end
+    end
+  end
+
+  def test_assert_table_should_fail_if_a_column_is_not_found
+    see_failure 'column <legs> not found in table <dogs>' do
+      assert_table :dogs do |t|
+        t.column :id,   :integer
+        t.column :tail, :string
+        t.column :legs, :integer
+      end
+    end
+  end
+
+  def test_assert_table_should_fail_if_wrong_options_not_specified
+    see_failure 'column <tail> in table <dogs> has wrong :default' do
+      assert_table :dogs do |t|
+        t.column :id,   :integer
+        t.column :tail, :string, :default => "blah"
+      end
+    end
+  end
+
+  def test_assert_table_should_fail_if_an_index_is_not_specified
+    see_failure 'wrong indexes for table: <dogs>' do
+      assert_table :dogs do |t|
+        declare_columns_on_table(t)
+      end
+    end
+  end
+
+  def test_assert_schema_should_fail_if_a_column_in_an_index_is_not_found
+    see_failure 'wrong indexes for table: <dogs>' do
+      assert_table :dogs do |t|
+        declare_columns_on_table(t)
+	t.index :legs, :name => 'index_legs_on_dogs'
+      end
+    end
+  end
+
+  def test_assert_schema_should_fail_if_wrong_options_on_an_index
+    see_failure 'wrong indexes for table: <dogs>' do
+      assert_table :dogs do |t|
+        declare_columns_on_table(t)
+	t.index :tail, :name => 'index_tail_on_dogs', :unique => true
+      end
+    end
+  end
+
+  def test_should_drop_all_tables
+    if Rails::VERSION::MAJOR >= 2 && Rails::VERSION::MINOR >= 1
+      schema_table = 'schema_migrations'
+    else
+      schema_table = 'schema_info'
+    end
+    assert_equal ['dogs', schema_table].sort, ActiveRecord::Base.connection.tables.sort
+    drop_all_tables
+    assert_equal [], ActiveRecord::Base.connection.tables
+    drop_all_tables
+    assert_equal [], ActiveRecord::Base.connection.tables
+  end
+
+  def test_should_migrate_to_highest_version
+    drop_all_tables
+    assert_schema do |s|
+    end
+
+    migrate :version => 1
+
+    assert_schema do |s|
+      s.table :top_dogs do |t|
+        t.column :id,      :integer
+        t.column :name,    :string
+      end
+    end
+
+    migrate :version => 2
+
+    assert_schema do |s|
+      s.table :top_dogs do |t|
+        t.column :id,      :integer
+        t.column :name,    :string
+      end
+      s.table :bottom_dogs do |t|
+        t.column :id,      :integer
+        t.column :name,    :string
+        t.column :sick,    :boolean
+      end
+    end
+
+    migrate :version => 3
+
+    assert_schema do |s|
+      s.table :top_dogs do |t|
+        t.column :id,      :integer
+        t.column :name,    :string
+      end
+      s.table :bottom_dogs do |t|
+        t.column :id,      :integer
+        t.column :name,    :string
+        t.column :sick,    :boolean
+      end
+      s.table :cats do |t|
+        t.column :id,      :integer
+        t.column :lives,   :integer
+      end
+    end
+
+    migrate :version => 0
+
+    assert_schema do |s|
+    end
+
+    migrate
+
+    assert_schema do |s|
+      s.table :top_dogs do |t|
+        t.column :id,      :integer
+        t.column :name,    :string
+      end
+      s.table :bottom_dogs do |t|
+        t.column :id,      :integer
+        t.column :name,    :string
+        t.column :sick,    :boolean
+      end
+      s.table :cats do |t|
+        t.column :id,      :integer
+        t.column :lives,   :integer
+      end
+    end
+  end
+
+  def test_should_have_default_migration_dir_set
+    Birdbath.migration_dir = nil
+    assert_equal File.expand_path(RAILS_ROOT + '/db/migrate'), Birdbath.migration_dir, 
+      "wrong default migration dir"
+    
+  end
+
+  def test_should_raise_error_if_migration_fails
+    Birdbath.migration_dir = plugin_path('test/db/migrate_bad')
+    drop_all_tables
+    err = assert_raise RuntimeError do
+      migrate
+    end
+    assert_match(//i, err.message)  
+  end
+end
+

data/test/config/database.yml

@@ -0,0 +1,3 @@
+test:
+  adapter: sqlite3
+  database: vendor/plugins/birdbath/test/db/test.sqlite3.db

data/test/db/migrate_bad/001_create_top_dogs.rb

@@ -0,0 +1,8 @@
+class CreateTopDogs < ActiveRecord::Migration
+  def self.up
+    raise 'boom'
+  end
+
+  def self.down
+  end
+end

data/test/db/migrate_good/001_create_top_dogs.rb

@@ -0,0 +1,11 @@
+class CreateTopDogs < ActiveRecord::Migration
+  def self.up
+    create_table :top_dogs do |t|
+      t.column :name, :string
+    end
+  end
+
+  def self.down
+    drop_table :top_dogs
+  end
+end

data/test/db/migrate_good/002_create_bottom_dogs.rb

@@ -0,0 +1,12 @@
+class CreateBottomDogs < ActiveRecord::Migration
+  def self.up
+    create_table :bottom_dogs do |t|
+      t.column :name, :string
+      t.column :sick, :boolean
+    end
+  end
+
+  def self.down
+    drop_table :bottom_dogs
+  end
+end

data/test/db/migrate_good/003_create_cats.rb

@@ -0,0 +1,11 @@
+class CreateCats < ActiveRecord::Migration
+  def self.up
+    create_table :cats do |t|
+      t.column :lives, :integer
+    end
+  end
+
+  def self.down
+    drop_table :cats
+  end
+end

data/test/db/schema.rb

@@ -0,0 +1,6 @@
+ActiveRecord::Schema.define(:version => 1) do
+  create_table "dogs", :force => true do |t|
+    t.column "tail", :string, :default => 'top dog', :limit => 187
+  end
+  add_index :dogs, :tail, :name => 'index_tail_on_dogs'
+end

data/test/foundry_only_helper.rb

@@ -0,0 +1,15 @@
+class Test::Unit::TestCase
+
+  def self.running_in_foundry
+    File.expand_path(File.dirname(__FILE__)) =~ /\/rails_plugin_foundry\//
+  end
+
+  def running_in_foundry
+    self.class.running_in_foundry
+  end
+
+  def self.in_foundry_should(behave,&block)
+    should(behave,&block) if running_in_foundry
+  end
+
+end

data/test/full_migration_test_generator_test.rb

@@ -0,0 +1,52 @@
+require File.expand_path(File.dirname(__FILE__) + '/helper') 
+require File.expand_path(File.dirname(__FILE__) + '/foundry_only_helper') 
+
+class FullMigrationTestGeneratorTest < Test::Unit::TestCase
+
+  def setup
+    return unless running_in_foundry
+    rm_migration_test_path
+  end
+
+  def teardown
+    return unless running_in_foundry
+    rm_migration_test_path
+  end
+
+
+  #
+  # HELPERS
+  #
+
+  def migration_test_file
+    migration_test_path + "/full_migration_test.rb"
+  end
+  
+  def run_generator
+    run_in_rails_root 'script/generate full_migration_test'
+  end
+
+  #
+  # TESTS
+  #
+
+  in_foundry_should "generate full migration test" do
+    run_generator
+    assert File.exists?(migration_test_file), "No test file made: #{migration_test_file}"
+    generated_code = File.read(migration_test_file)
+    source_code = File.read(plugin_path('generators/full_migration_test/templates/full_migration_test.rb'))
+    assert_equal source_code, generated_code, "Wrong code generated in #{migration_test_file}"
+  end
+
+  in_foundry_should "create a test that fails in expected fashion" do
+    run_generator
+    cd RAILS_ROOT do
+      @output = `ruby test/migration/full_migration_test.rb`
+    end
+
+    check_output '1 tests, 1 assertions, 1 failures, 0 errors'
+    check_output 'implement me'
+    check_output 'test_full_migration(FullMigrationTest)'
+  end
+
+end

data/test/helper.rb

@@ -0,0 +1,75 @@
+ENV["RAILS_ENV"] = "test"
+require File.expand_path(File.dirname(__FILE__) + '/../../../../config/environment')
+require 'logger'
+require 'fileutils'
+require 'test_help'
+
+plugin_path = RAILS_ROOT + "/vendor/plugins/birdbath"
+
+config_location = File.expand_path(plugin_path + "/test/config/database.yml")
+
+config = YAML::load(ERB.new(IO.read(config_location)).result)
+ActiveRecord::Base.logger = Logger.new(plugin_path + "/test/log/test.log")
+ActiveRecord::Base.establish_connection(config['test'])
+
+Test::Unit::TestCase.fixture_path = plugin_path + "/test/fixtures/"
+
+$LOAD_PATH.unshift(Test::Unit::TestCase.fixture_path)
+
+class Test::Unit::TestCase
+  include FileUtils
+  def plugin_path(path)
+    File.expand_path(File.dirname(__FILE__) + '/../' + path)  
+  end
+
+  def load_default_schema
+    ActiveRecord::Migration.suppress_messages do
+      schema_file = plugin_path("/test/db/schema.rb")
+      load(schema_file) if File.exist?(schema_file)
+    end
+  end
+
+  def migration_test_path
+    File.expand_path(RAILS_ROOT + "/test/migration")
+  end
+
+  def run_in_rails_root(command)
+    cd RAILS_ROOT do
+      @output = `#{command}`
+    end
+  end
+
+  def check_output(string_or_regexp)
+    assert_not_nil @output, "No output collected"
+    case string_or_regexp
+    when String
+      assert_match(/#{Regexp.escape(string_or_regexp)}/, @output)
+    when Regexp
+      assert_match(string_or_regexp, @output)
+    else
+      raise "Can't check output using oddball object #{string_or_regexp.inspect}"
+    end
+  end
+  
+  def rm_migration_test_path
+    rm_rf migration_test_path
+  end
+
+  def self.should(behave,&block)
+    return unless running_in_foundry
+    @context ||= nil
+    @context_setup ||= nil
+    context_string = @context.nil? ? '' : @context + ' '
+    mname = "test #{context_string}should #{behave}"
+    context_setup = @context_setup
+    if block_given?
+      define_method(mname) do
+        instance_eval(&context_setup) unless context_setup.nil?
+        instance_eval(&block)
+      end 
+    else
+      puts ">>> UNIMPLEMENTED CASE: #{name.sub(/Test$/,'')} should #{behave}"
+    end
+  end
+
+end

data/test/migration_test_generator_test.rb

@@ -0,0 +1,91 @@
+require File.expand_path(File.dirname(__FILE__) + '/helper') 
+require File.expand_path(File.dirname(__FILE__) + '/foundry_only_helper') 
+
+class MigrationTestGeneratorTest < Test::Unit::TestCase
+
+  def setup
+    return unless running_in_foundry
+    rm_migration_test_path
+    rm_migrations   
+  end
+
+  def teardown
+    return unless running_in_foundry
+    rm_migration_test_path
+    rm_migrations   
+  end
+
+
+  #
+  # HELPERS
+  #
+
+  def generate_migration(name)
+    run_in_rails_root "script/generate migration #{name}"
+    @output =~ /create  db\/migrate\/(\d+)_.*\.rb/
+    migration = $1
+    migration_file = "db/migrate/#{migration}_#{name}.rb"
+    assert File.exists?(migration_file), "Migration not there: #{migration_file}"
+    migration
+  end
+
+  def generate_migration_test(version)
+    run_in_rails_root "script/generate migration_test #{version}"
+  end
+
+  def rm_migrations
+    rm Dir["db/migrate/*.rb"]
+  end
+  
+  #
+  # TESTS
+  #
+  
+  in_foundry_should "generate a migration test for schema 1" do
+    migration = generate_migration 'create_the_dogs_table'
+
+    generate_migration_test migration
+
+    migration_test_file = "test/migration/#{migration}_create_the_dogs_table_test.rb"
+    assert File.exists?(migration_test_file), "Migration test not there: #{migration_test_file}"
+
+    run_in_rails_root "ruby #{migration_test_file}"
+    check_output 'test_migration_should_create_the_dogs_table(CreateTheDogsTableTest)'
+    check_output '1) Failure:'
+    check_output 'TODO: setup test data.'
+    check_output '1 tests, 1 assertions, 1 failures, 0 errors'
+  end
+
+  in_foundry_should "generate a migration test for schema 2" do
+    generate_migration 'create_the_dogs_table'
+    migration = generate_migration 'transfer_cats_data'
+    generate_migration_test migration
+    migration_test_file = "test/migration/#{migration}_transfer_cats_data_test.rb"
+    assert File.exists?(migration_test_file), "Migration test not there: #{migration_test_file}"
+
+    run_in_rails_root "ruby #{migration_test_file}"
+    check_output 'test_migration_should_transfer_cats_data(TransferCatsDataTest)'
+    check_output '1) Failure:'
+    check_output 'TODO: setup test data.'
+    check_output '1 tests, 1 assertions, 1 failures, 0 errors'
+  end
+
+  in_foundry_should "fail if no schema version is provided" do
+    generate_migration 'create_the_dogs_table'
+    generate_migration_test ''
+    check_output 'Usage: script/generate migration_test MigrationTestName [options]'
+  end
+
+  in_foundry_should "fail if bad schema version is provided" do
+    generate_migration 'create_the_dogs_table'
+    generate_migration_test 'oops'
+    check_output(/Invalid schema version 'oops'/i)
+  end
+
+  in_foundry_should "fail if schema version does not correspond to existing migration" do
+    migration = generate_migration 'create_the_dogs_table'
+    generate_migration_test '2'
+    check_output "No migration found for schema version 002"
+  end
+
+end

data/uninstall.rb

@@ -0,0 +1 @@
+# Uninstall hook code here

metadata

@@ -0,0 +1,78 @@
+--- !ruby/object:Gem::Specification
+name: birdbath
+version: !ruby/object:Gem::Version
+  version: 1.3.4
+  prerelease: 
+platform: ruby
+authors:
+- Micah Alles
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-08-09 00:00:00.000000000 -04:00
+default_executable: 
+dependencies: []
+description: ! "birdbath makes testing your migrations easier by \nadding helper methods
+  to Test::Unit::TestCase for asserting the \ncurrent state of the schema and executing
+  migrations against the \ntest database.\n"
+email:
+- micah@atomicobject.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .bundle/config
+- .gitignore
+- Gemfile
+- LICENSE
+- README.rdoc
+- Rakefile
+- birdbath.gemspec
+- generators/full_migration_test/USAGE
+- generators/full_migration_test/full_migration_test_generator.rb
+- generators/full_migration_test/templates/full_migration_test.rb
+- generators/migration_test/migration_test_generator.rb
+- generators/migration_test/templates/migration_test.rb
+- init.rb
+- install.rb
+- lib/birdbath.rb
+- lib/birdbath/version.rb
+- tasks/birdbath_tasks.rake
+- test/birdbath_test.rb
+- test/config/database.yml
+- test/db/migrate_bad/001_create_top_dogs.rb
+- test/db/migrate_good/001_create_top_dogs.rb
+- test/db/migrate_good/002_create_bottom_dogs.rb
+- test/db/migrate_good/003_create_cats.rb
+- test/db/schema.rb
+- test/foundry_only_helper.rb
+- test/full_migration_test_generator_test.rb
+- test/helper.rb
+- test/migration_test_generator_test.rb
+- uninstall.rb
+has_rdoc: true
+homepage: http://github.com/atomicobject/birdbath
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: birdbath
+rubygems_version: 1.6.2
+signing_key: 
+specification_version: 3
+summary: A Rails gem for testing migrations
+test_files: []