lib/unitf/logging/logger.rb in unitf-logging-0.1.0 vs lib/unitf/logging/logger.rb in unitf-logging-0.1.1

@@ -8,14 +8,13 @@
     WARN = ::Logger::Severity::WARN
     ERROR = ::Logger::Severity::ERROR
     FATAL = ::Logger::Severity::FATAL
 
     class Logger
-      def initialize(level = INFO)
+      def initialize
         @writers = []
-        @console = nil
-        @level = ENV['UNITF_LOG_LEVEL'] || level
+        self.level = ENV['UNITF_LOG_LEVEL'] || INFO
       end
 
       def add_writer(writer)
         @writers << writer
         writer.level = @level
@@ -39,15 +38,17 @@
 
       def fatal(message)
         write(FATAL, message)
       end
 
-      def level=(level)
-        level = UnitF::Logging.const_get(level.upcase) if level.is_a?(String)
-        @level = level
+      def level=(new_level)
+        new_level = UnitF::Logging.const_get(new_level.upcase) if new_level.is_a?(String)
+        @level = new_level
         @writers.each do |writer|
-          writer.level = level
+          writer.level = @level
         end
+      rescue ArgumentError, NameError
+        self.level = INFO
       end
 
       def console
         @writers.each do |writer|
           return writer if writer.is_a?(UnitF::Logging::ConsoleWriter)
\ No newline at end of file