1 /*
2  * Copyright 2020 HIMSA II K/S - www.himsa.com.
3  * Represented by EHIMA - www.ehima.com
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  *      http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  */
17 #pragma once
18 
19 #include <gmock/gmock.h>
20 
21 #include "stack/eatt/eatt.h"
22 #include "types/raw_address.h"
23 
24 using bluetooth::eatt::EattChannel;
25 using bluetooth::eatt::EattExtension;
26 
27 class MockEattExtension : public EattExtension {
28 public:
29   MockEattExtension() = default;
30   MockEattExtension(const MockEattExtension&) = delete;
31   MockEattExtension& operator=(const MockEattExtension&) = delete;
32 
33   ~MockEattExtension() override = default;
34 
35   static MockEattExtension* GetInstance();
36 
37   MOCK_METHOD((void), Connect, (const RawAddress& bd_addr));
38   MOCK_METHOD((void), Disconnect, (const RawAddress& bd_addr, uint16_t cid));
39   MOCK_METHOD((void), Reconfigure, (const RawAddress& bd_addr, uint16_t cid, uint16_t mtu));
40   MOCK_METHOD((void), ReconfigureAll, (const RawAddress& bd_addr, uint16_t mtu));
41 
42   MOCK_METHOD((bool), IsEattSupportedByPeer, (const RawAddress& bd_addr));
43   MOCK_METHOD((EattChannel*), FindEattChannelByCid, (const RawAddress& bd_addr, uint16_t cid));
44   MOCK_METHOD((EattChannel*), FindEattChannelByTransId,
45               (const RawAddress& bd_addr, uint32_t trans_id));
46   MOCK_METHOD((bool), IsIndicationPending, (const RawAddress& bd_addr, uint16_t indication_handle));
47   MOCK_METHOD((EattChannel*), GetChannelAvailableForIndication, (const RawAddress& bd_addr));
48   MOCK_METHOD((void), FreeGattResources, (const RawAddress& bd_addr));
49   MOCK_METHOD((bool), IsOutstandingMsgInSendQueue, (const RawAddress& bd_addr));
50   MOCK_METHOD((EattChannel*), GetChannelWithQueuedDataToSend, (const RawAddress& bd_addr));
51   MOCK_METHOD((EattChannel*), GetChannelAvailableForClientRequest, (const RawAddress& bd_addr));
52   MOCK_METHOD((void), StartIndicationConfirmationTimer, (const RawAddress& bd_addr, uint16_t cid));
53   MOCK_METHOD((void), StopIndicationConfirmationTimer, (const RawAddress& bd_addr, uint16_t cid));
54 
55   MOCK_METHOD((void), StartAppIndicationTimer, (const RawAddress& bd_addr, uint16_t cid));
56   MOCK_METHOD((void), StopAppIndicationTimer, (const RawAddress& bd_addr, uint16_t cid));
57 
58   MOCK_METHOD((void), Start, ());
59   MOCK_METHOD((void), Stop, ());
60 };
61