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 #include "securec.h"
18
19 #include "client_trans_pending.h"
20 #include "client_trans_proxy_file_manager.h"
21 #include "client_trans_session_manager.h"
22 #include "softbus_def.h"
23 #include "softbus_errcode.h"
24
25 #define TEST_SESSION_ID 0
26 #define TEST_SEQ 1
27 #define TEST_SESSION_ID_SECOND 2
28 #define TEST_SEQ_SECOND 2
29 #define TEST_WAIT_ACK_TIME 10
30
31 using namespace std;
32 using namespace testing::ext;
33
34 namespace OHOS {
35 class ClientTransPendingTest : public testing::Test {
36 public:
ClientTransPendingTest()37 ClientTransPendingTest() {}
~ClientTransPendingTest()38 ~ClientTransPendingTest() {}
39 static void SetUpTestCase(void);
40 static void TearDownTestCase(void);
SetUp()41 void SetUp() override {}
TearDown()42 void TearDown() override {}
43 };
44
SetUpTestCase(void)45 void ClientTransPendingTest::SetUpTestCase(void)
46 {
47 int ret = InitPendingPacket();
48 EXPECT_EQ(SOFTBUS_OK, ret);
49
50 ret = TransClientInit();
51 EXPECT_EQ(SOFTBUS_OK, ret);
52 }
TearDownTestCase(void)53 void ClientTransPendingTest::TearDownTestCase(void) {}
54
55 /**
56 * @tc.name: TransPendingTest001
57 * @tc.desc: client trans pending test,use the wrong or normal parameter.
58 * @tc.type: FUNC
59 * @tc.require:
60 */
61 HWTEST_F(ClientTransPendingTest, TransPendingTest, TestSize.Level0)
62 {
63 uint32_t id = 1;
64 uint64_t seq = 0;
65
66 int ret = CreatePendingPacket(id, seq);
67 EXPECT_EQ(SOFTBUS_OK, ret);
68
69 ret = CreatePendingPacket(TEST_SESSION_ID, seq);
70 EXPECT_EQ(SOFTBUS_OK, ret);
71
72 ret = CreatePendingPacket(id, TEST_SEQ);
73 EXPECT_EQ(SOFTBUS_OK, ret);
74
75 ret = CreatePendingPacket(id, seq);
76 EXPECT_EQ(SOFTBUS_ALREADY_EXISTED, ret);
77
78 uint32_t waitMillis = TEST_WAIT_ACK_TIME;
79 ret = GetPendingPacketData(id, seq, waitMillis, true, nullptr);
80 EXPECT_EQ(SOFTBUS_ERR, ret);
81
82 TransPendData pendDate = {0};
83 ret = GetPendingPacketData(TEST_SESSION_ID_SECOND, TEST_SEQ_SECOND, waitMillis, true, &pendDate);
84 EXPECT_EQ(SOFTBUS_NOT_FIND, ret);
85
86 ret = GetPendingPacketData(TEST_SESSION_ID_SECOND, seq, waitMillis, true, &pendDate);
87 EXPECT_EQ(SOFTBUS_NOT_FIND, ret);
88
89 ret = GetPendingPacketData(id, TEST_SEQ_SECOND, waitMillis, true, &pendDate);
90 EXPECT_EQ(SOFTBUS_NOT_FIND, ret);
91
92 ret = SetPendingPacketData(id, seq, &pendDate);
93 EXPECT_EQ(SOFTBUS_OK, ret);
94
95 ret = SetPendingPacketData(id, seq, nullptr);
96 EXPECT_EQ(SOFTBUS_OK, ret);
97
98 ret = SetPendingPacketData(TEST_SESSION_ID_SECOND, TEST_SEQ_SECOND, &pendDate);
99 EXPECT_EQ(SOFTBUS_ERR, ret);
100
101 ret = SetPendingPacketData(TEST_SESSION_ID_SECOND, seq, &pendDate);
102 EXPECT_EQ(SOFTBUS_ERR, ret);
103
104 ret = SetPendingPacketData(id, TEST_SEQ_SECOND, &pendDate);
105 EXPECT_EQ(SOFTBUS_ERR, ret);
106
107 ret = GetPendingPacketData(id, seq, waitMillis, false, &pendDate);
108 EXPECT_EQ(SOFTBUS_ALREADY_TRIGGERED, ret);
109
110 ret = GetPendingPacketData(id, TEST_SEQ, waitMillis, true, &pendDate);
111 EXPECT_EQ(SOFTBUS_TIMOUT, ret);
112
113 DeletePendingPacket(id, seq);
114
115 DeletePendingPacket(TEST_SESSION_ID, seq);
116
117 DeletePendingPacket(id, TEST_SEQ);
118
119 DeletePendingPacket(TEST_SESSION_ID_SECOND, TEST_SEQ_SECOND);
120 }
121 } // namespace OHOS nvv