Support arguments in single quotes at CsvSource (#790)

tsantalis · Oct 19, 2024 · 99ee636 · 99ee636
1 parent 5368196
commit 99ee636
Showing 1 changed file with 21 additions and 10 deletions.
diff --git a/src/main/java/gr/uom/java/xmi/util/CsvUtils.java b/src/main/java/gr/uom/java/xmi/util/CsvUtils.java
@@ -9,16 +9,27 @@
 public class CsvUtils {
     public static List<String> extractParametersFromCsv(String s) {
         List<String> parameters = new ArrayList<>();
-        String[] tokens = s.split(",");
-        for (String token : tokens) {
-            String trimmed = token.trim();
-            if (trimmed.startsWith("\"")) {
-                trimmed = trimmed.substring(1, trimmed.length());
-            }
-            if (trimmed.endsWith("\"")) {
-                trimmed = trimmed.substring(0, trimmed.length() - 1);
-            }
-            parameters.add(trimmed);
+        //fix for arguments with single quotes
+        if(s.contains("'")) {
+        	String[] tokens = s.split("'");
+        	for (String token : tokens) {
+        		if(!token.equals("\"") && !token.matches("\s*,\s*")) {
+        			parameters.add(token);
+        		}
+        	}
+        }
+        else {
+            String[] tokens = s.split(",");
+	        for (String token : tokens) {
+	            String trimmed = token.trim();
+	            if (trimmed.startsWith("\"")) {
+	                trimmed = trimmed.substring(1, trimmed.length());
+	            }
+	            if (trimmed.endsWith("\"")) {
+	                trimmed = trimmed.substring(0, trimmed.length() - 1);
+	            }
+	            parameters.add(trimmed);
+	        }
         }
         return parameters;
     }