Support running local unit test with transformer

Author: johnsonlee

booster-test-asm/build.gradle

@@ -0,0 +1,9 @@
+apply from: "$rootDir/gradle/booster.gradle"
+
+dependencies {
+    implementation project(':booster-test')
+    implementation project(':booster-transform-asm')
+
+    kaptTest 'com.google.auto.service:auto-service:1.0'
+    testImplementation 'com.github.stefanbirkner:system-rules:1.19.0'
+}

booster-test-asm/packages-zh-CN.md

@@ -0,0 +1,3 @@
+# Package com.didiglobal.booster.test.asm
+
+提供了基于 ASM 的单元测试支持

booster-test-asm/src/main/kotlin/com/didiglobal/booster/test/asm/BoosterTestRunnerWithAsm.kt

@@ -0,0 +1,15 @@
+package com.didiglobal.booster.test.asm
+
+import com.didiglobal.booster.test.BoosterTestRunner
+import com.didiglobal.booster.transform.asm.AsmTransformer
+import com.didiglobal.booster.transform.asm.ClassTransformer
+import com.didiglobal.booster.transform.util.TransformerClassLoader
+
+class BoosterTestRunnerWithAsm(
+        clazz: Class<*>
+) : BoosterTestRunner(clazz, TransformerClassLoader(
+        clazz.classLoader,
+        ClassTransformer::class.java::isAssignableFrom
+) {
+    AsmTransformer(it)
+})

booster-test-asm/src/test/kotlin/com/didiglobal/booster/test/asm/BoosterTestRunnerWithAsmTest.kt

@@ -0,0 +1,22 @@
+package com.didiglobal.booster.test.asm
+
+import org.junit.Rule
+import org.junit.contrib.java.lang.system.SystemOutRule
+import org.junit.runner.RunWith
+import kotlin.test.Test
+import kotlin.test.assertTrue
+
+@RunWith(BoosterTestRunnerWithAsm::class)
+class BoosterTestRunnerWithAsmTest {
+
+    @get:Rule
+    val systemOutRule = SystemOutRule().enableLog()
+
+    @Test
+    fun `test booster runner with asm`() {
+        assertTrue {
+            systemOutRule.log.isNotBlank()
+        }
+    }
+
+}

booster-test-asm/src/test/kotlin/com/didiglobal/booster/test/asm/SimpleClassTransformer.kt

@@ -0,0 +1,16 @@
+package com.didiglobal.booster.test.asm
+
+import com.didiglobal.booster.transform.TransformContext
+import com.didiglobal.booster.transform.asm.ClassTransformer
+import com.google.auto.service.AutoService
+import org.objectweb.asm.tree.ClassNode
+
+@AutoService(ClassTransformer::class)
+class SimpleClassTransformer : ClassTransformer {
+
+    override fun transform(context: TransformContext, klass: ClassNode): ClassNode {
+        println(klass.name)
+        return klass
+    }
+
+}

booster-test-javassist/build.gradle

@@ -0,0 +1,9 @@
+apply from: "$rootDir/gradle/booster.gradle"
+
+dependencies {
+    implementation project(':booster-test')
+    implementation project(':booster-transform-javassist')
+
+    kaptTest 'com.google.auto.service:auto-service:1.0'
+    testImplementation 'com.github.stefanbirkner:system-rules:1.19.0'
+}

booster-test-javassist/packages-zh-CN.md

@@ -0,0 +1,3 @@
+# Package com.didiglobal.booster.test.javassist
+
+提供了基于 Javassist 的单元测试支持

booster-test-javassist/src/main/kotlin/com/didiglobal/booster/test/javassist/BoosterTestRunnerWithJavassist.kt

@@ -0,0 +1,15 @@
+package com.didiglobal.booster.test.javassist
+
+import com.didiglobal.booster.test.BoosterTestRunner
+import com.didiglobal.booster.transform.javassist.ClassTransformer
+import com.didiglobal.booster.transform.javassist.JavassistTransformer
+import com.didiglobal.booster.transform.util.TransformerClassLoader
+
+class BoosterTestRunnerWithJavassist(
+        clazz: Class<*>
+) : BoosterTestRunner(clazz, TransformerClassLoader(
+        clazz.classLoader,
+        ClassTransformer::class.java::isAssignableFrom
+) {
+    JavassistTransformer(it)
+})

booster-test-javassist/src/test/kotlin/com/didiglobal/booster/test/javassist/BoosterTestRunnerWithJavassistTest.kt

@@ -0,0 +1,22 @@
+package com.didiglobal.booster.test.javassist
+
+import org.junit.Rule
+import org.junit.contrib.java.lang.system.SystemOutRule
+import org.junit.runner.RunWith
+import kotlin.test.Test
+import kotlin.test.assertTrue
+
+@RunWith(BoosterTestRunnerWithJavassist::class)
+class BoosterTestRunnerWithJavassistTest {
+
+    @get:Rule
+    val systemOutRule = SystemOutRule().enableLog()
+
+    @Test
+    fun `test booster runner with javassist`() {
+        assertTrue {
+            systemOutRule.log.isNotBlank()
+        }
+    }
+
+}

booster-test-javassist/src/test/kotlin/com/didiglobal/booster/test/javassist/SimpleClassTransformer.kt

@@ -0,0 +1,16 @@
+package com.didiglobal.booster.test.javassist
+
+import com.didiglobal.booster.transform.TransformContext
+import com.didiglobal.booster.transform.javassist.ClassTransformer
+import com.google.auto.service.AutoService
+import javassist.CtClass
+
+@AutoService(ClassTransformer::class)
+class SimpleClassTransformer : ClassTransformer {
+
+    override fun transform(context: TransformContext, klass: CtClass): CtClass {
+        println(klass.name)
+        return klass
+    }
+
+}