From fc2d3b8882026780b6dcc9ee25216ab93952ff0b Mon Sep 17 00:00:00 2001 From: morningman Date: Mon, 5 Dec 2022 14:46:41 +0800 Subject: [PATCH] add log --- .../doris/planner/StreamLoadPlannerTest.java | 71 ------------------- .../doris/utframe/TestWithFeService.java | 8 ++- 2 files changed, 7 insertions(+), 72 deletions(-) diff --git a/fe/fe-core/src/test/java/org/apache/doris/planner/StreamLoadPlannerTest.java b/fe/fe-core/src/test/java/org/apache/doris/planner/StreamLoadPlannerTest.java index d8ddcfe909a3b34..870dbbeb55721a5 100644 --- a/fe/fe-core/src/test/java/org/apache/doris/planner/StreamLoadPlannerTest.java +++ b/fe/fe-core/src/test/java/org/apache/doris/planner/StreamLoadPlannerTest.java @@ -17,90 +17,19 @@ package org.apache.doris.planner; -import org.apache.doris.analysis.Analyzer; import org.apache.doris.analysis.CompoundPredicate; import org.apache.doris.analysis.ImportColumnsStmt; import org.apache.doris.analysis.ImportWhereStmt; import org.apache.doris.analysis.SqlParser; import org.apache.doris.analysis.SqlScanner; -import org.apache.doris.catalog.Column; -import org.apache.doris.catalog.Database; -import org.apache.doris.catalog.OlapTable; -import org.apache.doris.catalog.Partition; -import org.apache.doris.catalog.PrimitiveType; -import org.apache.doris.common.UserException; import org.apache.doris.common.util.SqlParserUtils; -import org.apache.doris.task.StreamLoadTask; -import org.apache.doris.thrift.TFileFormatType; -import org.apache.doris.thrift.TFileType; -import org.apache.doris.thrift.TStreamLoadPutRequest; -import org.apache.doris.thrift.TUniqueId; -import com.google.common.collect.Lists; -import mockit.Expectations; -import mockit.Injectable; -import mockit.Mocked; import org.junit.Assert; import org.junit.Test; import java.io.StringReader; -import java.util.Arrays; -import java.util.List; public class StreamLoadPlannerTest { - @Injectable - Database db; - - @Injectable - OlapTable destTable; - - @Mocked - StreamLoadScanNode scanNode; - - @Mocked - OlapTableSink sink; - - @Mocked - Partition partition; - - @Test - public void testNormalPlan() throws UserException { - List columns = Lists.newArrayList(); - Column c1 = new Column("c1", PrimitiveType.BIGINT, false); - columns.add(c1); - Column c2 = new Column("c2", PrimitiveType.BIGINT, true); - columns.add(c2); - new Expectations() { - { - destTable.getBaseSchema(); - minTimes = 0; - result = columns; - destTable.getPartitions(); - minTimes = 0; - result = Arrays.asList(partition); - scanNode.init((Analyzer) any); - minTimes = 0; - scanNode.getChildren(); - minTimes = 0; - result = Lists.newArrayList(); - scanNode.getId(); - minTimes = 0; - result = new PlanNodeId(5); - partition.getId(); - minTimes = 0; - result = 0; - } - }; - TStreamLoadPutRequest request = new TStreamLoadPutRequest(); - request.setTxnId(1); - request.setLoadId(new TUniqueId(2, 3)); - request.setFileType(TFileType.FILE_STREAM); - request.setFormatType(TFileFormatType.FORMAT_CSV_PLAIN); - StreamLoadTask streamLoadTask = StreamLoadTask.fromTStreamLoadPutRequest(request); - StreamLoadPlanner planner = new StreamLoadPlanner(db, destTable, streamLoadTask); - planner.plan(streamLoadTask.getId()); - } - @Test public void testParseStmt() throws Exception { String sql = new String("COLUMNS (k1, k2, k3=abc(), k4=default_value())"); diff --git a/fe/fe-core/src/test/java/org/apache/doris/utframe/TestWithFeService.java b/fe/fe-core/src/test/java/org/apache/doris/utframe/TestWithFeService.java index 5ecce69719ca7be..e9dd0b4c96da727 100644 --- a/fe/fe-core/src/test/java/org/apache/doris/utframe/TestWithFeService.java +++ b/fe/fe-core/src/test/java/org/apache/doris/utframe/TestWithFeService.java @@ -95,6 +95,7 @@ import java.nio.file.Path; import java.nio.file.Paths; import java.util.Comparator; +import java.util.ConcurrentModificationException; import java.util.List; import java.util.Map; import java.util.UUID; @@ -494,7 +495,12 @@ protected ShowResultSet showCreateTableByName(String table) throws Exception { } public void createTable(String sql) throws Exception { - createTables(sql); + try { + createTables(sql); + } catch (ConcurrentModificationException e) { + e.printStackTrace(); + throw e; + } } public void dropTable(String table, boolean force) throws Exception {