@Before public void setUp() throws IOException, InterruptedException { FakeDatasetService.setUp(); BigQueryIO.clearCreatedTables(); fakeDatasetService.createDataset("project-id", "dataset-id", "", "", null); }
@Before public void setUp() throws IOException, InterruptedException { FakeDatasetService.setUp(); BigQueryIO.clearCreatedTables(); fakeDatasetService.createDataset("project-id", "dataset-id", "", "", null); }
@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!"); } }
fakeDatasetService.createDataset("project", "data_set", "", "", null); fakeDatasetService.createTable( new Table()
new TableRow().set("name", "e").set("number", 5L), new TableRow().set("name", "f").set("number", 6L)); fakeDatasetService.createDataset( tempTableReference.getProjectId(), tempTableReference.getDatasetId(), "", "", null); Table table =
fakeDatasetService.createDataset("project", "data_set", "", "", null); fakeDatasetService.createTable( new Table()
createTempTableReference( bqOptions.getProject(), createJobIdToken(bqOptions.getJobName(), stepUuid)); fakeDatasetService.createDataset( bqOptions.getProject(), tempTableReference.getDatasetId(), "", "", null); fakeDatasetService.createTable(
fakeDatasetService.createDataset("project", "data_set", "", "", null); fakeDatasetService.createTable( new Table()
sometable.setNumBytes(1024L * 1024L); FakeDatasetService fakeDatasetService = new FakeDatasetService(); fakeDatasetService.createDataset("non-executing-project", "somedataset", "", "", null); fakeDatasetService.createTable(sometable);