[PowerAssert] Add codegen tests

generators/build.gradle.kts

@@ -65,6 +65,7 @@ dependencies {
     testApi(projectTests(":kotlin-allopen-compiler-plugin"))
     testApi(projectTests(":kotlin-noarg-compiler-plugin"))
     testApi(projectTests(":kotlin-lombok-compiler-plugin"))
+    testApi(projectTests(":kotlin-power-assert-compiler-plugin"))
     testApi(projectTests(":kotlin-sam-with-receiver-compiler-plugin"))
     testApi(projectTests(":kotlin-assignment-compiler-plugin"))
     testApi(projectTests(":kotlinx-serialization-compiler-plugin"))

generators/tests/org/jetbrains/kotlin/generators/tests/GenerateTests.kt

@@ -41,6 +41,8 @@ import org.jetbrains.kotlin.kapt4.AbstractKotlinKapt4ContextTest
 import org.jetbrains.kotlin.lombok.*
 import org.jetbrains.kotlin.noarg.*
 import org.jetbrains.kotlin.parcelize.test.runners.*
+import org.jetbrains.kotlin.powerassert.AbstractFirLightTreeBlackBoxCodegenTestForPowerAssert
+import org.jetbrains.kotlin.powerassert.AbstractIrBlackBoxCodegenTestForPowerAssert
 import org.jetbrains.kotlin.samWithReceiver.*
 import org.jetbrains.kotlin.test.TargetBackend
 import org.jetbrains.kotlinx.atomicfu.AbstractAtomicfuJsIrTest
@@ -355,6 +357,15 @@ fun main(args: Array<String>) {
             }
         }
 
+        testGroup("plugins/power-assert/tests-gen", "plugins/power-assert/testData") {
+            testClass<AbstractIrBlackBoxCodegenTestForPowerAssert> {
+                model("codegen", excludedPattern = excludedFirTestdataPattern)
+            }
+            testClass<AbstractFirLightTreeBlackBoxCodegenTestForPowerAssert> {
+                model("codegen", excludedPattern = excludedFirTestdataPattern)
+            }
+        }
+
         testGroup("plugins/sam-with-receiver/tests-gen", "plugins/sam-with-receiver/testData") {
             testClass<AbstractSamWithReceiverTest> {
                 model("diagnostics", excludedPattern = excludedFirTestdataPattern)

plugins/power-assert/build.gradle.kts

@@ -5,6 +5,8 @@ plugins {
     id("jps-compatible")
 }
 
+val junit5Classpath by configurations.creating
+
 dependencies {
     embedded(project(":kotlin-power-assert-compiler-plugin.backend")) { isTransitive = false }
     embedded(project(":kotlin-power-assert-compiler-plugin.cli")) { isTransitive = false }
@@ -18,6 +20,8 @@ dependencies {
 
     testRuntimeOnly(commonDependency("org.codehaus.woodstox:stax2-api"))
     testRuntimeOnly(commonDependency("com.fasterxml:aalto-xml"))
+
+    junit5Classpath(libs.junit.jupiter.api)
 }
 
 optInToExperimentalCompilerApi()
@@ -41,4 +45,10 @@ projectTest(parallel = true) {
     dependsOn(":dist")
     workingDir = rootDir
     useJUnitPlatform()
+
+    val localJunit5Classpath: FileCollection = junit5Classpath
+
+    doFirst {
+        systemProperty("junit5.classpath", localJunit5Classpath.files.joinToString(",") { it.absolutePath })
+    }
 }

plugins/power-assert/power-assert.backend/src/org/jetbrains/kotlin/powerassert/diagram/SourceFile.kt

@@ -30,7 +30,26 @@ import java.io.File
 data class SourceFile(
     private val irFile: IrFile,
 ) {
-    private val source: String = File(irFile.path).readText()
+    companion object {
+        private val KOTLIN_POWER_ASSERT_ADD_SRC_ROOTS = System.getProperty("KOTLIN_POWER_ASSERT_ADD_SRC_ROOTS") ?: ""
+        private val sourceRoots = listOf(".") + KOTLIN_POWER_ASSERT_ADD_SRC_ROOTS.split(",").map { it.trim() }
+    }
+
+    private val source: String = run {
+        File(irFile.path).also { file ->
+            if (file.exists()) {
+                return@run file.readText()
+            }
+        }
+
+        for (root in sourceRoots) {
+            val file = File(root, irFile.path)
+            if (file.exists()) {
+                return@run file.readText()
+            }
+        }
+        throw IllegalArgumentException("Unable to find source for IrFile: ${irFile.path}")
+    }
         .replace("\r\n", "\n") // https://youtrack.jetbrains.com/issue/KT-41888
 
     fun getSourceRangeInfo(element: IrElement): SourceRangeInfo {

plugins/power-assert/test/org/jetbrains/kotlin/powerassert/JunitTestServices.kt

@@ -0,0 +1,42 @@
+/*
+ * Copyright 2010-2023 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.powerassert
+
+import org.jetbrains.kotlin.cli.jvm.config.addJvmClasspathRoot
+import org.jetbrains.kotlin.config.CompilerConfiguration
+import org.jetbrains.kotlin.test.builders.TestConfigurationBuilder
+import org.jetbrains.kotlin.test.model.TestModule
+import org.jetbrains.kotlin.test.services.EnvironmentConfigurator
+import org.jetbrains.kotlin.test.services.RuntimeClasspathProvider
+import org.jetbrains.kotlin.test.services.TestServices
+import java.io.File
+
+private val junit5Classpath = System.getProperty("junit5.classpath").split(",")
+
+fun TestConfigurationBuilder.enableJunit() {
+    useConfigurators(::JunitEnvironmentConfigurator)
+    useCustomRuntimeClasspathProviders(::JunitRuntimeClassPathProvider)
+}
+
+class JunitEnvironmentConfigurator(testServices: TestServices) : EnvironmentConfigurator(testServices) {
+    override fun configureCompilerConfiguration(configuration: CompilerConfiguration, module: TestModule) {
+        if (PowerAssertConfigurationDirectives.WITH_JUNIT5 in module.directives) {
+            for (file in junit5Classpath.map { File(it) }) {
+                configuration.addJvmClasspathRoot(file)
+            }
+        }
+    }
+}
+
+class JunitRuntimeClassPathProvider(testServices: TestServices) : RuntimeClasspathProvider(testServices) {
+    override fun runtimeClassPaths(module: TestModule): List<File> {
+        if (PowerAssertConfigurationDirectives.WITH_JUNIT5 in module.directives) {
+            return junit5Classpath.map { File(it) }
+        } else {
+            return emptyList()
+        }
+    }
+}

plugins/power-assert/test/org/jetbrains/kotlin/powerassert/PowerAssertConfigurationDirectives.kt

@@ -0,0 +1,17 @@
+/*
+ * Copyright 2010-2023 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.powerassert
+
+import org.jetbrains.kotlin.test.directives.model.SimpleDirectivesContainer
+
+object PowerAssertConfigurationDirectives : SimpleDirectivesContainer() {
+    val FUNCTION by stringDirective(
+        description = "Functions targeted by Power-Assert transformation",
+        multiLine = true,
+    )
+
+    val WITH_JUNIT5 by directive("Add JUnit5 to classpath")
+}

plugins/power-assert/test/org/jetbrains/kotlin/powerassert/PowerAssertTests.kt

@@ -0,0 +1,96 @@
+/*
+ * Copyright 2010-2023 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.powerassert
+
+import org.jetbrains.kotlin.backend.common.extensions.IrGenerationExtension
+import org.jetbrains.kotlin.cli.common.CLIConfigurationKeys
+import org.jetbrains.kotlin.cli.common.messages.MessageCollector
+import org.jetbrains.kotlin.compiler.plugin.CompilerPluginRegistrar
+import org.jetbrains.kotlin.config.CompilerConfiguration
+import org.jetbrains.kotlin.name.FqName
+import org.jetbrains.kotlin.test.builders.TestConfigurationBuilder
+import org.jetbrains.kotlin.test.directives.AdditionalFilesDirectives
+import org.jetbrains.kotlin.test.directives.ConfigurationDirectives.WITH_STDLIB
+import org.jetbrains.kotlin.test.directives.JvmEnvironmentConfigurationDirectives.FULL_JDK
+import org.jetbrains.kotlin.test.directives.model.DirectivesContainer
+import org.jetbrains.kotlin.test.directives.model.RegisteredDirectives
+import org.jetbrains.kotlin.test.model.TestFile
+import org.jetbrains.kotlin.test.model.TestModule
+import org.jetbrains.kotlin.test.runners.codegen.AbstractFirLightTreeBlackBoxCodegenTest
+import org.jetbrains.kotlin.test.runners.codegen.AbstractIrBlackBoxCodegenTest
+import org.jetbrains.kotlin.test.services.AdditionalSourceProvider
+import org.jetbrains.kotlin.test.services.EnvironmentConfigurator
+import org.jetbrains.kotlin.test.services.TestServices
+import java.io.File
+
+// ------------------------ codegen ------------------------
+
+open class AbstractIrBlackBoxCodegenTestForPowerAssert : AbstractIrBlackBoxCodegenTest() {
+    override fun configure(builder: TestConfigurationBuilder) {
+        super.configure(builder)
+        builder.configurePlugin()
+    }
+}
+
+open class AbstractFirLightTreeBlackBoxCodegenTestForPowerAssert : AbstractFirLightTreeBlackBoxCodegenTest() {
+    override fun configure(builder: TestConfigurationBuilder) {
+        super.configure(builder)
+        builder.configurePlugin()
+    }
+}
+
+// ------------------------ configuration ------------------------
+
+fun TestConfigurationBuilder.configurePlugin() {
+    // TODO there has got to be a better way?
+    val sourceRoots = File("plugins/power-assert/testData/")
+        .walkTopDown()
+        .filter { it.isDirectory }
+        .joinToString(",") { it.path }
+    System.setProperty("KOTLIN_POWER_ASSERT_ADD_SRC_ROOTS", sourceRoots)
+
+
+    defaultDirectives {
+        +FULL_JDK
+        +WITH_STDLIB
+    }
+    useDirectives(PowerAssertConfigurationDirectives)
+
+    useConfigurators(::PowerAssertEnvironmentConfigurator)
+
+    useAdditionalSourceProviders(
+        ::AdditionalSourceFilesProvider,
+    )
+
+    enableJunit()
+}
+
+class PowerAssertEnvironmentConfigurator(testServices: TestServices) : EnvironmentConfigurator(testServices) {
+    override fun CompilerPluginRegistrar.ExtensionStorage.registerCompilerExtensions(
+        module: TestModule,
+        configuration: CompilerConfiguration,
+    ) {
+        val functions = moduleStructure.allDirectives[PowerAssertConfigurationDirectives.FUNCTION]
+            .ifEmpty { listOf("kotlin.assert") }
+            .mapTo(mutableSetOf()) { FqName(it) }
+
+        val messageCollector = configuration.get(CLIConfigurationKeys.MESSAGE_COLLECTOR_KEY, MessageCollector.NONE)
+        IrGenerationExtension.registerExtension(PowerAssertIrGenerationExtension(messageCollector, functions))
+    }
+}
+
+class AdditionalSourceFilesProvider(testServices: TestServices) : AdditionalSourceProvider(testServices) {
+    override val directiveContainers: List<DirectivesContainer> =
+        listOf(AdditionalFilesDirectives)
+
+    override fun produceAdditionalFiles(globalDirectives: RegisteredDirectives, module: TestModule): List<TestFile> {
+        return buildList {
+            add(File("plugins/power-assert/testData/helpers/InfixDispatch.kt").toTestFile())
+            add(File("plugins/power-assert/testData/helpers/InfixExtension.kt").toTestFile())
+            add(File("plugins/power-assert/testData/helpers/utils.kt").toTestFile())
+        }
+    }
+}

plugins/power-assert/testData/codegen/CheckCustomMessage.box.txt

@@ -0,0 +1,4 @@
+the world is broken
+check(1 == 2) { "the world is broken" }
+        |
+        false

plugins/power-assert/testData/codegen/CheckCustomMessage.kt

@@ -0,0 +1,5 @@
+// FUNCTION: kotlin.check
+
+fun box() = expectThrowableMessage {
+    check(1 == 2) { "the world is broken" }
+}

plugins/power-assert/testData/codegen/Constants.box.txt

@@ -0,0 +1 @@
+Assertion failed

plugins/power-assert/testData/codegen/Constants.kt

@@ -0,0 +1,5 @@
+// FUNCTION: kotlin.require
+
+fun box() = expectThrowableMessage {
+    assert(false)
+}

plugins/power-assert/testData/codegen/CustomLocalVariableMessage.box.txt

@@ -0,0 +1,4 @@
+Not equal
+assert(1 == 2, lambda)
+         |
+         false

plugins/power-assert/testData/codegen/CustomLocalVariableMessage.kt

@@ -0,0 +1,4 @@
+fun box() = expectThrowableMessage {
+    val lambda = { "Not equal" }
+    assert(1 == 2, lambda)
+}

plugins/power-assert/testData/codegen/CustomMessage.box.txt

@@ -0,0 +1,4 @@
+Not equal
+assert(1 == 2) { "Not equal" }
+         |
+         false

plugins/power-assert/testData/codegen/CustomMessage.kt

@@ -0,0 +1,3 @@
+fun box() = expectThrowableMessage {
+    assert(1 == 2) { "Not equal" }
+}

plugins/power-assert/testData/codegen/MemberFunctions.box.txt

@@ -0,0 +1,8 @@
+Assertion failed
+assert(hello.length == "World".substring(1, 4).length)
+       |     |      |          |               |
+       |     |      |          |               3
+       |     |      |          orl
+       |     |      false
+       |     5
+       Hello

plugins/power-assert/testData/codegen/MemberFunctions.kt

@@ -0,0 +1,4 @@
+fun box() = expectThrowableMessage {
+    val hello = "Hello"
+    assert(hello.length == "World".substring(1, 4).length)
+}

plugins/power-assert/testData/codegen/Multiline.box.txt

@@ -0,0 +1,22 @@
+Assertion failed
+assert(
+    text
+    |
+    Hello
+        == null ||
+        |
+        false
+        (
+            text.length == 5 &&
+            |    |      |
+            |    |      true
+            |    5
+            Hello
+                text.toLowerCase() == text
+                |    |             |  |
+                |    |             |  Hello
+                |    |             false
+                |    hello
+                Hello
+            )
+)

plugins/power-assert/testData/codegen/Multiline.kt

@@ -0,0 +1,11 @@
+fun box() = expectThrowableMessage {
+    val text: String? = "Hello"
+    assert(
+        text
+            == null ||
+            (
+                text.length == 5 &&
+                    text.toLowerCase() == text
+                )
+    )
+}

plugins/power-assert/testData/codegen/RequireCustomMessage.box.txt

@@ -0,0 +1,4 @@
+the world is broken
+require(1 == 2) { "the world is broken" }
+          |
+          false

plugins/power-assert/testData/codegen/RequireCustomMessage.kt

@@ -0,0 +1,5 @@
+// FUNCTION: kotlin.require
+
+fun box() = expectThrowableMessage {
+    require(1 == 2) { "the world is broken" }
+}

plugins/power-assert/testData/codegen/Transformations.box.txt

@@ -0,0 +1,7 @@
+Assertion failed
+assert(hello.reversed() == emptyList<String>())
+       |     |          |  |
+       |     |          |  []
+       |     |          false
+       |     [World, Hello]
+       [Hello, World]

plugins/power-assert/testData/codegen/Transformations.kt

@@ -0,0 +1,4 @@
+fun box() = expectThrowableMessage {
+    val hello = listOf("Hello", "World")
+    assert(hello.reversed() == emptyList<String>())
+}

plugins/power-assert/testData/codegen/arithmetic/InlineAddition.box.txt

@@ -0,0 +1,5 @@
+Assertion failed
+assert(1 + 1 == 4)
+         |   |
+         |   false
+         2

plugins/power-assert/testData/codegen/arithmetic/InlineAddition.kt

@@ -0,0 +1,3 @@
+fun box() = expectThrowableMessage {
+    assert(1 + 1 == 4)
+}

plugins/power-assert/testData/codegen/arithmetic/InlineDivision.box.txt

@@ -0,0 +1,5 @@
+Assertion failed
+assert(2 / 1 == 4)
+         |   |
+         |   false
+         2

plugins/power-assert/testData/codegen/arithmetic/InlineDivision.kt

@@ -0,0 +1,3 @@
+fun box() = expectThrowableMessage {
+    assert(2 / 1 == 4)
+}

plugins/power-assert/testData/codegen/arithmetic/InlineMultiplication.box.txt

@@ -0,0 +1,5 @@
+Assertion failed
+assert(1 * 2 == 4)
+         |   |
+         |   false
+         2

plugins/power-assert/testData/codegen/arithmetic/InlineMultiplication.kt

@@ -0,0 +1,3 @@
+fun box() = expectThrowableMessage {
+    assert(1 * 2 == 4)
+}

plugins/power-assert/testData/codegen/arithmetic/InlinePostfixDecrement.box.txt

@@ -0,0 +1,5 @@
+Assertion failed
+assert(i-- == 4)
+       |   |
+       |   false
+       3

plugins/power-assert/testData/codegen/arithmetic/InlinePostfixDecrement.kt

@@ -0,0 +1,4 @@
+fun box() = expectThrowableMessage {
+    var i = 3
+    assert(i-- == 4)
+}

plugins/power-assert/testData/codegen/arithmetic/InlinePostfixIncrement.box.txt

@@ -0,0 +1,5 @@
+Assertion failed
+assert(i++ == 4)
+       |   |
+       |   false
+       1

plugins/power-assert/testData/codegen/arithmetic/InlinePostfixIncrement.kt

@@ -0,0 +1,4 @@
+fun box() = expectThrowableMessage {
+    var i = 1
+    assert(i++ == 4)
+}

plugins/power-assert/testData/codegen/arithmetic/InlinePrefixDecrement.box.txt

@@ -0,0 +1,5 @@
+Assertion failed
+assert(--i == 4)
+       |   |
+       |   false
+       2

plugins/power-assert/testData/codegen/arithmetic/InlinePrefixDecrement.kt

@@ -0,0 +1,4 @@
+fun box() = expectThrowableMessage {
+    var i = 3
+    assert(--i == 4)
+}

plugins/power-assert/testData/codegen/arithmetic/InlinePrefixIncrement.box.txt

@@ -0,0 +1,5 @@
+Assertion failed
+assert(++i == 4)
+       |   |
+       |   false
+       2

plugins/power-assert/testData/codegen/arithmetic/InlinePrefixIncrement.kt

@@ -0,0 +1,4 @@
+fun box() = expectThrowableMessage {
+    var i = 1
+    assert(++i == 4)
+}

plugins/power-assert/testData/codegen/arithmetic/InlineSubtraction.box.txt

@@ -0,0 +1,5 @@
+Assertion failed
+assert(3 - 1 == 4)
+         |   |
+         |   false
+         2

plugins/power-assert/testData/codegen/arithmetic/InlineSubtraction.kt

@@ -0,0 +1,3 @@
+fun box() = expectThrowableMessage {
+    assert(3 - 1 == 4)
+}

plugins/power-assert/testData/codegen/boolean/BooleanAnd.box.txt

@@ -0,0 +1,12 @@
+Assertion failed
+assert(text != null && text.length == 5 && text.toLowerCase() == text)
+       |    |          |    |      |       |    |             |  |
+       |    |          |    |      |       |    |             |  Hello
+       |    |          |    |      |       |    |             false
+       |    |          |    |      |       |    hello
+       |    |          |    |      |       Hello
+       |    |          |    |      true
+       |    |          |    5
+       |    |          Hello
+       |    true
+       Hello

plugins/power-assert/testData/codegen/boolean/BooleanAnd.kt

@@ -0,0 +1,4 @@
+fun box() = expectThrowableMessage {
+    val text: String? = "Hello"
+    assert(text != null && text.length == 5 && text.toLowerCase() == text)
+}

plugins/power-assert/testData/codegen/boolean/BooleanMixWithAndFirst.box.txt

@@ -0,0 +1,12 @@
+Assertion failed
+assert(text != null && (text.length == 1 || text.toLowerCase() == text))
+       |    |           |    |      |       |    |             |  |
+       |    |           |    |      |       |    |             |  Hello
+       |    |           |    |      |       |    |             false
+       |    |           |    |      |       |    hello
+       |    |           |    |      |       Hello
+       |    |           |    |      false
+       |    |           |    5
+       |    |           Hello
+       |    true
+       Hello

plugins/power-assert/testData/codegen/boolean/BooleanMixWithAndFirst.kt

@@ -0,0 +1,4 @@
+fun box() = expectThrowableMessage {
+    val text: String? = "Hello"
+    assert(text != null && (text.length == 1 || text.toLowerCase() == text))
+}

plugins/power-assert/testData/codegen/boolean/BooleanMixWithAndLast.box.txt

@@ -0,0 +1,10 @@
+Assertion failed
+assert((text.length == 1 || text.toLowerCase() == text) && text.length == 1)
+        |    |      |       |    |             |  |
+        |    |      |       |    |             |  Hello
+        |    |      |       |    |             false
+        |    |      |       |    hello
+        |    |      |       Hello
+        |    |      false
+        |    5
+        Hello

plugins/power-assert/testData/codegen/boolean/BooleanMixWithAndLast.kt

@@ -0,0 +1,4 @@
+fun box() = expectThrowableMessage {
+    val text = "Hello"
+    assert((text.length == 1 || text.toLowerCase() == text) && text.length == 1)
+}

plugins/power-assert/testData/codegen/boolean/BooleanMixWithOrFirst.box.txt

@@ -0,0 +1,12 @@
+Assertion failed
+assert(text == null || (text.length == 5 && text.toLowerCase() == text))
+       |    |           |    |      |       |    |             |  |
+       |    |           |    |      |       |    |             |  Hello
+       |    |           |    |      |       |    |             false
+       |    |           |    |      |       |    hello
+       |    |           |    |      |       Hello
+       |    |           |    |      true
+       |    |           |    5
+       |    |           Hello
+       |    false
+       Hello

plugins/power-assert/testData/codegen/boolean/BooleanMixWithOrFirst.kt

@@ -0,0 +1,4 @@
+fun box() = expectThrowableMessage {
+    val text: String? = "Hello"
+    assert(text == null || (text.length == 5 && text.toLowerCase() == text))
+}

plugins/power-assert/testData/codegen/boolean/BooleanMixWithOrLast.box.txt

@@ -0,0 +1,13 @@
+Assertion failed
+assert((text.length == 5 && text.toLowerCase() == text) || text.length == 1)
+        |    |      |       |    |             |  |        |    |      |
+        |    |      |       |    |             |  |        |    |      false
+        |    |      |       |    |             |  |        |    5
+        |    |      |       |    |             |  |        Hello
+        |    |      |       |    |             |  Hello
+        |    |      |       |    |             false
+        |    |      |       |    hello
+        |    |      |       Hello
+        |    |      true
+        |    5
+        Hello

plugins/power-assert/testData/codegen/boolean/BooleanMixWithOrLast.kt

@@ -0,0 +1,4 @@
+fun box() = expectThrowableMessage {
+    val text = "Hello"
+    assert((text.length == 5 && text.toLowerCase() == text) || text.length == 1)
+}

plugins/power-assert/testData/codegen/boolean/BooleanOr.box.txt

@@ -0,0 +1,12 @@
+Assertion failed
+assert(text == null || text.length == 1 || text.toLowerCase() == text)
+       |    |          |    |      |       |    |             |  |
+       |    |          |    |      |       |    |             |  Hello
+       |    |          |    |      |       |    |             false
+       |    |          |    |      |       |    hello
+       |    |          |    |      |       Hello
+       |    |          |    |      false
+       |    |          |    5
+       |    |          Hello
+       |    false
+       Hello

plugins/power-assert/testData/codegen/boolean/BooleanOr.kt

@@ -0,0 +1,4 @@
+fun box() = expectThrowableMessage {
+    val text: String? = "Hello"
+    assert(text == null || text.length == 1 || text.toLowerCase() == text)
+}

plugins/power-assert/testData/codegen/boolean/BooleanShortCircuit.box.txt

@@ -0,0 +1,5 @@
+Assertion failed
+assert(text != null && text.length == 1)
+       |    |
+       |    false
+       null

plugins/power-assert/testData/codegen/boolean/BooleanShortCircuit.kt

@@ -0,0 +1,4 @@
+fun box() = expectThrowableMessage {
+    val text: String? = null
+    assert(text != null && text.length == 1)
+}

plugins/power-assert/testData/codegen/cast/InstanceEquals.box.txt

@@ -0,0 +1,4 @@
+Assertion failed
+assert(null is String)
+            |
+            false

plugins/power-assert/testData/codegen/cast/InstanceEquals.kt

@@ -0,0 +1,3 @@
+fun box() = expectThrowableMessage {
+    assert(null is String)
+}

plugins/power-assert/testData/codegen/cast/InstanceNotEquals.box.txt

@@ -0,0 +1,4 @@
+Assertion failed
+assert("Hello, world!" !is String)
+                       |
+                       false

plugins/power-assert/testData/codegen/cast/InstanceNotEquals.kt

@@ -0,0 +1,3 @@
+fun box() = expectThrowableMessage {
+    assert("Hello, world!" !is String)
+}

plugins/power-assert/testData/codegen/cast/SmartCast.box.txt

@@ -0,0 +1,8 @@
+Assertion failed
+assert(greeting is String && greeting.length == 2)
+       |        |            |        |      |
+       |        |            |        |      false
+       |        |            |        5
+       |        |            hello
+       |        true
+       hello

plugins/power-assert/testData/codegen/cast/SmartCast.kt

@@ -0,0 +1,4 @@
+fun box() = expectThrowableMessage {
+    val greeting: Any = "hello"
+    assert(greeting is String && greeting.length == 2)
+}

plugins/power-assert/testData/codegen/dbg/DebugEntryFunction.box.txt

@@ -0,0 +1,6 @@
+sum=6
+dbg(operation, 1 + 2 + 3)
+    |            |   |
+    |            |   6
+    |            3
+    sum

plugins/power-assert/testData/codegen/dbg/DebugEntryFunction.kt

@@ -0,0 +1,9 @@
+// FUNCTION: dbg
+
+fun box() = expectThrowableMessage {
+    val operation = "sum"
+    dbg(operation, 1 + 2 + 3)
+}
+
+fun <T> dbg(key: Any, value: T): T = value
+fun <T> dbg(key: Any, value: T, msg: String): T = throw RuntimeException(key.toString() + "=" + value + "\n" + msg)

plugins/power-assert/testData/codegen/dbg/DebugEntryFunctionComplexBooleans.box.txt

@@ -0,0 +1,11 @@
+false=true
+dbg(
+    key = greeting != null && greeting.length == 5,
+          |        |
+          |        false
+          null
+    value = name == null || name.length == 5
+            |    |
+            |    true
+            null
+)

plugins/power-assert/testData/codegen/dbg/DebugEntryFunctionComplexBooleans.kt

@@ -0,0 +1,13 @@
+// FUNCTION: dbg
+
+fun box() = expectThrowableMessage {
+    val greeting: String? = null
+    val name: String? = null
+    dbg(
+        key = greeting != null && greeting.length == 5,
+        value = name == null || name.length == 5
+    )
+}
+
+fun <T> dbg(key: Any, value: T): T = value
+fun <T> dbg(key: Any, value: T, msg: String): T = throw RuntimeException(key.toString() + "=" + value + "\n" + msg)

plugins/power-assert/testData/codegen/dbg/DebugEntryFunctionMessage.box.txt

@@ -0,0 +1,7 @@
+sum=6
+Message:
+dbg(operation, 1 + 2 + 3, "Message:")
+    |            |   |
+    |            |   6
+    |            3
+    sum

plugins/power-assert/testData/codegen/dbg/DebugEntryFunctionMessage.kt

@@ -0,0 +1,9 @@
+// FUNCTION: dbg
+
+fun box() = expectThrowableMessage {
+    val operation = "sum"
+    dbg(operation, 1 + 2 + 3, "Message:")
+}
+
+fun <T> dbg(key: Any, value: T): T = value
+fun <T> dbg(key: Any, value: T, msg: String): T = throw RuntimeException(key.toString() + "=" + value + "\n" + msg)

plugins/power-assert/testData/codegen/dbg/DebugEntryFunctionMessageComplexBooleans.box.txt

@@ -0,0 +1,13 @@
+false=true
+Message:
+dbg(
+    key = greeting != null && greeting.length == 5,
+          |        |
+          |        false
+          null
+    value = name == null || name.length == 5,
+            |    |
+            |    true
+            null
+    msg = "Message:"
+)

plugins/power-assert/testData/codegen/dbg/DebugEntryFunctionMessageComplexBooleans.kt

@@ -0,0 +1,14 @@
+// FUNCTION: dbg
+
+fun box() = expectThrowableMessage {
+    val greeting: String? = null
+    val name: String? = null
+    dbg(
+        key = greeting != null && greeting.length == 5,
+        value = name == null || name.length == 5,
+        msg = "Message:"
+    )
+}
+
+fun <T> dbg(key: Any, value: T): T = value
+fun <T> dbg(key: Any, value: T, msg: String): T = throw RuntimeException(key.toString() + "=" + value + "\n" + msg)

plugins/power-assert/testData/codegen/dbg/DebugFunction.box.txt

@@ -0,0 +1,4 @@
+dbg(1 + 2 + 3)
+      |   |
+      |   6
+      3

plugins/power-assert/testData/codegen/dbg/DebugFunction.kt

@@ -0,0 +1,8 @@
+// FUNCTION: dbg
+
+fun box() = expectThrowableMessage {
+    dbg(1 + 2 + 3)
+}
+
+fun <T> dbg(value: T): T = value
+fun <T> dbg(value: T, msg: String): T = throw RuntimeException(msg)

plugins/power-assert/testData/codegen/dbg/DebugFunctionMessage.box.txt

@@ -0,0 +1,5 @@
+Message:
+dbg(1 + 2 + 3, "Message:")
+      |   |
+      |   6
+      3

plugins/power-assert/testData/codegen/dbg/DebugFunctionMessage.kt

@@ -0,0 +1,8 @@
+// FUNCTION: dbg
+
+fun box() = expectThrowableMessage {
+    dbg(1 + 2 + 3, "Message:")
+}
+
+fun <T> dbg(value: T): T = value
+fun <T> dbg(value: T, msg: String): T = throw RuntimeException(msg)

plugins/power-assert/testData/codegen/infix/DispatchInfix.box.txt

@@ -0,0 +1,5 @@
+Wrapper(1 + 1) mustEqual (2 + 4)
+|         |                 |
+|         |                 6
+|         2
+Wrapper

plugins/power-assert/testData/codegen/infix/DispatchInfix.kt

@@ -0,0 +1,7 @@
+// FUNCTION: infix.dispatch.Wrapper.mustEqual
+
+import infix.dispatch.*
+
+fun box() = expectThrowableMessage {
+    Wrapper(1 + 1) mustEqual (2 + 4)
+}

plugins/power-assert/testData/codegen/infix/DispatchInfixConstantParameter.box.txt

@@ -0,0 +1,4 @@
+Wrapper(1 + 1) mustEqual 6
+|         |
+|         2
+Wrapper

plugins/power-assert/testData/codegen/infix/DispatchInfixConstantParameter.kt

@@ -0,0 +1,7 @@
+// FUNCTION: infix.dispatch.Wrapper.mustEqual
+
+import infix.dispatch.*
+
+fun box() = expectThrowableMessage {
+    Wrapper(1 + 1) mustEqual 6
+}

plugins/power-assert/testData/codegen/infix/DispatchInfixConstantReceiver.box.txt

@@ -0,0 +1,4 @@
+Wrapper(1) mustEqual (2 + 4)
+|                       |
+|                       6
+Wrapper

plugins/power-assert/testData/codegen/infix/DispatchInfixConstantReceiver.kt

@@ -0,0 +1,7 @@
+// FUNCTION: infix.dispatch.Wrapper.mustEqual
+
+import infix.dispatch.*
+
+fun box() = expectThrowableMessage {
+    Wrapper(1) mustEqual (2 + 4)
+}

plugins/power-assert/testData/codegen/infix/DispatchInfixOnlyConstants.box.txt

@@ -0,0 +1,3 @@
+Wrapper(2) mustEqual 6
+|
+Wrapper

plugins/power-assert/testData/codegen/infix/DispatchInfixOnlyConstants.kt

@@ -0,0 +1,7 @@
+// FUNCTION: infix.dispatch.Wrapper.mustEqual
+
+import infix.dispatch.*
+
+fun box() = expectThrowableMessage {
+    Wrapper(2) mustEqual 6
+}

plugins/power-assert/testData/codegen/infix/DispatchNonInfix.box.txt

@@ -0,0 +1,5 @@
+Wrapper(1 + 1).mustEqual(2 + 4)
+|         |                |
+|         |                6
+|         2
+Wrapper

plugins/power-assert/testData/codegen/infix/DispatchNonInfix.kt

@@ -0,0 +1,7 @@
+// FUNCTION: infix.dispatch.Wrapper.mustEqual
+
+import infix.dispatch.*
+
+fun box() = expectThrowableMessage {
+    Wrapper(1 + 1).mustEqual(2 + 4)
+}

plugins/power-assert/testData/codegen/infix/DispatchNonInfixConstantParameter.box.txt

@@ -0,0 +1,4 @@
+Wrapper(1 + 1).mustEqual(6)
+|         |
+|         2
+Wrapper

plugins/power-assert/testData/codegen/infix/DispatchNonInfixConstantParameter.kt

@@ -0,0 +1,7 @@
+// FUNCTION: infix.dispatch.Wrapper.mustEqual
+
+import infix.dispatch.*
+
+fun box() = expectThrowableMessage {
+    Wrapper(1 + 1).mustEqual(6)
+}

plugins/power-assert/testData/codegen/infix/DispatchNonInfixConstantReceiver.box.txt

@@ -0,0 +1,4 @@
+Wrapper(1).mustEqual(2 + 4)
+|                      |
+|                      6
+Wrapper

plugins/power-assert/testData/codegen/infix/DispatchNonInfixConstantReceiver.kt

@@ -0,0 +1,7 @@
+// FUNCTION: infix.dispatch.Wrapper.mustEqual
+
+import infix.dispatch.*
+
+fun box() = expectThrowableMessage {
+    Wrapper(1).mustEqual(2 + 4)
+}

plugins/power-assert/testData/codegen/infix/DispatchNonInfixOnlyConstants.box.txt

@@ -0,0 +1,3 @@
+Wrapper(2).mustEqual(6)
+|
+Wrapper

plugins/power-assert/testData/codegen/infix/DispatchNonInfixOnlyConstants.kt

@@ -0,0 +1,7 @@
+// FUNCTION: infix.dispatch.Wrapper.mustEqual
+
+import infix.dispatch.*
+
+fun box() = expectThrowableMessage {
+    Wrapper(2).mustEqual(6)
+}

plugins/power-assert/testData/codegen/infix/ExtensionInfix.box.txt

@@ -0,0 +1,4 @@
+(1 + 1) mustEqual (2 + 4)
+   |                 |
+   |                 6
+   2

plugins/power-assert/testData/codegen/infix/ExtensionInfix.kt

@@ -0,0 +1,7 @@
+// FUNCTION: infix.extension.mustEqual
+
+import infix.extension.*
+
+fun box() = expectThrowableMessage {
+    (1 + 1) mustEqual (2 + 4)
+}

plugins/power-assert/testData/codegen/infix/ExtensionInfixConstantParameter.box.txt

@@ -0,0 +1,3 @@
+(1 + 1) mustEqual 6
+   |
+   2

plugins/power-assert/testData/codegen/infix/ExtensionInfixConstantParameter.kt

@@ -0,0 +1,7 @@
+// FUNCTION: infix.extension.mustEqual
+
+import infix.extension.*
+
+fun box() = expectThrowableMessage {
+    (1 + 1) mustEqual 6
+}

plugins/power-assert/testData/codegen/infix/ExtensionInfixConstantReceiver.box.txt

@@ -0,0 +1,3 @@
+1 mustEqual (2 + 4)
+               |
+               6

plugins/power-assert/testData/codegen/infix/ExtensionInfixConstantReceiver.kt

@@ -0,0 +1,7 @@
+// FUNCTION: infix.extension.mustEqual
+
+import infix.extension.*
+
+fun box() = expectThrowableMessage {
+    1 mustEqual (2 + 4)
+}

plugins/power-assert/testData/codegen/infix/ExtensionInfixOnlyConstants.box.txt

@@ -0,0 +1 @@
+Assertion failed

plugins/power-assert/testData/codegen/infix/ExtensionInfixOnlyConstants.kt

@@ -0,0 +1,7 @@
+// FUNCTION: infix.extension.mustEqual
+
+import infix.extension.*
+
+fun box() = expectThrowableMessage {
+    2 mustEqual 6
+}

plugins/power-assert/testData/codegen/infix/ExtensionNonInfix.box.txt

@@ -0,0 +1,4 @@
+(1 + 1).mustEqual(2 + 4)
+   |                |
+   |                6
+   2

plugins/power-assert/testData/codegen/infix/ExtensionNonInfix.kt

@@ -0,0 +1,7 @@
+// FUNCTION: infix.extension.mustEqual
+
+import infix.extension.*
+
+fun box() = expectThrowableMessage {
+    (1 + 1).mustEqual(2 + 4)
+}

plugins/power-assert/testData/codegen/infix/ExtensionNonInfixConstantParameter.box.txt

@@ -0,0 +1,3 @@
+(1 + 1).mustEqual(6)
+   |
+   2

plugins/power-assert/testData/codegen/infix/ExtensionNonInfixConstantParameter.kt

@@ -0,0 +1,7 @@
+// FUNCTION: infix.extension.mustEqual
+
+import infix.extension.*
+
+fun box() = expectThrowableMessage {
+    (1 + 1).mustEqual(6)
+}

plugins/power-assert/testData/codegen/infix/ExtensionNonInfixConstantReceiver.box.txt

@@ -0,0 +1,3 @@
+1.mustEqual(2 + 4)
+              |
+              6