bael-4909: update test case

sharifi · sharifi · commit d69e133c0e20 · 2021-05-12T08:41:27.000+04:30
diff --git a/core-java-modules/core-java-11-2/src/test/java/com/baeldung/version/VersionUnitTest.java b/core-java-modules/core-java-11-2/src/test/java/com/baeldung/version/VersionUnitTest.java
@@ -18,31 +18,29 @@ public void givenJava_whenUsingRuntime_thenGetVersion() {
     @Test
     @Disabled("Only valid for Java 8 and lower")
     public void givenJava_whenUsingCommonsLang_thenGetVersion() {
-        String expectedVersion = "8";
-        String version = SystemUtils.JAVA_SPECIFICATION_VERSION;
-        if (version.startsWith("1.")) {
-            version = version.substring(2, 3);
+        int expectedVersion = 8;
+        String[] versionElements = SystemUtils.JAVA_SPECIFICATION_VERSION.split("\\.");
+        int discard = Integer.parseInt(versionElements[0]);
+        int version;
+        if (discard == 1) {
+            version = Integer.parseInt(versionElements[1]);
         } else {
-            int separator = version.indexOf(".");
-            if (separator != -1) {
-                version = version.substring(0, separator);
-            }
+            version = discard;
         }
         Assertions.assertThat(version).isEqualTo(expectedVersion);
     }
 
     @Test
     @Disabled("Only valid for Java 8 and lower")
     public void givenJava_whenUsingSystemProp_thenGetVersion() {
-        String expectedVersion = "8";
-        String version = System.getProperty("java.version");
-        if (version.startsWith("1.")) {
-            version = version.substring(2, 3);
+        int expectedVersion = 8;
+        String[] versionElements = System.getProperty("java.version").split("\\.");
+        int discard = Integer.parseInt(versionElements[0]);
+        int version;
+        if (discard == 1) {
+            version = Integer.parseInt(versionElements[1]);
         } else {
-            int separator = version.indexOf(".");
-            if (separator != -1) {
-                version = version.substring(0, separator);
-            }
+            version = discard;
         }
         Assertions.assertThat(version).isEqualTo(expectedVersion);
     }
