1 /*
2  * Copyright (c) 2022-2024 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 "distributed_input_transbase_test.h"
17 
18 #include <cstdlib>
19 
20 #include "dinput_errcode.h"
21 #include "dinput_softbus_define.h"
22 
23 using namespace testing::ext;
24 using namespace OHOS::DistributedHardware::DistributedInput;
25 using namespace std;
26 namespace OHOS {
27 namespace DistributedHardware {
28 namespace DistributedInput {
29 namespace {
30     const std::string PEER_SESSION_NAME = "ohos.dhardware.dinput.session8647073e02e7a78f09473aa122";
31     const std::string REMOTE_DEV_ID = "f6d4c0864707aefte7a78f09473aa122ff57fc81c00981fcf5be989e7d112591";
32     const std::string DINPUT_PKG_NAME_TEST = "ohos.dhardware.dinput";
33 }
SetUp()34 void DistributedInputTransbaseTest::SetUp()
35 {
36 }
37 
TearDown()38 void DistributedInputTransbaseTest::TearDown()
39 {
40 }
41 
SetUpTestCase()42 void DistributedInputTransbaseTest::SetUpTestCase()
43 {
44 }
45 
TearDownTestCase()46 void DistributedInputTransbaseTest::TearDownTestCase()
47 {
48 }
49 
OnResult(const std::string & devId,const uint32_t status)50 void DistributedInputTransbaseTest::TestRegisterSessionStateCallbackStub::OnResult(const std::string &devId,
51     const uint32_t status)
52 {
53     (void)devId;
54     (void)status;
55 }
56 
57 HWTEST_F(DistributedInputTransbaseTest, Init01, testing::ext::TestSize.Level0)
58 {
59     int32_t ret = DistributedInputTransportBase::GetInstance().Init();
60     EXPECT_EQ(DH_SUCCESS, ret);
61 }
62 
63 HWTEST_F(DistributedInputTransbaseTest, StartSession01, testing::ext::TestSize.Level0)
64 {
65     std::string remoteDevId = "";
66     int32_t ret = DistributedInputTransportBase::GetInstance().StartSession(remoteDevId);
67     EXPECT_EQ(DH_SUCCESS, ret);
68 }
69 
70 HWTEST_F(DistributedInputTransbaseTest, StartSession02, testing::ext::TestSize.Level1)
71 {
72     std::string srcId = "f6d4c08647073e02e7a78f09473aa122ff57fc81c00981fcf5be989e7d112591";
73     int32_t sessionId = 2;
74     DistributedInputTransportBase::GetInstance().remoteDevSessionMap_[srcId] = sessionId;
75     int32_t ret = DistributedInputTransportBase::GetInstance().StartSession(srcId);
76     EXPECT_EQ(DH_SUCCESS, ret);
77 }
78 
79 HWTEST_F(DistributedInputTransbaseTest, GetDevIdBySessionId01, testing::ext::TestSize.Level0)
80 {
81     int32_t sessionId = 0;
82     std::string srcId = "f6d4c08647073e02e7a78f09473aa122ff57fc81c00981fcf5be989e7d112591";
83     DistributedInputTransportBase::GetInstance().remoteDevSessionMap_.clear();
84     DistributedInputTransportBase::GetInstance().remoteDevSessionMap_[srcId] = sessionId;
85     std::string ret = DistributedInputTransportBase::GetInstance().GetDevIdBySessionId(sessionId);
86     DistributedInputTransportBase::GetInstance().remoteDevSessionMap_.clear();
87     EXPECT_EQ(srcId, ret);
88     ret = DistributedInputTransportBase::GetInstance().GetDevIdBySessionId(sessionId);
89     EXPECT_EQ("", ret);
90 }
91 
92 HWTEST_F(DistributedInputTransbaseTest, OnSessionOpened01, testing::ext::TestSize.Level0)
93 {
94     int32_t sessionId = 0;
95     PeerSocketInfo peerSocketInfo = {
96         .name = const_cast<char*>(PEER_SESSION_NAME.c_str()),
97         .networkId = const_cast<char*>(REMOTE_DEV_ID.c_str()),
98         .pkgName = const_cast<char*>(DINPUT_PKG_NAME_TEST.c_str()),
99         .dataType = DATA_TYPE_BYTES
100     };
101     std::string srcId = "f6d4c08647073e02e7a78f09473aa122ff57fc81c00981fcf5be989e7d112591";
102     DistributedInputTransportBase::GetInstance().remoteDevSessionMap_[srcId] = sessionId;
103     int32_t ret = DistributedInputTransportBase::GetInstance().OnSessionOpened(sessionId, peerSocketInfo);
104     DistributedInputTransportBase::GetInstance().remoteDevSessionMap_.clear();
105     EXPECT_EQ(DH_SUCCESS, ret);
106 
107     ret = DistributedInputTransportBase::GetInstance().OnSessionOpened(sessionId, peerSocketInfo);
108     EXPECT_EQ(DH_SUCCESS, ret);
109 }
110 
111 HWTEST_F(DistributedInputTransbaseTest, HandleSession01, testing::ext::TestSize.Level1)
112 {
113     DistributedInputTransportBase::GetInstance().remoteDevSessionMap_.clear();
114     int32_t sessionId = 1;
115     nlohmann::json recMsg;
116     recMsg[DINPUT_SOFTBUS_KEY_CMD_TYPE] = TRANS_SINK_MSG_ON_RELAY_STOPTYPE;
117     std::string message = recMsg.dump();
118     DistributedInputTransportBase::GetInstance().HandleSession(sessionId, message);
119     EXPECT_EQ(0, DistributedInputTransportBase::GetInstance().remoteDevSessionMap_.size());
120 }
121 
122 HWTEST_F(DistributedInputTransbaseTest, HandleSession02, testing::ext::TestSize.Level1)
123 {
124     int32_t sessionId = 1;
125     nlohmann::json recMsg;
126     recMsg[DINPUT_SOFTBUS_KEY_CMD_TYPE] = TRANS_SOURCE_MSG_PREPARE;
127     std::string message = recMsg.dump();
128     DistributedInputTransportBase::GetInstance().HandleSession(sessionId, message);
129     EXPECT_EQ(0, DistributedInputTransportBase::GetInstance().remoteDevSessionMap_.size());
130 }
131 
132 HWTEST_F(DistributedInputTransbaseTest, CheckRecivedData01, testing::ext::TestSize.Level1)
133 {
134     std::string message = "";
135     DistributedInputTransportBase::GetInstance().CheckRecivedData(message);
136     message = "message_test";
137     DistributedInputTransportBase::GetInstance().CheckRecivedData(message);
138     nlohmann::json recMsg;
139     recMsg[DINPUT_SOFTBUS_KEY_CMD_TYPE] = "cmd_type_test";
140     DistributedInputTransportBase::GetInstance().CheckRecivedData(recMsg.dump());
141     recMsg[DINPUT_SOFTBUS_KEY_CMD_TYPE] = TRANS_SOURCE_TO_SOURCE_MSG_STOP_DHID_RESULT;
142     DistributedInputTransportBase::GetInstance().CheckRecivedData(recMsg.dump());
143     EXPECT_EQ(0, DistributedInputTransportBase::GetInstance().remoteDevSessionMap_.size());
144 }
145 
146 HWTEST_F(DistributedInputTransbaseTest, OnBytesReceived01, testing::ext::TestSize.Level1)
147 {
148     int32_t sessionId = -1;
149     char *data = nullptr;
150     uint32_t dataLen = -1;
151     DistributedInputTransportBase::GetInstance().OnBytesReceived(sessionId, data, dataLen);
152     sessionId = 1;
153     DistributedInputTransportBase::GetInstance().OnBytesReceived(sessionId, data, dataLen);
154     char dataMsg[10] = "dataMsg";
155     DistributedInputTransportBase::GetInstance().OnBytesReceived(sessionId, dataMsg, dataLen);
156     dataLen = 1;
157     DistributedInputTransportBase::GetInstance().OnBytesReceived(sessionId, dataMsg, dataLen);
158     EXPECT_EQ(0, DistributedInputTransportBase::GetInstance().remoteDevSessionMap_.size());
159 }
160 
161 HWTEST_F(DistributedInputTransbaseTest, SendMsg01, testing::ext::TestSize.Level1)
162 {
163     int32_t sessionId = 1;
164     std::string message(MSG_MAX_SIZE + 1, 'a');
165     int32_t ret = DistributedInputTransportBase::GetInstance().SendMsg(sessionId, message);
166     EXPECT_EQ(ERR_DH_INPUT_SERVER_SOURCE_TRANSPORT_SENDMESSSAGE, ret);
167 
168     std::string messageData(MSG_MAX_SIZE - 1, 'a');
169     ret = DistributedInputTransportBase::GetInstance().SendMsg(sessionId, messageData);
170     EXPECT_EQ(DH_SUCCESS, ret);
171 }
172 
173 HWTEST_F(DistributedInputTransbaseTest, Release01, testing::ext::TestSize.Level1)
174 {
175     DistributedInputTransportBase::GetInstance().Release();
176     EXPECT_EQ(0, DistributedInputTransportBase::GetInstance().channelStatusMap_.size());
177 }
178 
179 HWTEST_F(DistributedInputTransbaseTest, StopSession_001, testing::ext::TestSize.Level1)
180 {
181     DistributedInputTransportBase::GetInstance().remoteDevSessionMap_.clear();
182     std::string remoteDevId = "remoteDevId_test";
183     DistributedInputTransportBase::GetInstance().StopSession(remoteDevId);
184     EXPECT_EQ(0, DistributedInputTransportBase::GetInstance().remoteDevSessionMap_.size());
185 }
186 
187 HWTEST_F(DistributedInputTransbaseTest, RunSessionStateCallback_001, testing::ext::TestSize.Level1)
188 {
189     std::string remoteDevId = "remoteDevId_test";
190     uint32_t sessionState = 1;
191     sptr<TestRegisterSessionStateCallbackStub> callback(new TestRegisterSessionStateCallbackStub());
192     DistributedInputTransportBase::GetInstance().RegisterSessionStateCb(callback);
193     DistributedInputTransportBase::GetInstance().RunSessionStateCallback(remoteDevId, sessionState);
194     DistributedInputTransportBase::GetInstance().UnregisterSessionStateCb();
195     auto ret = DistributedInputTransportBase::GetInstance().CountSession(remoteDevId);
196     EXPECT_EQ(DH_SUCCESS, ret);
197 }
198 
199 HWTEST_F(DistributedInputTransbaseTest, OnSessionClosed_001, testing::ext::TestSize.Level1)
200 {
201     int32_t sessionId = 1;
202     std::string devId = "devId_741258";
203     ShutdownReason reason = ShutdownReason::SHUTDOWN_REASON_UNKNOWN;
204     DistributedInputTransportBase::GetInstance().OnSessionClosed(sessionId, reason);
205 
206     DistributedInputTransportBase::GetInstance().remoteDevSessionMap_[devId] = sessionId;
207     DistributedInputTransportBase::GetInstance().OnSessionClosed(sessionId, reason);
208 
209     DistributedInputTransportBase::GetInstance().sinkCallback_ = std::shared_ptr<DInputTransbaseSinkCallback>();
210     DistributedInputTransportBase::GetInstance().OnSessionClosed(sessionId, reason);
211 
212     DistributedInputTransportBase::GetInstance().srcCallback_ = std::shared_ptr<DInputTransbaseSourceCallback>();
213     DistributedInputTransportBase::GetInstance().OnSessionClosed(sessionId, reason);
214 
215     DistributedInputTransportBase::GetInstance().srcMgrCallback_ = std::shared_ptr<DInputSourceManagerCallback>();
216     DistributedInputTransportBase::GetInstance().OnSessionClosed(sessionId, reason);
217     auto ret = DistributedInputTransportBase::GetInstance().CountSession(devId);
218     EXPECT_EQ(DH_SUCCESS, ret);
219 }
220 
221 } // namespace DistributedInput
222 } // namespace DistributedHardware
223 } // namespace OHOS
224