diff --git a/spec/database_cleaner/moped/moped_examples.rb b/spec/database_cleaner/moped/moped_examples.rb index 0fd25df..fe8a96d 100644 --- a/spec/database_cleaner/moped/moped_examples.rb +++ b/spec/database_cleaner/moped/moped_examples.rb @@ -23,4 +23,10 @@ module MopedTest end class Gadget < ThingBase end + class System < ThingBase + def self.collection + super + @collection = @session['system_logs'] + end + end end diff --git a/spec/database_cleaner/moped/truncation_spec.rb b/spec/database_cleaner/moped/truncation_spec.rb index e84ae84..fab8326 100644 --- a/spec/database_cleaner/moped/truncation_spec.rb +++ b/spec/database_cleaner/moped/truncation_spec.rb @@ -39,12 +39,17 @@ module DatabaseCleaner MopedTest::Gadget.new({:name => 'some gadget'}.merge(attrs)).save! end + def create_system(attrs={}) + MopedTest::System.new({:name => 'some system'}.merge(attrs)).save! + end + it "truncates all collections by default" do create_widget create_gadget - ensure_counts(MopedTest::Widget => 1, MopedTest::Gadget => 1) + create_system + ensure_counts(MopedTest::Widget => 1, MopedTest::Gadget => 1, MopedTest::System => 1) truncation.clean - ensure_counts(MopedTest::Widget => 0, MopedTest::Gadget => 0) + ensure_counts(MopedTest::Widget => 0, MopedTest::Gadget => 0, MopedTest::System => 0) end context "when collections are provided to the :only option" do