some tests for #cleaning

spec/database_cleaner/base_spec.rb

@@ -451,6 +451,13 @@ module DatabaseCleaner
           subject.clean!
         end
       end
+
+      describe "cleaning" do
+        it "should proxy cleaning to the strategy" do
+          strategy.should_receive(:cleaning)
+          subject.cleaning { }
+        end
+      end
     end
 
     describe "auto_detected?" do

spec/database_cleaner/configuration_spec.rb

@@ -163,6 +163,11 @@ describe ::DatabaseCleaner do
       ::DatabaseCleaner.clean
     end
 
+    it 'should proxy cleaning' do
+      connection.should_receive(:cleaning)
+      ::DatabaseCleaner.cleaning { }
+    end
+
     it "should proxy clean_with" do
       stratagem = double("stratgem")
       connection.should_receive(:clean_with).with(stratagem, {})
@@ -203,6 +208,28 @@ describe ::DatabaseCleaner do
         ::DatabaseCleaner.clean
       end
 
+      it "should initiate cleaning on each connection, yield, and finish cleaning each connection" do
+        [active_record, data_mapper].each do |connection|
+          mc = class << connection; self; end
+          mc.send(:attr_reader, :started, :cleaned)
+          mc.send(:define_method, 'cleaning') do |&block|
+            @started = true
+            block.call
+            @cleaned = true
+          end
+        end
+
+        ::DatabaseCleaner.cleaning do
+          active_record.started.should == true
+          data_mapper.started.should == true
+          active_record.cleaned.should == nil
+          data_mapper.cleaned.should == nil
+          @yielded = true
+        end
+        active_record.cleaned.should == true
+        data_mapper.cleaned.should == true
+      end
+
       it "should proxy clean_with to all connections" do
         stratagem = double("stratgem")
         active_record.should_receive(:clean_with).with(stratagem)

spec/database_cleaner/shared_strategy.rb

@@ -5,9 +5,11 @@ end
 shared_examples_for "a generic truncation strategy" do
   it { should respond_to(:start) }
   it { should respond_to(:clean) }
+  it { should respond_to(:cleaning) }
 end
 
 shared_examples_for "a generic transaction strategy" do
   it { should respond_to(:start) }
   it { should respond_to(:clean) }
+  it { should respond_to(:cleaning) }
 end