From: Travis Fields <travis@puppetlabs.com>
Date: Fri, 23 Jan 2015 22:52:01 +0000 (-0800)
Subject: Fix issue with local testing firewallchain
X-Git-Tag: 1.4.0~7^2
X-Git-Url: https://review.fuel-infra.org/gitweb?a=commitdiff_plain;h=abfbc7630dd37b7fb30926d5940f0bb6fc9c5442;p=puppet-modules%2Fpuppetlabs-firewall.git

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 bd3095e..4139bf5 100755
--- 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