Fix invalid connection strings test

viniciusam · viniciusam · commit ce49bd896927 · 2017-06-18T09:08:20.000-03:00
diff --git a/src/main/java/io/github/utplsql/cli/ConnectionInfo.java b/src/main/java/io/github/utplsql/cli/ConnectionInfo.java
@@ -47,7 +47,7 @@ public ConnectionInfo parseConnectionString(String connectionString)
     private void parseCredentials(String str) throws ParameterException, IllegalArgumentException {
         int barIdx = str.indexOf("/");
 
-        if (barIdx == -1 || str.length() == 1)
+        if (barIdx == -1 || barIdx == 0 || barIdx == str.length() - 1)
             throw invalidConnectionString();
 
         this.setUser(str.substring(0, barIdx));
diff --git a/src/test/java/io/github/utplsql/cli/ConnectionInfoTest.java b/src/test/java/io/github/utplsql/cli/ConnectionInfoTest.java
@@ -17,7 +17,7 @@ public class ConnectionInfoTest {
      */
 
     @Test
-    public void connectionStr_Full() {
+    public void valid_Full() {
         try {
             ConnectionInfo ci = new ConnectionInfo()
                     .parseConnectionString("my_user/p@ss!@some.server.123-abc.com:3000/db_1.acme.com");
@@ -34,7 +34,7 @@ public void connectionStr_Full() {
     }
 
     @Test
-    public void connectionStr_WithoutPort() {
+    public void valid_WithoutPort() {
         try {
             ConnectionInfo ci = new ConnectionInfo()
                     .parseConnectionString("my_user/p@ss!@some.server.123-abc.com/db_1.acme.com");
@@ -51,7 +51,7 @@ public void connectionStr_WithoutPort() {
     }
 
     @Test
-    public void connectionStr_WithoutHostAndPort() {
+    public void valid_WithoutHostAndPort() {
         try {
             ConnectionInfo ci = new ConnectionInfo()
                     .parseConnectionString("my_user/p@ss!@127.0.0.1/db_1.acme.com");
@@ -68,14 +68,56 @@ public void connectionStr_WithoutHostAndPort() {
     }
 
     @Test
-    public void connectionStr_Invalid() {
+    public void invalid_WithoutDatabase_1() {
         try {
             new ConnectionInfo().parseConnectionString("user/pass@");
+            Assert.fail();
+        } catch (ParameterException ignored) {}
+    }
+
+    @Test
+    public void invalid_WithoutDatabase_2() {
+        try {
             new ConnectionInfo().parseConnectionString("user/pass");
-            new ConnectionInfo().parseConnectionString("user/@");
-            new ConnectionInfo().parseConnectionString("/pass@");
-            new ConnectionInfo().parseConnectionString("/@");
-            new ConnectionInfo().parseConnectionString("@");
+            Assert.fail();
+        } catch (ParameterException ignored) {}
+    }
+
+    @Test
+    public void invalid_WithoutDatabase_3() {
+        try {
+            new ConnectionInfo().parseConnectionString("user/pass@localhost:1521");
+            Assert.fail();
+        } catch (ParameterException ignored) {}
+    }
+
+    @Test
+    public void invalid_WithoutPassword() {
+        try {
+            new ConnectionInfo().parseConnectionString("user/@db");
+            Assert.fail();
+        } catch (ParameterException ignored) {}
+    }
+
+    @Test
+    public void invalid_WithoutUsername() {
+        try {
+            new ConnectionInfo().parseConnectionString("/pass@db");
+            Assert.fail();
+        } catch (ParameterException ignored) {}
+    }
+
+    @Test
+    public void invalid_WithoutUserPassDb_1() {
+        try {
+            new ConnectionInfo().parseConnectionString("/@db");
+            Assert.fail();
+        } catch (ParameterException ignored) {}
+    }
+
+    @Test
+    public void invalid_WithoutUserPass() {
+        try {
             new ConnectionInfo().parseConnectionString("@db");
             Assert.fail();
         } catch (ParameterException ignored) {}
