@Test public void testWriteAppend() throws IOException { onTableGet(basicTableSchema()); TableReference ref = BigQueryIO.parseTableSpec("project:dataset.table"); BigQueryTableInserter inserter = new BigQueryTableInserter(mockClient); inserter.getOrCreateTable( ref, BigQueryIO.Write.WriteDisposition.WRITE_APPEND, BigQueryIO.Write.CreateDisposition.CREATE_NEVER, null); verifyTableGet(); }
@Test public void testWriteEmpty() throws IOException { onTableGet(basicTableSchema()); TableDataList dataList = new TableDataList().setTotalRows(0L); onTableList(dataList); TableReference ref = BigQueryIO.parseTableSpec("project:dataset.table"); BigQueryTableInserter inserter = new BigQueryTableInserter(mockClient); inserter.getOrCreateTable( ref, BigQueryIO.Write.WriteDisposition.WRITE_EMPTY, BigQueryIO.Write.CreateDisposition.CREATE_NEVER, null); verifyTableGet(); verifyTabledataList(); }
@Test public void testWriteEmptyFail() throws IOException { thrown.expect(IOException.class); onTableGet(basicTableSchema()); TableDataList dataList = rawDataList(rawRow("Arthur", 42)); onTableList(dataList); TableReference ref = BigQueryIO.parseTableSpec("project:dataset.table"); BigQueryTableInserter inserter = new BigQueryTableInserter(mockClient); try { inserter.getOrCreateTable( ref, BigQueryIO.Write.WriteDisposition.WRITE_EMPTY, BigQueryIO.Write.CreateDisposition.CREATE_NEVER, null); } finally { verifyTableGet(); verifyTabledataList(); } }