1 /*
2  * Copyright (c) 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 <gtest/gtest.h>
17 
18 #include "freeze_util.h"
19 #include "hilog_tag_wrapper.h"
20 #include "ipc_object_stub.h"
21 #include "time_util.h"
22 using namespace testing;
23 using namespace testing::ext;
24 
25 namespace OHOS {
26 namespace AbilityRuntime {
27 class FreezeUtilTest : public testing::Test {
28 public:
29     static void SetUpTestCase();
30     static void TearDownTestCase();
31     void SetUp() override;
32     void TearDown() override;
33 };
34 
SetUpTestCase(void)35 void FreezeUtilTest::SetUpTestCase(void)
36 {}
37 
TearDownTestCase(void)38 void FreezeUtilTest::TearDownTestCase(void)
39 {}
40 
SetUp()41 void FreezeUtilTest::SetUp()
42 {}
43 
TearDown()44 void FreezeUtilTest::TearDown()
45 {}
46 
47 /*
48  * @tc.number    : FreezeUtilTest_001
49  * @tc.name      : FreezeUtilTest
50  * @tc.desc      : Test Function FreezeUtil::GetInstance() and AddLifecycleEvent() and GetLifecycleEvent()
51  */
52 HWTEST_F(FreezeUtilTest, FreezeUtilTest_001, TestSize.Level1)
53 {
54     FreezeUtil::LifecycleFlow flow;
55     EXPECT_EQ(FreezeUtil::GetInstance().GetLifecycleEvent(flow), "");
56     flow.state = FreezeUtil::TimeoutState::FOREGROUND;
57     FreezeUtil::GetInstance().AddLifecycleEvent(flow, "firstEntry");
58     EXPECT_EQ(FreezeUtil::GetInstance().GetLifecycleEvent(flow),
59         TimeUtil::DefaultCurrentTimeStr() + "; " + "firstEntry");
60 
61     FreezeUtil::GetInstance().AddLifecycleEvent(flow, "secondEntry");
62     EXPECT_EQ(FreezeUtil::GetInstance().GetLifecycleEvent(flow), TimeUtil::DefaultCurrentTimeStr() + "; " +
63         "firstEntry\n" + TimeUtil::DefaultCurrentTimeStr() + "; " + "secondEntry");
64     TAG_LOGI(AAFwkTag::TEST, "FreezeUtilTest_001 is end");
65 }
66 
67 /*
68  * @tc.number    : FreezeUtilTest_002
69  * @tc.name      : FreezeUtilTest
70  * @tc.desc      : Test Function FreezeUtil::GetInstance() and DeleteLifecycleEvent() and GetLifecycleEvent()
71  */
72 HWTEST_F(FreezeUtilTest, FreezeUtilTest_002, TestSize.Level1)
73 {
74     FreezeUtil::LifecycleFlow flow;
75     flow.state = FreezeUtil::TimeoutState::LOAD;
76     FreezeUtil::GetInstance().AddLifecycleEvent(flow, "testDeleteEntry");
77     EXPECT_EQ(FreezeUtil::GetInstance().GetLifecycleEvent(flow),
78         TimeUtil::DefaultCurrentTimeStr() + "; " + "testDeleteEntry");
79     FreezeUtil::GetInstance().DeleteLifecycleEvent(flow);
80     EXPECT_EQ(FreezeUtil::GetInstance().GetLifecycleEvent(flow), "");
81     TAG_LOGI(AAFwkTag::TEST, "FreezeUtilTest_002 is end");
82 }
83 
84 /*
85  * @tc.number    : FreezeUtilTest_003
86  * @tc.name      : FreezeUtilTest
87  * @tc.desc      : Test Function DeleteLifecycleEvent() and DeleteLifecycleEventInner()
88  */
89 HWTEST_F(FreezeUtilTest, FreezeUtilTest_003, TestSize.Level1)
90 {
91     sptr<IRemoteObject> token_(new IPCObjectStub());
92     FreezeUtil::LifecycleFlow foregroundFlow = { token_, FreezeUtil::TimeoutState::FOREGROUND };
93     FreezeUtil::GetInstance().AddLifecycleEvent(foregroundFlow, "testDeleteLifecyleEventForground");
94     EXPECT_EQ(FreezeUtil::GetInstance().GetLifecycleEvent(foregroundFlow),
95         TimeUtil::DefaultCurrentTimeStr() + "; " + "testDeleteLifecyleEventForground");
96 
97     FreezeUtil::LifecycleFlow backgroundFlow = { token_, FreezeUtil::TimeoutState::BACKGROUND };
98     FreezeUtil::GetInstance().AddLifecycleEvent(backgroundFlow, "testDeleteLifecyleEventBackground");
99     EXPECT_EQ(FreezeUtil::GetInstance().GetLifecycleEvent(backgroundFlow),
100         TimeUtil::DefaultCurrentTimeStr() + "; " + "testDeleteLifecyleEventBackground");
101 
102     FreezeUtil::GetInstance().DeleteLifecycleEvent(token_);
103     EXPECT_EQ(FreezeUtil::GetInstance().GetLifecycleEvent(foregroundFlow), "");
104     EXPECT_EQ(FreezeUtil::GetInstance().GetLifecycleEvent(backgroundFlow), "");
105     TAG_LOGI(AAFwkTag::TEST, "FreezeUtilTest_003 is end");
106 }
107 }
108 }
109