spec/lib/appsignal/capistrano3_spec.rb in appsignal-2.0.5 vs spec/lib/appsignal/capistrano3_spec.rb in appsignal-2.0.6

@@ -1,13 +1,12 @@
 if DependencyHelper.capistrano3_present?
   require 'capistrano/all'
   require 'capistrano/deploy'
   require 'appsignal/capistrano'
 
-  include Capistrano::DSL
-
   describe "Capistrano 3 integration" do
+    let(:capistrano) { Class.new.extend(Capistrano::DSL) }
     let(:config) { project_fixture_config }
     let(:out_stream) { std_stream }
     let(:output) { out_stream.read }
     let(:logger) { Logger.new(out_stream) }
     let!(:capistrano_config) do
@@ -24,10 +23,10 @@
     end
     before { Rake::Task['appsignal:deploy'].reenable }
 
     def run
       capture_std_streams(out_stream, out_stream) do
-        invoke('appsignal:deploy')
+        capistrano.invoke('appsignal:deploy')
       end
     end
 
     it "should have a deploy task" do
       Rake::Task.task_defined?('appsignal:deploy').should be_true