spec/lib/appsignal/capistrano3_spec.rb in appsignal-2.0.6 vs spec/lib/appsignal/capistrano3_spec.rb in appsignal-2.1.0.alpha.1

@@ -1,12 +1,13 @@
 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
@@ -23,10 +24,10 @@
     end
     before { Rake::Task['appsignal:deploy'].reenable }
 
     def run
       capture_std_streams(out_stream, out_stream) do
-        capistrano.invoke('appsignal:deploy')
+        invoke('appsignal:deploy')
       end
     end
 
     it "should have a deploy task" do
       Rake::Task.task_defined?('appsignal:deploy').should be_true