package org.jetbrains.dokka.tests

import org.jetbrains.dokka.Content
import org.jetbrains.dokka.NodeKind
import org.junit.Assert.assertEquals
import org.junit.Assert.assertTrue
import org.junit.Test
import kotlin.test.assertNotNull

class FunctionTest {
    @Test fun function() {
        verifyModel("testdata/functions/function.kt") { model ->
            with(model.members.single().members.single()) {
                assertEquals("fn", name)
                assertEquals(NodeKind.Function, kind)
                assertEquals("Function fn", content.summary.toTestString())
                assertEquals("Unit", detail(NodeKind.Type).name)
                assertTrue(members.none())
                assertTrue(links.none())
            }
        }
    }

    @Test fun functionWithReceiver() {
        verifyModel("testdata/functions/functionWithReceiver.kt") { model ->
            with(model.members.single().members.single()) {
                assertEquals("kotlin.String", name)
                assertEquals(NodeKind.ExternalClass, kind)
                assertEquals(2, members.count())
                with(members[0]) {
                    assertEquals("fn", name)
                    assertEquals(NodeKind.Function, kind)
                    assertEquals("Function with receiver", content.summary.toTestString())
                    assertEquals("public", details.elementAt(0).name)
                    assertEquals("final", details.elementAt(1).name)
                    with(details.elementAt(3)) {
                        assertEquals("<this>", name)
                        assertEquals(NodeKind.Receiver, kind)
                        assertEquals(Content.Empty, content)
                        assertEquals("String", details.single().name)
                        assertTrue(members.none())
                        assertTrue(links.none())
                    }
                    assertEquals("Unit", details.elementAt(4).name)
                    assertTrue(members.none())
                    assertTrue(links.none())
                }
                with(members[1]) {
                    assertEquals("fn", name)
                    assertEquals(NodeKind.Function, kind)
                }
            }
        }
    }

    @Test fun genericFunction() {
        verifyModel("testdata/functions/genericFunction.kt") { model ->
            with(model.members.single().members.single()) {
                assertEquals("generic", name)
                assertEquals(NodeKind.Function, kind)
                assertEquals("generic function", content.summary.toTestString())

                assertEquals("private", details.elementAt(0).name)
                assertEquals("final", details.elementAt(1).name)
                with(details.elementAt(3)) {
                    assertEquals("T", name)
                    assertEquals(NodeKind.TypeParameter, kind)
                    assertEquals(Content.Empty, content)
                    assertTrue(details.none())
                    assertTrue(members.none())
                    assertTrue(links.none())
                }
                assertEquals("Unit", details.elementAt(4).name)

                assertTrue(members.none())
                assertTrue(links.none())
            }
        }
    }
    @Test fun genericFunctionWithConstraints() {
        verifyModel("testdata/functions/genericFunctionWithConstraints.kt") { model ->
            with(model.members.single().members.single()) {
                assertEquals("generic", name)
                assertEquals(NodeKind.Function, kind)
                assertEquals("generic function", content.summary.toTestString())

                val functionDetails = details
                assertEquals("public", functionDetails.elementAt(0).name)
                assertEquals("final", functionDetails.elementAt(1).name)
                with(functionDetails.elementAt(3)) {
                    assertEquals("T", name)
                    assertEquals(NodeKind.TypeParameter, kind)
                    assertEquals(Content.Empty, content)
                    with(details.single()) {
                        assertEquals("R", name)
                        assertEquals(NodeKind.UpperBound, kind)
                        assertEquals(Content.Empty, content)
                        assertTrue(details.none())
                        assertTrue(members.none())
                        assertTrue(links.singleOrNull() == functionDetails.elementAt(4))
                    }
                    assertTrue(members.none())
                    assertTrue(links.none())
                }
                with(functionDetails.elementAt(4)) {
                    assertEquals("R", name)
                    assertEquals(NodeKind.TypeParameter, kind)
                    assertEquals(Content.Empty, content)
                    assertTrue(members.none())
                    assertTrue(links.none())
                }
                assertEquals("Unit", functionDetails.elementAt(5).name)

                assertTrue(members.none())
                assertTrue(links.none())
            }
        }
    }

    @Test fun functionWithParams() {
        verifyModel("testdata/functions/functionWithParams.kt") { model ->
            with(model.members.single().members.single()) {
                assertEquals("function", name)
                assertEquals(NodeKind.Function, kind)
                assertEquals("Multiline", content.summary.toTestString())
                assertEquals("""Function
Documentation""", content.description.toTestString())

                assertEquals("public", details.elementAt(0).name)
                assertEquals("final", details.elementAt(1).name)
                with(details.elementAt(3)) {
                    assertEquals("x", name)
                    assertEquals(NodeKind.Parameter, kind)
                    assertEquals("parameter", content.summary.toTestString())
                    assertEquals("Int", detail(NodeKind.Type).name)
                    assertTrue(members.none())
                    assertTrue(links.none())
                }
                assertEquals("Unit", details.elementAt(4).name)
                assertTrue(members.none())
                assertTrue(links.none())
            }
        }
    }

    @Test fun annotatedFunction() {
        verifyPackageMember("testdata/functions/annotatedFunction.kt", withKotlinRuntime = true) { func ->
            assertEquals(1, func.annotations.count())
            with(func.annotations[0]) {
                assertEquals("Strictfp", name)
                assertEquals(Content.Empty, content)
                assertEquals(NodeKind.Annotation, kind)
            }
        }
    }

    @Test fun functionWithNotDocumentedAnnotation() {
        verifyPackageMember("testdata/functions/functionWithNotDocumentedAnnotation.kt") { func ->
            assertEquals(0, func.annotations.count())
        }
    }

    @Test fun inlineFunction() {
        verifyPackageMember("testdata/functions/inlineFunction.kt") { func ->
            val modifiers = func.details(NodeKind.Modifier).map { it.name }
            assertTrue("inline" in modifiers)
        }
    }

    @Test fun functionWithAnnotatedParam() {
        verifyModel("testdata/functions/functionWithAnnotatedParam.kt") { model ->
            with(model.members.single().members.single { it.name == "function" }) {
                with(details(NodeKind.Parameter).first()) {
                    assertEquals(1, annotations.count())
                    with(annotations[0]) {
                        assertEquals("Fancy", name)
                        assertEquals(Content.Empty, content)
                        assertEquals(NodeKind.Annotation, kind)
                    }
                }
            }
        }
    }

    @Test fun functionWithNoinlineParam() {
        verifyPackageMember("testdata/functions/functionWithNoinlineParam.kt") { func ->
            with(func.details(NodeKind.Parameter).first()) {
                val modifiers = details(NodeKind.Modifier).map { it.name }
                assertTrue("noinline" in modifiers)
            }
        }
    }

    @Test fun annotatedFunctionWithAnnotationParameters() {
        verifyModel("testdata/functions/annotatedFunctionWithAnnotationParameters.kt") { model ->
            with(model.members.single().members.single { it.name == "f" }) {
                assertEquals(1, annotations.count())
                with(annotations[0]) {
                    assertEquals("Fancy", name)
                    assertEquals(Content.Empty, content)
                    assertEquals(NodeKind.Annotation, kind)
                    assertEquals(1, details.count())
                    with(details[0]) {
                        assertEquals(NodeKind.Parameter, kind)
                        assertEquals(1, details.count())
                        with(details[0]) {
                            assertEquals(NodeKind.Value, kind)
                            assertEquals("1", name)
                        }
                    }
                }
            }
        }
    }

    @Test fun functionWithDefaultParameter() {
        verifyModel("testdata/functions/functionWithDefaultParameter.kt") { model ->
            with(model.members.single().members.single()) {
                with(details.elementAt(3)) {
                    val value = details(NodeKind.Value)
                    assertEquals(1, value.count())
                    with(value[0]) {
                        assertEquals("\"\"", name)
                    }
                }
            }
        }
    }

    @Test fun sinceKotlin() {
        verifyModel("testdata/functions/sinceKotlin.kt") { model ->
            with(model.members.single().members.single()) {
                assertEquals(listOf("Kotlin 1.1"), platforms)
            }
        }
    }

    // Test for b/159470920, to ensure that we correctly parse annotated function types without resolving 'ERROR CLASS'
    // types. Note that the actual annotation is not included in the type information, this is tracked in b/145517104.
    @Test fun functionWithAnnotatedLambdaParam() {
        verifyModel("testdata/functions/functionWithAnnotatedLambdaParam.kt") { model ->
            with(model.members.single().members.single { it.name == "function" }) {
                with(details(NodeKind.Parameter).first()) {
                    with(details(NodeKind.Type).first()) {
                        assertEquals("Function0", name)
                    }
                }
            }
        }
    }
}