Cache CtMethod's TestFramework (#113)

Signed-off-by: André Silva <[email protected]>

src/main/java/eu/stamp_project/testrunner/test_framework/TestFramework.java

@@ -39,6 +39,8 @@ public static void init(Factory factory) {
         _instance = new TestFramework(factory);
     }
 
+    private HashMap<CtMethod<?>, TestFrameworkSupport> cachedTestFrameworks;
+
     private List<TestFrameworkSupport> testFrameworkSupportList;
 
     public static TestFramework get() {
@@ -49,6 +51,7 @@ public static TestFramework get() {
 
     private TestFramework(Factory factory) {
         this.factory = factory;
+        this.cachedTestFrameworks = new HashMap<>();
         this.testFrameworkSupportList = new ArrayList<>();
         this.testFrameworkSupportList.add(new JUnit3Support());
         this.testFrameworkSupportList.add(new JUnit4Support());
@@ -163,8 +166,10 @@ private TestFrameworkSupport getTestFramework(CtMethod<?> testMethod) {
     		DSpotCache.getTestFrameworkCache().put(TypeUtils.getQualifiedName(originalMethod), tfs);
     	}
     	return tfs;*/
-        // TODO Enable the cache in test-runner
-        return getTestFrameworkImpl(testMethod);
+        if (!cachedTestFrameworks.containsKey(testMethod)) {
+            cachedTestFrameworks.put(testMethod, getTestFrameworkImpl(testMethod));
+        }
+        return cachedTestFrameworks.get(testMethod);
     }