Move iptables test code to a new IptablesBaseTest class.
Bug: 25691379
Bug: 21725996
Change-Id: Ia0598e60ad24714d53470e05849929831ba9dbf6
diff --git a/server/IptablesBaseTest.cpp b/server/IptablesBaseTest.cpp
new file mode 100644
index 0000000..b3065f5
--- /dev/null
+++ b/server/IptablesBaseTest.cpp
@@ -0,0 +1,68 @@
+/*
+ * Copyright 2016 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ * IptablesBaseTest.cpp - utility class for tests that use iptables
+ */
+
+#include <string>
+#include <vector>
+
+#include <gtest/gtest.h>
+
+#include "IptablesBaseTest.h"
+#include "NetdConstants.h"
+
+IptablesBaseTest::IptablesBaseTest() {
+ sCmds.clear();
+ sRestoreCmds.clear();
+}
+
+int IptablesBaseTest::fake_android_fork_exec(int argc, char* argv[], int *status, bool, bool) {
+ std::string cmd = argv[0];
+ for (int i = 1; i < argc; i++) {
+ cmd += " ";
+ cmd += argv[i];
+ }
+ sCmds.push_back(cmd);
+ *status = 0;
+ return 0;
+}
+
+int IptablesBaseTest::fakeExecIptablesRestore(IptablesTarget target, const std::string& commands) {
+ EXPECT_EQ(V4V6, target);
+ sRestoreCmds.push_back(commands);
+ return 0;
+}
+
+void IptablesBaseTest::expectIptablesCommands(const std::vector<std::string>& expectedCmds) {
+ EXPECT_EQ(expectedCmds.size() * 2, sCmds.size());
+ if (expectedCmds.size() * 2 != sCmds.size()) return;
+
+ for (size_t i = 0; i < expectedCmds.size(); i ++) {
+ EXPECT_EQ("/system/bin/iptables -w " + expectedCmds[i], sCmds[2 * i]);
+ EXPECT_EQ("/system/bin/ip6tables -w " + expectedCmds[i], sCmds[2 * i + 1]);
+ }
+
+ sCmds.clear();
+}
+
+void IptablesBaseTest::expectIptablesRestoreCommands(const std::vector<std::string>& expectedCmds) {
+ EXPECT_EQ(expectedCmds.size(), sRestoreCmds.size());
+ EXPECT_EQ(expectedCmds, sRestoreCmds);
+ sRestoreCmds.clear();
+}
+
+std::vector<std::string> IptablesBaseTest::sCmds = {};
+std::vector<std::string> IptablesBaseTest::sRestoreCmds = {};