1 files changed, 89 insertions, 0 deletions
diff --git a/src/test/java/com/code_intelligence/jazzer/instrumentor/AfterHooksPatchTest.kt b/src/test/java/com/code_intelligence/jazzer/instrumentor/AfterHooksPatchTest.kt
new file mode 100644
index 00000000..55263786
--- /dev/null
+++ b/src/test/java/com/code_intelligence/jazzer/instrumentor/AfterHooksPatchTest.kt
@@ -0,0 +1,89 @@
+// Copyright 2021 Code Intelligence GmbH
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package com.code_intelligence.jazzer.instrumentor
+
+import com.code_intelligence.jazzer.instrumentor.PatchTestUtils.bytecodeToClass
+import com.code_intelligence.jazzer.instrumentor.PatchTestUtils.classToBytecode
+import org.junit.Test
+import java.io.File
+
+private fun getOriginalAfterHooksTargetInstance(): AfterHooksTargetContract {
+    return AfterHooksTarget()
+}
+
+private fun getNoHooksAfterHooksTargetInstance(): AfterHooksTargetContract {
+    val originalBytecode = classToBytecode(AfterHooksTarget::class.java)
+    // Let the bytecode pass through the hooking logic, but don't apply any hooks.
+    val patchedBytecode = HookInstrumentor(emptyList(), false, null).instrument(
+        AfterHooksTarget::class.java.name.replace('.', '/'),
+        originalBytecode,
+    )
+    val patchedClass = bytecodeToClass(AfterHooksTarget::class.java.name, patchedBytecode)
+    return patchedClass.getDeclaredConstructor().newInstance() as AfterHooksTargetContract
+}
+
+private fun getPatchedAfterHooksTargetInstance(classWithHooksEnabledField: Class<*>?): AfterHooksTargetContract {
+    val originalBytecode = classToBytecode(AfterHooksTarget::class.java)
+    val hooks = Hooks.loadHooks(emptyList(), setOf(AfterHooks::class.java.name)).first().hooks
+    val patchedBytecode = HookInstrumentor(
+        hooks,
+        false,
+        classWithHooksEnabledField = classWithHooksEnabledField?.name?.replace('.', '/'),
+    ).instrument(AfterHooksTarget::class.java.name.replace('.', '/'), originalBytecode)
+    // Make the patched class available in bazel-testlogs/.../test.outputs for manual inspection.
+    val outDir = System.getenv("TEST_UNDECLARED_OUTPUTS_DIR")
+    File("$outDir/${AfterHooksTarget::class.java.simpleName}.class").writeBytes(originalBytecode)
+    File("$outDir/${AfterHooksTarget::class.java.simpleName}.patched.class").writeBytes(patchedBytecode)
+    val patchedClass = bytecodeToClass(AfterHooksTarget::class.java.name, patchedBytecode)
+    return patchedClass.getDeclaredConstructor().newInstance() as AfterHooksTargetContract
+}
+
+class AfterHooksPatchTest {
+
+    @Test
+    fun testOriginal() {
+        assertSelfCheck(getOriginalAfterHooksTargetInstance(), false)
+    }
+
+    @Test
+    fun testPatchedWithoutHooks() {
+        assertSelfCheck(getNoHooksAfterHooksTargetInstance(), false)
+    }
+
+    @Test
+    fun testPatched() {
+        assertSelfCheck(getPatchedAfterHooksTargetInstance(null), true)
+    }
+
+    object HooksEnabled {
+        @Suppress("unused")
+        const val hooksEnabled = true
+    }
+
+    object HooksDisabled {
+        @Suppress("unused")
+        const val hooksEnabled = false
+    }
+
+    @Test
+    fun testPatchedWithConditionalHooksEnabled() {
+        assertSelfCheck(getPatchedAfterHooksTargetInstance(HooksEnabled::class.java), true)
+    }
+
+    @Test
+    fun testPatchedWithConditionalHooksDisabled() {
+        assertSelfCheck(getPatchedAfterHooksTargetInstance(HooksDisabled::class.java), false)
+    }
+}