@@ -105,8 +105,7 @@ protected Client.Builder newClient() {
105105 .useHttpCompression (useHttpCompression )
106106 .setDefaultDatabase (ClickHouseServerForTest .getDatabase ())
107107 .serverSetting (ServerSettings .ASYNC_INSERT , "0" )
108- .serverSetting (ServerSettings .WAIT_END_OF_QUERY , "1" )
109- .useNewImplementation (System .getProperty ("client.tests.useNewImplementation" , "true" ).equals ("true" ));
108+ .serverSetting (ServerSettings .WAIT_END_OF_QUERY , "1" );
110109 }
111110
112111 @ AfterMethod (groups = { "integration" })
@@ -162,7 +161,7 @@ public void insertPOJOWithJSON() throws Exception {
162161 client .execute ("DROP TABLE IF EXISTS " + tableName , commandSettings ).get (EXECUTE_CMD_TIMEOUT , TimeUnit .SECONDS );
163162 client .execute (createSQL , commandSettings ).get (EXECUTE_CMD_TIMEOUT , TimeUnit .SECONDS );
164163
165- client .register (PojoWithJSON .class , client .getTableSchema (tableName , "default" ));
164+ client .register (PojoWithJSON .class , client .getTableSchema (tableName ));
166165 PojoWithJSON pojo = new PojoWithJSON ();
167166 pojo .setEventPayload (originalJsonStr );
168167 List <Object > data = Arrays .asList (pojo );
@@ -189,7 +188,7 @@ public void insertPOJOAndReadBack() throws Exception {
189188
190189 initTable (tableName , createSQL );
191190
192- client .register (SamplePOJO .class , client .getTableSchema (tableName , "default" ));
191+ client .register (SamplePOJO .class , client .getTableSchema (tableName ));
193192
194193 System .out .println ("Inserting POJO: " + pojo );
195194 try (InsertResponse response = client .insert (tableName , Collections .singletonList (pojo ), settings ).get (EXECUTE_CMD_TIMEOUT , TimeUnit .SECONDS )) {
@@ -231,7 +230,7 @@ public void testInsertingPOJOWithNullValueForNonNullableColumn() throws Exceptio
231230
232231 initTable (tableName , createSQL );
233232
234- client .register (SamplePOJO .class , client .getTableSchema (tableName , "default" ));
233+ client .register (SamplePOJO .class , client .getTableSchema (tableName ));
235234
236235 try (InsertResponse response = client .insert (tableName , Collections .singletonList (pojo ), settings ).get (30 , TimeUnit .SECONDS )) {
237236 fail ("Should have thrown an exception" );
0 commit comments