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 #ifndef SECURITY_GUARD_COLLECTOR_MANAGER_MOCK_H
17 #define SECURITY_GUARD_COLLECTOR_MANAGER_MOCK_H
18 
19 #include "gmock/gmock.h"
20 
21 #include "security_event_ruler.h"
22 #include "security_collector_subscribe_info.h"
23 #include "security_event.h"
24 
25 namespace OHOS::Security::SecurityCollector {
26 class ICollectorSubscriber {
27 public:
28     ICollectorSubscriber(const Event &event, int64_t duration = -1, bool isNotify = false) {};
29     virtual ~ICollectorSubscriber() = default;
30     virtual int32_t OnNotify(const Event &event) = 0;
31 };
32 
33 class BaseCollectorManager {
34 public:
35     virtual ~BaseCollectorManager() = default;
36     virtual int32_t Subscribe(const std::shared_ptr<ICollectorSubscriber> &subscriber) = 0;
37     virtual int32_t Unsubscribe(const std::shared_ptr<ICollectorSubscriber> &subscriber) = 0;
38     virtual int32_t QuerySecurityEvent(const std::vector<SecurityEventRuler> rulers,
39         std::vector<SecurityEvent> &events) = 0;
40     virtual int32_t CollectorStart(const SecurityCollector::SecurityCollectorSubscribeInfo &subscriber) = 0;
41     virtual int32_t CollectorStop(const SecurityCollector::SecurityCollectorSubscribeInfo &subscriber) = 0;
42 };
43 
44 class CollectorManager : public BaseCollectorManager {
45 public:
GetInstance()46     static CollectorManager &GetInstance()
47     {
48         static CollectorManager instance;
49         return instance;
50     };
51     CollectorManager() = default;
52     ~CollectorManager() override = default;
53     MOCK_METHOD1(Subscribe, int32_t(const std::shared_ptr<ICollectorSubscriber> &subscriber));
54     MOCK_METHOD1(Unsubscribe, int32_t(const std::shared_ptr<ICollectorSubscriber> &subscriber));
55     MOCK_METHOD2(QuerySecurityEvent, int32_t(const std::vector<SecurityEventRuler> rulers,
56         std::vector<SecurityEvent> &events));
57     MOCK_METHOD1(CollectorStart, int32_t(const SecurityCollector::SecurityCollectorSubscribeInfo &subscriber));
58     MOCK_METHOD1(CollectorStop, int32_t(const SecurityCollector::SecurityCollectorSubscribeInfo &subscriber));
59 };
60 } // OHOS::Security::SecurityCollector
61 
62 #endif // SECURITY_GUARD_COLLECTOR_MANAGER_MOCK_H