Make fake iptables' Save operation more realistic

da924fc3 · Shyam Jeedigunta · f038c549 · da924fc3
Commit da924fc3 authored May 16, 2017 by Shyam Jeedigunta
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

fake.go pkg/util/iptables/testing/fake.go +4 -2

No files found.
--- a/pkg/util/iptables/testing/fake.go
+++ b/pkg/util/iptables/testing/fake.go
@@ -72,8 +72,10 @@ func (*FakeIPTables) IsIpv6() bool {
 	return false
 }
-func (*FakeIPTables) Save(table iptables.Table) ([]byte, error) {
+func (f *FakeIPTables) Save(table iptables.Table) ([]byte, error) {
-	return make([]byte, 0), nil
+	lines := make([]byte, len(f.Lines))
+	copy(lines, f.Lines)
+	return lines, nil
 }
 func (*FakeIPTables) SaveAll() ([]byte, error) {