1 /*
2  * Copyright (C) 2016 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #ifndef WIFICOND_TEST_MOCK_NETLINK_MANAGER_H_
18 #define WIFICOND_TEST_MOCK_NETLINK_MANAGER_H_
19 
20 #include <vector>
21 
22 #include <gmock/gmock.h>
23 
24 #include "wificond/net/netlink_manager.h"
25 #include "wificond/net/nl80211_packet.h"
26 
27 namespace android {
28 namespace wificond {
29 
30 class MockNetlinkManager : public NetlinkManager {
31  public:
32   MockNetlinkManager();
33   ~MockNetlinkManager() override = default;
34 
35   MOCK_METHOD0(GetSequenceNumber, uint32_t());
36   MOCK_METHOD0(GetFamilyId, uint16_t());
37   MOCK_METHOD0(Start, bool());
38   MOCK_CONST_METHOD0(IsStarted, bool());
39   MOCK_METHOD2(SendMessageAndGetResponses,
40       bool(const NL80211Packet&, std::vector<std::unique_ptr<const NL80211Packet>>*));
41   MOCK_METHOD2(RegisterHandlerAndSendMessage,
42       bool(const NL80211Packet&, std::function<void(std::unique_ptr<const NL80211Packet>)>));
43 };  // class MockNetlinkManager
44 
45 }  // namespace wificond
46 }  // namespace android
47 
48 #endif  // WIFICOND_TEST_MOCK_NETLINK_MANAGER_H
49