1094332d3Sopenharmony_ci/*
2094332d3Sopenharmony_ci * Copyright (c) 2023 Huawei Device Co., Ltd.
3094332d3Sopenharmony_ci * Licensed under the Apache License, Version 2.0 (the "License");
4094332d3Sopenharmony_ci * you may not use this file except in compliance with the License.
5094332d3Sopenharmony_ci * You may obtain a copy of the License at
6094332d3Sopenharmony_ci *
7094332d3Sopenharmony_ci *     http://www.apache.org/licenses/LICENSE-2.0
8094332d3Sopenharmony_ci *
9094332d3Sopenharmony_ci * Unless required by applicable law or agreed to in writing, software
10094332d3Sopenharmony_ci * distributed under the License is distributed on an "AS IS" BASIS,
11094332d3Sopenharmony_ci * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12094332d3Sopenharmony_ci * See the License for the specific language governing permissions and
13094332d3Sopenharmony_ci * limitations under the License.
14094332d3Sopenharmony_ci */
15094332d3Sopenharmony_ci
16094332d3Sopenharmony_ci#include "hdi_death_test.h"
17094332d3Sopenharmony_ci#include <chrono>
18094332d3Sopenharmony_ci#include <cinttypes>
19094332d3Sopenharmony_ci#include <algorithm>
20094332d3Sopenharmony_ci#include "display_test.h"
21094332d3Sopenharmony_ci#include "display_test_utils.h"
22094332d3Sopenharmony_ci#include "hdi_test_device.h"
23094332d3Sopenharmony_ci#include "hdi_test_device_common.h"
24094332d3Sopenharmony_ci#include "hdi_test_display.h"
25094332d3Sopenharmony_ci
26094332d3Sopenharmony_ciusing namespace OHOS::HDI::Display::Composer::V1_2;
27094332d3Sopenharmony_ciusing namespace OHOS::HDI::Display::TEST;
28094332d3Sopenharmony_ciusing namespace testing::ext;
29094332d3Sopenharmony_ci
30094332d3Sopenharmony_cistatic bool g_isServiceDead = false;
31094332d3Sopenharmony_ci
32094332d3Sopenharmony_civoid ComposerDiedRecipient::OnRemoteDied(const wptr<IRemoteObject>& remote)
33094332d3Sopenharmony_ci{
34094332d3Sopenharmony_ci    if (remote == nullptr) {
35094332d3Sopenharmony_ci        return;
36094332d3Sopenharmony_ci    }
37094332d3Sopenharmony_ci    EXPECT_EQ(g_isServiceDead, true);
38094332d3Sopenharmony_ci}
39094332d3Sopenharmony_ci
40094332d3Sopenharmony_ciHWTEST_F(DeathTest, test_AddDeathRecipient, TestSize.Level1)
41094332d3Sopenharmony_ci{
42094332d3Sopenharmony_ci    displayComposer_ = Composer::V1_2::IDisplayComposerInterface::Get();
43094332d3Sopenharmony_ci    ASSERT_TRUE(displayComposer_ != nullptr);
44094332d3Sopenharmony_ci    sptr<IRemoteObject::DeathRecipient> recipient = new ComposerDiedRecipient();
45094332d3Sopenharmony_ci    ASSERT_TRUE(recipient != nullptr);
46094332d3Sopenharmony_ci    auto ret = displayComposer_->AddDeathRecipient(recipient);
47094332d3Sopenharmony_ci    EXPECT_EQ(ret, true);
48094332d3Sopenharmony_ci    g_isServiceDead = true;
49094332d3Sopenharmony_ci    system("killall composer_host");
50094332d3Sopenharmony_ci}
51094332d3Sopenharmony_ci
52094332d3Sopenharmony_ciHWTEST_F(DeathTest, test_RemoveDeathRecipient, TestSize.Level1)
53094332d3Sopenharmony_ci{
54094332d3Sopenharmony_ci    displayComposer_ = Composer::V1_2::IDisplayComposerInterface::Get();
55094332d3Sopenharmony_ci    ASSERT_TRUE(displayComposer_ != nullptr);
56094332d3Sopenharmony_ci    sptr<IRemoteObject::DeathRecipient> recipient = new ComposerDiedRecipient();
57094332d3Sopenharmony_ci    ASSERT_TRUE(recipient != nullptr);
58094332d3Sopenharmony_ci    auto ret = displayComposer_->AddDeathRecipient(recipient);
59094332d3Sopenharmony_ci    EXPECT_EQ(ret, true);
60094332d3Sopenharmony_ci    ret = displayComposer_->RemoveDeathRecipient();
61094332d3Sopenharmony_ci    EXPECT_EQ(ret, true);
62094332d3Sopenharmony_ci    g_isServiceDead = true;
63094332d3Sopenharmony_ci    system("killall composer_host");
64094332d3Sopenharmony_ci}