1 /*
2 * Copyright 2017 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 *
16 * IdletimerControllerTest.cpp - unit tests for IdletimerController.cpp
17 */
18
19 #include <gtest/gtest.h>
20
21 #include <android-base/strings.h>
22 #include <android-base/stringprintf.h>
23
24 #include "IdletimerController.h"
25 #include "IptablesBaseTest.h"
26
27 using android::base::Join;
28 using android::base::StringPrintf;
29
30 class IdletimerControllerTest : public IptablesBaseTest {
31 protected:
IdletimerControllerTest()32 IdletimerControllerTest() {
33 IdletimerController::execIptablesRestore = fakeExecIptablesRestore;
34 }
35 IdletimerController mIt;
36 };
37
TEST_F(IdletimerControllerTest,TestSetupIptablesHooks)38 TEST_F(IdletimerControllerTest, TestSetupIptablesHooks) {
39 mIt.setupIptablesHooks();
40 expectIptablesRestoreCommands(ExpectedIptablesCommands{});
41 }
42
TEST_F(IdletimerControllerTest,TestEnableDisable)43 TEST_F(IdletimerControllerTest, TestEnableDisable) {
44 std::vector<std::string> expected = {
45 "*raw\n"
46 ":idletimer_raw_PREROUTING -\n"
47 "COMMIT\n"
48 "*mangle\n"
49 ":idletimer_mangle_POSTROUTING -\n"
50 "COMMIT\n",
51 };
52
53 mIt.enableIdletimerControl();
54 expectIptablesRestoreCommands(expected);
55
56 mIt.enableIdletimerControl();
57 expectIptablesRestoreCommands(expected);
58
59 mIt.disableIdletimerControl();
60 expectIptablesRestoreCommands(expected);
61
62 mIt.disableIdletimerControl();
63 expectIptablesRestoreCommands(expected);
64 }
65
makeAddRemoveCommands(bool add)66 const std::vector<std::string> makeAddRemoveCommands(bool add) {
67 const char *op = add ? "-A" : "-D";
68 std::vector<std::string> cmds = {
69 "*raw",
70 StringPrintf("%s idletimer_raw_PREROUTING -i wlan0 -j IDLETIMER"
71 " --timeout 12345 --label hello --send_nl_msg 1", op),
72 "COMMIT",
73 "*mangle",
74 StringPrintf("%s idletimer_mangle_POSTROUTING -o wlan0 -j IDLETIMER"
75 " --timeout 12345 --label hello --send_nl_msg 1", op),
76 "COMMIT\n",
77 };
78 return { Join(cmds, '\n') };
79 }
80
TEST_F(IdletimerControllerTest,TestAddRemove)81 TEST_F(IdletimerControllerTest, TestAddRemove) {
82 auto expected = makeAddRemoveCommands(true);
83 mIt.addInterfaceIdletimer("wlan0", 12345, "hello");
84 expectIptablesRestoreCommands(expected);
85
86 mIt.addInterfaceIdletimer("wlan0", 12345, "hello");
87 expectIptablesRestoreCommands(expected);
88
89 expected = makeAddRemoveCommands(false);
90 mIt.removeInterfaceIdletimer("wlan0", 12345, "hello");
91 expectIptablesRestoreCommands(expected);
92
93 mIt.removeInterfaceIdletimer("wlan0", 12345, "hello");
94 expectIptablesRestoreCommands(expected);
95 }
96