Method main

testing/postgres-client-tests/java/PostgresTest.java:58–123

(String[] args)

    };

    public static void main(String[] args) {
        Connection conn = null;

        String user = args[0];
        String port = args[1];

        try {
            String url = "jdbc:postgresql://127.0.0.1:" + port + "/postgres";
            String password = "password";

            conn = DriverManager.getConnection(url, user, password);
            Statement st = conn.createStatement(ResultSet.TYPE_SCROLL_INSENSITIVE, ResultSet.CONCUR_READ_ONLY);

            for (int i = 0; i < tests.length; i++) {
                PostgresTest t = tests[i];
                String query    = t.getQuery();
                if ( st.execute(query) ) {
                    String[] expectedResults = t.getExpectedResults();
                    Object fieldAccessor = t.getFieldAccessor();
                    ResultSet rs = st.getResultSet();
                    int rowCount = getRowCount(rs);
                    // Compare the row count to the length of the array
                    if (rowCount != expectedResults.length) {
                        System.out.println("Row count in ResultSet: " + rowCount + " does not match length of the expected results: " + expectedResults.length);
                        System.exit(1);
                    }

                    Integer j = 0;
                    if (rs.next()) {
                        String expected = expectedResults[j];
                        String result = "";
                        if (fieldAccessor instanceof String) {
                            result = rs.getString((String)fieldAccessor);
                        } else if (fieldAccessor instanceof Integer) {
                            result = rs.getString((Integer)fieldAccessor);
                        } else {
                            System.out.println("Unsupported field accessor value: " + fieldAccessor);
                            System.exit(1);
                        }

                        if (!expected.equals(result) && !(query.contains("dolt_commit")) && !(query.contains("dolt_merge"))) {
                            System.out.println("Query: \n" + query);
                            System.out.println("Expected:\n'" + expected + "'");
                            System.out.println("Result:\n'" + result + "'");
                            System.exit(1);
                        }
                        j++;
                    }
                } else {
                    Integer expectedUpdateCount = t.getExpectedUpdateCount();
                    Integer result = st.getUpdateCount();
                    if ( !expectedUpdateCount.equals(result) ) {
                        System.out.println("Query: \n" + query);
                        System.out.println("Expected:\n" + expectedUpdateCount);
                        System.out.println("Rows Updated:\n" + result);
                        System.exit(1);
                    }
                }