add ut

Author: imnutz

src/test/java/com/treasuredata/client/TestTDClient.java

@@ -396,6 +396,58 @@ public void submitJob()
             }
         });
     }
+    @Test
+    public void submitTrinoJob()
+            throws Exception
+    {
+        String jobId = client.submit(TDJobRequest.newTrinoQuery("sample_datasets", "-- td-client-java test\nselect count(*) cnt from nasdaq"));
+        logger.debug("job id: " + jobId);
+
+        int retryCount = 0;
+
+        TDJobSummary tdJob = waitJobCompletion(jobId);
+        TDJob jobInfo = client.jobInfo(jobId);
+        logger.debug("job show result: " + tdJob);
+        logger.debug("job info: " + jobInfo);
+        Optional<String> schema = jobInfo.getResultSchema();
+        assertTrue(schema.isPresent());
+        assertEquals("[[\"cnt\", \"bigint\"]]", schema.get());
+
+        String[] array = client.jobResult(jobId, TDResultFormat.JSON, input -> {
+            try (BufferedReader reader = new BufferedReader(new InputStreamReader(input, StandardCharsets.UTF_8))) {
+                String result = reader.lines().collect(Collectors.joining());
+                logger.info("result:\n" + result);
+                return objectMapper.readValue(result, String[].class);
+            }
+            catch (Exception e) {
+                throw new RuntimeException(e);
+            }
+        });
+        assertEquals(1, array.length);
+        assertEquals(1, jobInfo.getNumRecords());
+        assertTrue(Long.parseLong(array[0]) > 0);
+
+        // test msgpack.gz format
+        client.jobResult(jobId, TDResultFormat.MESSAGE_PACK_GZ, input -> {
+            try {
+                logger.debug("Reading job result in msgpack.gz");
+                MessageUnpacker unpacker = MessagePack.newDefaultUnpacker(new GZIPInputStream(input));
+                int rowCount = 0;
+                while (unpacker.hasNext()) {
+                    ArrayValue array1 = unpacker.unpackValue().asArrayValue();
+                    assertEquals(1, array1.size());
+                    int numRows = array1.get(0).asIntegerValue().toInt();
+                    assertTrue(numRows > 0);
+                    rowCount++;
+                }
+                assertEquals(rowCount, 1);
+                return null;
+            }
+            catch (IOException e) {
+                throw new RuntimeException(e);
+            }
+        });
+    }
 
     @Test
     public void submitJobWithInvalidEngineVersionPresto()
@@ -506,6 +558,17 @@ public void submitJobWithResultOutput()
         client.existsTable(SAMPLE_DB, "sample_output");
     }
 
+    @Test
+    public void submitTrinoJobWithResultOutput()
+            throws Exception
+    {
+        client.deleteTableIfExists(SAMPLE_DB, "sample_output");
+        String resultOutput = String.format("td://%s@/%s/sample_output?mode=replace", client.config.apiKey.get(), SAMPLE_DB);
+        String jobId = client.submit(TDJobRequest.newTrinoQuery("sample_datasets", "-- td-client-java test\nselect count(*) from nasdaq", resultOutput));
+        TDJobSummary tdJob = waitJobCompletion(jobId);
+        client.existsTable(SAMPLE_DB, "sample_output");
+    }
+
     @Test
     public void submitJobWithPoolName()
             throws Exception
@@ -531,7 +594,6 @@ public void submitPrestoJobWithPoolName()
             assertEquals(400, e.getStatusCode());
         }
     }
-
     @Test
     public void submitPrestoJobWithInvalidPoolName()
             throws Exception