Merge pull request #478 from cyberious/FirewallChain

Fix issue with firewallchain unit tests
strider · Jan 23, 2015 · 82c2917 · 82c2917
2 parents ca980a5 + abfbc76
commit 82c2917
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/spec/unit/puppet/type/firewallchain_spec.rb b/spec/unit/puppet/type/firewallchain_spec.rb
@@ -142,7 +142,7 @@
 
   describe 'purge iptables rules' do
     before(:each) do
-      allow(Puppet::Type.type(:firewall).provider(:iptables)).to receive(:iptables_save).and_return(<<EOS
+      stub_return = <<EOS
 # Completed on Sun Jan  5 19:30:21 2014
 # Generated by iptables-save v1.4.12 on Sun Jan  5 19:30:21 2014
 *filter
@@ -162,7 +162,8 @@
 COMMIT
 # Completed on Sun Jan  5 19:30:21 2014
 EOS
-)
+      allow(Puppet::Type.type(:firewall).provider(:iptables)).to receive(:iptables_save).and_return(stub_return)
+      allow(Puppet::Type.type(:firewall).provider(:ip6tables)).to receive(:ip6tables_save).and_return(stub_return)
     end
 
     it 'should generate iptables resources' do