webmachine_spec.rb in appsignal-2.2.0.beta.1

- old
+ new

@@ -5,11 +5,11 @@
     let(:request) do
       Webmachine::Request.new("GET", "http://google.com:80/foo", {}, nil)
     end
     let(:resource)    { double(:trace? => false, :handle_exception => true) }
     let(:response)    { double }
-    let(:transaction) { double(:set_action => true) }
+    let(:transaction) { double(:set_action_if_nil => true) }
     let(:fsm) { Webmachine::Decision::FSM.new(resource, request, response) }
     before(:context) { start_agent }
 
     # Make sure the request responds to the method we need to get query params.
     describe "request" do
@@ -34,10 +34,10 @@
           :params_method => :query
         ).and_return(transaction)
       end
 
       it "should set the action" do
-        expect(transaction).to receive(:set_action).with("RSpec::Mocks::Mock#GET")
+        expect(transaction).to receive(:set_action_if_nil).with("RSpec::Mocks::Mock#GET")
       end
 
       it "should call the original method" do
         expect(fsm).to receive(:run_without_appsignal)
       end