module RSpec::Generator
  module Helpers
    def check_migration(file_name, &block)
      self.should generate_file(file_name, :migration) do |content|  
        yield content if block
      end
    end
    alias_method :migration, :check_migration
  end
end

module RSpec::RubyContent
  module Helpers
    def check_up &block
      self.should have_up, &block
    end
    alias_method :up, :check_up      

    def check_down &block
      self.should have_down, &block 
    end
    alias_method :down, :check_down      

    def check_create_table name, &block
      self.should create_table name, &block
    end
    alias_method :creates_table, :check_create_table      

    def check_change_table name, &block
      self.should change_table name, &block
    end
    alias_method :changes_table, :check_change_table      

    def check_drop_table name
      self.should drop_table name, &block
    end
    alias_method :drops_table, :check_drop_table

    def check_rename_table name
      self.should rename_table name, &block
    end
    alias_method :renames_table, :check_rename_table
    
    def check_remove_column name
      self.should remove_colum name, &block
    end
    alias_method :removes_column, :check_remove_column

    def check_add_column name, type='string'
      self.should add_column name, type, &block
    end
    alias_method :adds_column, :check_add_column

    def check_change_column name, type='string'
      self.should change_column name, type, &block
    end
    alias_method :changes_column, :check_change_column

    def check_add_tbl_column tbl_name, name, type='string'
      self.should add_tbl_column tbl_name, name, type
    end
    alias_method :adds_tbl_column, :check_add_tbl_column      

    def check_change_tbl_column tbl_name, name, type='string'
      self.should change_tbl_column tbl_name, name, type
    end
    alias_method :changes_tbl_column, :check_change_tbl_column      

    def check_rename_tbl_column tbl_name, new_name
      self.should rename_tbl_column tbl_name, new_name
    end
    alias_method :renames_tbl_column, :check_rename_tbl_column      

    def check_remove_tbl_column tbl_name, name
      self.should remove_tbl_column tbl_name, name
    end
    alias_method :removes_tbl_column, :check_remove_tbl_column      

    def check_add_tbl_index tbl_name, name
      self.should add_tbl_index tbl_name, name
    end
    alias_method :adds_tbl_index, :check_add_tbl_index      

    def check_remove_tbl_index tbl_name, name
      self.should remove_tbl_index tbl_name, name
    end
    alias_method :removes_tbl_index, :check_remove_tbl_index
  end
end