@@ -167,6 +167,16 @@ struct CommissionerInitParams : public ControllerInitParams
167
167
Credentials::DeviceAttestationVerifier * deviceAttestationVerifier = nullptr ;
168
168
};
169
169
170
+ // Interface class for DeviceController methods that need to be mocked
171
+ class IDeviceController
172
+ {
173
+ public:
174
+ virtual ~IDeviceController () = default ;
175
+ virtual CHIP_ERROR GetConnectedDevice (NodeId peerNodeId, chip::Callback::Callback<OnDeviceConnected> * onConnection,
176
+ chip::Callback::Callback<OnDeviceConnectionFailure> * onFailure,
177
+ TransportPayloadCapability transportPayloadCapability) = 0;
178
+ };
179
+
170
180
/* *
171
181
* @brief
172
182
* Controller applications can use this class to communicate with already paired CHIP devices. The
@@ -175,7 +185,7 @@ struct CommissionerInitParams : public ControllerInitParams
175
185
* and device pairing information for individual devices). Alternatively, this class can retrieve the
176
186
* relevant information when the application tries to communicate with the device
177
187
*/
178
- class DLL_EXPORT DeviceController : public AbstractDnssdDiscoveryController
188
+ class DLL_EXPORT DeviceController : public AbstractDnssdDiscoveryController, IDeviceController
179
189
{
180
190
public:
181
191
DeviceController ();
@@ -243,9 +253,10 @@ class DLL_EXPORT DeviceController : public AbstractDnssdDiscoveryController
243
253
* An error return from this function means that neither callback has been
244
254
* called yet, and neither callback will be called in the future.
245
255
*/
246
- CHIP_ERROR GetConnectedDevice (NodeId peerNodeId, Callback::Callback<OnDeviceConnected> * onConnection,
247
- chip::Callback::Callback<OnDeviceConnectionFailure> * onFailure,
248
- TransportPayloadCapability transportPayloadCapability = TransportPayloadCapability::kMRPPayload )
256
+ CHIP_ERROR
257
+ GetConnectedDevice (NodeId peerNodeId, Callback::Callback<OnDeviceConnected> * onConnection,
258
+ chip::Callback::Callback<OnDeviceConnectionFailure> * onFailure,
259
+ TransportPayloadCapability transportPayloadCapability = TransportPayloadCapability::kMRPPayload ) override
249
260
{
250
261
VerifyOrReturnError (mState == State::Initialized, CHIP_ERROR_INCORRECT_STATE);
251
262
mSystemState ->CASESessionMgr ()->FindOrEstablishSession (ScopedNodeId (peerNodeId, GetFabricIndex ()), onConnection, onFailure,
0 commit comments