kotlin-fork/compiler/testData/diagnostics/testsWithStdLib/contracts/smartcasts/nullabilitySmartcastWhenNullability.fir.kt

// !LANGUAGE: +AllowContractsForCustomFunctions +UseReturnsEffect
// !USE_EXPERIMENTAL: kotlin.contracts.ExperimentalContracts
// !DIAGNOSTICS: -INVISIBLE_REFERENCE -INVISIBLE_MEMBER
// !WITH_NEW_INFERENCE

import kotlin.contracts.*

fun nullWhenNull(x: Int?): Int? {
    contract {
        returnsNotNull() implies (x != null)
    }
    return x?.inc()
}

fun testNullWhenNull(x: Int?) {
    if (nullWhenNull(x) == null) {
        x.dec()
    }
    else {
        x.<!AMBIGUITY!>dec<!>()
    }

    if (nullWhenNull(x) != null) {
        x.<!AMBIGUITY!>dec<!>()
    }
    else {
        x.dec()
    }

    x.<!AMBIGUITY!>dec<!>()
}

// NB. it is the same function as `nullWhenNull`, but annotations specifies other facet of the function behaviour
fun notNullWhenNotNull (x: Int?): Int? {
    contract {
        returns(null) implies (x == null)
    }
    return x?.inc()
}

fun testNotNullWhenNotNull (x: Int?) {
    if (notNullWhenNotNull(x) == null) {
        x == null
    }
    else {
        x.<!AMBIGUITY!>dec<!>()
    }

    if (notNullWhenNotNull(x) != null) {
        x.<!AMBIGUITY!>dec<!>()
    }
    else {
        x == null
    }

    x.<!AMBIGUITY!>dec<!>()
}