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

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

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

class FilenameRuleTest {

    @Test
    fun testMatchingSingleClassName() {
        for (src in listOf(
            "class A",
            "data class A(val v: Int)",
            "sealed class A",
            "interface A",
            "object A",
            "enum class A {A}",
            "typealias A = Set<Network.Node>",
            // >1 declaration case
            "class B\nfun A.f() {}"
        )) {
            assertThat(FilenameRule().lint(
                """
                /*
                 * license
                 */
                @file:JvmName("Foo")
                package x
                import y.Z
                $src
                //
                """.trimIndent(),
                fileName("/some/path/A.kt")
            )).isEmpty()
        }
    }

    @Test
    fun testNonMatchingSingleClassName() {
        for (src in mapOf(
            "class A" to "class",
            "data class A(val v: Int)" to "class",
            "sealed class A" to "class",
            "interface A" to "interface",
            "object A" to "object",
            "enum class A {A}" to "class",
            "typealias A = Set<Network.Node>" to "typealias"
        )) {
            assertThat(FilenameRule().lint(
                """
                /*
                 * license
                 */
                @file:JvmName("Foo")
                package x
                import y.Z
                ${src.key}
                //
                """.trimIndent(),
                fileName("/some/path/B.kt")
            )).isEqualTo(listOf(
                LintError(1, 1, "filename", "${src.value} A should be declared in a file named A.kt")
            ))
        }
    }

    @Test
    fun testFileWithoutTopLevelDeclarations() {
        assertThat(FilenameRule().lint(
            """
            /*
             * copyright
             */
            """.trimIndent(),
            fileName("A.kt")
        )).isEmpty()
    }

    @Test
    fun testMultipleTopLevelClasses() {
        assertThat(FilenameRule().lint(
            """
            class B
            class C
            """.trimIndent(),
            fileName("A.kt")
        )).isEmpty()
    }

    @Test
    fun testMultipleNonTopLevelClasses() {
        assertThat(FilenameRule().lint(
            """
            class B {
                class C
                class D
            }
            """.trimIndent(),
            fileName("A.kt")
        )).isEqualTo(listOf(
            LintError(1, 1, "filename", "class B should be declared in a file named B.kt")
        ))
    }

    @Test
    fun testCaseSensitiveMatching() {
        assertThat(FilenameRule().lint(
            """
            interface Woohoo
            """.trimIndent(),
            fileName("woohoo.kt")
        )).isEqualTo(listOf(
            LintError(1, 1, "filename", "interface Woohoo should be declared in a file named Woohoo.kt")
        ))
    }

    @Test
    fun testIgnoreKotlinScriptFiles() {
        assertThat(FilenameRule().lint(
            """
            class B
            """.trimIndent(),
            fileName("A.kts")
        )).isEmpty()
    }

    private fun fileName(fileName: String) = mapOf("file_path" to fileName)
}