diff --git a/spec/database_cleaner/active_record/base_spec.rb b/spec/database_cleaner/active_record/base_spec.rb index 0bed4f3..bf0d83a 100644 --- a/spec/database_cleaner/active_record/base_spec.rb +++ b/spec/database_cleaner/active_record/base_spec.rb @@ -3,6 +3,12 @@ require 'active_record' require 'database_cleaner/active_record/base' require 'database_cleaner/shared_strategy' +class FakeModel + def self.connection + :fake_connection + end +end + module DatabaseCleaner describe ActiveRecord do it { should respond_to(:available_strategies) } @@ -121,6 +127,11 @@ my_db: subject.connection_class.should == ::ActiveRecord::Base end + it "allows for database models to be passed in" do + subject.db = FakeModel + subject.connection_class.should == FakeModel + end + context "when connection_hash is set" do let(:hash) { mock("hash") } before { ::ActiveRecord::Base.stub!(:respond_to?).and_return(false)}