diff options
| author | Lorenzo Colitti <lorenzo@google.com> | 2016-03-26 22:42:07 +0900 |
|---|---|---|
| committer | Lorenzo Colitti <lorenzo@google.com> | 2016-03-28 02:40:56 +0900 |
| commit | a0dc8a54c3d1a805294c5608dbeb78292014d2a5 (patch) | |
| tree | ba86101ee37acf000273a139c5a1b9ba3db2bed3 /server/BandwidthControllerTest.cpp | |
| parent | 464eabecf1174154b8f61845610c3f4f0ca294b3 (diff) | |
Add more test coverage for BandwidthController.
Bug: 25691379
Bug: 21725996
Change-Id: I196f568221b3bad98ba2d60e019c2b24c59a9cbc
Diffstat (limited to 'server/BandwidthControllerTest.cpp')
| -rw-r--r-- | server/BandwidthControllerTest.cpp | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/server/BandwidthControllerTest.cpp b/server/BandwidthControllerTest.cpp index 82518302..6d261edd 100644 --- a/server/BandwidthControllerTest.cpp +++ b/server/BandwidthControllerTest.cpp @@ -64,6 +64,30 @@ public: }; +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 @@ TEST_F(BandwidthControllerTest, TestEnableBandwidthControl) { 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 = { |