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