integration-testing/src/debugAgentTest/kotlin/PrecompiledDebugProbesTest.kt

/*
 * Copyright 2016-2021 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license.
 */
import org.junit.Test
import java.io.*
import kotlin.test.*

/*
 * This is intentionally put here instead of coreAgentTest to avoid accidental classpath replacing
 * and ruining core agent test.
 */
class PrecompiledDebugProbesTest {

    private val overwrite = java.lang.Boolean.getBoolean("overwrite.probes")

    @Test
    fun testClassFileContent() {
        val clz = Class.forName("kotlin.coroutines.jvm.internal.DebugProbesKt")
        val classFileResourcePath = clz.name.replace(".", "/") + ".class"
        val array = clz.classLoader.getResourceAsStream(classFileResourcePath).use { it.readBytes() }
        assertJava8Compliance(array)
        // we expect the integration testing project to be in a subdirectory of the main kotlinx.coroutines project
        val base = File("").absoluteFile.parentFile
        val probes = File(base, "kotlinx-coroutines-core/jvm/resources/DebugProbesKt.bin")
        val binContent = probes.readBytes()
        if (overwrite) {
            FileOutputStream(probes).use { it.write(array) }
            println("Content was successfully overwritten!")
        } else {
            assertTrue(
                array.contentEquals(binContent),
                "Compiled DebugProbesKt.class does not match the file shipped as a resource in kotlinx-coroutines-core. " +
                        "Typically it happens because of the Kotlin version update (-> binary metadata). " +
                        "In that case, run the same test with -Poverwrite.probes=true."
            )
        }
    }

    private fun assertJava8Compliance(classBytes: ByteArray) {
        DataInputStream(classBytes.inputStream()).use {
            val magic: Int = it.readInt()
            if (magic != -0x35014542) throw IllegalArgumentException("Not a valid class!")
            val minor: Int = it.readUnsignedShort()
            val major: Int = it.readUnsignedShort()
            assertEquals(52, major)
            assertEquals(0, minor)
        }
    }
}