kotovalexarian-likes-github/thoughtbot--shoulda-matchers
1
0
Fork 0
mirror of https://github.com/thoughtbot/shoulda-matchers.git synced 2022-11-09 12:01:38 -05:00
Code Releases Activity

Changes to ModelBuilder

Browse source 
* create_table and define_model_class accepts optional `connection' parameter for custom db connections
This commit is contained in:
Joey Cheng 2019-04-13 19:58:58 +08:00 • committed by Elliot Winkler
parent 25dea722d9
commit b53433f96d
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
spec/support/unit/helpers/model_builder.rb
View file

@ -35,7 +35,8 @@ module UnitTests
end
def create_table(table_name, options = {}, &block)
connection = ::ActiveRecord::Base.connection
connection =
options.delete(:connection) || ::ActiveRecord::Base.connection
begin
connection.execute("DROP TABLE IF EXISTS #{table_name}")
@ -48,8 +49,8 @@ module UnitTests
end
end
def define_model_class(class_name, &block)
ClassBuilder.define_class(class_name, ::ActiveRecord::Base, &block)
def define_model_class(class_name, parent_class: ::ActiveRecord::Base, &block)
ClassBuilder.define_class(class_name, parent_class, &block)
end
def define_active_model_class(class_name, options = {}, &block)