spec/support/logger_support.rb in boty-0.1.0 vs spec/support/logger_support.rb in boty-0.1.1

@@ -1,20 +1,17 @@
 module LoggerSupport
   def self.included(base)
     base.instance_eval do
       let(:logger) { Boty::Logger.adapter }
+      before do change_to_memory_adapter end
+      around(&:run)
+      after do Boty::Logger.adapter = @adapter end
+    end
+  end
 
-      before do
-        @adapter = Boty::Logger.adapter
-        Boty::Logger.adapter = Boty::Logger::Memory.new
-      end
+  private
 
-      around do |example|
-        example.run
-      end
-
-      after do
-        Boty::Logger.adapter = @adapter
-      end
-    end
+  def change_to_memory_adapter
+    @adapter = Boty::Logger.adapter
+    Boty::Logger.adapter = Boty::Logger::Memory.new
   end
 end