lib/simple_segment/configuration.rb in simple_segment-0.1.1 vs lib/simple_segment/configuration.rb in simple_segment-0.2.0

@@ -1,14 +1,20 @@
+require 'simple_segment/logging'
+
 module SimpleSegment
   class Configuration
     include SimpleSegment::Utils
+    include SimpleSegment::Logging
 
-    attr_reader :write_key, :on_error
+    attr_reader :write_key, :on_error, :stub, :logger
 
     def initialize(settings = {})
       symbolized_settings = symbolize_keys(settings)
       @write_key = symbolized_settings[:write_key]
       @on_error = symbolized_settings[:on_error] || proc {}
-      raise ArgumentError, 'Missing required option :write_key' unless @write_key
+      @stub = symbolized_settings[:stub]
+      @logger = default_logger(symbolized_settings[:logger])
+      raise ArgumentError, 'Missing required option :write_key' \
+        unless @write_key
     end
   end
 end