ktlint-ruleset-standard/src/test/kotlin/com/github/shyiko/ktlint/ruleset/standard/SpacingAroundKeywordRuleTest.kt

package com.github.shyiko.ktlint.ruleset.standard

import com.github.shyiko.ktlint.core.LintError
import com.github.shyiko.ktlint.test.format
import com.github.shyiko.ktlint.test.lint
import org.assertj.core.api.Assertions.assertThat
import org.testng.annotations.Test

class SpacingAroundKeywordRuleTest {

    @Test
    fun testLint() {
        assertThat(SpacingAroundKeywordRule().lint(
            """
            fun main() {
                if(true) {}
                while(true) {}
                do {} while(true)

                if (true) {}
                else {}

                try { "".trim() }
                catch (e: Exception) {}
                finally {}

                if (true) {
                } else {}
                if (true) {} else {}
                if (true)
                    println()
                else
                    entry.value

                try {
                    "".trim()
                } catch (e: Exception) {
                } finally {}
                try { "".trim() } catch (e: Exception) {} finally {}

                val v = 0
                when (v) {
                    1 -> println("")
                    else -> println("")
                }

                if (V)
                    V.let { "" }
                else
                    V
            }
            """.trimIndent()
        )).isEqualTo(listOf(
            LintError(2, 7, "keyword-spacing", "Missing spacing after \"if\""),
            LintError(3, 10, "keyword-spacing", "Missing spacing after \"while\""),
            LintError(4, 16, "keyword-spacing", "Missing spacing after \"while\""),
            LintError(7, 5, "keyword-spacing", "Unexpected newline before \"else\""),
            LintError(10, 5, "keyword-spacing", "Unexpected newline before \"catch\""),
            LintError(11, 5, "keyword-spacing", "Unexpected newline before \"finally\"")
        ))
    }

    @Test
    fun testFormat() {
        assertThat(SpacingAroundKeywordRule().format(
            """
            fun main() {
                if(true) {}
                if (true) {}
                while(true) {}
                do {} while(true)

                if (true) {}
                else {}
                if (true) {
                }
                else {}

                try { "".trim() }
                catch (e: Exception) {}
                finally {}
            }
            """.trimIndent()
        )).isEqualTo(
            """
            fun main() {
                if (true) {}
                if (true) {}
                while (true) {}
                do {} while (true)

                if (true) {} else {}
                if (true) {
                } else {}

                try { "".trim() } catch (e: Exception) {} finally {}
            }
            """.trimIndent()
        )
    }

    @Test
    fun getterAndSetterFunction() {
        assertThat(SpacingAroundKeywordRule().format(
            """
            var x: String
			    get () {
				    return ""
			    }
			    private set (value) {
				    x = value
			    }
            """.trimIndent()
        )).isEqualTo(
            """
            var x: String
			    get() {
				    return ""
			    }
			    private set(value) {
				    x = value
			    }
            """.trimIndent()
        )
    }

    @Test
    fun visibilityOrInjectProperty() {
        assertThat(SpacingAroundKeywordRule().lint(
            """
        var setterVisibility: String = "abc"
            private set
        var setterWithAnnotation: Any? = null
            @Inject set
        var setterOnNextLine: String
            private set
            (value) { setterOnNextLine = value}
            """
        )).isEqualTo(listOf(
            LintError(7, 21, "keyword-spacing", "Unexpected spacing after \"set\"")
        ))
    }
}