Home | History | Annotate | Download | only in server 
      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:
     32     IdletimerControllerTest() {
     33         IdletimerController::execIptablesRestore = fakeExecIptablesRestore;
     34     }
     35     IdletimerController mIt;
     36 };
     37 
     38 TEST_F(IdletimerControllerTest, TestSetupIptablesHooks) {
     39     mIt.setupIptablesHooks();
     40     expectIptablesRestoreCommands(ExpectedIptablesCommands{});
     41 }
     42 
     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 
     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 
     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