This is an automated email from the ASF dual-hosted git repository. anton pushed a commit to branch master in repository https://gitbox.apache.org/repos/asf/beam.git
The following commit(s) were added to refs/heads/master by this push: new 4142f55 [BEAM-6468] Allow creating empty TestBoundedTable. new 50d0760 Merge pull request #7568 from amaliujia/rui_wang-fix_testbounded_table 4142f55 is described below commit 4142f5509ca0fce2e57b8b1c3420bad58baa8611 Author: amaliujia <amaliu...@gmail.com> AuthorDate: Fri Jan 18 12:15:20 2019 -0800 [BEAM-6468] Allow creating empty TestBoundedTable. --- .../sql/meta/provider/test/TestBoundedTable.java | 4 +- .../sdk/extensions/sql/TestBoundedTableTest.java | 43 ++++++++++++++++++++++ 2 files changed, 46 insertions(+), 1 deletion(-) diff --git a/sdks/java/extensions/sql/src/main/java/org/apache/beam/sdk/extensions/sql/meta/provider/test/TestBoundedTable.java b/sdks/java/extensions/sql/src/main/java/org/apache/beam/sdk/extensions/sql/meta/provider/test/TestBoundedTable.java index 9a64791..5c92c47 100644 --- a/sdks/java/extensions/sql/src/main/java/org/apache/beam/sdk/extensions/sql/meta/provider/test/TestBoundedTable.java +++ b/sdks/java/extensions/sql/src/main/java/org/apache/beam/sdk/extensions/sql/meta/provider/test/TestBoundedTable.java @@ -93,7 +93,9 @@ public class TestBoundedTable extends TestTable { @Override public PCollection<Row> buildIOReader(PBegin begin) { return begin - .apply("MockedBoundedTable_Reader_" + COUNTER.incrementAndGet(), Create.of(rows)) + .apply( + "MockedBoundedTable_Reader_" + COUNTER.incrementAndGet(), + Create.of(rows).withRowSchema(schema)) .setRowSchema(getSchema()); } diff --git a/sdks/java/extensions/sql/src/test/java/org/apache/beam/sdk/extensions/sql/TestBoundedTableTest.java b/sdks/java/extensions/sql/src/test/java/org/apache/beam/sdk/extensions/sql/TestBoundedTableTest.java new file mode 100644 index 0000000..2657b30 --- /dev/null +++ b/sdks/java/extensions/sql/src/test/java/org/apache/beam/sdk/extensions/sql/TestBoundedTableTest.java @@ -0,0 +1,43 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.beam.sdk.extensions.sql; + +import org.apache.beam.sdk.extensions.sql.meta.provider.test.TestBoundedTable; +import org.apache.beam.sdk.schemas.Schema; +import org.apache.beam.sdk.testing.TestPipeline; +import org.apache.beam.sdk.values.PBegin; +import org.junit.Rule; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.junit.runners.JUnit4; + +/** test TestBoundedTable class. */ +@RunWith(JUnit4.class) +public class TestBoundedTableTest { + + @Rule public final TestPipeline pipeline = TestPipeline.create(); + + @Test + public void testCreatingEmptyTable() { + TestBoundedTable emptyTable = + TestBoundedTable.of( + Schema.builder().addInt32Field("ColId").addStringField("Value").build()); + emptyTable.buildIOReader(PBegin.in(pipeline)); + pipeline.run(); + } +}