1 /*
2  * Copyright (c) 2021-2022 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 <unistd.h>
17 #include "gtest/gtest.h"
18 
19 #include "mock_ability_connect_callback_stub.h"
20 #include "ability_connect_callback_proxy.h"
21 
22 using namespace testing;
23 using namespace testing::ext;
24 using namespace OHOS;
25 using namespace OHOS::AppExecFwk;
26 using OHOS::iface_cast;
27 using OHOS::sptr;
28 using testing::_;
29 using testing::Invoke;
30 using testing::InvokeWithoutArgs;
31 
32 namespace OHOS {
33 namespace AAFwk {
34 class IpcAbilityConnectModuleTest : public testing::Test {
35 public:
36     static void SetUpTestCase();
37     static void TearDownTestCase();
38     void SetUp();
39     void TearDown();
40     static constexpr int COUNT = 5;
41 };
42 
SetUpTestCase()43 void IpcAbilityConnectModuleTest::SetUpTestCase()
44 {}
45 
TearDownTestCase()46 void IpcAbilityConnectModuleTest::TearDownTestCase()
47 {}
48 
SetUp()49 void IpcAbilityConnectModuleTest::SetUp()
50 {}
51 
TearDown()52 void IpcAbilityConnectModuleTest::TearDown()
53 {}
54 
55 /*
56  * Feature: AAFwk
57  * Function: AbilityConnect
58  * SubFunction: IPC of client and server
59  * FunctionPoints: OnAbilityConnectDone
60  * EnvConditions: NA
61  * CaseDescription: verify OnAbilityConnectDone IPC between client and server.
62  */
63 HWTEST_F(IpcAbilityConnectModuleTest, AbilityConnectCallBack_IPC_001, TestSize.Level1)
64 {
65     GTEST_LOG_(INFO) << "IpcAbilityConnectModuleTest AbilityConnectCallBack_IPC_001 start";
66     for (int i = 0; i < COUNT; i++) {
67         sptr<MockAbilityConnectCallbackStub> mockAbilityConnectStub(new MockAbilityConnectCallbackStub());
68         sptr<AbilityConnectionProxy> callback(new AbilityConnectionProxy(mockAbilityConnectStub));
69 
70         AppExecFwk::ElementName element;
71         EXPECT_CALL(*mockAbilityConnectStub, OnAbilityConnectDone(_, _, _))
72             .Times(1)
73             .WillOnce(InvokeWithoutArgs(mockAbilityConnectStub.GetRefPtr(), &MockAbilityConnectCallbackStub::PostVoid));
74         callback->OnAbilityConnectDone(element, mockAbilityConnectStub, 0);
75 
76         mockAbilityConnectStub->Wait();
77     }
78 
79     GTEST_LOG_(INFO) << "IpcAbilityConnectModuleTest AbilityConnectCallBack_IPC_001 end";
80 }
81 
82 /*
83  * Feature: AAFwk
84  * Function: AbilityConnect
85  * SubFunction: IPC of client and server
86  * FunctionPoints: OnAbilityDisconnectDone
87  * EnvConditions: NA
88  * CaseDescription: verify OnAbilityDisconnectDone IPC between client and server.
89  */
90 HWTEST_F(IpcAbilityConnectModuleTest, AbilityConnectCallBack_IPC_002, TestSize.Level1)
91 {
92     GTEST_LOG_(INFO) << "IpcAbilityConnectModuleTest AbilityConnectCallBack_IPC_002 start";
93     for (int i = 0; i < COUNT; i++) {
94         sptr<MockAbilityConnectCallbackStub> mockAbilityConnectStub(new MockAbilityConnectCallbackStub());
95         sptr<AbilityConnectionProxy> callback(new AbilityConnectionProxy(mockAbilityConnectStub));
96 
97         AppExecFwk::ElementName element;
98         EXPECT_CALL(*mockAbilityConnectStub, OnAbilityDisconnectDone(_, _))
99             .Times(1)
100             .WillOnce(InvokeWithoutArgs(mockAbilityConnectStub.GetRefPtr(), &MockAbilityConnectCallbackStub::PostVoid));
101         callback->OnAbilityDisconnectDone(element, 0);
102 
103         mockAbilityConnectStub->Wait();
104     }
105 
106     GTEST_LOG_(INFO) << "IpcAbilityConnectModuleTest AbilityConnectCallBack_IPC_002 end";
107 }
108 }  // namespace AAFwk
109 }  // namespace OHOS
110