88import com .clickhouse .client .ClickHouseProtocol ;
99import com .clickhouse .client .ClickHouseRequest ;
1010import com .clickhouse .client .ClickHouseResponse ;
11+ import com .clickhouse .client .ClickHouseServerForTest ;
1112import com .clickhouse .client .api .Client ;
13+ import com .clickhouse .client .api .enums .Protocol ;
14+ import com .clickhouse .client .api .internal .ServerSettings ;
1215import com .clickhouse .client .api .metadata .DefaultColumnToMethodMatchingStrategy ;
1316import com .clickhouse .client .api .metadata .TableSchema ;
1417import com .clickhouse .data .ClickHouseColumn ;
@@ -27,22 +30,17 @@ public class MetadataTests extends BaseIntegrationTest {
2730
2831 @ BeforeMethod (groups = { "integration" })
2932 public void setUp () {
30- ClickHouseNode node = getServer (ClickHouseProtocol .HTTP );
31- client = new Client .Builder ()
32- .addEndpoint (node .getBaseUri () + "default" )
33- .setUsername ("default" )
34- .setPassword ("" )
35- .build ();
33+ client = newClient ().build ();
3634 }
3735
3836 @ Test (groups = { "integration" })
3937 public void testGetTableSchema () {
4038 prepareDataSet ("describe_table" );
4139
42- TableSchema schema = client .getTableSchema ("describe_table" , "default" );
40+ TableSchema schema = client .getTableSchema ("describe_table" , ClickHouseServerForTest . getDatabase () );
4341
4442 Assert .assertEquals (schema .getTableName (), "describe_table" );
45- Assert .assertEquals (schema .getDatabaseName (), "default" );
43+ Assert .assertEquals (schema .getDatabaseName (), ClickHouseServerForTest . getDatabase () );
4644
4745 Assert .assertEquals (schema .getColumns ().size (), 2 );
4846
@@ -54,10 +52,10 @@ public void testGetTableSchema() {
5452 private void prepareDataSet (String tableName ) {
5553
5654 try {
57- String sql = "CREATE TABLE IF NOT EXISTS default. " + tableName + " (param1 UInt32, param2 UInt16) ENGINE = Memory" ;
55+ String sql = "CREATE TABLE IF NOT EXISTS " + tableName + " (param1 UInt32, param2 UInt16) ENGINE = Memory" ;
5856 client .execute (sql ).get ();
5957
60- sql = "INSERT INTO default. " + tableName + " VALUES (1, 2), (3, 4), (5, 6)" ;
58+ sql = "INSERT INTO " + tableName + " VALUES (1, 2), (3, 4), (5, 6)" ;
6159 client .execute (sql ).get ();
6260 } catch (Exception e ) {
6361 Assert .fail ("Failed to prepare data set" , e );
@@ -87,4 +85,16 @@ public Object[][] testMatchingNormalizationData() {
8785
8886 };
8987 }
88+
89+ protected Client .Builder newClient () {
90+ ClickHouseNode node = getServer (ClickHouseProtocol .HTTP );
91+ boolean isSecure = isCloud ();
92+ return new Client .Builder ()
93+ .addEndpoint (Protocol .HTTP , node .getHost (), node .getPort (), isSecure )
94+ .setUsername ("default" )
95+ .setPassword (ClickHouseServerForTest .getPassword ())
96+ .setDefaultDatabase (ClickHouseServerForTest .getDatabase ())
97+ .serverSetting (ServerSettings .WAIT_END_OF_QUERY , "1" )
98+ .useNewImplementation (System .getProperty ("client.tests.useNewImplementation" , "true" ).equals ("true" ));
99+ }
90100}
0 commit comments