[webnfc] Handle IllegalStateException during read / write operations

device/nfc/android/junit/src/org/chromium/device/nfc/NFCTest.java

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 package org.chromium.device.nfc;
 
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertNotEquals;
+import static org.junit.Assert.assertNotNull;
 import static org.junit.Assert.assertNull;
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.ArgumentMatchers.anyInt;
 import static org.mockito.ArgumentMatchers.anyString;
 import static org.mockito.ArgumentMatchers.isNull;
 import static org.mockito.Mockito.doNothing;
 import static org.mockito.Mockito.doReturn;
+import static org.mockito.Mockito.doThrow;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.times;
 import static org.mockito.Mockito.verify;
 
 import android.app.Activity;
 import android.content.Context;
 import android.content.pm.PackageManager;
 import android.nfc.FormatException;
 import android.nfc.NdefMessage;
 import android.nfc.NdefRecord;
@@ skipping 546 matching lines @@
     public void testCancelAllWatchesWithNoWathcers() {
         TestNfcImpl nfc = new TestNfcImpl(mContext);
         nfc.setActivityForTesting(mActivity);
         CancelAllWatchesResponse mockCallback = mock(CancelAllWatchesResponse.class);
         nfc.cancelAllWatches(mockCallback);
 
         verify(mockCallback).call(mErrorCaptor.capture());
         assertEquals(NfcErrorType.NOT_FOUND, mErrorCaptor.getValue().errorType);
     }
 
+    /**
+     * Test that when tag is disconnected during read operation, IllegalStateException is handled.
+     */
+    @Test
+    @Feature({"NFCTest"})
+    public void testTagDisconnectedDuringRead() throws IOException, FormatException {
+        TestNfcImpl nfc = new TestNfcImpl(mContext);
+        nfc.setActivityForTesting(mActivity);
+        nfc.setClient(mNfcClient);
+        WatchResponse mockWatchCallback = mock(WatchResponse.class);
+        nfc.watch(createNfcWatchOptions(), mockWatchCallback);
+
+        // Force read operation to fail
+        doThrow(IllegalStateException.class).when(mNfcTagHandler).read();
+
+        // Mocks 'NFC tag found' event.
+        nfc.processPendingOperationsForTesting(mNfcTagHandler);
+
+        // Check that client was not notified.
+        verify(mNfcClient, times(0))
+                .onWatch(mOnWatchCallbackCaptor.capture(), any(NfcMessage.class));
+    }
+
+    /**
+     * Test that when tag is disconnected during write operation, IllegalStateException is handled.
+     */
+    @Test
+    @Feature({"NFCTest"})
+    public void testTagDisconnectedDuringWrite() throws IOException, FormatException {
+        TestNfcImpl nfc = new TestNfcImpl(mContext);
+        nfc.setActivityForTesting(mActivity);
+        PushResponse mockCallback = mock(PushResponse.class);
+
+        // Force write operation to fail
+        doThrow(IllegalStateException.class).when(mNfcTagHandler).write(any(NdefMessage.class));
+        nfc.push(createNfcMessage(), createNfcPushOptions(), mockCallback);
+        nfc.processPendingOperationsForTesting(mNfcTagHandler);
+        verify(mockCallback).call(mErrorCaptor.capture());
+
+        // Test that correct error is returned.
+        assertNotNull(mErrorCaptor.getValue());
+        assertEquals(NfcErrorType.IO_ERROR, mErrorCaptor.getValue().errorType);
+    }
+
     private NfcPushOptions createNfcPushOptions() {
         NfcPushOptions pushOptions = new NfcPushOptions();
         pushOptions.target = NfcPushTarget.ANY;
         pushOptions.timeout = 0;
         pushOptions.ignoreRead = false;
         return pushOptions;
     }
 
     private NfcWatchOptions createNfcWatchOptions() {
         NfcWatchOptions options = new NfcWatchOptions();
@@ skipping 31 matching lines @@
         urlNfcRecord.mediaType = TEXT_MIME;
         urlNfcRecord.data = TEST_URL.getBytes();
         NfcMessage urlNfcMessage = createNfcMessage(TEST_URL, urlNfcRecord);
         try {
             return NfcTypeConverter.toNdefMessage(urlNfcMessage);
         } catch (InvalidNfcMessageException e) {
             return null;
         }
     }
 }