From 5c43d023578d86cc9d03e178e00fc07465391a61 Mon Sep 17 00:00:00 2001
From: Gabriel Tofvesson <gabriel.tofvesson@gmail.com>
Date: Sun, 19 Apr 2020 16:13:28 +0200
Subject: [PATCH] Update tests for method weaving

---
 test/MergeInject.java | 18 ++++++++++++++++--
 test/MergeTest.java   |  2 +-
 2 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/test/MergeInject.java b/test/MergeInject.java
index 0380481..beda8b3 100644
--- a/test/MergeInject.java
+++ b/test/MergeInject.java
@@ -17,8 +17,22 @@ public class MergeInject extends MergeTest implements Runnable {
         number = 10;
     }
 
-    @Inject(REPLACE)
-    public String test(){
+    @Inject(value = AFTER, target = "test()Ljava/lang/String;", acceptOriginalReturn = true)
+    public String test(String retVal){
+        System.out.println("Got retval: "+retVal);
+
+        if (retVal.equals("Test")) {
+            System.out.println("Test");
+        }
+
+        retVal = "ASDF";
+
+        String a = "retVal";
+
+        System.out.println(a);
+        a = "Test";
+        System.out.println(a);
+
         System.out.println(s);
 
         if(s.endsWith("e!!")) {
diff --git a/test/MergeTest.java b/test/MergeTest.java
index 74a4d78..2f09556 100644
--- a/test/MergeTest.java
+++ b/test/MergeTest.java
@@ -12,7 +12,7 @@ public class MergeTest {
     }
 
     public String test(){
-        return s;
+        return s + "Test";
     }
 
     public String test1(){