test/jdk/java/sql/testng/util/BaseTest.java - master - githubmirror/galahad

test/jdk/java/sql/testng/util/BaseTest.java
195 строк7 KB
Lance Andersen
8369432: Add Support for JDBC 4.5 MR
30 ноя 2025, 15:53
30 ноя 2025, 15:53282f339
/*
 * Copyright (c) 2014, 2025, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */
package util;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.sql.JDBCType;
import java.sql.SQLException;

import org.testng.annotations.DataProvider;

public class BaseTest {

    protected final String reason = "reason";
    protected final String state = "SQLState";
    protected final String cause = "java.lang.Throwable: cause";
    protected final Throwable t = new Throwable("cause");
    protected final Throwable t1 = new Throwable("cause 1");
    protected final Throwable t2 = new Throwable("cause 2");
    protected final int errorCode = 21;
    protected final String[] msgs = {"Exception 1", "cause 1", "Exception 2",
        "Exception 3", "cause 2"};
    private static final String MAX_LENGTH_IDENTIFIER = "a".repeat(128);

    /*
     * Take some form of SQLException, serialize and deserialize it
     */
    @SuppressWarnings("unchecked")
    protected <T extends SQLException> T
            createSerializedException(T ex)
            throws IOException, ClassNotFoundException {
        return (T) serializeDeserializeObject(ex);
    }

    /*
     * Utility method to serialize and deserialize an object
     */
    @SuppressWarnings("unchecked")
    protected <T> T serializeDeserializeObject(T o)
            throws IOException, ClassNotFoundException {
        T o1;
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        try (ObjectOutputStream oos = new ObjectOutputStream(baos)) {
            oos.writeObject(o);
        }
        try (ObjectInputStream ois
                = new ObjectInputStream(new ByteArrayInputStream(baos.toByteArray()))) {
            o1 = (T) ois.readObject();
        }
        return o1;
    }

    /*
     * DataProvider used to specify the value to set and check for
     * methods using boolean values
     */
    @DataProvider(name = "trueFalse")
    protected Object[][] trueFalse() {
        return new Object[][]{
            {true},
            {false}
        };
    }

    /*
     * DataProvider used to specify the standard JDBC Types
     */
    @DataProvider(name = "jdbcTypes")
    protected Object[][] jdbcTypes() {
        Object[][] o = new Object[JDBCType.values().length][1];
        int pos = 0;
        for (JDBCType c : JDBCType.values()) {
            o[pos++][0] = c.getVendorTypeNumber();
        }
        return o;
    }

    /*
     * DataProvider used to provide strings that will be used to validate
     * that enquoteLiteral converts a string to a literal and every instance of
     * a single quote will be converted into two single quotes in the literal.
     */
    @DataProvider(name = "validEnquotedLiteralValues")
    protected Object[][] validEnquotedLiteralValues() {
        return new Object[][]{
                {"Hello", "'Hello'"},
                {"G'Day", "'G''Day'"},
                {"'G''Day'", "'''G''''Day'''"},
                {"I'''M", "'I''''''M'"},
                {"The Dark Knight", "'The Dark Knight'"},
        };
    }

    /*
     * DataProvider used to provide strings that will be used to validate
     * that enqouteIdentifier returns a simple SQL Identifier or a
     * quoted identifier
     */
    @DataProvider(name = "validIdentifierValues")
    protected Object[][] validEnquotedIdentifierValues() {
        return new Object[][]{
                {"b", false, "b"},
                {"b", true, "\"b\""},
                {MAX_LENGTH_IDENTIFIER, false, MAX_LENGTH_IDENTIFIER},
                {MAX_LENGTH_IDENTIFIER, true, "\"" + MAX_LENGTH_IDENTIFIER + "\""},
                {"Hello", false, "Hello"},
                {"Hello", true, "\"Hello\""},
                {"G'Day", false, "\"G'Day\""},
                {"G'Day", true, "\"G'Day\""},
                {"Bruce Wayne", false, "\"Bruce Wayne\""},
                {"Bruce Wayne", true, "\"Bruce Wayne\""},
                {"select", false, "\"select\""},
                {"table", true, "\"table\""},
                {"GoodDay$", false, "\"GoodDay$\""},
                {"GoodDay$", true, "\"GoodDay$\""},};
    }

    /*
     * DataProvider used to provide strings are invalid for enquoteIdentifier
     * resulting in a SQLException being thrown
     */
    @DataProvider(name = "invalidIdentifierValues")
    protected Object[][] invalidEnquotedIdentifierValues() {
        return new Object[][]{
                {"Hel\"lo", false},
                {"\"Hel\"lo\"", true},
                {"Hello" + '\0', false},
                {"", false},
                {MAX_LENGTH_IDENTIFIER + 'a', false},};
    }

    /*
     * DataProvider used to provide strings that will be used to validate
     * that isSimpleIdentifier returns the correct value based on the
     * identifier specified.
     */
    @DataProvider(name = "simpleIdentifierValues")
    protected Object[][] simpleIdentifierValues() {
        return new Object[][]{
                {"b", true},
                {"Hello", true},
                {"\"Gotham\"", false},
                {"G'Day", false},
                {"Bruce Wayne", false},
                {"GoodDay$", false},
                {"Dick_Grayson", true},
                {"Batmobile1966", true},
                {MAX_LENGTH_IDENTIFIER, true},
                {MAX_LENGTH_IDENTIFIER + 'a', false},
                {"", false},
                {"select", false}
            };
    }

    /*
     * DataProvider used to provide strings that will be used to validate
     * that enquoteNCharLiteral converts a string to a National Character
     * literal and every instance of
     * a single quote will be converted into two single quotes in the literal.
     */
    @DataProvider(name = "validEnquotedNCharLiteralValues")
    protected Object[][] validEnquotedNCharLiteralValues() {
        return new Object[][]{
                {"Hello", "N'Hello'"},
                {"G'Day", "N'G''Day'"},
                {"'G''Day'", "N'''G''''Day'''"},
                {"I'''M", "N'I''''''M'"},
                {"N'Hello'", "N'N''Hello'''"},
                {"The Dark Knight", "N'The Dark Knight'"}
        };
    }
}