Make sure all things template

jordansissel · Aug 7, 2012 · 537c2ce · 537c2ce
1 parent eb301e4
commit 537c2ce
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/spec/fpm/package_spec.rb b/spec/fpm/package_spec.rb
@@ -145,15 +145,27 @@
   context "#script (internal method)" do
     it "should template when :template_scripts? is true" do
       subject.scripts[:after_install] = "<%= name %>"
+      subject.scripts[:before_install] = "<%= name %>"
+      subject.scripts[:after_remove] = "<%= name %>"
+      subject.scripts[:before_remove] = "<%= name %>"
       subject.attributes[:template_scripts?] = true
       subject.name = "Example"
       insist { subject.script(:after_install) } == subject.name
+      insist { subject.script(:before_install) } == subject.name
+      insist { subject.script(:after_remove) } == subject.name
+      insist { subject.script(:before_remove) } == subject.name
     end
 
     it "should not template when :template_scripts? is false" do
+      subject.scripts[:after_install] = "<%= name %>"
+      subject.scripts[:before_install] = "<%= name %>"
+      subject.scripts[:after_remove] = "<%= name %>"
       subject.scripts[:after_install] = "<%= name %>"
       subject.attributes[:template_scripts?] = false
       insist { subject.script(:after_install) } == subject.scripts[:after_install]
+      insist { subject.script(:before_install) } == subject.scripts[:before_install]
+      insist { subject.script(:after_remove) } == subject.scripts[:after_remove]
+      insist { subject.script(:before_remove) } == subject.scripts[:before_remove]
     end
 
     it "should not template by default" do