Skip to content

Commit faa7dfc

Browse files
committed
test: move BaseTest4 to testkit module
1 parent dbf2847 commit faa7dfc

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

pgjdbc/src/test/java/org/postgresql/test/jdbc2/BaseTest4.java renamed to testkit/src/main/java/org/postgresql/test/jdbc2/BaseTest4.java

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
import org.postgresql.core.Version;
1616
import org.postgresql.jdbc.PreferQueryMode;
1717
import org.postgresql.test.TestUtil;
18+
import org.postgresql.util.internal.Nullness;
1819

1920
import org.checkerframework.checker.nullness.qual.Nullable;
2021
import org.junit.jupiter.api.AfterEach;
@@ -72,7 +73,7 @@ public final void setBinaryMode(BinaryMode binaryMode) {
7273
this.binaryMode = binaryMode;
7374
}
7475

75-
public StringType getStringType() {
76+
public @Nullable StringType getStringType() {
7677
return stringType;
7778
}
7879

@@ -161,24 +162,28 @@ public void assumeNotSimpleQueryMode() {
161162
* Shorthand for {@code Assume.assumeTrue(TestUtil.haveMinimumServerVersion(conn, version)}.
162163
*/
163164
public void assumeMinimumServerVersion(String message, Version version) throws SQLException {
165+
Connection con = Nullness.castNonNull(this.con, "connection");
164166
assumeTrue(TestUtil.haveMinimumServerVersion(con, version), message);
165167
}
166168

167169
/**
168170
* Shorthand for {@code Assume.assumeTrue(TestUtil.haveMinimumServerVersion(conn, version)}.
169171
*/
170172
public void assumeMinimumServerVersion(Version version) throws SQLException {
173+
Connection con = Nullness.castNonNull(this.con, "connection");
171174
assumeTrue(TestUtil.haveMinimumServerVersion(con, version));
172175
}
173176

174177
protected void assertBinaryForReceive(int oid, boolean expected, Supplier<String> message) throws SQLException {
178+
Connection con = Nullness.castNonNull(this.con, "connection");
175179
assertEquals(
176180
expected,
177181
con.unwrap(BaseConnection.class).getQueryExecutor().useBinaryForReceive(oid),
178182
() -> message.get() + ", useBinaryForReceive(oid=" + oid + ")");
179183
}
180184

181185
protected void assertBinaryForSend(int oid, boolean expected, Supplier<String> message) throws SQLException {
186+
Connection con = Nullness.castNonNull(this.con, "connection");
182187
assertEquals(
183188
expected,
184189
con.unwrap(BaseConnection.class).getQueryExecutor().useBinaryForSend(oid),

0 commit comments

Comments
 (0)