тестирование и установление соединения
Вопрос
Как мне написать тесты для проектов, которые используют create_connection
в модели для подключения к другой базе данных?
Решение
Когда вы устанавливаете коннект для определенных моделей для подключения к другой базе данных, одна из проблем, с которой вы столкнетесь при тестировании этих таблиц, заключается в том, что созданные вами тестовые данные не будут автоматически откатываться.
Фактический код для создания точки сохранения транзакции и отката данных для тестовых жизней в rails / activerecord / lib / active_record / fixtures.rb
. И особенно это два метода setup_fixtures
и teardown_fixtures
. Код в этих методах прост. Они просто создают точку сохранения и выполняют откат для каждого теста. Но это только для соединения ActiveRecord :: Base
. Р>
То, что вам нужно сделать, это «патч обезьяны» эти методы, чтобы в дополнение к соединению ActiveRecord :: Base
выполнялся тот же набор операций для вашего дополнительного соединения с базой данных.
Вот пример кода для того же самого:
## database.yml
development:
database: dev
test:
database: test
#...
my_connection_development:
database: my_connection_dev
my_connection_test:
database: my_connection_test
#...
## my_connection_base.rb
class MyConnectionBase < ActiveRecord::Base
establish_connection(ActiveRecord::Base.configurations["my_connection_#{RAILS_ENV}"])
self.abstract_class = true
end
## my_model.rb
class MyModel < MyConnectionBase
end
## my_another_model.rb
class MyAnotherModel < MyConnectionBase
end
## test_case_patch.rb
module ActiveSupport
class TestCase
def setup_fixtures
return unless defined?(ActiveRecord) && !ActiveRecord::Base.configurations.blank?
if pre_loaded_fixtures && !use_transactional_fixtures
raise RuntimeError, 'pre_loaded_fixtures requires use_transactional_fixtures'
end
@fixture_cache = {}
@@already_loaded_fixtures ||= {}
# Load fixtures once and begin transaction.
if run_in_transaction?
if @@already_loaded_fixtures[self.class]
@loaded_fixtures = @@already_loaded_fixtures[self.class]
else
load_fixtures
@@already_loaded_fixtures[self.class] = @loaded_fixtures
end
ActiveRecord::Base.connection.increment_open_transactions
ActiveRecord::Base.connection.transaction_joinable = false
ActiveRecord::Base.connection.begin_db_transaction
MyConnectionBase.connection.increment_open_transactions
MyConnectionBase.connection.transaction_joinable = false
MyConnectionBase.connection.begin_db_transaction
# Load fixtures for every test.
else
Fixtures.reset_cache
@@already_loaded_fixtures[self.class] = nil
load_fixtures
end
# Instantiate fixtures for every test if requested.
instantiate_fixtures if use_instantiated_fixtures
end
def teardown_fixtures
return unless defined?(ActiveRecord) && !ActiveRecord::Base.configurations.blank?
unless run_in_transaction?
Fixtures.reset_cache
end
# Rollback changes if a transaction is active.
if run_in_transaction? && MyConnectionBase.connection.open_transactions != 0
MyConnectionBase.connection.rollback_db_transaction
MyConnectionBase.connection.decrement_open_transactions
end
# Rollback changes if a transaction is active.
if run_in_transaction? && ActiveRecord::Base.connection.open_transactions != 0
ActiveRecord::Base.connection.rollback_db_transaction
ActiveRecord::Base.connection.decrement_open_transactions
end
MyConnectionBase.clear_active_connections!
ActiveRecord::Base.clear_active_connections!
end
end
end
Другие советы
Я не вижу необходимости менять тестовые коды после того, как вы поместите в вашу модель assign_connection, поскольку вы все еще тестируете ту же функциональность модели. Р>
Вам не нужно будет проверять метод create_connection, так как он является методом activrecord и хорошо протестирован перед выпуском.
хотя, если вы все еще хотите это сделать, вызовите метод в другом методе и посмотрите, сможете ли вы подключиться к соответствующим таблицам в этой БД.