11package com.woocommerce.android.cardreader.internal.connection
22
33import android.app.Application
4- import com.stripe.stripeterminal.external.callable.Callback
5- import com.stripe.stripeterminal.external.callable.ReaderCallback
64import com.stripe.stripeterminal.external.models.DeviceType
75import com.stripe.stripeterminal.external.models.Reader
86import com.stripe.stripeterminal.external.models.TerminalErrorCode
@@ -60,7 +58,6 @@ class ConnectionManagerTest : CardReaderBaseUnitTest() {
6058 val defaultReaderStatus: StateFlow <CardReaderStatus > = MutableStateFlow (CardReaderStatus .NotConnected ())
6159 whenever(terminalListenerImpl.readerStatus).thenReturn(defaultReaderStatus)
6260
63- // uses the previously created mock objects
6461 connectionManager = ConnectionManager (
6562 terminalWrapper,
6663 bluetoothReaderListener,
@@ -246,9 +243,7 @@ class ConnectionManagerTest : CardReaderBaseUnitTest() {
246243 val cardReader: CardReaderImpl = mock {
247244 on { cardReader }.thenReturn(reader)
248245 }
249- whenever(terminalWrapper.connectToReader(any(), any(), any())).thenAnswer {
250- (it.arguments[2 ] as ReaderCallback ).onSuccess(mock())
251- }
246+ whenever(terminalWrapper.connectToReader(any(), any())).thenReturn(mock())
252247
253248 connectionManager.startConnectionToReader(cardReader, " location_id" )
254249
@@ -270,9 +265,7 @@ class ConnectionManagerTest : CardReaderBaseUnitTest() {
270265 on { errorMessage }.thenReturn(message)
271266 on { this .errorCode }.thenReturn(errorCode)
272267 }
273- whenever(terminalWrapper.connectToReader(any(), any(), any())).thenAnswer {
274- (it.arguments[2 ] as ReaderCallback ).onFailure(exception)
275- }
268+ whenever(terminalWrapper.connectToReader(any(), any())).thenAnswer { throw exception }
276269
277270 connectionManager.startConnectionToReader(cardReader, " location_id" )
278271
@@ -299,9 +292,7 @@ class ConnectionManagerTest : CardReaderBaseUnitTest() {
299292 on { errorMessage }.thenReturn(message)
300293 on { this .errorCode }.thenReturn(errorCode)
301294 }
302- whenever(terminalWrapper.connectToReader(any(), any(), any())).thenAnswer {
303- (it.arguments[2 ] as ReaderCallback ).onFailure(exception)
304- }
295+ whenever(terminalWrapper.connectToReader(any(), any())).thenAnswer { throw exception }
305296
306297 connectionManager.startConnectionToReader(cardReader, " location_id" )
307298
@@ -322,9 +313,7 @@ class ConnectionManagerTest : CardReaderBaseUnitTest() {
322313 val cardReader: CardReaderImpl = mock {
323314 on { cardReader }.thenReturn(reader)
324315 }
325- whenever(terminalWrapper.connectToReader(any(), any(), any())).thenAnswer {
326- (it.arguments[2 ] as ReaderCallback ).onSuccess(cardReader.cardReader)
327- }
316+ whenever(terminalWrapper.connectToReader(any(), any())).thenReturn(reader)
328317
329318 connectionManager.startConnectionToReader(cardReader, " location_id" )
330319
@@ -336,9 +325,7 @@ class ConnectionManagerTest : CardReaderBaseUnitTest() {
336325
337326 @Test
338327 fun `when disconnect succeeds, then status updated with not connected` () = testBlocking {
339- whenever(terminalWrapper.disconnectReader(any())).thenAnswer {
340- (it.arguments[0 ] as Callback ).onSuccess()
341- }
328+ whenever(terminalWrapper.disconnectReader()).thenReturn(Unit )
342329
343330 connectionManager.disconnectReader()
344331
@@ -347,9 +334,7 @@ class ConnectionManagerTest : CardReaderBaseUnitTest() {
347334
348335 @Test
349336 fun `when disconnect succeeds, then true is returned` () = testBlocking {
350- whenever(terminalWrapper.disconnectReader(any())).thenAnswer {
351- (it.arguments[0 ] as Callback ).onSuccess()
352- }
337+ whenever(terminalWrapper.disconnectReader()).thenReturn(Unit )
353338
354339 val result = connectionManager.disconnectReader()
355340
@@ -358,9 +343,7 @@ class ConnectionManagerTest : CardReaderBaseUnitTest() {
358343
359344 @Test
360345 fun `when disconnect fails, then false is returned` () = testBlocking {
361- whenever(terminalWrapper.disconnectReader(any())).thenAnswer {
362- (it.arguments[0 ] as Callback ).onFailure(mock())
363- }
346+ whenever(terminalWrapper.disconnectReader()).thenAnswer { throw mock<TerminalException >() }
364347
365348 val result = connectionManager.disconnectReader()
366349
@@ -369,9 +352,7 @@ class ConnectionManagerTest : CardReaderBaseUnitTest() {
369352
370353 @Test
371354 fun `when disconnect fails, then false with not connected` () = testBlocking {
372- whenever(terminalWrapper.disconnectReader(any())).thenAnswer {
373- (it.arguments[0 ] as Callback ).onFailure(mock())
374- }
355+ whenever(terminalWrapper.disconnectReader()).thenAnswer { throw mock<TerminalException >() }
375356
376357 connectionManager.disconnectReader()
377358
0 commit comments