1 /*
2  * Copyright (c) 2022-2023 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 #include <securec.h>
18 
19 #include "bus_center_manager.h"
20 #include "bus_center_info_key.h"
21 #include "softbus_access_token_test.h"
22 #include "lnn_local_net_ledger.h"
23 #include "softbus_common.h"
24 #include "softbus_bus_center.h"
25 #include "softbus_errcode.h"
26 
27 constexpr char PKG_NAME[] = "com.softbus.test";
28 constexpr char NETWORK_ID[] = "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA";
29 constexpr uint16_t DATA_CHANGE_FLAG = 65535;
30 constexpr int DATA_CHANGE_FLAG_BUF_NUM = 2;
31 
32 namespace OHOS {
33 using namespace testing::ext;
34 
35 class ClientBusCenterSdkTest : public testing::Test {
36 public:
37     static void SetUpTestCase();
38     static void TearDownTestCase();
39     void SetUp();
40     void TearDown();
41     void AddPermission();
42 };
43 
SetUpTestCase()44 void ClientBusCenterSdkTest::SetUpTestCase()
45 {
46     SetAceessTokenPermission("busCenterTest");
47 }
48 
TearDownTestCase()49 void ClientBusCenterSdkTest::TearDownTestCase()
50 {
51 }
52 
SetUp()53 void ClientBusCenterSdkTest::SetUp()
54 {
55 }
56 
TearDown()57 void ClientBusCenterSdkTest::TearDown()
58 {
59 }
60 
61 /*
62 * @tc.name: SOFTBUS_CENTER_SDK_TEST_001
63 * @tc.desc: test set dataChangeFlag and get dataChangeFlag
64 * @tc.type: FUNC
65 * @tc.require: I5TQW0
66 */
67 HWTEST_F(ClientBusCenterSdkTest, SOFTBUS_CENTER_SDK_TEST_001, TestSize.Level0)
68 {
69     NodeBasicInfo info;
70     (void)memset_s(&info, sizeof(NodeBasicInfo), 0, sizeof(NodeBasicInfo));
71     EXPECT_TRUE(GetLocalNodeDeviceInfo(PKG_NAME, &info) == SOFTBUS_OK);
72     EXPECT_TRUE(SetNodeDataChangeFlag(PKG_NAME, info.networkId, DATA_CHANGE_FLAG) == SOFTBUS_OK);
73     uint16_t dataChangeFlag = 0;
74     EXPECT_TRUE(GetNodeKeyInfo(PKG_NAME, info.networkId, NODE_KEY_DATA_CHANGE_FLAG,
75         (uint8_t *)&dataChangeFlag, DATA_CHANGE_FLAG_BUF_NUM) == SOFTBUS_OK);
76     EXPECT_TRUE(dataChangeFlag == DATA_CHANGE_FLAG);
77 }
78 
79 /*
80 * @tc.name: SOFTBUS_CENTER_SDK_TEST_002
81 * @tc.desc: test SetNodeDataChangeFlag fail
82 * @tc.type: FUNC
83 * @tc.require: I5TQW0
84 */
85 HWTEST_F(ClientBusCenterSdkTest, SOFTBUS_CENTER_SDK_TEST_002, TestSize.Level0)
86 {
87     NodeBasicInfo info;
88     (void)memset_s(&info, sizeof(NodeBasicInfo), 0, sizeof(NodeBasicInfo));
89     EXPECT_TRUE(GetLocalNodeDeviceInfo(PKG_NAME, &info) == SOFTBUS_OK);
90     EXPECT_TRUE(SetNodeDataChangeFlag(nullptr, info.networkId, DATA_CHANGE_FLAG) == SOFTBUS_INVALID_PARAM);
91     EXPECT_TRUE(SetNodeDataChangeFlag(PKG_NAME, NETWORK_ID, DATA_CHANGE_FLAG) == SOFTBUS_INVALID_PARAM);
92 }
93 
94 /*
95 * @tc.name: SOFTBUS_CENTER_SDK_TEST_003
96 * @tc.desc: after joinLNN test get remote dataChangeFlag
97 * @tc.type: FUNC
98 * @tc.require: I5TQW0
99 */
100 HWTEST_F(ClientBusCenterSdkTest, SOFTBUS_CENTER_SDK_TEST_003, TestSize.Level0)
101 {
102     NodeBasicInfo *info = nullptr;
103     int32_t infoNum = 0;
104     uint16_t dataChangeFlag = 1;
105     EXPECT_TRUE(GetAllNodeDeviceInfo(PKG_NAME, &info, &infoNum) == SOFTBUS_OK);
106     if (info != nullptr && infoNum != 0)
107     {
108         EXPECT_TRUE(GetNodeKeyInfo(PKG_NAME, info->networkId, NODE_KEY_DATA_CHANGE_FLAG,
109             (uint8_t *)&dataChangeFlag, DATA_CHANGE_FLAG_BUF_NUM) == SOFTBUS_OK);
110         EXPECT_TRUE(dataChangeFlag == DATA_CHANGE_FLAG);
111     }
112 }
113 }
114