Examples with DistributedLogManager org.apache.distributedlog.api.DistributedLogManager used on opensource projects

Example 76 with DistributedLogManager

use of org.apache.distributedlog.api.DistributedLogManager in project incubator-pulsar by apache.

the class DLInputStreamTest method testRead.

/**
 * Test Case: read records from the input stream.
 */
@Test
public void testRead() throws Exception {
    DistributedLogManager dlm = mock(DistributedLogManager.class);
    LogReader reader = mock(LogReader.class);
    when(dlm.getInputStream(any(DLSN.class))).thenReturn(reader);
    byte[] data = "test-read".getBytes(UTF_8);
    LogRecordWithDLSN record = mock(LogRecordWithDLSN.class);
    when(record.getPayLoadInputStream()).thenReturn(new ByteArrayInputStream(data));
    when(reader.readNext(anyBoolean())).thenReturn(record).thenThrow(new EndOfStreamException("eos"));
    DLInputStream in = new DLInputStream(dlm);
    int numReads = 0;
    int readByte;
    while ((readByte = in.read()) != -1) {
        assertEquals(data[numReads], readByte);
        ++numReads;
    }
    assertEquals(data.length, numReads);
}

Example 77 with DistributedLogManager

use of org.apache.distributedlog.api.DistributedLogManager in project incubator-pulsar by apache.

the class DLOutputStreamTest method testClose.

/**
 * Test Case: close output stream.
 */
@Test
public void testClose() throws Exception {
    DistributedLogManager dlm = mock(DistributedLogManager.class);
    AppendOnlyStreamWriter writer = mock(AppendOnlyStreamWriter.class);
    DLOutputStream out = new DLOutputStream(dlm, writer);
    out.close();
    verify(writer, times(1)).markEndOfStream();
    verify(writer, times(1)).close();
    verify(dlm, times(1)).close();
}

Example 78 with DistributedLogManager

use of org.apache.distributedlog.api.DistributedLogManager in project incubator-pulsar by apache.

the class DLOutputStreamTest method testFlush.

/**
 * Test Case: flush should force writing the data.
 */
@Test
public void testFlush() throws Exception {
    DistributedLogManager dlm = mock(DistributedLogManager.class);
    AppendOnlyStreamWriter writer = mock(AppendOnlyStreamWriter.class);
    DLOutputStream out = new DLOutputStream(dlm, writer);
    out.flush();
    verify(writer, times(1)).force(eq(false));
}

Example 79 with DistributedLogManager

use of org.apache.distributedlog.api.DistributedLogManager in project incubator-pulsar by apache.

the class DLOutputStreamTest method testWrite.

/**
 * Test Case: test writing the data.
 */
@Test
public void testWrite() throws Exception {
    DistributedLogManager dlm = mock(DistributedLogManager.class);
    AppendOnlyStreamWriter writer = mock(AppendOnlyStreamWriter.class);
    DLOutputStream out = new DLOutputStream(dlm, writer);
    byte[] data = new byte[16];
    out.write(data);
    verify(writer, times(1)).write(data);
}

Example 80 with DistributedLogManager

use of org.apache.distributedlog.api.DistributedLogManager in project incubator-pulsar by apache.

the class Utils method downloadFromBookkeeper.

public static void downloadFromBookkeeper(Namespace namespace, OutputStream outputStream, String packagePath) throws Exception {
    DistributedLogManager dlm = namespace.openLog(packagePath);
    InputStream in = new DLInputStream(dlm);
    int read = 0;
    byte[] bytes = new byte[1024];
    while ((read = in.read(bytes)) != -1) {
        outputStream.write(bytes, 0, read);
    }
    outputStream.flush();
}