浏览代码

Ukonczony krok trzeci

Adam Matuszewski 1 年之前
父节点
当前提交
ae59642120

+ 3 - 2
StringCalculator/src/main/java/pl/dmcs/Calculator.java

@@ -5,10 +5,11 @@ public class Calculator {
         if (numbers == null || numbers.isEmpty()) {
             return 0;
         }
-        String[] numArray = numbers.split(",");
+        // Split the string using both commas and new lines as delimiters
+        String[] numArray = numbers.split("[,\n]");
         int sum = 0;
         for (String num : numArray) {
-            sum += Integer.parseInt(num);
+            sum += Integer.parseInt(num.trim());
         }
         return sum;
     }

+ 13 - 0
StringCalculator/src/test/java/pl/dmcs/CalculatorTest.java

@@ -22,5 +22,18 @@ public class CalculatorTest {
         assertEquals(3, calculator.add("1,2"));
     }
 
+    @Test
+    public void testAddMultipleNumbers() {
+        assertEquals(6, calculator.add("1,2,3"));
+        assertEquals(10, calculator.add("1,2,3,4"));
+        assertEquals(15, calculator.add("1,2,3,4,5"));
+    }
+
+    @Test
+    public void testAddNumbersWithNewLines() {
+        assertEquals(6, calculator.add("1\n2,3"));
+        assertEquals(10, calculator.add("1\n2\n3,4"));
+        assertEquals(15, calculator.add("1,2\n3\n4,5"));
+    }
 
 }