1 /*
2 * Copyright (C) 2022-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 "executor_messenger_stub_test.h"
17
18 #include "message_parcel.h"
19
20 #include "iam_ptr.h"
21 #include "mock_executor_messenger_service.h"
22
23 namespace OHOS {
24 namespace UserIam {
25 namespace UserAuth {
26 using namespace testing;
27 using namespace testing::ext;
28
SetUpTestCase()29 void ExecutorMessengerStubTest::SetUpTestCase()
30 {
31 }
32
TearDownTestCase()33 void ExecutorMessengerStubTest::TearDownTestCase()
34 {
35 }
36
SetUp()37 void ExecutorMessengerStubTest::SetUp()
38 {
39 }
40
TearDown()41 void ExecutorMessengerStubTest::TearDown()
42 {
43 }
44
45 HWTEST_F(ExecutorMessengerStubTest, TestSendDataStub_001, TestSize.Level0)
46 {
47 uint64_t scheduleId = 6598;
48 ExecutorRole dstRole = COLLECTOR;
49 std::vector<uint8_t> message = {1, 2, 4, 6};
50
51 MessageParcel data;
52 MessageParcel reply;
53 MessageOption option(MessageOption::TF_SYNC);
54 uint32_t code = ExecutorMessengerInterfaceCode::CO_AUTH_SEND_DATA;
55
56 EXPECT_TRUE(data.WriteInterfaceToken(ExecutorMessengerInterface::GetDescriptor()));
57 EXPECT_TRUE(data.WriteUint64(scheduleId));
58 EXPECT_TRUE(data.WriteInt32(dstRole));
59 EXPECT_TRUE(data.WriteUInt8Vector(message));
60
61 auto service = Common::MakeShared<MockExecutorMessengerService>();
62 EXPECT_NE(service, nullptr);
63 EXPECT_CALL(*service, SendData(_, _, _)).Times(1);
64
65 EXPECT_EQ(service->OnRemoteRequest(code, data, reply, option), SUCCESS);
66 }
67
68 HWTEST_F(ExecutorMessengerStubTest, TestFinishStub_001, TestSize.Level0)
69 {
70 uint64_t scheduleId = 6598;
71 ResultCode resultCode = SUCCESS;
72 std::vector<uint8_t> finalResult = {1, 2, 4, 6};
73
74 MessageParcel data;
75 MessageParcel reply;
76 MessageOption option(MessageOption::TF_SYNC);
77 uint32_t code = ExecutorMessengerInterfaceCode::CO_AUTH_FINISH;
78
79 EXPECT_TRUE(data.WriteInterfaceToken(ExecutorMessengerInterface::GetDescriptor()));
80 EXPECT_TRUE(data.WriteUint64(scheduleId));
81 EXPECT_TRUE(data.WriteInt32(resultCode));
82 EXPECT_TRUE(data.WriteUInt8Vector(finalResult));
83
84 auto service = Common::MakeShared<MockExecutorMessengerService>();
85 EXPECT_NE(service, nullptr);
86 EXPECT_CALL(*service, Finish(_, _, _)).Times(1);
87
88 EXPECT_EQ(service->OnRemoteRequest(code, data, reply, option), SUCCESS);
89 }
90 } // namespace UserAuth
91 } // namespace UserIam
92 } // namespace OHOS
93