Skip to content

Commit ac6e4df

Browse files
committed
Fix error CIs
1 parent fd16677 commit ac6e4df

File tree

3 files changed

+6
-20
lines changed

3 files changed

+6
-20
lines changed

integration-test/src/test/java/org/apache/iotdb/ainode/it/AINodeCallInferenceIT.java

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -115,13 +115,4 @@ public void callInferenceTest(Statement statement, AINodeTestUtils.FakeModelInfo
115115
}
116116
}
117117
}
118-
119-
@Test
120-
public void errorCallInferenceTestInTree() throws SQLException {
121-
try (Connection connection = EnvFactory.getEnv().getConnection(BaseEnv.TREE_SQL_DIALECT);
122-
Statement statement = connection.createStatement()) {
123-
String sql = "CALL INFERENCE(notFound404, \"select s0,s1,s2 from root.AI\")";
124-
errorTest(statement, sql, "1505: model [notFound404] has not been created.");
125-
}
126-
}
127118
}

integration-test/src/test/java/org/apache/iotdb/ainode/it/AINodeForecastIT.java

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -38,32 +38,27 @@
3838
import java.sql.Statement;
3939

4040
import static org.apache.iotdb.ainode.utils.AINodeTestUtils.BUILTIN_MODEL_MAP;
41-
import static org.apache.iotdb.db.it.utils.TestUtils.prepareData;
4241

4342
@RunWith(IoTDBTestRunner.class)
4443
@Category({AIClusterIT.class})
4544
public class AINodeForecastIT {
4645

47-
private static final String[] WRITE_SQL_IN_TABLE =
48-
new String[] {
49-
"CREATE DATABASE db",
50-
"CREATE TABLE db.AI (s0 FLOAT FIELD, s1 DOUBLE FIELD, s2 INT32 FIELD, s3 INT64 FIELD)",
51-
};
52-
5346
private static final String FORECAST_TABLE_FUNCTION_SQL_TEMPLATE =
5447
"SELECT * FROM FORECAST(model_id=>'%s', input=>(SELECT time, s%d FROM db.AI) ORDER BY time)";
5548

5649
@BeforeClass
5750
public static void setUp() throws Exception {
5851
// Init 1C1D1A cluster environment
5952
EnvFactory.getEnv().initClusterEnvironment(1, 1);
60-
prepareData(WRITE_SQL_IN_TABLE);
6153
try (Connection connection = EnvFactory.getEnv().getConnection(BaseEnv.TABLE_SQL_DIALECT);
6254
Statement statement = connection.createStatement()) {
55+
statement.execute("CREATE DATABASE db");
56+
statement.execute(
57+
"CREATE TABLE db.AI (s0 FLOAT FIELD, s1 DOUBLE FIELD, s2 INT32 FIELD, s3 INT64 FIELD)");
6358
for (int i = 0; i < 2880; i++) {
6459
statement.execute(
6560
String.format(
66-
"INSERT INTO root.AI(time,s0,s1,s2,s3) VALUES(%d,%f,%f,%d,%d)",
61+
"INSERT INTO db.AI(time,s0,s1,s2,s3) VALUES(%d,%f,%f,%d,%d)",
6762
i, (float) i, (double) i, i, i));
6863
}
6964
}

integration-test/src/test/java/org/apache/iotdb/ainode/it/AINodeModelManageIT.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -131,15 +131,15 @@ private void userDefinedModelManagementTest(Statement statement)
131131
public void dropBuiltInModelErrorTestInTree() throws SQLException {
132132
try (Connection connection = EnvFactory.getEnv().getConnection(BaseEnv.TREE_SQL_DIALECT);
133133
Statement statement = connection.createStatement()) {
134-
errorTest(statement, "drop model sundial", "1510: Built-in model sundial can't be removed");
134+
errorTest(statement, "drop model sundial", "1510: Cannot delete built-in model: sundial");
135135
}
136136
}
137137

138138
@Test
139139
public void dropBuiltInModelErrorTestInTable() throws SQLException {
140140
try (Connection connection = EnvFactory.getEnv().getConnection(BaseEnv.TABLE_SQL_DIALECT);
141141
Statement statement = connection.createStatement()) {
142-
errorTest(statement, "drop model sundial", "1510: Built-in model sundial can't be removed");
142+
errorTest(statement, "drop model sundial", "1510: Cannot delete built-in model: sundial");
143143
}
144144
}
145145

0 commit comments

Comments
 (0)