lib/fog/bouncer.rb in fog-bouncer-0.2.0 vs lib/fog/bouncer.rb in fog-bouncer-0.2.1

@@ -9,10 +9,17 @@
 require "fog/bouncer/group_manager"
 require "fog/bouncer/source_manager"
 
 module Fog
   module Bouncer
+    module Logger
+      def self.log(data, &block)
+        STDOUT.puts data
+        yield if block_given?
+      end
+    end
+
     # Public: An AWS account ID
     #
     # Example
     #
     #   Fog::Bouncer.aws_account_id
@@ -79,10 +86,10 @@
 
     # Public: The logging location
     #
     # Returns an Object
     def self.logger
-      @logger || STDOUT.method(:puts)
+      @logger || Fog::Bouncer::Logger.method(:log)
     end
 
     # Public: Load a file for evaluation
     #
     # Example