spec/modules/audit/trainer_spec.rb in arachni-0.4.2 vs spec/modules/audit/trainer_spec.rb in arachni-0.4.3

@@ -1,10 +1,12 @@
-require_relative '../../spec_helper'
+require 'spec_helper'
 
 describe name_from_filename do
     include_examples 'module'
 
+    before( :each ){ framework.sitemap.clear }
+
     def self.targets
         %w(Generic)
     end
 
     def self.elements
@@ -17,9 +19,17 @@
             audit element.to_sym, false
 
             urls = framework.auditstore.sitemap
             urls.include?( options.url + "#{element}/straight/trained" ).should be_true
             urls.include?( options.url + "#{element}/append/trained" ).should be_true
+        end
+    end
+
+    context 'when the link count limit has been reached' do
+        it 'does not run' do
+            framework.opts.link_count_limit = 0
+            audit :form, false
+            framework.auditstore.sitemap.should be_empty
         end
     end
 
 end