Skip to content

Commit 3d98862

Browse files
committed
added default checks
1 parent 337cb64 commit 3d98862

File tree

2 files changed

+24
-9
lines changed

2 files changed

+24
-9
lines changed

jdbc-v2/src/test/java/com/clickhouse/jdbc/SQLTests.java

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
import com.fasterxml.jackson.databind.DeserializationFeature;
66
import com.fasterxml.jackson.databind.ObjectMapper;
77
import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
8+
import lombok.AllArgsConstructor;
89
import lombok.Data;
910
import lombok.NoArgsConstructor;
1011
import org.apache.commons.lang3.tuple.Pair;
@@ -16,6 +17,7 @@
1617
import java.sql.Connection;
1718
import java.sql.ResultSet;
1819
import java.sql.Statement;
20+
import java.util.ArrayList;
1921
import java.util.Arrays;
2022
import java.util.HashMap;
2123
import java.util.List;
@@ -41,10 +43,19 @@ public void testSQLQueryWithResultSet(Map<String, TestDataset> tables, SQLTestCa
4143
checkCount += rsMetadataChecks(resultSet, testCase, tables);
4244
checkCount += dataCheck(resultSet, testCase, tables);
4345
Assert.assertEquals(checkCount, testCase.checks.size(), "Check count does not match");
46+
Assert.assertTrue(checkCount > 0, "Test without checks");
4447
}
4548
}
4649
}
4750

51+
public static final ArrayList<TestResultCheckModel> DEFAULT_CHECKS = new ArrayList<>();
52+
static {
53+
DEFAULT_CHECKS.add(new TestResultCheckModel("row_count", "${events.rowCount}"));
54+
DEFAULT_CHECKS.add(new TestResultCheckModel("column_count", "${events.columnsCount}"));
55+
DEFAULT_CHECKS.add(new TestResultCheckModel("column_names", "${events.columnNames}"));
56+
DEFAULT_CHECKS.add(new TestResultCheckModel("column_types", "${events.columnTypes}"));
57+
}
58+
4859
@DataProvider(name = "testSQLQueryWithResultSetDP")
4960
public static Object[][] testSQLQueryWithResultSetDP() throws Exception {
5061
ClassLoader classLoader = SQLTests.class.getClassLoader();
@@ -61,6 +72,9 @@ public static Object[][] testSQLQueryWithResultSetDP() throws Exception {
6172
Object[][] testData = new Object[testCases.length][];
6273
for (int i = 0; i < testCases.length; i++) {
6374
SQLTestCase testCase = testCases[i];
75+
if (testCase.getChecks() == null || testCase.getChecks().isEmpty()) {
76+
testCase.setChecks(DEFAULT_CHECKS);
77+
}
6478
Map<String, TestDataset> testDatasetMap = new HashMap<>();
6579
for (Map.Entry<String, TestDataset> entry : datasetMap.entrySet()) {
6680
String table = testCase.getDatasets().get(entry.getKey());
@@ -203,6 +217,7 @@ public void setTables(Map<String, String> tables) {
203217

204218
@Data
205219
@NoArgsConstructor
220+
@AllArgsConstructor
206221
public static final class TestResultCheckModel {
207222
private String name;
208223
private Object expected;

jdbc-v2/src/test/resources/SQLTests.yaml

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -3,15 +3,15 @@
33
query: SELECT * FROM events
44
tables:
55
events: datasets/events
6-
checks:
7-
- name: row_count
8-
expected: ${events.rowCount}
9-
- name: column_count
10-
expected: ${events.columnsCount}
11-
- name: column_names
12-
expected: ${events.columnNames}
13-
- name: column_types
14-
expected: ${events.columnTypes}
6+
# checks:
7+
# - name: row_count
8+
# expected: ${events.rowCount}
9+
# - name: column_count
10+
# expected: ${events.columnsCount}
11+
# - name: column_names
12+
# expected: ${events.columnNames}
13+
# - name: column_types
14+
# expected: ${events.columnTypes}
1515
- name: cte_select_events_with_date
1616
query: |
1717
WITH

0 commit comments

Comments
 (0)