embulk 0.8.39-java → 0.10.28-java

data/embulk-core/src/test/java/org/embulk/spi/TestFileOutputRunner.java

@@ -1,221 +0,0 @@
-package org.embulk.spi;
-
-import static org.junit.Assert.assertEquals;
-
-import java.util.ArrayList;
-import java.util.List;
-
-import org.embulk.EmbulkTestRuntime;
-import org.embulk.config.TaskReport;
-import org.embulk.config.ConfigSource;
-import org.embulk.config.ConfigDiff;
-import org.embulk.config.Task;
-import org.embulk.config.TaskSource;
-import org.embulk.spi.time.Timestamp;
-import org.embulk.spi.Schema;
-import org.junit.Rule;
-import org.junit.Test;
-
-import com.google.common.collect.ImmutableList;
-import com.google.common.collect.ImmutableMap;
-import org.msgpack.value.ImmutableMapValue;
-import static org.msgpack.value.ValueFactory.newBoolean;
-import static org.msgpack.value.ValueFactory.newInteger;
-import static org.msgpack.value.ValueFactory.newMap;
-import static org.msgpack.value.ValueFactory.newString;
-
-public class TestFileOutputRunner
-{
-    @Rule
-    public EmbulkTestRuntime runtime = new EmbulkTestRuntime();
-
-    public interface PluginTask extends Task
-    {
-    }
-
-    private static class MockFileOutputPlugin implements FileOutputPlugin
-    {
-        Boolean transactionCompleted = null;
-
-        @Override
-        public ConfigDiff transaction(ConfigSource config, int taskCount,
-                FileOutputPlugin.Control control)
-        {
-            PluginTask task = config.loadConfig(PluginTask.class);
-            control.run(task.dump());
-            return Exec.newConfigDiff();
-        }
-
-        @Override
-        public ConfigDiff resume(TaskSource taskSource,
-                int taskCount,
-                FileOutputPlugin.Control control)
-        {
-            throw new UnsupportedOperationException();
-        }
-
-        @Override
-        public void cleanup(TaskSource taskSource,
-                int taskCount,
-                List<TaskReport> successTaskReports)
-        {
-        }
-
-        @Override
-        public TransactionalFileOutput open(TaskSource taskSource,
-                final int taskIndex)
-        {
-            return new TransactionalFileOutput()
-            {
-
-                @Override
-                public void nextFile()
-                {
-                }
-
-                @Override
-                public void add(Buffer buffer)
-                {
-                }
-
-                @Override
-                public void finish()
-                {
-                }
-
-                @Override
-                public void close()
-                {
-                }
-
-                @Override
-                public void abort()
-                {
-                    transactionCompleted = false;
-                }
-
-                @Override
-                public TaskReport commit()
-                {
-                    transactionCompleted = true;
-                    return Exec.newTaskReport();
-                }
-            };
-        }
-    }
-
-    @Test
-    public void testMockFormatterIteration()
-    {
-        MockFileOutputPlugin fileOutputPlugin = new MockFileOutputPlugin();
-        final FileOutputRunner runner = new FileOutputRunner(fileOutputPlugin);
-
-        ImmutableList<ImmutableMap<String, Object>> columns = ImmutableList.of(
-                ImmutableMap.<String,Object>of("name", "col1", "type", "boolean", "option", ImmutableMap.of()),
-                ImmutableMap.<String,Object>of("name", "col2", "type", "long", "option", ImmutableMap.of()),
-                ImmutableMap.<String,Object>of("name", "col3", "type", "double", "option", ImmutableMap.of()),
-                ImmutableMap.<String,Object>of("name", "col4", "type", "string", "option", ImmutableMap.of()),
-                ImmutableMap.<String,Object>of("name", "col5", "type", "timestamp", "option", ImmutableMap.of()),
-                ImmutableMap.<String,Object>of("name", "col6", "type", "json", "option", ImmutableMap.of()));
-        ConfigSource config = Exec
-                .newConfigSource()
-                .set("type", "unused?")
-                .set("formatter",
-                        ImmutableMap.of("type", "mock", "columns", columns));
-        final Schema schema = config.getNested("formatter")
-                .loadConfig(MockParserPlugin.PluginTask.class)
-                .getSchemaConfig().toSchema();
-
-        runner.transaction(config, schema, 1, new OutputPlugin.Control()
-        {
-            public List<TaskReport> run(final TaskSource outputTask)
-            {
-                TransactionalPageOutput tran = runner.open(outputTask, schema,
-                        1);
-                boolean committed = false;
-                try {
-                    ImmutableMapValue jsonValue = newMap(
-                            newString("_c1"), newBoolean(true),
-                            newString("_c2"), newInteger(10),
-                            newString("_c3"), newString("embulk"),
-                            newString("_c4"), newMap(newString("k"), newString("v"))
-                    );
-                    for (Page page : PageTestUtils.buildPage(
-                            runtime.getBufferAllocator(), schema, true, 2L,
-                            3.0D, "45", Timestamp.ofEpochMilli(678L), jsonValue, true, 2L,
-                            3.0D, "45", Timestamp.ofEpochMilli(678L), jsonValue)) {
-                        tran.add(page);
-                    }
-                    tran.commit();
-                    committed = true;
-                } finally {
-                    if (!committed) {
-                        tran.abort();
-                    }
-                    tran.close();
-                }
-                return new ArrayList<TaskReport>();
-            }
-        });
-
-        assertEquals(true, fileOutputPlugin.transactionCompleted);
-        assertEquals(2, MockFormatterPlugin.records.size());
-        for (List<Object> record : MockFormatterPlugin.records) {
-            assertEquals(Boolean.TRUE, record.get(0));
-            assertEquals(2L, record.get(1));
-            assertEquals(3.0D, (Double) record.get(2), 0.1D);
-            assertEquals("45", record.get(3));
-            assertEquals(678L, ((Timestamp) record.get(4)).toEpochMilli());
-            assertEquals("{\"_c1\":true,\"_c2\":10,\"_c3\":\"embulk\",\"_c4\":{\"k\":\"v\"}}", record.get(5).toString());
-        }
-    }
-
-    @Test
-    public void testTransactionAborted()
-    {
-        MockFileOutputPlugin fileOutputPlugin = new MockFileOutputPlugin();
-        final FileOutputRunner runner = new FileOutputRunner(fileOutputPlugin);
-
-        ImmutableList<ImmutableMap<String, Object>> columns = ImmutableList.of(
-                ImmutableMap.<String,Object>of("name", "col1", "type", "boolean", "option", ImmutableMap.of()),
-                ImmutableMap.<String,Object>of("name", "col2", "type", "long", "option", ImmutableMap.of()),
-                ImmutableMap.<String,Object>of("name", "col3", "type", "double", "option", ImmutableMap.of()),
-                ImmutableMap.<String,Object>of("name", "col4", "type", "string", "option", ImmutableMap.of()),
-                ImmutableMap.<String,Object>of("name", "col5", "type", "timestamp", "option", ImmutableMap.of()),
-                ImmutableMap.<String,Object>of("name", "col6", "type", "json", "option", ImmutableMap.of()));
-        ConfigSource config = Exec
-                .newConfigSource()
-                .set("type", "unused?")
-                .set("formatter",
-                        ImmutableMap.of("type", "mock", "columns", columns));
-        final Schema schema = config.getNested("formatter")
-                .loadConfig(MockParserPlugin.PluginTask.class)
-                .getSchemaConfig().toSchema();
-
-        try {
-            runner.transaction(config, schema, 1, new OutputPlugin.Control()
-            {
-                public List<TaskReport> run(final TaskSource outputTask)
-                {
-                    TransactionalPageOutput tran = runner.open(outputTask,
-                            schema, 1);
-                    boolean committed = false;
-                    try {
-                        tran.add(null);
-                        tran.commit();
-                        committed = true;
-                    } finally {
-                        if (!committed) {
-                            tran.abort();
-                        }
-                        tran.close();
-                    }
-                    return new ArrayList<TaskReport>();
-                }
-            });
-        } catch (NullPointerException npe) {
-        }
-
-        assertEquals(false, fileOutputPlugin.transactionCompleted);
-    }
-}

data/embulk-core/src/test/java/org/embulk/spi/TestInputStreamFileInput.java

@@ -1,188 +0,0 @@
-package org.embulk.spi;
-
-import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.fail;
-import java.io.ByteArrayInputStream;
-import java.io.IOException;
-import java.io.InputStream;
-import org.junit.Rule;
-import org.junit.Test;
-import com.google.common.collect.ImmutableList;
-import org.embulk.EmbulkTestRuntime;
-import org.embulk.spi.util.InputStreamFileInput;
-
-public class TestInputStreamFileInput
-{
-    @Rule
-    public EmbulkTestRuntime runtime = new EmbulkTestRuntime();
-
-    @Test
-    public void testSingleProvider() throws IOException
-    {
-        InputStreamFileInput subject = new InputStreamFileInput(
-                runtime.getBufferAllocator(),
-                provider(new ByteArrayInputStream("abcdef".getBytes("UTF-8"))));
-        assertEquals(true, subject.nextFile());
-        assertEquals("abcdef", bufferToString(subject.poll()));
-        assertEquals(null, subject.poll());
-        subject.close();
-    }
-
-    @Test
-    public void testMultipleProvider() throws IOException
-    {
-        InputStreamFileInput subject = new InputStreamFileInput(
-                runtime.getBufferAllocator(), provider(
-                        new ByteArrayInputStream("abcdef".getBytes("UTF-8")),
-                        new ByteArrayInputStream("ghijkl".getBytes("UTF-8")),
-                        new ByteArrayInputStream("mnopqr".getBytes("UTF-8"))));
-        assertEquals(true, subject.nextFile());
-        assertEquals("abcdef", bufferToString(subject.poll()));
-        assertEquals(true, subject.nextFile());
-        assertEquals("ghijkl", bufferToString(subject.poll()));
-        assertEquals(true, subject.nextFile());
-        assertEquals("mnopqr", bufferToString(subject.poll()));
-        subject.close();
-    }
-
-    @Test
-    public void testEmptyStream() throws IOException
-    {
-        InputStreamFileInput subject = new InputStreamFileInput(
-                runtime.getBufferAllocator(),
-                provider(new ByteArrayInputStream(new byte[0])));
-        assertEquals(true, subject.nextFile());
-        assertEquals(null, subject.poll());
-        subject.close();
-    }
-
-    @Test
-    public void testPollFirstException() throws IOException
-    {
-        InputStreamFileInput subject = new InputStreamFileInput(
-                runtime.getBufferAllocator(),
-                provider(new ByteArrayInputStream("abcdef".getBytes("UTF-8"))));
-        try {
-            subject.poll();
-            fail();
-        } catch (IllegalStateException ile) {
-            // OK
-        }
-        subject.close();
-    }
-
-    @Test
-    public void testEmptyProvider() throws IOException
-    {
-        InputStreamFileInput subject = new InputStreamFileInput(
-                runtime.getBufferAllocator(), provider(new InputStream[0]));
-        assertEquals(false, subject.nextFile());
-        subject.close();
-    }
-
-    @Test
-    public void testProviderOpenNextException()
-    {
-        InputStreamFileInput subject = new InputStreamFileInput(
-                runtime.getBufferAllocator(),
-                new InputStreamFileInput.Provider()
-                {
-                    @Override
-                    public InputStream openNext() throws IOException
-                    {
-                        throw new IOException("emulated exception");
-                    }
-
-                    @Override
-                    public void close() throws IOException
-                    {
-                    }
-                });
-
-        try {
-            subject.nextFile();
-            fail();
-        } catch (RuntimeException re) {
-            // OK
-        }
-        subject.close();
-    }
-
-    @Test
-    public void testProviderCloseException()
-    {
-        InputStreamFileInput subject = new InputStreamFileInput(
-                runtime.getBufferAllocator(),
-                new InputStreamFileInput.Provider()
-                {
-                    @Override
-                    public InputStream openNext() throws IOException
-                    {
-                        return new ByteArrayInputStream(new byte[0]);
-                    }
-
-                    @Override
-                    public void close() throws IOException
-                    {
-                        throw new IOException("emulated exception");
-                    }
-                });
-
-        try {
-            subject.close();
-            fail();
-        } catch (RuntimeException re) {
-            // OK
-        }
-    }
-
-    @Test
-    public void testInputStreamReadException()
-    {
-        InputStreamFileInput subject = new InputStreamFileInput(
-                runtime.getBufferAllocator(),
-                new InputStreamFileInput.Provider()
-                {
-                    @Override
-                    public InputStream openNext() throws IOException
-                    {
-                        return new InputStream()
-                        {
-                            @Override
-                            public int read() throws IOException
-                            {
-                                throw new IOException("emulated exception");
-                            }
-                        };
-                    }
-
-                    @Override
-                    public void close() throws IOException
-                    {
-                    }
-                });
-
-        assertEquals(true, subject.nextFile());
-        try {
-            subject.poll();
-            fail();
-        } catch (RuntimeException re) {
-            // OK
-        }
-        subject.close();
-    }
-
-    private InputStreamFileInput.IteratorProvider provider(
-            InputStream... inputStreams) throws IOException
-    {
-        return new InputStreamFileInput.IteratorProvider(
-                ImmutableList.copyOf(inputStreams));
-    }
-
-    private String bufferToString(Buffer buffer) throws IOException
-    {
-        byte[] buf = new byte[buffer.limit()];
-        buffer.getBytes(0, buf, 0, buffer.limit());
-        return new String(buf, "UTF-8");
-    }
-}

data/embulk-core/src/test/java/org/embulk/spi/TestPageBuilderReader.java

@@ -1,411 +0,0 @@
-package org.embulk.spi;
-
-import static org.embulk.spi.type.Types.BOOLEAN;
-import static org.embulk.spi.type.Types.DOUBLE;
-import static org.embulk.spi.type.Types.JSON;
-import static org.embulk.spi.type.Types.LONG;
-import static org.embulk.spi.type.Types.STRING;
-import static org.embulk.spi.type.Types.TIMESTAMP;
-import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.assertFalse;
-import static org.junit.Assert.assertTrue;
-import org.msgpack.value.ImmutableMapValue;
-import static org.msgpack.value.ValueFactory.newBoolean;
-import static org.msgpack.value.ValueFactory.newInteger;
-import static org.msgpack.value.ValueFactory.newMap;
-import static org.msgpack.value.ValueFactory.newString;
-import java.util.ArrayList;
-import java.util.List;
-import org.embulk.spi.time.Timestamp;
-import org.embulk.spi.Schema;
-import org.embulk.EmbulkTestRuntime;
-import org.msgpack.value.ImmutableMapValue;
-import org.msgpack.value.Value;
-import org.junit.After;
-import org.junit.Before;
-import org.junit.Rule;
-import org.junit.Test;
-
-public class TestPageBuilderReader
-{
-    @Rule
-    public EmbulkTestRuntime runtime = new EmbulkTestRuntime();
-
-    public static class MockPageOutput implements PageOutput
-    {
-        public List<Page> pages;
-
-        public MockPageOutput()
-        {
-            this.pages = new ArrayList<>();
-        }
-
-        @Override
-        public void add(Page page)
-        {
-            pages.add(page);
-        }
-
-        @Override
-        public void finish()
-        {
-        }
-
-        @Override
-        public void close()
-        {
-        }
-    }
-
-    private BufferAllocator bufferAllocator;
-    private PageReader reader;
-    private PageBuilder builder;
-
-    @Before
-    public void setup()
-    {
-        this.bufferAllocator = runtime.getBufferAllocator();
-    }
-
-    @After
-    public void destroy()
-    {
-        if (reader != null) {
-            reader.close();
-            reader = null;
-        }
-        if (builder != null) {
-            builder.close();
-            builder = null;
-        }
-    }
-
-    @Test
-    public void testBoolean()
-    {
-        check(Schema.builder().add("col1", BOOLEAN).build(),
-                false, true, true);
-    }
-
-    @Test
-    public void testLong()
-    {
-        check(Schema.builder().add("col1", LONG).build(),
-                1L, Long.MIN_VALUE, Long.MAX_VALUE);
-    }
-
-    @Test
-    public void testDouble()
-    {
-        check(Schema.builder().add("col1", DOUBLE).build(),
-                8.1, 3.141592, 4.3);
-    }
-
-    @Test
-    public void testUniqueStrings()
-    {
-        check(Schema.builder().add("col1", STRING).build(),
-                "test1", "test2", "test0");
-    }
-
-    @Test
-    public void testDuplicateStrings()
-    {
-        check(Schema.builder().add("col1", STRING).build(),
-            "test1", "test1", "test1");
-    }
-
-    @Test
-    public void testDuplicateStringsMultiColumns()
-    {
-        check(Schema.builder().add("col1", STRING).add("col1", STRING).build(),
-                "test2", "test1",
-                "test1", "test2",
-                "test2", "test0",
-                "test1", "test1");
-    }
-
-    @Test
-    public void testTimestamp()
-    {
-        check(Schema.builder().add("col1", TIMESTAMP).build(),
-                Timestamp.ofEpochMilli(0), Timestamp.ofEpochMilli(10));
-    }
-
-    @Test
-    public void testJson()
-    {
-        check(Schema.builder().add("col1", JSON).build(), getJsonSampleData());
-    }
-
-    @Test
-    public void testNull()
-    {
-        check(Schema.builder()
-                    .add("col3", DOUBLE)
-                    .add("col1", STRING)
-                    .add("col3", LONG)
-                    .add("col3", BOOLEAN)
-                    .add("col2", TIMESTAMP)
-                    .add("col4", JSON)
-                    .build(),
-                null, null, null, null, null, null,
-                null, null, null, null, null, null);
-    }
-
-    @Test
-    public void testMixedTypes()
-    {
-        check(Schema.builder()
-                    .add("col3", DOUBLE)
-                    .add("col1", STRING)
-                    .add("col3", LONG)
-                    .add("col3", BOOLEAN)
-                    .add("col2", TIMESTAMP)
-                    .add("col4", JSON)
-                    .build(),
-                8122.0, "val1", 3L, false, Timestamp.ofEpochMilli(0), getJsonSampleData(),
-                140.15, "val2", Long.MAX_VALUE, true, Timestamp.ofEpochMilli(10), getJsonSampleData());
-    }
-
-    private void check(Schema schema, Object... objects)
-    {
-        Page page = buildPage(schema, objects);
-        checkPage(schema, page, objects);
-    }
-
-    private Page buildPage(Schema schema, final Object... objects)
-    {
-        List<Page> pages = buildPages(schema, objects);
-        assertEquals(1, pages.size());
-        return pages.get(0);
-    }
-
-    private List<Page> buildPages(Schema schema, final Object... objects)
-    {
-        MockPageOutput output = new MockPageOutput();
-        this.builder = new PageBuilder(bufferAllocator, schema, output);
-        int idx = 0;
-        while (idx < objects.length) {
-            for (int column = 0; column < builder.getSchema().getColumnCount(); ++column) {
-                Object value = objects[idx++];
-                if (value == null) {
-                    builder.setNull(column);
-                } else if (value instanceof Boolean) {
-                    builder.setBoolean(column, (Boolean) value);
-                } else if (value instanceof Double) {
-                    builder.setDouble(column, (Double) value);
-                } else if (value instanceof Long) {
-                    builder.setLong(column, (Long) value);
-                } else if (value instanceof String) {
-                    builder.setString(column, (String) value);
-                } else if (value instanceof Timestamp) {
-                    builder.setTimestamp(column, (Timestamp) value);
-                } else if (value instanceof Value) {
-                    builder.setJson(column, (Value) value);
-                } else {
-                    throw new IllegalStateException(
-                            "Unsupported type in test utils: "
-                                    + value.toString());
-                }
-            }
-            builder.addRecord();
-        }
-        builder.flush();
-        builder.close();
-        return output.pages;
-    }
-
-    private void checkPage(Schema schema, Page page, final Object... objects)
-    {
-        this.reader = new PageReader(schema);
-        reader.setPage(page);
-        int idx = 0;
-        while (idx < objects.length && reader.nextRecord()) {
-            for (int column = 0; column < reader.getSchema().getColumnCount(); ++column) {
-                Object value = objects[idx++];
-                if (value == null) {
-                    assertEquals(true, reader.isNull(column));
-                } else if (value instanceof Boolean) {
-                    assertEquals(value, reader.getBoolean(column));
-                } else if (value instanceof Double) {
-                    assertEquals(value, reader.getDouble(column));
-                } else if (value instanceof Long) {
-                    assertEquals(value, reader.getLong(column));
-                } else if (value instanceof String) {
-                    assertEquals(value, reader.getString(column));
-                } else if (value instanceof Timestamp) {
-                    assertEquals(value, reader.getTimestamp(column));
-                } else if (value instanceof Value) {
-                    assertEquals(value, reader.getJson(column));
-                } else {
-                    throw new IllegalStateException(
-                            "Unsupported type in test utils: "
-                                    + value.toString());
-                }
-            }
-        }
-    }
-
-    private ImmutableMapValue getJsonSampleData()
-    {
-        return newMap(
-                newString("_c1"), newBoolean(true),
-                newString("_c2"), newInteger(10),
-                newString("_c3"), newString("embulk"),
-                newString("_c4"), newMap(newString("k"), newString("v"))
-        );
-    }
-
-    @Test
-    public void testEmptySchema()
-    {
-        MockPageOutput output = new MockPageOutput();
-        this.builder = new PageBuilder(bufferAllocator, Schema.builder().build(), output);
-        builder.addRecord();
-        builder.addRecord();
-        builder.flush();
-        builder.close();
-        this.reader = new PageReader(Schema.builder().build());
-        assertEquals(1, output.pages.size());
-        reader.setPage(output.pages.get(0));
-        assertTrue(reader.nextRecord());
-        assertTrue(reader.nextRecord());
-        assertFalse(reader.nextRecord());
-    }
-
-    @Test
-    public void testRenewPage()
-    {
-        this.bufferAllocator = new BufferAllocator()
-        {
-            @Override
-            public Buffer allocate()
-            {
-                return Buffer.allocate(1);
-            }
-
-            @Override
-            public Buffer allocate(int minimumCapacity)
-            {
-                return Buffer.allocate(minimumCapacity);
-            }
-        };
-        assertEquals(
-                9,
-                buildPages(Schema.builder().add("col1", LONG).build(),
-                    0L, 1L, 2L, 3L, 4L, 5L, 6L, 7L, 8L).size());
-    }
-
-    @Test
-    public void testRenewPageWithStrings()
-    {
-        this.bufferAllocator = new BufferAllocator()
-        {
-            @Override
-            public Buffer allocate()
-            {
-                return Buffer.allocate(1);
-            }
-
-            @Override
-            public Buffer allocate(int minimumCapacity)
-            {
-                return Buffer.allocate(minimumCapacity);
-            }
-        };
-        assertEquals(
-                3,
-                buildPages(
-                        Schema.builder()
-                            .add("col1", LONG)
-                            .add("col1", STRING)
-                            .build(),
-                        0L, "record0",
-                        1L, "record1",
-                        3L, "record3"
-                ).size());
-    }
-
-    @Test
-    public void testDoubleWriteStringsToRow()
-    {
-        MockPageOutput output = new MockPageOutput();
-        Schema schema = Schema.builder()
-                .add("col0", STRING)
-                .add("col1", STRING)
-                .add("col2", STRING)
-                .build();
-
-        builder = new PageBuilder(bufferAllocator, schema, output);
-        builder.setString(0, "v0");
-        builder.setString(1, "v1");
-        builder.setNull(2);
-        builder.setString(0, "v2"); // stored to page for col0
-        builder.setNull(1);         // null is stored to page for col1
-        builder.setString(2, "v3"); // stored to page for col2
-        builder.addRecord();
-        builder.finish();
-        builder.close();
-
-        reader = new PageReader(schema);
-        reader.setPage(output.pages.get(0));
-        assertTrue(reader.nextRecord());
-        assertEquals(reader.getString(0), "v2");
-        assertTrue(reader.isNull(1));
-        assertEquals(reader.getString(2), "v3");
-        assertFalse(reader.nextRecord());
-        reader.close();
-    }
-
-    @Test
-    public void testDoubleWriteJsonsToRow()
-    {
-        MockPageOutput output = new MockPageOutput();
-        Schema schema = Schema.builder()
-                .add("col0", JSON)
-                .add("col1", JSON)
-                .add("col2", JSON)
-                .build();
-
-        builder = new PageBuilder(bufferAllocator, schema, output);
-        builder.setJson(0, newString("v0"));
-        builder.setJson(1, newString("v1"));
-        builder.setNull(2);
-        builder.setJson(0, newString("v2")); // store to page for col0
-        builder.setNull(1);                  // null is stored to page for col1
-        builder.setJson(2, newString("v3")); // store to page for col2
-        builder.addRecord();
-        builder.finish();
-        builder.close();
-
-        reader = new PageReader(schema);
-        reader.setPage(output.pages.get(0));
-        assertTrue(reader.nextRecord());
-        assertEquals(reader.getJson(0), newString("v2"));
-        assertTrue(reader.isNull(1));
-        assertEquals(reader.getJson(2), newString("v3"));
-        assertFalse(reader.nextRecord());
-        reader.close();
-    }
-
-    @Test
-    public void testRepeatableClose()
-    {
-        MockPageOutput output = new MockPageOutput();
-        this.builder = new PageBuilder(bufferAllocator,
-                Schema.builder().add("col1", STRING).build(), output);
-        builder.close();
-        builder.close();
-    }
-
-    @Test
-    public void testRepeatableFlush()
-    {
-        MockPageOutput output = new MockPageOutput();
-        this.builder = new PageBuilder(bufferAllocator,
-                Schema.builder().add("col1", STRING).build(), output);
-        builder.flush();
-        builder.flush();
-    }
-}