Extract with_max_links to Helper module

The with_max_links helper was already being used across multiple spec
files, but wasn't properly being included for reuse. (It was defined
_globally_ in one single spec, which only worked if that specific test
just _happened_ to be loaded before the other.)

Helpers like this need to be included wherever they are used (if not
included globally).

Author: jasonkarns

spec/sitemap_generator/interpreter_spec.rb

@@ -2,6 +2,8 @@
 require 'sitemap_generator/interpreter'
 
 RSpec.describe SitemapGenerator::Interpreter do
+  include SitemapHelpers
+
   let(:link_set)    { SitemapGenerator::LinkSet.new }
   let(:interpreter) { SitemapGenerator::Interpreter.new(:link_set => link_set) }
 

spec/sitemap_generator/sitemap_generator_spec.rb

@@ -7,15 +7,9 @@ class << self
   end
 end
 
-def with_max_links(num)
-  original = SitemapGenerator::Sitemap.max_sitemap_links
-  SitemapGenerator::Sitemap.max_sitemap_links = num
-  yield
-ensure
-  SitemapGenerator::Sitemap.max_sitemap_links = original
-end
-
 RSpec.describe 'SitemapGenerator' do
+  include SitemapHelpers
+
   describe 'reset!' do
     before do
       SitemapGenerator::Sitemap.default_host # Force initialization of the LinkSet

spec/spec_helper.rb

@@ -11,6 +11,7 @@
 # Load support files
 require_relative 'support/file_macros'
 require_relative 'support/xml_macros'
+require_relative 'support/sitemap_helpers'
 
 # Configure webmock
 WebMock.disable_net_connect!

spec/support/sitemap_helpers.rb

@@ -0,0 +1,9 @@
+module SitemapHelpers
+  def with_max_links(num)
+    original = SitemapGenerator::Sitemap.max_sitemap_links
+    SitemapGenerator::Sitemap.max_sitemap_links = num
+    yield
+  ensure
+    SitemapGenerator::Sitemap.max_sitemap_links = original
+  end
+end