1 /*
2  * Copyright (c) 2022 Huawei Device Co., Ltd.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at
6  *
7  *     http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15 
16 #include <gtest/gtest.h>
17 
18 #include <securec.h>
19 
20 #include "bus_center_event.h"
21 #include "bus_center_manager.h"
22 #include "lnn_decision_center.h"
23 #include "lnn_heartbeat_ctrl.h"
24 #include "message_handler.h"
25 #include "softbus_access_token_test.h"
26 #include "softbus_bus_center.h"
27 #include "softbus_errcode.h"
28 
29 namespace OHOS {
30 using namespace testing::ext;
31 
32 static constexpr char TEST_PKG_NAME1[] = "com.softbus.test";
33 static constexpr char TEST_PKG_NAME2[] = "com.softbus.test1";
34 
35 class BusCenterHeartbeatSdkTest : public testing::Test {
36 public:
37     static void SetUpTestCase();
38     static void TearDownTestCase();
39     void SetUp();
40     void TearDown();
41 };
42 
SetUpTestCase()43 void BusCenterHeartbeatSdkTest::SetUpTestCase()
44 {
45     SetAceessTokenPermission("busCenterTest");
46     LnnInitLnnLooper();
47     LooperInit();
48     LnnInitBusCenterEvent();
49     LnnInitHeartbeat();
50 }
51 
TearDownTestCase()52 void BusCenterHeartbeatSdkTest::TearDownTestCase()
53 {
54     LnnDeinitHeartbeat();
55     LnnDeinitBusCenterEvent();
56     LooperDeinit();
57     LnnDeinitLnnLooper();
58 }
59 
SetUp()60 void BusCenterHeartbeatSdkTest::SetUp()
61 {
62 }
63 
TearDown()64 void BusCenterHeartbeatSdkTest::TearDown()
65 {
66 }
67 
68 /*
69 * @tc.name: Shift_Lnn_Gear_Test_001
70 * @tc.desc: heart beat parameter adjust test
71 * @tc.type: FUNC
72 * @tc.require: I5HMXC
73 */
74 HWTEST_F(BusCenterHeartbeatSdkTest, Shift_Lnn_Gear_Test_001, TestSize.Level1)
75 {
76     char networkId1[] = "012345678B987654321001234A678998";
77     char networkId2[] = "012345678B987654321001234A67899876543210012E4567899876543210012FFFFFFFFF012345678B9876";
78     const char *callerId1 = "1";
79     const char *callerId2 = "2";
80     const char *callerId3 = "3";
81     const char *callerId4 = "4";
82     const char *callerId5 = "5";
83     const char *callerId6 = "";
84     const char *callerId7 = "012345678B987654321001234A67899876543210012E4567899876543210012FFFFFFFFF012345678B9876"
85         "54321001234A67899876543210012E4567899876543210012FFFFFFFFF";
86     GearMode mode1 = { .cycle = MID_FREQ_CYCLE, .duration = DEFAULT_DURATION, .wakeupFlag = false };
87     GearMode mode2 = { .cycle = HIGH_FREQ_CYCLE, .duration = DEFAULT_DURATION, .wakeupFlag = true };
88     GearMode mode3 = { .cycle = LOW_FREQ_CYCLE, .duration = LONG_DURATION, .wakeupFlag = true };
89     GearMode mode4 = { .cycle = HIGH_FREQ_CYCLE, .duration = NORMAL_DURATION, .wakeupFlag = true };
90     GearMode mode5 = { .cycle = MID_FREQ_CYCLE, .duration = LONG_DURATION, .wakeupFlag = false };
91 
92     int32_t ret = ShiftLNNGear(TEST_PKG_NAME1, callerId1, NULL, &mode1);
93     if (ret != SOFTBUS_NOT_IMPLEMENT) {
94         EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId1, networkId1, &mode1), SOFTBUS_INVALID_PARAM);
95         EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId1, networkId2, &mode1), SOFTBUS_INVALID_PARAM);
96         EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId1, NULL, &mode1), SOFTBUS_OK);
97         EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME2, callerId1, NULL, &mode1), SOFTBUS_OK);
98         EXPECT_EQ(ShiftLNNGear(NULL, callerId1, NULL, &mode1), SOFTBUS_INVALID_PARAM);
99         EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId1, NULL, NULL), SOFTBUS_INVALID_PARAM);
100 
101         EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId1, NULL, &mode2), SOFTBUS_OK);
102         EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId2, NULL, &mode2), SOFTBUS_OK);
103         EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId3, NULL, &mode3), SOFTBUS_OK);
104         EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId4, NULL, &mode4), SOFTBUS_OK);
105         EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId5, NULL, &mode5), SOFTBUS_OK);
106         EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId6, NULL, &mode5), SOFTBUS_INVALID_PARAM);
107         EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId7, NULL, &mode5), SOFTBUS_INVALID_PARAM);
108     }
109 }
110 } // namespace OHOS
111