135 lines
3.4 KiB
Ruby
135 lines
3.4 KiB
Ruby
module ModelBuilder
|
|
TMP_VIEW_PATH =
|
|
File.expand_path(File.join(File.dirname(__FILE__), 'rails3_root', 'tmp', 'views')).freeze
|
|
|
|
def self.included(example_group)
|
|
example_group.class_eval do
|
|
before do
|
|
@created_tables ||= []
|
|
@defined_constants ||= []
|
|
end
|
|
|
|
after { teardown_defined_constants }
|
|
end
|
|
end
|
|
|
|
def create_table(table_name, &block)
|
|
connection = ActiveRecord::Base.connection
|
|
|
|
begin
|
|
connection.execute("DROP TABLE IF EXISTS #{table_name}")
|
|
connection.create_table(table_name, &block)
|
|
@created_tables << table_name
|
|
connection
|
|
rescue Exception => e
|
|
connection.execute("DROP TABLE IF EXISTS #{table_name}")
|
|
raise e
|
|
end
|
|
end
|
|
|
|
def define_constant(class_name, base, &block)
|
|
class_name = class_name.to_s.camelize
|
|
|
|
klass = Class.new(base)
|
|
Object.const_set(class_name, klass)
|
|
|
|
klass.class_eval(&block) if block_given?
|
|
|
|
@defined_constants << class_name
|
|
|
|
klass
|
|
end
|
|
|
|
def define_model_class(class_name, &block)
|
|
define_constant(class_name, ActiveRecord::Base, &block)
|
|
end
|
|
|
|
def define_model(name, columns = {}, &block)
|
|
class_name = name.to_s.pluralize.classify
|
|
table_name = class_name.tableize
|
|
|
|
create_table(table_name) do |table|
|
|
columns.each do |name, type|
|
|
table.column name, type
|
|
end
|
|
end
|
|
|
|
define_model_class(class_name, &block)
|
|
end
|
|
|
|
def define_mailer(name, paths, &block)
|
|
class_name = name.to_s.pluralize.classify
|
|
klass = define_constant(class_name, ActionMailer::Base, &block)
|
|
end
|
|
|
|
def define_controller(class_name, &block)
|
|
class_name = class_name.to_s
|
|
class_name << 'Controller' unless class_name =~ /Controller$/
|
|
define_constant(class_name, ActionController::Base, &block)
|
|
end
|
|
|
|
def define_routes(&block)
|
|
Rails.application.routes.draw(&block)
|
|
@routes = Rails.application.routes
|
|
class << self
|
|
include ActionDispatch::Assertions
|
|
end
|
|
end
|
|
|
|
def build_response(opts = {}, &block)
|
|
action = opts[:action] || 'example'
|
|
klass = define_controller('Examples')
|
|
block ||= lambda { render :nothing => true }
|
|
klass.class_eval { layout false; define_method(action, &block) }
|
|
define_routes do |map|
|
|
map.connect 'examples', :controller => 'examples', :action => action
|
|
end
|
|
|
|
create_view("examples/#{action}.html.erb", "abc")
|
|
klass.view_paths = [TMP_VIEW_PATH]
|
|
|
|
@controller = klass.new
|
|
@request = ActionController::TestRequest.new
|
|
@response = ActionController::TestResponse.new
|
|
@controller.send :assign_shortcuts, @request, @response
|
|
@controller.send :initialize_template_class, @response
|
|
|
|
class << self
|
|
include ActionController::TestCase::Behavior
|
|
end
|
|
@routes = Rails.application.routes
|
|
|
|
get action
|
|
|
|
@controller
|
|
end
|
|
|
|
def create_view(path, contents)
|
|
full_path = File.join(TMP_VIEW_PATH, path)
|
|
FileUtils.mkdir_p(File.dirname(full_path))
|
|
File.open(full_path, 'w') { |file| file.write(contents) }
|
|
end
|
|
|
|
def teardown_defined_constants
|
|
@defined_constants.each do |class_name|
|
|
Object.send(:remove_const, class_name)
|
|
end
|
|
@defined_constants = []
|
|
|
|
@created_tables.each do |table_name|
|
|
ActiveRecord::Base.
|
|
connection.
|
|
execute("DROP TABLE IF EXISTS #{table_name}")
|
|
end
|
|
|
|
FileUtils.rm_rf(TMP_VIEW_PATH)
|
|
|
|
Rails.application.routes_reloader.reload!
|
|
end
|
|
end
|
|
|
|
RSpec.configure do |config|
|
|
config.include ModelBuilder
|
|
end
|
|
|