1 /*
2 * Copyright (c) 2021 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 "heartbeat_ut.h"
16 #include "securec.h"
17 #include "serial_struct.h"
18
19 using namespace testing::ext;
20 namespace Hdc {
SetUpTestCase()21 void HdcHeartbeatTest::SetUpTestCase() {}
TearDownTestCase()22 void HdcHeartbeatTest::TearDownTestCase() {}
SetUp()23 void HdcHeartbeatTest::SetUp() {}
TearDown()24 void HdcHeartbeatTest::TearDown() {}
25
26 HWTEST_F(HdcHeartbeatTest, TestAddHeartbeatCount, TestSize.Level0)
27 {
28 HdcHeartbeat heartbeat;
29 int count = heartbeat.heartbeatCount;
30 heartbeat.AddHeartbeatCount();
31 ASSERT_EQ(heartbeat.heartbeatCount, count + 1);
32 }
33
34 HWTEST_F(HdcHeartbeatTest, TestHandleMessageCount, TestSize.Level0)
35 {
36 HdcHeartbeat heartbeat;
37 int count = heartbeat.messageCount;
38 ASSERT_EQ(heartbeat.HandleMessageCount(), true);
39 ASSERT_EQ(heartbeat.messageCount, count + 1);
40 auto now = std::chrono::steady_clock::now();
41 heartbeat.lastTime = now - std::chrono::hours(2);
42 ASSERT_EQ(heartbeat.HandleMessageCount(), false);
43 }
44
45 HWTEST_F(HdcHeartbeatTest, TestGetHeartbeatCount, TestSize.Level0)
46 {
47 HdcHeartbeat heartbeat;
48 int count = heartbeat.heartbeatCount;
49 ASSERT_EQ(heartbeat.GetHeartbeatCount(), count);
50 }
51
52 HWTEST_F(HdcHeartbeatTest, TestToString, TestSize.Level0)
53 {
54 HdcHeartbeat heartbeat;
55 std::string str = heartbeat.ToString();
56 std::stringstream ss;
57 ss << "heartbeat count is " << heartbeat.heartbeatCount << " and messages count is " << heartbeat.messageCount;
58 ASSERT_EQ(str, ss.str());
59 }
60
61 HWTEST_F(HdcHeartbeatTest, TestHandleRecvHeartbeatMsg, TestSize.Level0)
62 {
63 HdcSessionBase::HeartbeatMsg heartbeatMsg;
64 HdcHeartbeat heartbeat;
65 int count = rand();
66 heartbeatMsg.heartbeatCount = count;
67 std::string s = SerialStruct::SerializeToString(heartbeatMsg);
68 std::string str = heartbeat.HandleRecvHeartbeatMsg(
69 reinterpret_cast<uint8_t *>(const_cast<char *>(s.c_str())), s.size());
70 std::stringstream ss;
71 ss << "heartbeat count is " << count;
72 ASSERT_EQ(str, ss.str());
73
74 std::string strError = heartbeat.HandleRecvHeartbeatMsg(
75 reinterpret_cast<uint8_t *>(const_cast<char *>(s.c_str())), 0);
76 ASSERT_EQ("invalid heartbeat message", strError);
77 }
78
79 HWTEST_F(HdcHeartbeatTest, TestSupportHeartbeat, TestSize.Level0)
80 {
81 HdcHeartbeat heartbeat;
82 bool supportHeartbeat = true;
83 heartbeat.SetSupportHeartbeat(supportHeartbeat);
84 ASSERT_EQ(heartbeat.supportHeartbeat, supportHeartbeat);
85 ASSERT_EQ(heartbeat.GetSupportHeartbeat(), supportHeartbeat);
86 }
87 }