Fix UT

apache · Oct 17, 2024 · 86d0232 · 86d0232
1 parent cf32e54
commit 86d0232
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/core/src/test/java/org/apache/gravitino/storage/TestEntityStorage.java b/core/src/test/java/org/apache/gravitino/storage/TestEntityStorage.java
@@ -578,10 +578,10 @@ void testEntityDelete(String type) throws IOException {
           createSchemaEntity(1L, Namespace.of("metalake", "catalog"), "schema1", auditInfo);
       ColumnEntity column1 =
           createColumnEntity(
-              RandomIdGenerator.INSTANCE.nextId(), "column1", Types.StringType.get(), auditInfo);
+              RandomIdGenerator.INSTANCE.nextId(), "column1", 0, Types.StringType.get(), auditInfo);
       ColumnEntity column2 =
           createColumnEntity(
-              RandomIdGenerator.INSTANCE.nextId(), "column2", Types.StringType.get(), auditInfo);
+              RandomIdGenerator.INSTANCE.nextId(), "column2", 1, Types.StringType.get(), auditInfo);
       TableEntity table1 =
           createTableEntityWithColumns(
               1L,
@@ -600,10 +600,10 @@ void testEntityDelete(String type) throws IOException {
           createSchemaEntity(2L, Namespace.of("metalake", "catalog"), "schema2", auditInfo);
       ColumnEntity column3 =
           createColumnEntity(
-              RandomIdGenerator.INSTANCE.nextId(), "column3", Types.StringType.get(), auditInfo);
+              RandomIdGenerator.INSTANCE.nextId(), "column3", 2, Types.StringType.get(), auditInfo);
       ColumnEntity column4 =
           createColumnEntity(
-              RandomIdGenerator.INSTANCE.nextId(), "column4", Types.StringType.get(), auditInfo);
+              RandomIdGenerator.INSTANCE.nextId(), "column4", 3, Types.StringType.get(), auditInfo);
       TableEntity table1InSchema2 =
           createTableEntityWithColumns(
               2L,
@@ -1257,10 +1257,11 @@ public static SchemaEntity createSchemaEntity(
   }
 
   public static ColumnEntity createColumnEntity(
-      Long id, String name, Type dataType, AuditInfo auditInfo) {
+      Long id, String name, int position, Type dataType, AuditInfo auditInfo) {
     return ColumnEntity.builder()
         .withId(id)
         .withName(name)
+        .withPosition(position)
         .withComment("")
         .withDataType(dataType)
         .withNullable(true)