1 /*
2 * Copyright 2018 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 #pragma once
18
19 #include <gmock/gmock.h>
20
21 #include <type_traits>
22
23 #include "avrcp.h"
24 #include "avrcp_internal.h"
25 #include "avrcp_packet.h"
26 #include "stack/include/a2dp_api.h"
27 #include "stack/include/bt_hdr.h"
28 #include "tests/packet_test_helper.h"
29 #include "types/bluetooth/uuid.h"
30 #include "types/raw_address.h"
31
32 namespace bluetooth {
33 namespace avrcp {
34
35 using AvrcpResponse = std::unique_ptr<::bluetooth::PacketBuilder>;
36 using TestAvrcpPacket = TestPacketType<Packet>;
37 using TestBrowsePacket = TestPacketType<BrowsePacket>;
38
39 class MockMediaInterface : public MediaInterface {
40 public:
41 MOCK_METHOD2(SendKeyEvent, void(uint8_t, KeyState));
42 MOCK_METHOD1(GetSongInfo, void(MediaInterface::SongInfoCallback));
43 MOCK_METHOD1(GetPlayStatus, void(MediaInterface::PlayStatusCallback));
44 MOCK_METHOD1(GetNowPlayingList, void(MediaInterface::NowPlayingCallback));
45 MOCK_METHOD1(GetMediaPlayerList, void(MediaInterface::MediaListCallback));
46 MOCK_METHOD1(GetAddressedPlayer, void(MediaInterface::GetAddressedPlayerCallback));
47 MOCK_METHOD3(GetFolderItems, void(uint16_t, std::string, MediaInterface::FolderItemsCallback));
48 MOCK_METHOD3(SetBrowsedPlayer,
49 void(uint16_t, std::string, MediaInterface::SetBrowsedPlayerCallback));
50 MOCK_METHOD2(SetAddressedPlayer, void(uint16_t, MediaInterface::SetAddressedPlayerCallback));
51 MOCK_METHOD3(PlayItem, void(uint16_t, bool, std::string));
52 MOCK_METHOD1(SetActiveDevice, void(const RawAddress&));
53 MOCK_METHOD1(RegisterUpdateCallback, void(MediaCallbacks*));
54 MOCK_METHOD1(UnregisterUpdateCallback, void(MediaCallbacks*));
55 };
56
57 class MockVolumeInterface : public VolumeInterface {
58 public:
59 MOCK_METHOD1(DeviceConnected, void(const RawAddress&));
60 MOCK_METHOD2(DeviceConnected, void(const RawAddress&, VolumeChangedCb));
61 MOCK_METHOD1(DeviceDisconnected, void(const RawAddress&));
62 MOCK_METHOD1(SetVolume, void(int8_t));
63 };
64
65 class MockPlayerSettingsInterface : public PlayerSettingsInterface {
66 public:
67 MOCK_METHOD1(ListPlayerSettings, void(ListPlayerSettingsCallback));
68 MOCK_METHOD2(ListPlayerSettingValues, void(PlayerAttribute, ListPlayerSettingValuesCallback));
69 MOCK_METHOD2(GetCurrentPlayerSettingValue,
70 void(std::vector<PlayerAttribute>, GetCurrentPlayerSettingValueCallback));
71 MOCK_METHOD3(SetPlayerSettings, void(std::vector<PlayerAttribute>, std::vector<uint8_t>,
72 SetPlayerSettingValueCallback));
73 };
74
75 class MockAvrcpInterface : public AvrcpInterface {
76 public:
77 MOCK_METHOD0(GetAvrcpVersion, uint16_t(void));
78 MOCK_METHOD8(AddRecord, uint16_t(uint16_t, const char*, const char*, uint16_t, uint32_t, bool,
79 uint16_t, uint16_t));
80 MOCK_METHOD1(RemoveRecord, uint16_t(uint32_t));
81 MOCK_METHOD4(FindService,
82 uint16_t(uint16_t, const RawAddress&, tAVRC_SDP_DB_PARAMS*, tAVRC_FIND_CBACK));
83 MOCK_METHOD3(Open, uint16_t(uint8_t*, tAVRC_CONN_CB*, const RawAddress&));
84 MOCK_METHOD2(OpenBrowse, uint16_t(uint8_t, tAVCT_ROLE));
85 MOCK_METHOD1(GetPeerMtu, uint16_t(uint8_t));
86 MOCK_METHOD1(GetBrowseMtu, uint16_t(uint8_t));
87 MOCK_METHOD1(Close, uint16_t(uint8_t));
88 MOCK_METHOD1(CloseBrowse, uint16_t(uint8_t));
89 MOCK_METHOD4(MsgReq, uint16_t(uint8_t, uint8_t, uint8_t, BT_HDR*));
90 MOCK_METHOD2(SaveControllerVersion, void(const RawAddress&, uint16_t));
91 };
92
93 class MockA2dpInterface : public A2dpInterface {
94 public:
95 MOCK_METHOD1(event_open, void(const RawAddress&));
96 MOCK_METHOD1(event_close, void(const RawAddress&));
97 MOCK_METHOD0(active_peer, RawAddress());
98 MOCK_METHOD1(is_peer_in_silence_mode, bool(const RawAddress&));
99 MOCK_METHOD2(connect_audio_sink_delayed, void(uint8_t, const RawAddress&));
100 MOCK_METHOD2(find_audio_sink_service, uint16_t(const RawAddress&, tA2DP_FIND_CBACK));
101 };
102
103 class MockSdpInterface : public SdpInterface {
104 public:
105 MOCK_METHOD6(InitDiscoveryDb, bool(tSDP_DISCOVERY_DB*, uint32_t, uint16_t, const bluetooth::Uuid*,
106 uint16_t, uint16_t*));
107 MOCK_METHOD3(ServiceSearchAttributeRequest,
108 bool(const RawAddress&, tSDP_DISCOVERY_DB*, tSDP_DISC_CMPL_CB*));
109 MOCK_METHOD3(FindServiceInDb, tSDP_DISC_REC*(tSDP_DISCOVERY_DB*, uint16_t, t_sdp_disc_rec*));
110 MOCK_METHOD2(FindAttributeInRec, tSDP_DISC_ATTR*(t_sdp_disc_rec*, uint16_t));
111 MOCK_METHOD3(FindProfileVersionInRec, bool(t_sdp_disc_rec*, uint16_t, uint16_t*));
112 };
113
ACTION_TEMPLATE(InvokeCb,HAS_1_TEMPLATE_PARAMS (int,k),AND_1_VALUE_PARAMS (input))114 ACTION_TEMPLATE(InvokeCb, HAS_1_TEMPLATE_PARAMS(int, k), AND_1_VALUE_PARAMS(input)) {
115 ::testing::get<k>(args).Run(input);
116 }
117
ACTION_TEMPLATE(InvokeCb,HAS_1_TEMPLATE_PARAMS (int,k),AND_2_VALUE_PARAMS (a,b))118 ACTION_TEMPLATE(InvokeCb, HAS_1_TEMPLATE_PARAMS(int, k), AND_2_VALUE_PARAMS(a, b)) {
119 ::testing::get<k>(args).Run(a, b);
120 }
121
ACTION_TEMPLATE(InvokeCb,HAS_1_TEMPLATE_PARAMS (int,k),AND_3_VALUE_PARAMS (a,b,c))122 ACTION_TEMPLATE(InvokeCb, HAS_1_TEMPLATE_PARAMS(int, k), AND_3_VALUE_PARAMS(a, b, c)) {
123 ::testing::get<k>(args).Run(a, b, c);
124 }
125
126 template <class PacketType>
127 class PacketMatcher : public ::testing::MatcherInterface<const AvrcpResponse&> {
128 public:
129 AvrcpResponse pkt_to_compare_to_;
130
PacketMatcher(AvrcpResponse && pkt)131 PacketMatcher(AvrcpResponse&& pkt) { pkt_to_compare_to_ = std::move(pkt); }
132
MatchAndExplain(const AvrcpResponse & r,::testing::MatchResultListener * listener)133 bool MatchAndExplain(const AvrcpResponse& r,
134 ::testing::MatchResultListener* listener) const override {
135 auto packet1 = TestPacketType<PacketType>::Make();
136 r->Serialize(packet1);
137
138 auto packet2 = TestPacketType<PacketType>::Make();
139 pkt_to_compare_to_->Serialize(packet2);
140
141 if (packet1->GetData() != packet2->GetData()) {
142 *listener << "\nPacket to compare to: \n";
143 *listener << packet2->ToString();
144 *listener << "\nActual packet: \n";
145 *listener << packet1->ToString();
146 }
147
148 return packet1->GetData() == packet2->GetData();
149 }
150
DescribeTo(::std::ostream * os)151 void DescribeTo(::std::ostream* os) const override { *os << "Packets match"; }
152 };
153
matchPacket(std::unique_ptr<PacketBuilder> && arg)154 inline ::testing::Matcher<const AvrcpResponse&> matchPacket(std::unique_ptr<PacketBuilder>&& arg) {
155 return MakeMatcher(new PacketMatcher<Packet>(std::move(arg)));
156 }
157
matchPacket(std::unique_ptr<BrowsePacketBuilder> && arg)158 inline ::testing::Matcher<const AvrcpResponse&> matchPacket(
159 std::unique_ptr<BrowsePacketBuilder>&& arg) {
160 return MakeMatcher(new PacketMatcher<BrowsePacket>(std::move(arg)));
161 }
162
163 } // namespace avrcp
164 } // namespace bluetooth
165