在Rails中加載定製燈具

Question

我們試圖加載一組使用與ActiveRecord :: Base不同的數據庫連接的模型（本例中從Foo :: Base繼承而來）。

我們創建了這個模塊，包含在ActiveSupport :: TestCase中，並指定了.yml文件的路徑，例如foo_fitures :all。這對於運行的第一個測試非常有用。定義夾具訪問器並在數據庫中找到記錄。但對於後續測試，數據庫中沒有記錄。

module Foo::Fixtures 
    extend ActiveSupport::Concern 

    included do 
    setup :setup_foo_fixtures 
    teardown :teardown_foo_fixtures 

    class_attribute :foo_fixture_path 
    class_attribute :foo_fixture_table_names 

    self.foo_fixture_table_names = [] 
    end 

    module ClassMethods 
    def foo_fixtures(*fixture_names) 
     if fixture_names.first == :all 
     fixture_names = Dir[foo_fixture_path.join("**/*.yml")].map { |f| File.basename(f, ".yml") } 
     else 
     fixture_names = fixture_names.flatten.map { |n| n.to_s } 
     end 

     self.foo_fixture_table_names |= fixture_names 
     require_fixture_classes(fixture_names) 
     setup_fixture_accessors(fixture_names) 
    end 
    end 

    def setup_foo_fixtures 
    @loaded_fixtures.merge!(load_foo_fixtures) 
    end 

    def teardown_foo_fixtures 
    Foo::Base.clear_active_connections! 
    end 

    private 

    def load_foo_fixtures 
    foo_classes = Foo::Base.subclasses.flat_map { |klass| klass.abstract_class ? klass.subclasses : klass } 
    class_names = foo_classes.each_with_object({}) do |klass, memo| 
     memo[klass.table_name.to_sym] = klass if klass.table_name.present? && foo_fixture_table_names.include?(klass.table_name) 
    end 
    foo_fixtures = ActiveRecord::Fixtures.create_fixtures(foo_fixture_path, foo_fixture_table_names, class_names) do 
     Foo::Base.connection 
    end 
    Hash[foo_fixtures.map { |f| [f.name, f] }] 
    end 
end 

Rails的夾具系統是有點令人費解，我無法弄清楚什麼我們缺少的，以確保我們的額外的設備被加載。

Answer 1

好的。它看起來可能是從數據庫中刪除我們的設備的交易。我的猜測是交易在我們的代碼加載到燈具之前就已經開始了，所以這就是爲什麼他們在第一次測試中出現，但是第二次測試結束。

所以我們改變了策略，現在我們只需要登錄到load_fixtures和fixtures。這工作得很好。

module FooFixtures 
    module ClassMethods 
    def foo_fixture_classes 
     collect_subclasses = ->(k) { k.abstract_class ? k.subclasses.flat_map(&collect_subclasses) : k } 
     Foo::Base.subclasses.flat_map(&collect_subclasses) 
    end 

    def foo_fixture_path 
     Rails.root.join("test/foo_fixtures") 
    end 

    def foo_fixture_table_names 
     Dir[foo_fixture_path.join("**/*.yml")].map { |f| File.basename(f, ".yml") } 
    end 

    def fixtures(*fixture_names) 
     super 
     if fixture_names.first == :all 
     require_fixture_classes(foo_fixture_table_names) 
     setup_fixture_accessors(foo_fixture_table_names) 
     end 
    end 
    end 

    private 

    def load_fixtures 
    foo_fixture_path = self.class.foo_fixture_path 
    foo_fixture_table_names = self.class.foo_fixture_table_names 
    class_names = self.class.foo_fixture_classes.each_with_object({}) do |klass, memo| 
     memo[klass.table_name.to_sym] = klass if klass.table_name.present? && foo_fixture_table_names.include?(klass.table_name) 
    end 
    foo_fixtures = ActiveRecord::Fixtures.create_fixtures(foo_fixture_path, foo_fixture_table_names, class_names) do 
     Foo::Base.connection 
    end 

    super.merge(Hash[foo_fixtures.map { |f| [f.name, f] }]) 
    end 
end 

class ActiveSupport::TestCase 
    extend FooFixtures::ClassMethods 
    prepend FooFixtures 

    self.foo_fixture_classes.each do |fixture_class| 
    set_fixture_class fixture_class.table_name.to_sym => fixture_class 
    end 

    ... 
end