3b8e29fe56
Signed-off-by: Jeremy Kemper <jeremy@bitsweat.net>
103 lines
2.7 KiB
Ruby
103 lines
2.7 KiB
Ruby
# TODO: Fix this RAILS_ENV stuff
|
|
RAILS_ENV = 'test' unless defined?(RAILS_ENV)
|
|
require 'abstract_unit'
|
|
|
|
module Rails
|
|
def self.root
|
|
@root ||= File.expand_path(File.join(File.dirname(__FILE__), '..', 'fixtures'))
|
|
end
|
|
end
|
|
Rails.application.config.root = Rails.root
|
|
|
|
require 'rails/generators'
|
|
require 'rubygems'
|
|
require 'active_record'
|
|
require 'action_dispatch'
|
|
|
|
CURRENT_PATH = File.expand_path(Dir.pwd)
|
|
Rails::Generators.no_color!
|
|
|
|
class GeneratorsTestCase < ActiveSupport::TestCase
|
|
include FileUtils
|
|
|
|
def destination_root
|
|
File.join(Rails.root, "tmp")
|
|
end
|
|
|
|
def setup
|
|
cd CURRENT_PATH
|
|
rm_rf(destination_root)
|
|
mkdir_p(destination_root)
|
|
end
|
|
|
|
def test_truth
|
|
# don't complain, test/unit
|
|
end
|
|
|
|
def capture(stream)
|
|
begin
|
|
stream = stream.to_s
|
|
eval "$#{stream} = StringIO.new"
|
|
yield
|
|
result = eval("$#{stream}").string
|
|
ensure
|
|
eval("$#{stream} = #{stream.upcase}")
|
|
end
|
|
|
|
result
|
|
end
|
|
alias :silence :capture
|
|
|
|
def assert_file(relative, *contents)
|
|
absolute = File.join(destination_root, relative)
|
|
assert File.exists?(absolute), "Expected file #{relative.inspect} to exist, but does not"
|
|
|
|
read = File.read(absolute) if block_given? || !contents.empty?
|
|
yield read if block_given?
|
|
|
|
contents.each do |content|
|
|
case content
|
|
when String
|
|
assert_equal content, read
|
|
when Regexp
|
|
assert_match content, read
|
|
end
|
|
end
|
|
end
|
|
|
|
def assert_no_file(relative)
|
|
absolute = File.join(destination_root, relative)
|
|
assert !File.exists?(absolute), "Expected file #{relative.inspect} to not exist, but does"
|
|
end
|
|
|
|
def assert_migration(relative, *contents, &block)
|
|
file_name = migration_file_name(relative)
|
|
assert file_name, "Expected migration #{relative} to exist, but was not found"
|
|
assert_file File.join(File.dirname(relative), file_name), *contents, &block
|
|
end
|
|
|
|
def assert_no_migration(relative)
|
|
file_name = migration_file_name(relative)
|
|
assert_nil file_name, "Expected migration #{relative} to not exist, but found #{file_name}"
|
|
end
|
|
|
|
def assert_class_method(content, method, &block)
|
|
assert_instance_method content, "self.#{method}", &block
|
|
end
|
|
|
|
def assert_instance_method(content, method)
|
|
assert content =~ /def #{method}(\(.+\))?(.*?)\n end/m, "Expected to have method #{method}"
|
|
yield $2.strip if block_given?
|
|
end
|
|
|
|
protected
|
|
|
|
def migration_file_name(relative)
|
|
absolute = File.join(destination_root, relative)
|
|
dirname, file_name = File.dirname(absolute), File.basename(absolute).sub(/\.rb$/, '')
|
|
|
|
migration = Dir.glob("#{dirname}/[0-9]*_*.rb").grep(/\d+_#{file_name}.rb$/).first
|
|
File.basename(migration) if migration
|
|
end
|
|
end
|