Fix issue with local testing firewallchain

diff --git a/spec/unit/puppet/type/firewallchain_spec.rb b/spec/unit/puppet/type/firewallchain_spec.rb
index bd3095e0993f8fa69b592e179a687185046fe69a..4139bf55d5cd7492e00233d00807b663f449dab3 100755 (executable)
--- a/spec/unit/puppet/type/firewallchain_spec.rb
+++ b/spec/unit/puppet/type/firewallchain_spec.rb
@@ -142,7 +142,7 @@ describe firewallchain do
 
   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 @@ describe firewallchain do
 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