commit a0dc8a54c3d1a805294c5608dbeb78292014d2a5
author Lorenzo Colitti <lorenzo@google.com> Sat Mar 26 22:42:07 2016 +0900
committer Lorenzo Colitti <lorenzo@google.com> Mon Mar 28 02:40:56 2016 +0900
tree ba86101ee37acf000273a139c5a1b9ba3db2bed3
parent 464eabecf1174154b8f61845610c3f4f0ca294b3

Add more test coverage for BandwidthController.

Bug: 25691379
Bug: 21725996
Change-Id: I196f568221b3bad98ba2d60e019c2b24c59a9cbc

server/BandwidthControllerTest.cpp

diff --git a/server/BandwidthControllerTest.cpp b/server/BandwidthControllerTest.cpp
index 8251830..6d261ed 100644
--- a/server/BandwidthControllerTest.cpp
+++ b/server/BandwidthControllerTest.cpp
@@ -64,6 +64,30 @@
 };
 
 
+TEST_F(BandwidthControllerTest, TestSetupIptablesHooks) {
+    mBw.setupIptablesHooks();
+    std::vector<std::string> expected = {
+        "-F bw_INPUT",
+        "-F bw_OUTPUT",
+        "-F bw_FORWARD",
+        "-F bw_happy_box",
+        "-F bw_penalty_box",
+        "-F bw_data_saver",
+        "-F bw_costly_shared",
+        "-t raw -F bw_raw_PREROUTING",
+        "-t mangle -F bw_mangle_POSTROUTING",
+        "-X bw_happy_box",
+        "-X bw_penalty_box",
+        "-X bw_data_saver",
+        "-X bw_costly_shared",
+        "-N bw_happy_box",
+        "-N bw_penalty_box",
+        "-N bw_data_saver",
+        "-N bw_costly_shared",
+    };
+    expectIptablesCommands(expected);
+}
+
 TEST_F(BandwidthControllerTest, TestEnableBandwidthControl) {
     mBw.enableBandwidthControl(false);
     std::vector<std::string> expected = {
@@ -89,6 +113,22 @@
     expectIptablesCommands(expected);
 }
 
+TEST_F(BandwidthControllerTest, TestDisableBandwidthControl) {
+    mBw.disableBandwidthControl();
+    std::vector<std::string> expected = {
+        "-F bw_INPUT",
+        "-F bw_OUTPUT",
+        "-F bw_FORWARD",
+        "-F bw_happy_box",
+        "-F bw_penalty_box",
+        "-F bw_data_saver",
+        "-F bw_costly_shared",
+        "-t raw -F bw_raw_PREROUTING",
+        "-t mangle -F bw_mangle_POSTROUTING",
+    };
+    expectIptablesCommands(expected);
+}
+
 TEST_F(BandwidthControllerTest, TestEnableDataSaver) {
     mBw.enableDataSaver(true);
     std::vector<std::string> expected = {