spec/action/stop_server_spec.rb in vagrant-notify-0.6.0 vs spec/action/stop_server_spec.rb in vagrant-notify-0.6.1

@@ -2,14 +2,14 @@
 
 require 'vagrant-notify/action/stop_server'
 
 describe Vagrant::Notify::Action::StopServer do
   let(:app)          { lambda { |env| } }
-  let(:config)       { mock(notify: stub(enable: true)) }
-  let(:communicator) { mock(sudo: true) }
-  let(:ui)           { mock(success: true) }
-  let(:machine)      { mock(communicate: communicator, state: stub(id: :running), ui: ui, config: config) }
+  let(:config)       { double(notify: double(enable: true)) }
+  let(:communicator) { double(sudo: true) }
+  let(:ui)           { double(success: true) }
+  let(:machine)      { double(communicate: communicator, state: double(id: :running), ui: ui, config: config) }
   let(:env)          { {notify_data: {pid: pid, port: 1234}, machine: machine} }
   let(:pid)          { '42' }
   let(:port)         { described_class::PORT }
 
   subject { described_class.new(app, env) }
@@ -18,10 +18,10 @@
     Process.stub(kill: true)
     subject.call(env)
   end
 
   it 'kills the notification server' do
-    Process.should have_received(:kill).with('KILL', pid.to_i)
+    Process.should have_received(:kill).with('KILL', pid.to_i).at_most(2).times
   end
 
   it "removes server PID from notify data" do
     env[:notify_data][:pid].should be_nil
   end