2010-12-13 22:28:59 +00:00
|
|
|
module ModelBuilder
|
2010-02-13 03:05:12 +00:00
|
|
|
TMP_VIEW_PATH =
|
2011-05-25 19:34:27 +00:00
|
|
|
File.expand_path(File.join(TESTAPP_ROOT, 'tmp', 'views')).freeze
|
2010-02-13 03:05:12 +00:00
|
|
|
|
2010-12-13 22:28:59 +00:00
|
|
|
def self.included(example_group)
|
|
|
|
example_group.class_eval do
|
|
|
|
before do
|
|
|
|
@created_tables ||= []
|
|
|
|
end
|
|
|
|
|
2012-03-09 16:53:24 +00:00
|
|
|
after do
|
|
|
|
teardown_defined_constants
|
|
|
|
end
|
2010-12-13 22:28:59 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2011-09-12 10:25:31 +00:00
|
|
|
def create_table(table_name, options = {}, &block)
|
2008-12-15 18:13:56 +00:00
|
|
|
connection = ActiveRecord::Base.connection
|
2010-06-22 18:14:52 +00:00
|
|
|
|
2008-12-15 18:13:56 +00:00
|
|
|
begin
|
|
|
|
connection.execute("DROP TABLE IF EXISTS #{table_name}")
|
2011-09-12 10:25:31 +00:00
|
|
|
connection.create_table(table_name, options, &block)
|
2009-01-24 21:23:11 +00:00
|
|
|
@created_tables << table_name
|
2009-01-25 21:19:07 +00:00
|
|
|
connection
|
2008-12-15 18:13:56 +00:00
|
|
|
rescue Exception => e
|
2009-01-24 21:23:11 +00:00
|
|
|
connection.execute("DROP TABLE IF EXISTS #{table_name}")
|
2008-12-15 18:13:56 +00:00
|
|
|
raise e
|
|
|
|
end
|
2009-01-24 21:23:11 +00:00
|
|
|
end
|
2008-12-15 18:13:56 +00:00
|
|
|
|
2012-03-16 16:15:11 +00:00
|
|
|
def define_constant(class_name, base = Object, &block)
|
2009-02-03 23:22:57 +00:00
|
|
|
class_name = class_name.to_s.camelize
|
|
|
|
|
2012-03-09 16:53:24 +00:00
|
|
|
Class.new(base).tap do |constant_class|
|
|
|
|
Object.const_set(class_name, constant_class)
|
|
|
|
constant_class.unloadable
|
2008-12-15 18:13:56 +00:00
|
|
|
|
2012-03-09 16:53:24 +00:00
|
|
|
if block_given?
|
|
|
|
constant_class.class_eval(&block)
|
|
|
|
end
|
2008-12-15 18:13:56 +00:00
|
|
|
|
2012-03-09 16:53:24 +00:00
|
|
|
if constant_class.respond_to?(:reset_column_information)
|
|
|
|
constant_class.reset_column_information
|
|
|
|
end
|
|
|
|
end
|
2008-12-15 18:13:56 +00:00
|
|
|
end
|
|
|
|
|
2009-01-31 23:42:48 +00:00
|
|
|
def define_model_class(class_name, &block)
|
|
|
|
define_constant(class_name, ActiveRecord::Base, &block)
|
|
|
|
end
|
|
|
|
|
2011-01-24 16:19:34 +00:00
|
|
|
def define_active_model_class(class_name, options = {}, &block)
|
|
|
|
define_constant(class_name, Object) do
|
|
|
|
include ActiveModel::Validations
|
|
|
|
|
|
|
|
options[:accessors].each do |column|
|
|
|
|
attr_accessor column.to_sym
|
|
|
|
end
|
|
|
|
|
2012-03-09 16:53:24 +00:00
|
|
|
if block_given?
|
|
|
|
class_eval(&block)
|
|
|
|
end
|
2011-01-24 16:19:34 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2009-01-24 21:23:11 +00:00
|
|
|
def define_model(name, columns = {}, &block)
|
|
|
|
class_name = name.to_s.pluralize.classify
|
|
|
|
table_name = class_name.tableize
|
|
|
|
|
2009-01-24 21:34:09 +00:00
|
|
|
create_table(table_name) do |table|
|
2009-01-24 21:23:11 +00:00
|
|
|
columns.each do |name, type|
|
|
|
|
table.column name, type
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
define_model_class(class_name, &block)
|
|
|
|
end
|
|
|
|
|
2010-06-08 16:15:52 +00:00
|
|
|
def define_mailer(name, paths, &block)
|
|
|
|
class_name = name.to_s.pluralize.classify
|
2012-03-09 16:53:24 +00:00
|
|
|
define_constant(class_name, ActionMailer::Base, &block)
|
2010-06-08 16:15:52 +00:00
|
|
|
end
|
|
|
|
|
2009-01-31 23:42:48 +00:00
|
|
|
def define_controller(class_name, &block)
|
2009-02-03 23:22:57 +00:00
|
|
|
class_name = class_name.to_s
|
|
|
|
class_name << 'Controller' unless class_name =~ /Controller$/
|
2009-01-31 23:42:48 +00:00
|
|
|
define_constant(class_name, ActionController::Base, &block)
|
|
|
|
end
|
|
|
|
|
|
|
|
def define_routes(&block)
|
2010-06-07 21:18:36 +00:00
|
|
|
Rails.application.routes.draw(&block)
|
2010-12-13 22:28:59 +00:00
|
|
|
@routes = Rails.application.routes
|
|
|
|
class << self
|
|
|
|
include ActionDispatch::Assertions
|
|
|
|
end
|
2009-01-31 23:42:48 +00:00
|
|
|
end
|
|
|
|
|
2010-02-13 03:05:12 +00:00
|
|
|
def build_response(opts = {}, &block)
|
|
|
|
action = opts[:action] || 'example'
|
2011-10-15 11:33:51 +00:00
|
|
|
partial = opts[:partial] || '_partial'
|
2009-01-31 23:42:48 +00:00
|
|
|
klass = define_controller('Examples')
|
|
|
|
block ||= lambda { render :nothing => true }
|
2010-06-07 21:18:36 +00:00
|
|
|
klass.class_eval { layout false; define_method(action, &block) }
|
2010-12-14 23:35:12 +00:00
|
|
|
define_routes do
|
|
|
|
match 'examples', :to => "examples##{action}"
|
2009-02-05 03:04:08 +00:00
|
|
|
end
|
2009-01-31 23:42:48 +00:00
|
|
|
|
2010-02-13 03:05:12 +00:00
|
|
|
create_view("examples/#{action}.html.erb", "abc")
|
2011-10-15 11:33:51 +00:00
|
|
|
create_view("examples/#{partial}.html.erb", "partial")
|
2010-02-13 03:05:12 +00:00
|
|
|
klass.view_paths = [TMP_VIEW_PATH]
|
|
|
|
|
2009-01-31 23:42:48 +00:00
|
|
|
@controller = klass.new
|
|
|
|
@request = ActionController::TestRequest.new
|
|
|
|
@response = ActionController::TestResponse.new
|
2010-06-07 21:18:36 +00:00
|
|
|
@controller.send :assign_shortcuts, @request, @response
|
|
|
|
@controller.send :initialize_template_class, @response
|
2010-12-13 22:28:59 +00:00
|
|
|
|
|
|
|
class << self
|
|
|
|
include ActionController::TestCase::Behavior
|
|
|
|
end
|
|
|
|
@routes = Rails.application.routes
|
|
|
|
|
2010-02-13 03:05:12 +00:00
|
|
|
get action
|
2009-01-31 23:42:48 +00:00
|
|
|
|
|
|
|
@controller
|
|
|
|
end
|
|
|
|
|
2010-02-13 03:05:12 +00:00
|
|
|
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
|
|
|
|
|
2010-12-13 22:28:59 +00:00
|
|
|
def teardown_defined_constants
|
2010-12-14 23:35:12 +00:00
|
|
|
ActiveSupport::Dependencies.clear
|
2009-01-24 21:23:11 +00:00
|
|
|
|
2012-03-09 16:53:24 +00:00
|
|
|
connection = ActiveRecord::Base.connection
|
|
|
|
|
2010-12-13 22:28:59 +00:00
|
|
|
@created_tables.each do |table_name|
|
2012-03-09 16:53:24 +00:00
|
|
|
connection.execute("DROP TABLE IF EXISTS #{table_name}")
|
2009-01-24 21:23:11 +00:00
|
|
|
end
|
|
|
|
|
2010-02-13 03:05:12 +00:00
|
|
|
FileUtils.rm_rf(TMP_VIEW_PATH)
|
|
|
|
|
2010-12-14 23:35:12 +00:00
|
|
|
Rails.application.reload_routes!
|
2008-12-15 18:13:56 +00:00
|
|
|
end
|
|
|
|
end
|
2010-12-13 22:28:59 +00:00
|
|
|
|
|
|
|
RSpec.configure do |config|
|
|
|
|
config.include ModelBuilder
|
|
|
|
end
|