Examples with ServiceNotAvailableException com.couchbase.client.core.ServiceNotAvailableException used on opensource projects

Example 1 with ServiceNotAvailableException

use of com.couchbase.client.core.ServiceNotAvailableException in project nifi by apache.

the class TestGetCouchbaseKey method testCouchbaseFailure.

@Test
public void testCouchbaseFailure() throws Exception {
    Bucket bucket = mock(Bucket.class);
    String inFileDataStr = "doc-in";
    when(bucket.get(inFileDataStr, RawJsonDocument.class)).thenThrow(new ServiceNotAvailableException());
    setupMockBucket(bucket);
    byte[] inFileData = inFileDataStr.getBytes(StandardCharsets.UTF_8);
    testRunner.enqueue(inFileData);
    try {
        testRunner.run();
        fail("ProcessException should be thrown.");
    } catch (AssertionError e) {
        Assert.assertTrue(e.getCause().getClass().equals(ProcessException.class));
    }
    testRunner.assertTransferCount(REL_SUCCESS, 0);
    testRunner.assertTransferCount(REL_ORIGINAL, 0);
    testRunner.assertTransferCount(REL_RETRY, 0);
    testRunner.assertTransferCount(REL_FAILURE, 0);
}

Example 2 with ServiceNotAvailableException

use of com.couchbase.client.core.ServiceNotAvailableException in project nifi by apache.

the class TestPutCouchbaseKey method testCouchbaseFailure.

@Test
public void testCouchbaseFailure() throws Exception {
    String docId = "doc-a";
    String inFileData = "{\"key\":\"value\"}";
    byte[] inFileDataBytes = inFileData.getBytes(StandardCharsets.UTF_8);
    Bucket bucket = mock(Bucket.class);
    when(bucket.upsert(any(RawJsonDocument.class), eq(PersistTo.NONE), eq(ReplicateTo.ONE))).thenThrow(new ServiceNotAvailableException());
    setupMockBucket(bucket);
    testRunner.enqueue(inFileDataBytes);
    testRunner.setProperty(DOC_ID, docId);
    testRunner.setProperty(PutCouchbaseKey.REPLICATE_TO, ReplicateTo.ONE.toString());
    try {
        testRunner.run();
        fail("ProcessException should be thrown.");
    } catch (AssertionError e) {
        Assert.assertTrue(e.getCause().getClass().equals(ProcessException.class));
    }
    verify(bucket, times(1)).upsert(any(RawJsonDocument.class), eq(PersistTo.NONE), eq(ReplicateTo.ONE));
    testRunner.assertAllFlowFilesTransferred(REL_FAILURE);
    testRunner.assertTransferCount(REL_SUCCESS, 0);
    testRunner.assertTransferCount(REL_RETRY, 0);
    testRunner.assertTransferCount(REL_FAILURE, 0);
}