spec/timber/logger_spec.rb in timber-2.1.0.rc2 vs spec/timber/logger_spec.rb in timber-2.1.0.rc3

@@ -1,8 +1,29 @@
 require "spec_helper"
 
 describe Timber::Logger, :rails_23 => true do
+  describe "#initialize" do
+    it "shoud select the augmented formatter" do
+      logger = described_class.new(nil)
+      expect(logger.formatter).to be_kind_of(Timber::Logger::AugmentedFormatter)
+    end
+
+    context "development environment" do
+      around(:each) do |example|
+        old_env = Timber::Config.instance.environment
+        Timber::Config.instance.environment = "development"
+        example.run
+        Timber::Config.instance.environment = old_env
+      end
+
+      it "shoud select the augmented formatter" do
+        logger = described_class.new(nil)
+        expect(logger.formatter).to be_kind_of(Timber::Logger::MessageOnlyFormatter)
+      end
+    end
+  end
+
   describe "#add" do
     let(:time) { Time.utc(2016, 9, 1, 12, 0, 0) }
     let(:io) { StringIO.new }
     let(:logger) { Timber::Logger.new(io) }
 
@@ -119,9 +140,22 @@
     it "should set the formatter" do
       logger = Timber::Logger.new(STDOUT)
       formatter = ::Logger::Formatter.new
       logger.formatter = formatter
       expect(logger.formatter).to eq(formatter)
+    end
+  end
+
+  describe "#silence" do
+    let(:io) { StringIO.new }
+    let(:logger) { Timber::Logger.new(io) }
+
+    it "should silence the logs" do
+      logger.silence do
+        logger.info("test")
+      end
+
+      expect(io.string).to eq("")
     end
   end
 
   describe "#with_context" do
     let(:io) { StringIO.new }
\ No newline at end of file