Fix issue with local testing firewallchain

strider · Jan 23, 2015 · abfbc76 · abfbc76
1 parent 4b1c215
commit abfbc76
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