Fix specs

spec/timber/probes/action_controller_base_spec.rb

@@ -24,6 +24,7 @@ def method_for_action(action_name)
       Object.send(:remove_const, :PagesPlainController)
     end
 
+    let(:logger_context_class) { Timber::Contexts::Logger }
     let(:rack_request_context_class) { Timber::Contexts::HTTPRequests::Rack }
     let(:request_context_class) { Timber::Contexts::HTTPRequests::ActionControllerSpecific }
     let(:organization_context_class) { Timber::Contexts::Organizations::ActionController }
@@ -32,6 +33,7 @@ def method_for_action(action_name)
 
     describe "#process" do
       it "should set the context" do
+        allow(Timber::CurrentContext).to receive(:add).with(kind_of(logger_context_class))
         expect(Timber::CurrentContext).to receive(:add).with(kind_of(rack_request_context_class)).and_yield.once
         expect(Timber::CurrentContext).to receive(:add).with(kind_of(request_context_class), kind_of(organization_context_class), kind_of(user_context_class), kind_of(response_context_class)).and_yield.once
         dispatch_rails_request("/pages_plain")

spec/timber/probes/action_controller_log_subscriber/action_view_spec.rb

@@ -23,6 +23,7 @@ def method_for_action(action_name)
     Object.send(:remove_const, :UsersController)
   end
 
+  let(:logger_context_class) { Timber::Contexts::Logger }
   let(:rack_request_context_class) { Timber::Contexts::HTTPRequests::Rack }
   let(:request_context_class) { Timber::Contexts::HTTPRequests::ActionControllerSpecific }
   let(:organization_context_class) { Timber::Contexts::Organizations::ActionController }
@@ -33,6 +34,7 @@ def method_for_action(action_name)
 
   describe "#process_action" do
     it "should set the context" do
+      allow(Timber::CurrentContext).to receive(:add).with(kind_of(logger_context_class))
       expect(Timber::CurrentContext).to receive(:add).with(kind_of(rack_request_context_class)).and_yield.once
       expect(Timber::CurrentContext).to receive(:add).with(kind_of(request_context_class), kind_of(organization_context_class), kind_of(user_context_class), kind_of(response_context_class)).and_yield.once
       expect(Timber::CurrentContext).to receive(:add).with(kind_of(action_view_context_class), kind_of(action_view_specific_context_class)).and_yield.once

spec/timber/probes/action_dispatch_debug_exceptions_spec.rb

@@ -34,7 +34,7 @@ def method_for_action(action_name)
       let(:exception_context_class) { Timber::Contexts::Exception }
 
       it "should set the context" do
-        expect(Timber::CurrentContext).to receive(:add).with(kind_of(logger_context_class)).and_yield.once
+        expect(Timber::CurrentContext).to receive(:add).with(kind_of(logger_context_class)).and_yield.at_least(:once)
         expect(Timber::CurrentContext).to receive(:add).with(kind_of(rack_request_context_class)).and_yield.once
         expect(Timber::CurrentContext).to receive(:add).with(kind_of(request_context_class), kind_of(organization_context_class), kind_of(user_context_class), kind_of(response_context_class)).and_yield.once
         expect(Timber::CurrentContext).to receive(:add).with(kind_of(exception_context_class)).and_yield