107 lines
2.9 KiB
Ruby
107 lines
2.9 KiB
Ruby
require 'test/unit'
|
|
require 'fileutils'
|
|
|
|
fixtures = File.expand_path(File.join(File.dirname(__FILE__), '..', 'fixtures'))
|
|
if defined?(RAILS_ROOT)
|
|
RAILS_ROOT.replace fixtures
|
|
else
|
|
RAILS_ROOT = fixtures
|
|
end
|
|
|
|
$LOAD_PATH.unshift "#{File.dirname(__FILE__)}/../../lib"
|
|
require 'rails/generators'
|
|
|
|
require 'rubygems'
|
|
$LOAD_PATH.unshift "#{File.dirname(__FILE__)}/../../../activerecord/lib"
|
|
$LOAD_PATH.unshift "#{File.dirname(__FILE__)}/../../../actionpack/lib"
|
|
require 'active_record'
|
|
require 'action_dispatch'
|
|
|
|
CURRENT_PATH = File.expand_path(Dir.pwd)
|
|
Rails::Generators.no_color!
|
|
|
|
class GeneratorsTestCase < Test::Unit::TestCase
|
|
include FileUtils
|
|
|
|
def destination_root
|
|
@destination_root ||= File.expand_path(File.join(File.dirname(__FILE__),
|
|
'..', 'fixtures', '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
|