1 /*
2  * Copyright (c) 2023 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 #define private public
19 #include "fault_data.h"
20 #include "message_parcel.h"
21 #include "mock_app_scheduler.h"
22 #undef private
23 
24 using namespace testing;
25 using namespace testing::ext;
26 
27 namespace OHOS {
28 namespace AppExecFwk {
29 class AppSchedulerHostTest : public testing::Test {
30 public:
31     static void SetUpTestCase();
32     static void TearDownTestCase();
33     void SetUp() override;
34     void TearDown() override;
35 
36     sptr<MockAppScheduler> mockAppScheduler_;
37 
38     void WriteInterfaceToken(MessageParcel& data);
39 };
40 
SetUpTestCase(void)41 void AppSchedulerHostTest::SetUpTestCase(void)
42 {}
43 
TearDownTestCase(void)44 void AppSchedulerHostTest::TearDownTestCase(void)
45 {}
46 
SetUp()47 void AppSchedulerHostTest::SetUp()
48 {
49     GTEST_LOG_(INFO) << "AppSchedulerHostTest::SetUp()";
50 
51     mockAppScheduler_ = new MockAppScheduler();
52 }
53 
TearDown()54 void AppSchedulerHostTest::TearDown()
55 {}
56 
WriteInterfaceToken(MessageParcel & data)57 void AppSchedulerHostTest::WriteInterfaceToken(MessageParcel& data)
58 {
59     GTEST_LOG_(INFO) << "AppSchedulerHostTest::WriteInterfaceToken()";
60 
61     data.WriteInterfaceToken(AppSchedulerHost::GetDescriptor());
62 }
63 
64 /**
65  * @tc.name: HandleNotifyAppFault_001
66  * @tc.desc: Verify that the HandleNotifyAppFault interface calls normally
67  * @tc.type: FUNC
68  */
69 HWTEST_F(AppSchedulerHostTest, HandleNotifyAppFault_001, TestSize.Level1)
70 {
71     MessageParcel data;
72     MessageParcel reply;
73     MessageOption option;
74     WriteInterfaceToken(data);
75     FaultData faultData;
76     faultData.errorObject.name = "testName";
77     faultData.errorObject.message = "testMessage";
78     faultData.errorObject.stack = "testStack";
79     faultData.faultType = FaultDataType::UNKNOWN;
80     data.WriteParcelable(&faultData);
81     EXPECT_CALL(*mockAppScheduler_, ScheduleNotifyAppFault(_)).Times(1);
82     auto result = mockAppScheduler_->OnRemoteRequest(
83         static_cast<uint32_t>(IAppScheduler::Message::SCHEDULE_NOTIFY_FAULT), data, reply, option);
84     EXPECT_EQ(result, NO_ERROR);
85 }
86 
87 /**
88  * @tc.name: ScheduleChangeAppGcState_001
89  * @tc.desc: Verify that the ScheduleChangeAppGcState interface calls normally
90  * @tc.type: FUNC
91  */
92 HWTEST_F(AppSchedulerHostTest, ScheduleChangeAppGcState_001, TestSize.Level1)
93 {
94     MessageParcel data;
95     MessageParcel reply;
96     MessageOption option;
97     WriteInterfaceToken(data);
98     data.WriteInt32(0);
99     EXPECT_CALL(*mockAppScheduler_, ScheduleChangeAppGcState(_)).Times(1);
100     auto result = mockAppScheduler_->OnRemoteRequest(
101             static_cast<uint32_t>(IAppScheduler::Message::APP_GC_STATE_CHANGE), data, reply, option);
102     EXPECT_EQ(result, NO_ERROR);
103 }
104 
105 /**
106  * @tc.name: HandleAttachAppDebug_001
107  * @tc.desc: Verify that HandleAttachAppDebug interface calls normally.
108  * @tc.type: FUNC
109  */
110 HWTEST_F(AppSchedulerHostTest, HandleAttachAppDebug_001, TestSize.Level1)
111 {
112     EXPECT_NE(mockAppScheduler_, nullptr);
113     EXPECT_CALL(*mockAppScheduler_, AttachAppDebug()).Times(1);
114     MessageParcel data;
115     MessageParcel reply;
116     MessageOption option;
117     WriteInterfaceToken(data);
118 
119     auto result = mockAppScheduler_->OnRemoteRequest(
120         static_cast<uint32_t>(IAppScheduler::Message::SCHEDULE_ATTACH_APP_DEBUG), data, reply, option);
121     EXPECT_EQ(result, NO_ERROR);
122 }
123 
124 /**
125  * @tc.name: HandleAttachAppDebug_002
126  * @tc.desc: Check null descriptor.
127  * @tc.type: FUNC
128  */
129 HWTEST_F(AppSchedulerHostTest, HandleAttachAppDebug_002, TestSize.Level1)
130 {
131     EXPECT_NE(mockAppScheduler_, nullptr);
132     EXPECT_CALL(*mockAppScheduler_, AttachAppDebug()).Times(0);
133     MessageParcel data;
134     MessageParcel reply;
135     MessageOption option;
136 
137     auto result = mockAppScheduler_->OnRemoteRequest(
138         static_cast<uint32_t>(IAppScheduler::Message::SCHEDULE_ATTACH_APP_DEBUG), data, reply, option);
139     EXPECT_EQ(result, ERR_INVALID_STATE);
140 }
141 
142 /**
143  * @tc.name: HandleDetachAppDebug_001
144  * @tc.desc: Verify that HandleDetachAppDebug interface calls normally.
145  * @tc.type: FUNC
146  */
147 HWTEST_F(AppSchedulerHostTest, HandleDetachAppDebug_001, TestSize.Level1)
148 {
149     EXPECT_NE(mockAppScheduler_, nullptr);
150     EXPECT_CALL(*mockAppScheduler_, DetachAppDebug()).Times(1);
151     MessageParcel data;
152     MessageParcel reply;
153     MessageOption option;
154     WriteInterfaceToken(data);
155 
156     auto result = mockAppScheduler_->OnRemoteRequest(
157         static_cast<uint32_t>(IAppScheduler::Message::SCHEDULE_DETACH_APP_DEBUG), data, reply, option);
158     EXPECT_EQ(result, NO_ERROR);
159 }
160 
161 /**
162  * @tc.name: HandleDetachAppDebug_002
163  * @tc.desc: Check unnown message code.
164  * @tc.type: FUNC
165  */
166 HWTEST_F(AppSchedulerHostTest, HandleDetachAppDebug_002, TestSize.Level1)
167 {
168     EXPECT_NE(mockAppScheduler_, nullptr);
169     EXPECT_CALL(*mockAppScheduler_, DetachAppDebug()).Times(0);
170     MessageParcel data;
171     MessageParcel reply;
172     MessageOption option;
173     WriteInterfaceToken(data);
174     uint32_t UNKNOWN_CODE = -1;
175 
176     auto result = mockAppScheduler_->OnRemoteRequest(UNKNOWN_CODE, data, reply, option);
177     EXPECT_EQ(result, IPC_STUB_UNKNOW_TRANS_ERR);
178 }
179 } // namespace AppExecFwk
180 } // namespace OHOS
181