@Test public void testRemoveTemporaryTables() throws Exception { FakeDatasetService datasetService = new FakeDatasetService(); String projectId = "project"; String datasetId = "dataset"; datasetService.createDataset(projectId, datasetId, "", "", null); List<TableReference> tableRefs = Lists.newArrayList( BigQueryHelpers.parseTableSpec( String.format("%s:%s.%s", projectId, datasetId, "table1")), BigQueryHelpers.parseTableSpec( String.format("%s:%s.%s", projectId, datasetId, "table2")), BigQueryHelpers.parseTableSpec( String.format("%s:%s.%s", projectId, datasetId, "table3"))); for (TableReference tableRef : tableRefs) { datasetService.createTable(new Table().setTableReference(tableRef)); } // Add one more table to delete that does not actually exist. tableRefs.add( BigQueryHelpers.parseTableSpec(String.format("%s:%s.%s", projectId, datasetId, "table4"))); WriteRename.removeTemporaryTables(datasetService, tableRefs); for (TableReference ref : tableRefs) { loggedWriteRename.verifyDebug("Deleting table " + toJsonString(ref)); checkState(datasetService.getTable(ref) == null, "Table " + ref + " was not deleted!"); } }
private JobStatus runQueryJob(JobConfigurationQuery query) throws IOException, InterruptedException { List<TableRow> rows = FakeBigQueryServices.rowsFromEncodedQuery(query.getQuery()); datasetService.createTable(new Table().setTableReference(query.getDestinationTable())); datasetService.insertAll(query.getDestinationTable(), rows, null); return new JobStatus().setState("DONE"); }
.setDatasetId("dataset-id") .setTableId(String.format("%s_%05d_%05d", jobIdToken, i, j)); fakeDatasetService.createTable(new Table().setTableReference(tempTable));
existingTable = existingTable.setTimePartitioning(load.getTimePartitioning()); datasetService.createTable(existingTable);
source.getProjectId(), source.getDatasetId(), source.getTableId())); datasetService.createTable( new Table() .setTableReference(destination)
fakeDatasetService.createTable( new Table() .setTableReference(table)
new TableFieldSchema().setName("name").setType("STRING"), new TableFieldSchema().setName("number").setType("INTEGER")))); fakeDatasetService.createTable(table);
fakeDatasetService.createTable( new Table() .setTableReference(table)
fakeDatasetService.createDataset( bqOptions.getProject(), tempTableReference.getDatasetId(), "", "", null); fakeDatasetService.createTable( new Table() .setTableReference(tempTableReference)
fakeDatasetService.createTable( new Table() .setTableReference(table)
FakeDatasetService fakeDatasetService = new FakeDatasetService(); fakeDatasetService.createDataset("non-executing-project", "somedataset", "", "", null); fakeDatasetService.createTable(sometable);
.setDatasetId("dataset-id") .setTableId(tableId); fakeDatasetService.createTable( new Table() .setTableReference(tableReference)