fix: quote and dquote are discarded for the random char

STAMP-project · Dec 18, 2016 · 2e42624 · 2e42624
1 parent eb9a750
commit 2e42624
Showing 1 changed file with 7 additions and 10 deletions.
diff --git a/src/main/java/fr/inria/diversify/dspot/amp/TestDataMutator.java b/src/main/java/fr/inria/diversify/dspot/amp/TestDataMutator.java
@@ -129,16 +129,11 @@ protected List<CtMethod> createAllStringMutant(CtMethod method, CtLiteral litera
     }
 
 
-    protected CtMethod createStringMutant(CtMethod method, int original_lit_index, String newValue) {
+    private CtMethod createStringMutant(CtMethod method, int original_lit_index, String newValue) {
         dataCount++;
-        //clone the method
         CtMethod cloned_method = AmplificationHelper.cloneMethod(method, "_literalMutation");
-        //get the lit_indexth literal of the cloned method
-        CtLiteral newLiteral = Query.getElements(cloned_method, new TypeFilter<CtLiteral>(CtLiteral.class))
-                .get(original_lit_index);
-
-        newLiteral.setValue(newValue);
-
+        Query.getElements(cloned_method, new TypeFilter<>(CtLiteral.class))
+                .get(original_lit_index).replace(cloned_method.getFactory().Code().createLiteral(newValue));
         return cloned_method;
     }
 
@@ -170,8 +165,10 @@ protected Set<String> stringMutated(CtLiteral literal) {
         return values;
     }
 
-    protected char getRandomChar() {
-        return (char) (AmplificationHelper.getRandom().nextInt(94) + 32);
+    private char getRandomChar() {
+        int value = AmplificationHelper.getRandom().nextInt(94) + 32;
+        char c = (char) ((value == 34 || value == 39) ? value + (AmplificationHelper.getRandom().nextBoolean() ? 1 : -1) : value);
+        return c;//excluding " and '
     }
 
     protected Set<? extends Number> numberMutated(CtLiteral literal) {