Merge remote-tracking branch 'upstream/1.5.x'

[puppet-modules/puppetlabs-apt.git] / spec / defines / source_spec.rb

diff --git a/spec/defines/source_spec.rb b/spec/defines/source_spec.rb
index bf5e50e401286d58a44fc1c63ef2cbd6a2b0ce25..7c7ae399e84768fe3d51daca47c67a49c35b8855 100644 (file)
--- a/spec/defines/source_spec.rb
+++ b/spec/defines/source_spec.rb
@@ -74,7 +74,12 @@ describe 'apt::source', :type => :define do
       end
 
       let :content do
-        content = "#file generated by puppet\n# #{title}"
+        content = "#file generated by puppet\n"
+       if param_hash[:comment]
+         content << "# #{comment}"
+       else
+         content << "# #{title}"
+       end
         if param_hash[:architecture]
           arch = "[arch=#{param_hash[:architecture]}] "
         end