X-Git-Url: https://review.fuel-infra.org/gitweb?a=blobdiff_plain;f=spec%2Fdefines%2Fsource_spec.rb;h=7c7ae399e84768fe3d51daca47c67a49c35b8855;hb=23149c9a3a25b347cebcc70c5c501b02709ce69a;hp=bf5e50e401286d58a44fc1c63ef2cbd6a2b0ce25;hpb=3fad56d26627c3e19561c319ddf124d22639c801;p=puppet-modules%2Fpuppetlabs-apt.git

diff --git a/spec/defines/source_spec.rb b/spec/defines/source_spec.rb
index bf5e50e..7c7ae39 100644
--- 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