improved insert async test

Author: chernser

client-v2/src/test/java/com/clickhouse/client/insert/InsertTests.java

@@ -27,6 +27,7 @@
 import com.clickhouse.data.ClickHouseFormat;
 import com.clickhouse.data.ClickHouseVersion;
 import com.clickhouse.data.format.BinaryStreamUtils;
+import lombok.Data;
 import net.jpountz.lz4.LZ4Compressor;
 import net.jpountz.lz4.LZ4Factory;
 import net.jpountz.lz4.LZ4SafeDecompressor;
@@ -270,16 +271,16 @@ public void insertRawData() throws Exception {
         assertEquals(records.size(), 1000);
     }
 
-    @Test(groups = { "integration" }, enabled = true)
-    public void insertRawDataAsync() throws Exception {
+    @Test(groups = { "integration" }, dataProvider = "insertRawDataAsyncProvider", dataProviderClass = InsertTests.class)
+    public void insertRawDataAsync(boolean async) throws Exception {
         final String tableName = "raw_data_table_async";
         final String createSQL = "CREATE TABLE " + tableName +
                 " (Id UInt32, event_ts Timestamp, name String, p1 Int64, p2 String) ENGINE = MergeTree() ORDER BY ()";
 
         initTable(tableName, createSQL);
 
         InsertSettings localSettings = new InsertSettings(settings.getAllSettings());
-        localSettings.setOption(ClientConfigProperties.ASYNC_OPERATIONS.getKey(), true);
+        localSettings.setOption(ClientConfigProperties.ASYNC_OPERATIONS.getKey(), async);
         ByteArrayOutputStream data = new ByteArrayOutputStream();
         PrintWriter writer = new PrintWriter(data);
         for (int i = 0; i < 1000; i++) {
@@ -293,10 +294,20 @@ public void insertRawDataAsync() throws Exception {
 
                 List<GenericRecord> records = client.queryAll("SELECT * FROM " + tableName);
                 assertEquals(records.size(), 1000);
+                assertTrue(Thread.currentThread().getName()
+                        .startsWith(async ? "ForkJoinPool.commonPool" : "main"), "Threads starts with " + Thread.currentThread().getName());
         })
                 .join(); // wait operation complete. only for tests
     }
 
+    @DataProvider
+    public static Object[][] insertRawDataAsyncProvider(){
+        return new Object[][] {
+                {true}, // async
+                {false} // blocking
+        };
+    }
+
     @Test(groups = { "integration" }, dataProvider = "insertRawDataSimpleDataProvider", dataProviderClass = InsertTests.class)
     public void insertRawDataSimple(String tableName) throws Exception {
 //        final String tableName = "raw_data_table";