46
46
import static com .facebook .presto .hive .HiveCommonSessionProperties .NODE_SELECTION_STRATEGY ;
47
47
import static com .facebook .presto .iceberg .IcebergQueryRunner .ICEBERG_CATALOG ;
48
48
import static com .facebook .presto .iceberg .IcebergSessionProperties .PUSHDOWN_FILTER_ENABLED ;
49
- import static com .facebook .presto .iceberg .IcebergSessionProperties .TARGET_SPLIT_SIZE ;
49
+ import static com .facebook .presto .iceberg .IcebergSessionProperties .TARGET_SPLIT_SIZE_BYTES ;
50
50
import static com .facebook .presto .spi .connector .NotPartitionedPartitionHandle .NOT_PARTITIONED ;
51
51
import static org .testng .Assert .assertEquals ;
52
52
import static org .testng .Assert .assertNotNull ;
@@ -178,7 +178,7 @@ private void testGetSplitsByNonIdentityPartitionColumns(String tableName, boolea
178
178
public void testSplitSchedulingWithTablePropertyAndSession ()
179
179
{
180
180
Session session = Session .builder (getSession ())
181
- .setCatalogSessionProperty ("iceberg" , IcebergSessionProperties .TARGET_SPLIT_SIZE , "0" )
181
+ .setCatalogSessionProperty ("iceberg" , IcebergSessionProperties .TARGET_SPLIT_SIZE_BYTES , "0" )
182
182
.build ();
183
183
assertQuerySucceeds ("CREATE TABLE test_split_size as SELECT * FROM UNNEST(sequence(1, 512)) as t(i)" );
184
184
// verify that the session property hasn't propagated into the table
@@ -198,7 +198,7 @@ public void testSplitSchedulingWithTablePropertyAndSession()
198
198
// Set it to 1 with the session property to override the table value and verify we get the
199
199
// same number of splits as when the table value is set to 1.
200
200
Session minSplitSession = Session .builder (session )
201
- .setCatalogSessionProperty ("iceberg" , TARGET_SPLIT_SIZE , "1" )
201
+ .setCatalogSessionProperty ("iceberg" , TARGET_SPLIT_SIZE_BYTES , "1" )
202
202
.build ();
203
203
assertEquals (getSplitsForSql (minSplitSession , selectQuery ).size (), maxSplits );
204
204
assertQuerySucceeds ("DROP TABLE test_split_size" );
@@ -209,7 +209,7 @@ public void testSoftAffinitySchedulingSectionConfig()
209
209
{
210
210
Session maxIdentifiers = Session .builder (getSession ())
211
211
.setCatalogSessionProperty ("iceberg" , AFFINITY_SCHEDULING_FILE_SECTION_SIZE , "1B" )
212
- .setCatalogSessionProperty ("iceberg" , TARGET_SPLIT_SIZE , "1" )
212
+ .setCatalogSessionProperty ("iceberg" , TARGET_SPLIT_SIZE_BYTES , "1" )
213
213
.setCatalogSessionProperty ("iceberg" , NODE_SELECTION_STRATEGY , "SOFT_AFFINITY" )
214
214
.build ();
215
215
assertQuerySucceeds ("CREATE TABLE test_affinity_section_scheduling as SELECT * FROM UNNEST(sequence(1, 512)) as t(i)" );
0 commit comments