1 files changed, 27 insertions, 0 deletions
diff --git a/agent/src/test/java/com/code_intelligence/jazzer/instrumentor/ReplaceHooks.java b/agent/src/test/java/com/code_intelligence/jazzer/instrumentor/ReplaceHooks.java
index a71e1180..7e31b77b 100644
--- a/agent/src/test/java/com/code_intelligence/jazzer/instrumentor/ReplaceHooks.java
+++ b/agent/src/test/java/com/code_intelligence/jazzer/instrumentor/ReplaceHooks.java
@@ -18,6 +18,7 @@ import com.code_intelligence.jazzer.api.HookType;
 import com.code_intelligence.jazzer.api.MethodHook;
 import java.lang.invoke.MethodHandle;
 
+@SuppressWarnings("unused")
 public class ReplaceHooks {
   @MethodHook(type = HookType.REPLACE,
       targetClassName = "com.code_intelligence.jazzer.instrumentor.ReplaceHooksTarget",
@@ -106,4 +107,30 @@ public class ReplaceHooks {
   patchAbstractListGet(MethodHandle method, Object thisObject, Object[] arguments, int hookId) {
     return true;
   }
+
+  @MethodHook(type = HookType.REPLACE, targetClassName = "java.util.Set", targetMethod = "contains",
+      targetMethodDescriptor = "(Ljava/lang/Object;)Z")
+  public static boolean
+  patchSetGet(MethodHandle method, Object thisObject, Object[] arguments, int hookId) {
+    return true;
+  }
+
+  @MethodHook(type = HookType.REPLACE,
+      targetClassName = "com.code_intelligence.jazzer.instrumentor.ReplaceHooksInit",
+      targetMethod = "<init>", targetMethodDescriptor = "()V")
+  public static ReplaceHooksInit
+  patchInit(MethodHandle method, Object thisObject, Object[] arguments, int hookId) {
+    // Test with subclass
+    return new ReplaceHooksInit() {
+      { initialized = true; }
+    };
+  }
+
+  @MethodHook(type = HookType.REPLACE,
+      targetClassName = "com.code_intelligence.jazzer.instrumentor.ReplaceHooksInit",
+      targetMethod = "<init>", targetMethodDescriptor = "(ZLjava/lang/String;)V")
+  public static ReplaceHooksInit
+  patchInitWithParams(MethodHandle method, Object thisObject, Object[] arguments, int hookId) {
+    return new ReplaceHooksInit(true, "");
+  }
 }