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_heartbeat_ctrl.h"
23 #include "message_handler.h"
24 #include "softbus_access_token_test.h"
25 #include "softbus_bus_center.h"
26 #include "softbus_client_frame_manager.h"
27 #include "softbus_error_code.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
TearDownTestCase()45 void BusCenterHeartbeatSdkTest::TearDownTestCase() { }
46
SetUp()47 void BusCenterHeartbeatSdkTest::SetUp()
48 {
49 SetAccessTokenPermission("device_manager");
50 EXPECT_EQ(InitSoftBus(TEST_PKG_NAME1), SOFTBUS_OK);
51 uint64_t tokenId = SetTokenIdByProcessName("device_manager");
52 printf("SetTokenIdByProcessName tokenId:%ju\n", tokenId);
53 LnnInitLnnLooper();
54 LooperInit();
55 LnnInitBusCenterEvent();
56 LnnInitHeartbeat();
57 }
58
TearDown()59 void BusCenterHeartbeatSdkTest::TearDown()
60 {
61 LnnDeinitHeartbeat();
62 LnnDeinitBusCenterEvent();
63 LooperDeinit();
64 LnnDeinitLnnLooper();
65 }
66
67 /*
68 * @tc.name: Shift_Lnn_Gear_Test_001
69 * @tc.desc: heart beat parameter adjust test
70 * @tc.type: FUNC
71 * @tc.require: I5HMXC
72 */
73 HWTEST_F(BusCenterHeartbeatSdkTest, Shift_Lnn_Gear_Test_001, TestSize.Level1)
74 {
75 char networkId1[] = "012345678B987654321001234A678998";
76 char networkId2[] = "012345678B987654321001234A67899876543210012E4567899876543210012FFFFFFFFF012345678B9876";
77 const char *callerId1 = "1";
78 const char *callerId2 = "2";
79 const char *callerId3 = "3";
80 const char *callerId4 = "4";
81 const char *callerId5 = "5";
82 const char *callerId6 = "";
83 const char *callerId7 = "012345678B987654321001234A67899876543210012E4567899876543210012FFFFFFFFF012345678B9876"
84 "54321001234A67899876543210012E4567899876543210012FFFFFFFFF";
85 GearMode mode1 = { .cycle = MID_FREQ_CYCLE, .duration = DEFAULT_DURATION, .wakeupFlag = false };
86 GearMode mode2 = { .cycle = HIGH_FREQ_CYCLE, .duration = DEFAULT_DURATION, .wakeupFlag = true };
87 GearMode mode3 = { .cycle = LOW_FREQ_CYCLE, .duration = LONG_DURATION, .wakeupFlag = true };
88 GearMode mode4 = { .cycle = HIGH_FREQ_CYCLE, .duration = NORMAL_DURATION, .wakeupFlag = true };
89 GearMode mode5 = { .cycle = MID_FREQ_CYCLE, .duration = LONG_DURATION, .wakeupFlag = false };
90
91 int32_t ret = ShiftLNNGear(TEST_PKG_NAME1, callerId1, nullptr, &mode1);
92 if (ret != SOFTBUS_NOT_IMPLEMENT) {
93 EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId1, networkId1, &mode1), SOFTBUS_INVALID_PARAM);
94 EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId1, networkId2, &mode1), SOFTBUS_INVALID_PARAM);
95 EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId1, nullptr, &mode1), SOFTBUS_NETWORK_HB_START_STRATEGY_FAIL);
96 EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME2, callerId1, nullptr, &mode1), SOFTBUS_NETWORK_HB_START_STRATEGY_FAIL);
97 EXPECT_EQ(ShiftLNNGear(nullptr, callerId1, nullptr, &mode1), SOFTBUS_INVALID_PARAM);
98 EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId1, nullptr, nullptr), SOFTBUS_INVALID_PARAM);
99
100 EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId1, nullptr, &mode2), SOFTBUS_NETWORK_HB_START_STRATEGY_FAIL);
101 EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId2, nullptr, &mode2), SOFTBUS_NETWORK_HB_START_STRATEGY_FAIL);
102 EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId3, nullptr, &mode3), SOFTBUS_NETWORK_HB_START_STRATEGY_FAIL);
103 EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId4, nullptr, &mode4), SOFTBUS_NETWORK_HB_START_STRATEGY_FAIL);
104 EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId5, nullptr, &mode5), SOFTBUS_NETWORK_HB_START_STRATEGY_FAIL);
105 EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId6, nullptr, &mode5), SOFTBUS_INVALID_PARAM);
106 EXPECT_EQ(ShiftLNNGear(TEST_PKG_NAME1, callerId7, nullptr, &mode5), SOFTBUS_INVALID_PARAM);
107 }
108 }
109 } // namespace OHOS
110