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 "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 {
SetUp()29 void DistributedInputTransbaseTest::SetUp()
30 {
31 sourceTransport_ = new DistributedInputSourceTransport();
32 statusSourceListener_ =
33 std::make_shared<DistributedInputSourceTransport::DInputTransbaseSourceListener>(sourceTransport_);
34 DistributedInputTransportBase::GetInstance().RegisterSrcHandleSessionCallback(statusSourceListener_);
35
36 sinkTransport_ = new DistributedInputSinkTransport();
37 statusSinkListener_ = std::make_shared<DistributedInputSinkTransport::DInputTransbaseSinkListener>(sinkTransport_);
38 DistributedInputTransportBase::GetInstance().RegisterSinkHandleSessionCallback(statusSinkListener_);
39 }
40
TearDown()41 void DistributedInputTransbaseTest::TearDown()
42 {
43 }
44
SetUpTestCase()45 void DistributedInputTransbaseTest::SetUpTestCase()
46 {
47 }
48
TearDownTestCase()49 void DistributedInputTransbaseTest::TearDownTestCase()
50 {
51 }
52
53 HWTEST_F(DistributedInputTransbaseTest, Init01, testing::ext::TestSize.Level0)
54 {
55 int32_t ret = DistributedInputTransportBase::GetInstance().Init();
56 EXPECT_EQ(DH_SUCCESS, ret);
57 }
58
59 HWTEST_F(DistributedInputTransbaseTest, StartSession01, testing::ext::TestSize.Level0)
60 {
61 std::string remoteDevId = "";
62 int32_t ret = DistributedInputTransportBase::GetInstance().StartSession(remoteDevId);
63 EXPECT_EQ(ERR_DH_INPUT_SERVER_SOURCE_TRANSPORT_OPEN_SESSION_FAIL, ret);
64 }
65
66 HWTEST_F(DistributedInputTransbaseTest, StartSession02, testing::ext::TestSize.Level1)
67 {
68 std::string srcId = "f6d4c08647073e02e7a78f09473aa122ff57fc81c00981fcf5be989e7d112591";
69 int32_t sessionId = 2;
70 DistributedInputTransportBase::GetInstance().remoteDevSessionMap_[srcId] = sessionId;
71 int32_t ret = DistributedInputTransportBase::GetInstance().StartSession(srcId);
72 EXPECT_EQ(DH_SUCCESS, ret);
73 }
74
75 HWTEST_F(DistributedInputTransbaseTest, GetDevIdBySessionId01, testing::ext::TestSize.Level0)
76 {
77 int32_t sessionId = 0;
78 std::string srcId = "f6d4c08647073e02e7a78f09473aa122ff57fc81c00981fcf5be989e7d112591";
79 DistributedInputTransportBase::GetInstance().remoteDevSessionMap_[srcId] = sessionId;
80 std::string ret = DistributedInputTransportBase::GetInstance().GetDevIdBySessionId(sessionId);
81 DistributedInputTransportBase::GetInstance().remoteDevSessionMap_.clear();
82 EXPECT_EQ(srcId, ret);
83 ret = DistributedInputTransportBase::GetInstance().GetDevIdBySessionId(sessionId);
84 EXPECT_EQ("", ret);
85 }
86
87 HWTEST_F(DistributedInputTransbaseTest, OnSessionOpened01, testing::ext::TestSize.Level0)
88 {
89 int32_t sessionId = 0;
90 int32_t result = ERR_DH_INPUT_SERVER_SOURCE_TRANSPORT_STOP_FAIL;
91 std::string srcId = "f6d4c08647073e02e7a78f09473aa122ff57fc81c00981fcf5be989e7d112591";
92 DistributedInputTransportBase::GetInstance().remoteDevSessionMap_[srcId] = sessionId;
93 int32_t ret = DistributedInputTransportBase::GetInstance().OnSessionOpened(sessionId, result);
94 DistributedInputTransportBase::GetInstance().remoteDevSessionMap_.clear();
95 EXPECT_EQ(DH_SUCCESS, ret);
96
97 result = DH_SUCCESS;
98 ret = DistributedInputTransportBase::GetInstance().OnSessionOpened(sessionId, result);
99 EXPECT_EQ(DH_SUCCESS, ret);
100 }
101
102 HWTEST_F(DistributedInputTransbaseTest, HandleSession01, testing::ext::TestSize.Level1)
103 {
104 DistributedInputTransportBase::GetInstance().remoteDevSessionMap_.clear();
105 int32_t sessionId = 1;
106 nlohmann::json recMsg;
107 recMsg[DINPUT_SOFTBUS_KEY_CMD_TYPE] = TRANS_SINK_MSG_ON_RELAY_STOPTYPE;
108 std::string message = recMsg.dump();
109 DistributedInputTransportBase::GetInstance().HandleSession(sessionId, message);
110 EXPECT_EQ(0, DistributedInputTransportBase::GetInstance().remoteDevSessionMap_.size());
111 }
112
113 HWTEST_F(DistributedInputTransbaseTest, HandleSession02, testing::ext::TestSize.Level1)
114 {
115 int32_t sessionId = 1;
116 nlohmann::json recMsg;
117 recMsg[DINPUT_SOFTBUS_KEY_CMD_TYPE] = TRANS_SOURCE_MSG_PREPARE;
118 std::string message = recMsg.dump();
119 DistributedInputTransportBase::GetInstance().HandleSession(sessionId, message);
120 EXPECT_EQ(0, DistributedInputTransportBase::GetInstance().remoteDevSessionMap_.size());
121 }
122
123 HWTEST_F(DistributedInputTransbaseTest, CheckRecivedData01, testing::ext::TestSize.Level1)
124 {
125 std::string message = "";
126 DistributedInputTransportBase::GetInstance().CheckRecivedData(message);
127 message = "message_test";
128 DistributedInputTransportBase::GetInstance().CheckRecivedData(message);
129 nlohmann::json recMsg;
130 recMsg[DINPUT_SOFTBUS_KEY_CMD_TYPE] = "cmd_type_test";
131 DistributedInputTransportBase::GetInstance().CheckRecivedData(recMsg.dump());
132 recMsg[DINPUT_SOFTBUS_KEY_CMD_TYPE] = TRANS_SOURCE_TO_SOURCE_MSG_STOP_DHID_RESULT;
133 DistributedInputTransportBase::GetInstance().CheckRecivedData(recMsg.dump());
134 EXPECT_EQ(0, DistributedInputTransportBase::GetInstance().remoteDevSessionMap_.size());
135 }
136
137 HWTEST_F(DistributedInputTransbaseTest, OnBytesReceived01, testing::ext::TestSize.Level1)
138 {
139 int32_t sessionId = -1;
140 char *data = nullptr;
141 uint32_t dataLen = -1;
142 DistributedInputTransportBase::GetInstance().OnBytesReceived(sessionId, data, dataLen);
143 sessionId = 1;
144 DistributedInputTransportBase::GetInstance().OnBytesReceived(sessionId, data, dataLen);
145 char dataMsg[10] = "dataMsg";
146 DistributedInputTransportBase::GetInstance().OnBytesReceived(sessionId, dataMsg, dataLen);
147 dataLen = 1;
148 DistributedInputTransportBase::GetInstance().OnBytesReceived(sessionId, dataMsg, dataLen);
149 EXPECT_EQ(0, DistributedInputTransportBase::GetInstance().remoteDevSessionMap_.size());
150 }
151
152 HWTEST_F(DistributedInputTransbaseTest, SendMsg01, testing::ext::TestSize.Level1)
153 {
154 int32_t sessionId = 1;
155 std::string message(MSG_MAX_SIZE + 1, 'a');
156 int32_t ret = DistributedInputTransportBase::GetInstance().SendMsg(sessionId, message);
157 EXPECT_EQ(ERR_DH_INPUT_SERVER_SOURCE_TRANSPORT_SENDMESSSAGE, ret);
158
159 std::string messageData(MSG_MAX_SIZE - 1, 'a');
160 ret = DistributedInputTransportBase::GetInstance().SendMsg(sessionId, messageData);
161 EXPECT_EQ(DH_SUCCESS, ret);
162 }
163
164 HWTEST_F(DistributedInputTransbaseTest, Release01, testing::ext::TestSize.Level1)
165 {
166 DistributedInputTransportBase::GetInstance().Release();
167 EXPECT_EQ(0, DistributedInputTransportBase::GetInstance().channelStatusMap_.size());
168 }
169
170 } // namespace DistributedInput
171 } // namespace DistributedHardware
172 } // namespace OHOS
173