class AnnotationValidationTest {
    @Test
    fun `classes with 'RestController' annotation should have 'Controller' suffix`() {
        Konsist
            .scopeFromProject()
            .classes()
            .withAnnotationOf(RestController::class)
            .assert { it.hasNameEndingWithOrNone("Controller") }
    }

    @Test
    fun `classes with 'Service' annotation should have 'Service' suffix`() {
        Konsist
            .scopeFromProject()
            .classes()
            .withAnnotationOf(Service::class)
            .assert { it.hasNameEndingWithOrNone("Service") }
    }

    @Test
    fun `classes with 'Repository' annotation should have 'Repository' suffix`() {
        Konsist
            .scopeFromProject()
            .classes()
            .withAnnotationOf(Repository::class)
            .assert { it.hasNameEndingWithOrNone("Repository") }
    }

    private fun KoClassDeclaration.hasNameEndingWithOrNone(vararg names: String): Boolean {
        if (this.name.isEmpty()) {
            return true
        }

        return names.all { name ->
            LocationUtil.resideInLocation("..$name", name)
        }
    }
}