Add unit tests for static creditials

Author: alex268

src/test/java/tech/ydb/jdbc/YdbDriverStaticCredsTest.java

@@ -0,0 +1,97 @@
+package tech.ydb.jdbc;
+
+import java.sql.Connection;
+import java.sql.DriverManager;
+import java.sql.SQLException;
+import java.sql.Statement;
+import java.util.Properties;
+
+import org.junit.jupiter.api.AfterAll;
+import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.RegisterExtension;
+
+import tech.ydb.jdbc.common.QueryType;
+import tech.ydb.jdbc.impl.helper.ExceptionAssert;
+import tech.ydb.jdbc.impl.helper.JdbcUrlHelper;
+import tech.ydb.test.junit5.YdbHelperExtension;
+
+/**
+ *
+ * @author Aleksandr Gorshenin
+ */
+public class YdbDriverStaticCredsTest {
+    @RegisterExtension
+    private static final YdbHelperExtension ydb = new YdbHelperExtension();
+
+    private static final JdbcUrlHelper jdbcURL = new JdbcUrlHelper(ydb);
+
+    @BeforeAll
+    public static void createUsers() throws SQLException {
+        try (Connection connection = DriverManager.getConnection(jdbcURL.build())) {
+            try (Statement statement = connection.createStatement()) {
+                statement.execute(QueryType.SCHEME_QUERY.getPrefix() + "\n"
+                        + "CREATE USER user1 PASSWORD NULL;"
+                        + "CREATE USER user2 PASSWORD 'pwss';"
+                );
+            }
+        }
+    }
+
+    @AfterAll
+    public static void dropUsers() throws SQLException {
+        try (Connection connection = DriverManager.getConnection(jdbcURL.build())) {
+            try (Statement statement = connection.createStatement()) {
+                statement.execute(QueryType.SCHEME_QUERY.getPrefix() + "\n"
+                        + "DROP USER IF EXISTS user1, user2;"
+                );
+            }
+        }
+    }
+
+    private ConnectionSupplier connectByProperties(String username, String password) {
+        final Properties props = new Properties();
+        props.put("user", username);
+        if (password != null) {
+            props.put("password", password);
+        }
+        return () -> DriverManager.getConnection(jdbcURL.disableToken().build(), props);
+    }
+
+    private ConnectionSupplier connectByAuthority(String username, String password) {
+        return () -> DriverManager.getConnection(jdbcURL.disableToken().withAutority(username, password).build());
+    }
+
+    private void testConnection(ConnectionSupplier connectionSupplier) throws SQLException {
+        try (Connection connection = connectionSupplier.get()) {
+            try (Statement statement = connection.createStatement()) {
+                statement.execute("SELECT 1;");
+            }
+        }
+    }
+
+    @Test
+    public void connectOK() throws SQLException {
+        testConnection(connectByProperties("user1", "pwss"));
+        testConnection(connectByAuthority("user1", "pwss"));
+
+        testConnection(connectByProperties("user2", ""));
+        testConnection(connectByAuthority("user2", ""));
+
+        testConnection(connectByProperties("user2", null));
+        testConnection(connectByAuthority("user2", null));
+    }
+
+    @Test
+    public void connectWring() throws SQLException {
+        ExceptionAssert.ydbConfiguration("can't connect", () -> testConnection(connectByProperties("user1", "")));
+        ExceptionAssert.ydbConfiguration("can't connect", () -> testConnection(connectByProperties("user1", null)));
+        ExceptionAssert.ydbConfiguration("can't connect", () -> testConnection(connectByProperties("user1", "pass")));
+
+        ExceptionAssert.ydbConfiguration("can't connect", () -> testConnection(connectByProperties("user2", "a")));
+    }
+
+    interface ConnectionSupplier {
+        Connection get() throws SQLException;
+    }
+}

src/test/java/tech/ydb/jdbc/impl/helper/JdbcConnectionExtention.java

@@ -23,14 +23,12 @@
 public class JdbcConnectionExtention implements
         BeforeEachCallback, BeforeAllCallback, AfterEachCallback, AfterAllCallback {
 
-    private final YdbHelperExtension ydb;
     private final JdbcUrlHelper jdbcURL;
 
     private final Map<ExtensionContext, Connection> map = new HashMap<>();
     private final Stack<Connection> stack = new Stack<>();
 
     public JdbcConnectionExtention(YdbHelperExtension ydb, boolean autoCommit) {
-        this.ydb = ydb;
         this.jdbcURL = new JdbcUrlHelper(ydb)
                 .withArg("failOnTruncatedResult", "true")
                 .withArg("autoCommit", String.valueOf(autoCommit));

src/test/java/tech/ydb/jdbc/impl/helper/JdbcUrlHelper.java

@@ -10,29 +10,51 @@
 public class JdbcUrlHelper {
     private final YdbHelperExtension ydb;
     private final String extra;
+    private final String authority;
+    private final boolean disableToken;
 
     public JdbcUrlHelper(YdbHelperExtension ydb) {
+        this(ydb, "", "", false);
+    }
+
+    private JdbcUrlHelper(YdbHelperExtension ydb, String extra, String authority, boolean disableToken) {
         this.ydb = ydb;
-        this.extra = "";
+        this.extra = extra;
+        this.authority = authority;
+        this.disableToken = disableToken;
     }
 
-    private JdbcUrlHelper(JdbcUrlHelper other, String extra) {
-        this.ydb = other.ydb;
-        this.extra = other.extra.isEmpty() ? extra : other.extra +  "&" + extra;
+    public JdbcUrlHelper disableToken() {
+        return new JdbcUrlHelper(ydb, extra, authority, true);
     }
 
     public JdbcUrlHelper withArg(String arg, String value) {
-        return new JdbcUrlHelper(this, arg + "=" + value);
+        String newExtra = new StringBuilder(extra)
+                .append(extra.isEmpty() ? "" : "&")
+                .append(arg)
+                .append("=")
+                .append(value)
+                .toString();
+        return new JdbcUrlHelper(ydb, newExtra, authority, disableToken);
+    }
+
+    public JdbcUrlHelper withAutority(String username, String password) {
+        StringBuilder newAuthority = new StringBuilder(username);
+        if (password != null && !password.isEmpty()) {
+            newAuthority = newAuthority.append(":").append(password);
+        }
+        return new JdbcUrlHelper(ydb, extra, newAuthority.append("@").toString(), disableToken);
     }
 
     public String build() {
         StringBuilder jdbc = new StringBuilder("jdbc:ydb:")
                 .append(ydb.useTls() ? "grpcs://" : "grpc://")
+                .append(authority)
                 .append(ydb.endpoint())
                 .append(ydb.database());
 
         char splitter = '?';
-        if (ydb.authToken() != null) {
+        if (ydb.authToken() != null && !disableToken) {
             jdbc.append(splitter).append("token=").append(ydb.authToken());
             splitter = '&';
         }