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 #include "mock_bundle_mgr.h"
18 #include "scan_service_ability.h"
19 #include "scan_constant.h"
20 #include "system_ability_definition.h"
21 #include "scan_utils.h"
22 #include "mock_scan_callback_proxy.h"
23
24 namespace OHOS {
25 namespace Scan {
26 class ScanServiceStubTest : public testing::Test {
27 public:
28 static void SetUpTestCase(void);
29 static void TearDownTestCase(void);
30 void SetUp();
31 void TearDown();
32 std::shared_ptr<ScanServiceStub> stubSpr;
33
34 using ScanCmdHandler = bool (ScanServiceStub::*)(MessageParcel &, MessageParcel &);
35 std::map<uint32_t, ScanCmdHandler> cmdMap_;
36 sptr<ScanServiceAbility> scanSa;
37 };
38
SetUpTestCase(void)39 void ScanServiceStubTest::SetUpTestCase(void) {}
40
TearDownTestCase(void)41 void ScanServiceStubTest::TearDownTestCase(void) {}
42
SetUp(void)43 void ScanServiceStubTest::SetUp(void)
44 {
45 stubSpr = std::make_shared<ScanServiceStub>();
46 EXPECT_NE(stubSpr, nullptr);
47 cmdMap_[CMD_INIT_SCAN] = &ScanServiceStub::OnInitScan;
48 cmdMap_[CMD_EXIT_SCAN] = &ScanServiceStub::OnExitScan;
49 cmdMap_[CMD_GET_SCANNER_LIST] = &ScanServiceStub::OnGetScannerList;
50 cmdMap_[CMD_OPEN_SCANNER] = &ScanServiceStub::OnOpenScanner;
51 cmdMap_[CMD_CLOSE_SCANNER] = &ScanServiceStub::OnCloseScanner;
52 cmdMap_[CMD_GET_SCAN_OPTION_DESC] = &ScanServiceStub::OnGetScanOptionDesc;
53 cmdMap_[CMD_OP_SCAN_OPTION_VALUE] = &ScanServiceStub::OnOpScanOptionValue;
54 cmdMap_[CMD_GET_SCAN_PARAMETERS] = &ScanServiceStub::OnGetScanParameters;
55 cmdMap_[CMD_START_SCAN] = &ScanServiceStub::OnStartScan;
56 cmdMap_[CMD_CANCEL_SCAN] = &ScanServiceStub::OnCancelScan;
57 cmdMap_[CMD_GET_SCAN_PROGRESS] = &ScanServiceStub::OnGetScanProgress;
58 cmdMap_[CMD_ON] = &ScanServiceStub::OnEventOn;
59 cmdMap_[CMD_OFF] = &ScanServiceStub::OnEventOff;
60 scanSa = ScanServiceAbility::GetInstance();
61 EXPECT_TRUE(scanSa != nullptr);
62 EXPECT_TRUE(scannerId != nullptr && scannerId != "")
63 }
64
TearDown(void)65 void ScanServiceStubTest::TearDown(void) {}
66
67 /**
68 * @tc.name: ScanMdnsServiceTest_0001
69 * @tc.desc: Verify the capability function.
70 * @tc.type: FUNC
71 * @tc.require:
72 */
73 HWTEST_F(ScanServiceStubTest, VerfifyAppCount_When_CallInitScanOneTime, TestSize.Level1) {
74 auto itFunc = cmdMap_.find(CMD_INIT_SCAN);
75 if (itFunc != cmdMap_.end()) {
76 auto requestFunc = itFunc->second;
77 if (requestFunc != nullptr) {
78 (this->*requestFunc)(data, reply);
79 }
80 }
81 EXPECT_EQ(scanSa->appCount_, 1);
82 }
83
84 /**
85 * @tc.name: ScanMdnsServiceTest_0002
86 * @tc.desc: Verify the capability function.
87 * @tc.type: FUNC
88 * @tc.require:
89 */
90 HWTEST_F(ScanServiceStubTest, VerfifyAppCount_When_CallEXITScanOneTime, TestSize.Level1) {
91 auto itFunc = cmdMap_.find(CMD_EXIT_SCAN);
92 if (itFunc != cmdMap_.end()) {
93 auto requestFunc = itFunc->second;
94 if (requestFunc != nullptr) {
95 (this->*requestFunc)(data, reply);
96 }
97 }
98 EXPECT_EQ(scanSa->appCount_, 0);
99 }
100
101 /**
102 * @tc.name: ScanMdnsServiceTest_0003
103 * @tc.desc: Verify the capability function.
104 * @tc.type: FUNC
105 * @tc.require:
106 */
107 HWTEST_F(ScanServiceStubTest, VerfifyAppCount_When_CallInitAndExitBothOneTime, TestSize.Level1) {
108 auto itFunc = cmdMap_.find(CMD_INIT_SCAN);
109 if (itFunc != cmdMap_.end()) {
110 auto requestFunc = itFunc->second;
111 if (requestFunc != nullptr) {
112 (this->*requestFunc)(data, reply);
113 }
114 }
115 itFunc = cmdMap_.find(CMD_EXIT_SCAN);
116 if (itFunc != cmdMap_.end()) {
117 auto requestFunc = itFunc->second;
118 if (requestFunc != nullptr) {
119 (this->*requestFunc)(data, reply);
120 }
121 }
122 EXPECT_GE(scanSa->appCount_, 0);
123 }
124
125 /**
126 * @tc.name: ScanMdnsServiceTest_0004
127 * @tc.desc: Verify the capability function.
128 * @tc.type: FUNC
129 * @tc.require:
130 */
131 HWTEST_F(ScanServiceStubTest, VerfifyAppCount_When_CallInitFiveTimes, TestSize.Level1) {
132
133 for (int i = 0; i < 5; i++) {
134 auto itFunc = cmdMap_.find(CMD_INIT_SCAN);
135 if (itFunc != cmdMap_.end()) {
136 auto requestFunc = itFunc->second;
137 if (requestFunc != nullptr) {
138 (this->*requestFunc)(data, reply);
139 }
140 }
141 }
142 EXPECT_EQ(scanSa->appCount_, 5);
143 }
144
145 /**
146 * @tc.name: ScanMdnsServiceTest_0005
147 * @tc.desc: Verify the capability function.
148 * @tc.type: FUNC
149 * @tc.require:
150 */
151 HWTEST_F(ScanServiceStubTest, VerfifyAppCount_When_CallExitFiveTimes, TestSize.Level1) {
152
153 for (int i = 0; i < 5; i++) {
154 auto itFunc = cmdMap_.find(CMD_EXIT_SCAN);
155 if (itFunc != cmdMap_.end()) {
156 auto requestFunc = itFunc->second;
157 if (requestFunc != nullptr) {
158 (this->*requestFunc)(data, reply);
159 }
160 }
161 }
162 EXPECT_EQ(scanSa->appCount_, 0);
163 }
164 } // namespace Scan
165 } // namespace OHOS