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 #include "gtest/gtest.h"
16 #include <securec.h>
17
18 #include "bus_center_event.h"
19 #include "lnn_bt_monitor.c"
20 #include "lnn_event_monitor_impl.h"
21 #include "lnn_netlink_monitor.c"
22 #include "network_mock.h"
23 #include "softbus_error_code.h"
24
25 namespace OHOS {
26 #define TEST_LEN 5
27 #define TEST_LEN2 65
28 #define TEST_LISTENER_ID 20
29 #define TEST_LISTENER_ID2 (-1)
30 #define TEST_STATE1 1
31 #define TEST_STATE2 3
32 #define TEST_STATE3 5
33 #define TEST_STATE4 7
34 #define TEST_ACL_STATE 0
35 #define TEST_ACL_STATE2 1
36 using namespace testing::ext;
37 using namespace testing;
38 class AdapterDsoftbusNetworkTest : public testing::Test {
39 protected:
40 static void SetUpTestCase(void);
41 static void TearDownTestCase(void);
42 void SetUp();
43 void TearDown();
44 };
SetUpTestCase(void)45 void AdapterDsoftbusNetworkTest::SetUpTestCase(void) { }
TearDownTestCase(void)46 void AdapterDsoftbusNetworkTest::TearDownTestCase(void) { }
SetUp()47 void AdapterDsoftbusNetworkTest::SetUp() { }
TearDown()48 void AdapterDsoftbusNetworkTest::TearDown() { }
49
50 /*
51 * @tc.name: CreateNetlinkSocket
52 * @tc.desc: softbus network test
53 * @tc.type: FUNC
54 * @tc.require:
55 */
56 HWTEST_F(AdapterDsoftbusNetworkTest, CreateNetlinkSocketTest001, TestSize.Level1)
57 {
58 NiceMock<NetworkInterfaceMock> networkMock;
59 ON_CALL(networkMock, SoftBusSocketCreate).WillByDefault(Return(SOFTBUS_OK));
60 ON_CALL(networkMock, SoftBusSocketSetOpt).WillByDefault(Return(SOFTBUS_OK));
61 ON_CALL(networkMock, SoftBusSocketClose).WillByDefault(Return(SOFTBUS_OK));
62 ON_CALL(networkMock, SoftBusSocketBind).WillByDefault(Return(SOFTBUS_OK));
63 int32_t ret = CreateNetlinkSocket();
64 EXPECT_TRUE(ret == SOFTBUS_OK);
65 EXPECT_CALL(networkMock, SoftBusSocketCreate)
66 .WillOnce(Return(SOFTBUS_ADAPTER_ERR))
67 .WillRepeatedly(Return(SOFTBUS_OK));
68 ret = CreateNetlinkSocket();
69 EXPECT_TRUE(ret == SOFTBUS_NETWORK_CREATE_SOCKET_FAILED);
70 EXPECT_CALL(networkMock, SoftBusSocketSetOpt)
71 .WillOnce(Return(SOFTBUS_ADAPTER_ERR))
72 .WillOnce(Return(SOFTBUS_ADAPTER_ERR))
73 .WillRepeatedly(Return(SOFTBUS_OK));
74 ret = CreateNetlinkSocket();
75 EXPECT_TRUE(ret == SOFTBUS_NETWORK_SET_SOCKET_OPTION_FAILED);
76 EXPECT_CALL(networkMock, SoftBusSocketBind)
77 .WillOnce(Return(SOFTBUS_ADAPTER_ERR))
78 .WillRepeatedly(Return(SOFTBUS_OK));
79 ret = CreateNetlinkSocket();
80 EXPECT_TRUE(ret == SOFTBUS_NETWORK_BIND_SOCKET_FAILED);
81 }
82
83 /*
84 * @tc.name: ProcessAddrEvent
85 * @tc.desc: softbus network test
86 * @tc.type: FUNC
87 * @tc.require:
88 */
89 HWTEST_F(AdapterDsoftbusNetworkTest, ProcessAddrEventTest001, TestSize.Level1)
90 {
91 nlmsghdr nlh1 = {
92 .nlmsg_len = TEST_LEN,
93 };
94 NiceMock<NetworkInterfaceMock> networkMock;
95 ON_CALL(networkMock, LnnGetNetIfTypeByName).WillByDefault(Return(SOFTBUS_OK));
96 ON_CALL(networkMock, SoftBusSocketCreate).WillByDefault(Return(SOFTBUS_OK));
97 ON_CALL(networkMock, SoftBusSocketSetOpt).WillByDefault(Return(SOFTBUS_OK));
98 ON_CALL(networkMock, SoftBusSocketClose).WillByDefault(Return(SOFTBUS_OK));
99 ON_CALL(networkMock, SoftBusSocketBind).WillByDefault(Return(SOFTBUS_OK));
100 ON_CALL(networkMock, SoftBusSocketRecv).WillByDefault(Return(TEST_LISTENER_ID2));
101 ProcessAddrEvent(&nlh1);
102 nlmsghdr nlh2 = {
103 .nlmsg_len = TEST_LEN2,
104 };
105 ProcessAddrEvent(&nlh2);
106 EXPECT_TRUE(NetlinkOnConnectEvent(NETLINK, -1, nullptr) == SOFTBUS_OK);
107 EXPECT_TRUE(NetlinkOnDataEvent(NETLINK, SOFTBUS_SOCKET_OUT, -1) == SOFTBUS_INVALID_PARAM);
108 EXPECT_CALL(networkMock, SoftBusSocketRecv).WillOnce(Return(TEST_LEN)).WillRepeatedly(Return(TEST_LISTENER_ID2));
109 EXPECT_TRUE(NetlinkOnDataEvent(NETLINK, SOFTBUS_SOCKET_IN, 0) == SOFTBUS_SOCKET_EXCEPTION);
110 }
111
112 /*
113 * @tc.name: LnnOnBtStateChanged
114 * @tc.desc: softbus network test
115 * @tc.type: FUNC
116 * @tc.require:
117 */
118 HWTEST_F(AdapterDsoftbusNetworkTest, LnnOnBtStateChangedTest001, TestSize.Level1)
119 {
120 NiceMock<NetworkInterfaceMock> networkMock;
121 ON_CALL(networkMock, LnnAsyncCallbackHelper).WillByDefault(Return(SOFTBUS_OK));
122 ON_CALL(networkMock, SoftBusAddBtStateListener).WillByDefault(Return(SOFTBUS_OK));
123
124 int32_t listenerId = TEST_LISTENER_ID;
125 int32_t state = TEST_STATE1;
126 SoftBusBtAddr addr;
127 int32_t aclState = TEST_ACL_STATE;
128 LnnOnBtStateChanged(listenerId, state);
129 state = TEST_STATE2;
130 LnnOnBtStateChanged(listenerId, state);
131 state = TEST_STATE3;
132 LnnOnBtStateChanged(listenerId, state);
133 state = TEST_STATE4;
134 LnnOnBtStateChanged(listenerId, state);
135 EXPECT_CALL(networkMock, LnnAsyncCallbackHelper).WillRepeatedly(Return(SOFTBUS_INVALID_PARAM));
136 LnnOnBtStateChanged(listenerId, state);
137 LnnOnBtAclStateChanged(listenerId, &addr, aclState, 0);
138 aclState = TEST_ACL_STATE2;
139 LnnOnBtAclStateChanged(listenerId, &addr, aclState, 0);
140 aclState = TEST_STATE1;
141 LnnOnBtAclStateChanged(listenerId, &addr, aclState, 0);
142 int32_t ret = LnnInitBtStateMonitorImpl();
143 EXPECT_TRUE(ret == SOFTBUS_OK);
144 EXPECT_CALL(networkMock, SoftBusAddBtStateListener).
145 WillRepeatedly(Return(SOFTBUS_COMM_BLUETOOTH_ADD_STATE_LISTENER_ERR));
146 ret = LnnInitBtStateMonitorImpl();
147 EXPECT_TRUE(ret == SOFTBUS_COMM_BLUETOOTH_ADD_STATE_LISTENER_ERR);
148 }
149 } // namespace OHOS