ktlint-ruleset-standard/src/test/kotlin/com/github/shyiko/ktlint/ruleset/standard/NoConsecutiveBlankLinesRuleTest.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 NoConsecutiveBlankLinesRuleTest {

    @Test
    fun testLintInDeclarations() {
        assertThat(NoConsecutiveBlankLinesRule().lint(
            """fun a() {

            }


            fun b() {

            }"""
        )).isEqualTo(listOf(
            LintError(5, 1, "no-consecutive-blank-lines", "Needless blank line(s)")))
    }

    @Test
    fun testLintInCode() {
        assertThat(NoConsecutiveBlankLinesRule().lint(
            """fun main() {
                fun a()
                fun b()


                fun c()
            }"""
        )).isEqualTo(listOf(
            LintError(5, 1, "no-consecutive-blank-lines", "Needless blank line(s)")))
    }

    @Test
    fun testLintAtTheEndOfFile() {
        assertThat(NoConsecutiveBlankLinesRule().lint(
            """
            fun main() {
            }


            """.trimIndent()
        )).isEqualTo(listOf(
            LintError(4, 1, "no-consecutive-blank-lines", "Needless blank line(s)")))
    }

    @Test
    fun testLintInString() {
        assertThat(NoConsecutiveBlankLinesRule().lint(
            "fun main() {println(\"\"\"\n\n\n\"\"\")}")).isEmpty()
    }

    @Test
    fun testFormatInDeclarations() {
        assertThat(NoConsecutiveBlankLinesRule().format(
            """
            fun a() {

            }


            fun b() {

            }
            """
        )).isEqualTo(
            """
            fun a() {

            }

            fun b() {

            }
            """
        )
    }

    @Test
    fun testFormatInCode() {
        assertThat(NoConsecutiveBlankLinesRule().format(
            """
            fun main() {
                fun a()
                fun b()


                fun c()

            }
            """
        )).isEqualTo(
            """
            fun main() {
                fun a()
                fun b()

                fun c()

            }
            """
        )
    }

    @Test
    fun testFormatAtTheEndOfFile() {
        assertThat(NoConsecutiveBlankLinesRule().format(
            """
            fun main() {
            }


            """,
            script = true
        )).isEqualTo(
            """
            fun main() {
            }

            """
        )
    }
}