From 41afc5d9c862e80b59813fe19997e22b12b23905 Mon Sep 17 00:00:00 2001
From: danglotb <bdanglot@gmail.com>
Date: Thu, 25 Jan 2018 15:59:14 +0100
Subject: [PATCH] feat: returns an empty list if the given test method does not
 contain any literals

---
 src/main/java/fr/inria/stamp/Ex2Amplifier.java | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/main/java/fr/inria/stamp/Ex2Amplifier.java b/src/main/java/fr/inria/stamp/Ex2Amplifier.java
index b180b6e..551baed 100644
--- a/src/main/java/fr/inria/stamp/Ex2Amplifier.java
+++ b/src/main/java/fr/inria/stamp/Ex2Amplifier.java
@@ -18,6 +18,7 @@
 import spoon.reflect.visitor.filter.TypeFilter;
 
 import java.io.File;
+import java.util.Collections;
 import java.util.Iterator;
 import java.util.List;
 import java.util.stream.Collectors;
@@ -37,6 +38,9 @@ public Ex2Amplifier(InputConfiguration configuration) {
 
     @Override
     public List<CtMethod> apply(CtMethod ctMethod) {
+        if (ctMethod.getElements(new TypeFilter<CtLiteral>(CtLiteral.class)).isEmpty()) {
+            return Collections.emptyList();
+        }
         final CtMethod<?> mainMethodFromTestMethod =
                 MainGenerator.generateMainMethodFromTestMethod(ctMethod);
         final CtClass testClass = ctMethod.getParent(CtClass.class);