jetbrains-ultimate/tst/software/aws/toolkits/jetbrains/services/lambda/nodejs/NodeJsLambdaHandlerResolverTest.kt

// Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0

package software.aws.toolkits.jetbrains.services.lambda.nodejs

import org.junit.Rule
import org.junit.Test
import software.aws.toolkits.jetbrains.utils.rules.NodeJsCodeInsightTestFixtureRule
import software.aws.toolkits.jetbrains.utils.rules.addLambdaHandler
import software.aws.toolkits.jetbrains.utils.rules.addPackageJsonFile

class NodeJsLambdaHandlerResolverTest {
    @Rule
    @JvmField
    val projectRule = NodeJsCodeInsightTestFixtureRule()

    @Test
    fun determineHandler() {
        projectRule.fixture.addPackageJsonFile()
        val handlerElement = projectRule.fixture.addLambdaHandler()

        assertDetermineHandler(handlerElement, "app.lambdaHandler")
    }

    @Test
    fun determineHandlernoExportsReturnsNull() {
        projectRule.fixture.addPackageJsonFile()
        val handlerElement = projectRule.fixture.addLambdaHandler(
            fileContent = """
                lambdaHandler = async (event, context) => {
                    return "Hello World"
                }
            """.trimIndent()
        )

        assertDetermineHandler(handlerElement, null)
    }

    @Test
    fun determineHandleroneParameter() {
        projectRule.fixture.addPackageJsonFile()
        val handlerElement = projectRule.fixture.addLambdaHandler(
            fileContent = """
                exports.lambdaHandler = function(event) {
                    return "Hello World"
                }
            """.trimIndent()
        )

        assertDetermineHandler(handlerElement, "app.lambdaHandler")
    }

    @Test
    fun determineHandlerTooManyParametersReturnsNull() {
        projectRule.fixture.addPackageJsonFile()
        val handlerElement = projectRule.fixture.addLambdaHandler(
            fileContent = """
                exports.lambdaHandler = function(event, context, callback, foo) {
                    return "Hello World"
                }
            """.trimIndent()
        )

        assertDetermineHandler(handlerElement, null)
    }

    @Test
    fun determineHandleroneParameterForAsync() {
        projectRule.fixture.addPackageJsonFile()
        val handlerElement = projectRule.fixture.addLambdaHandler(
            fileContent = """
                exports.lambdaHandler = async (event) => {
                    return "Hello World"
                }
            """.trimIndent()
        )

        assertDetermineHandler(handlerElement, "app.lambdaHandler")
    }

    @Test
    fun determineHandlerThreeParametersForAsyncReturnsNull() {
        projectRule.fixture.addPackageJsonFile()
        val handlerElement = projectRule.fixture.addLambdaHandler(
            fileContent = """
                exports.lambdaHandler = async (event, context, callback) => {
                    return "Hello World"
                }
            """.trimIndent()
        )

        assertDetermineHandler(handlerElement, null)
    }

    @Test
    fun determineHandleres5Code() {
        projectRule.fixture.addPackageJsonFile()
        val handlerElement = projectRule.fixture.addLambdaHandler(
            fileContent = """
                exports.lambdaHandler = function(event, context, callback) {
                    return "Hello World"
                }
            """.trimIndent()
        )

        assertDetermineHandler(handlerElement, "app.lambdaHandler")
    }

    @Test
    fun determineHandlerinSubFolder() {
        projectRule.fixture.addPackageJsonFile()
        val handlerElement = projectRule.fixture.addLambdaHandler(
            subPath = "foo/bar",
            fileContent = """
                exports.lambdaHandler = async (event, context) => {
                    return "Hello World"
                }
            """.trimIndent()
        )

        assertDetermineHandler(handlerElement, "foo/bar/app.lambdaHandler")
    }

    @Test
    fun determineHandlerpackageJsonFolderAsSourceRoot() {
        projectRule.fixture.addPackageJsonFile("foo")
        val handlerElement = projectRule.fixture.addLambdaHandler(
            subPath = "foo/bar",
            fileContent = """
                exports.lambdaHandler = async (event, context) => {
                    return "Hello World"
                }
            """.trimIndent()
        )

        assertDetermineHandler(handlerElement, "bar/app.lambdaHandler")
    }

    @Test
    fun determineHandlernotAFunction() {
        projectRule.fixture.addPackageJsonFile()
        val handlerElement = projectRule.fixture.addLambdaHandler(
            subPath = "foo/bar",
            fileContent = """
                exports.lambdaHandler = "foo"
            """.trimIndent()
        )

        assertDetermineHandler(handlerElement, null)
    }

    @Test
    fun findPsiElementexportsAsync1Parameter() {
        val fileContent =
            """
            exports.lambdaHandler = async (event) => {
                return "Hello World";
            }
            """.trimIndent()

        projectRule.fixture.addFileToProject("app.js", fileContent)
        projectRule.fixture.addPackageJsonFile()
        assertFindPsiElements(projectRule, "app.lambdaHandler", true)
    }

    @Test
    fun findPsiElementexportsAsync2Parameters() {
        val fileContent =
            """
            exports.lambdaHandler = async (event, context) => {
                return "Hello World";
            }
            """.trimIndent()

        projectRule.fixture.addFileToProject("app.js", fileContent)
        projectRule.fixture.addPackageJsonFile()
        assertFindPsiElements(projectRule, "app.lambdaHandler", true)
    }

    @Test
    fun findPsiElementexportsAsync3Parameters() {
        val fileContent = """
            exports.lambdaHandler = async (event, context, callback) => {
                return "Hello World";
            }
        """.trimIndent()

        projectRule.fixture.addFileToProject("app.js", fileContent)
        projectRule.fixture.addPackageJsonFile()
        assertFindPsiElements(projectRule, "app.lambdaHandler", false)
    }

    @Test
    fun findPsiElementnoExports() {
        val fileContent =
            """
            var lambdaHandler = async (event, context) => {
                return "Hello World";
            }
            """.trimIndent()

        projectRule.fixture.addFileToProject("app.js", fileContent)
        projectRule.fixture.addPackageJsonFile()
        assertFindPsiElements(projectRule, "app.lambdaHandler", false)
    }

    @Test
    fun findPsiElementnotAnAssignment() {
        val fileContent =
            """
            async function lambdaHandler(event, context) {
                return "Hello World";
            }
            """.trimIndent()

        projectRule.fixture.addFileToProject("app.js", fileContent)
        projectRule.fixture.addPackageJsonFile()
        assertFindPsiElements(projectRule, "app.lambdaHandler", false)
    }

    @Test
    fun findPsiElementinSubFolderWithNoPackageJson() {
        val fileContent =
            """
            exports.lambdaHandler = async (event, context) => {
                return "Hello World";
            }
            """.trimIndent()

        projectRule.fixture.addFileToProject("foo/app.js", fileContent)
        assertFindPsiElements(projectRule, "foo/app.lambdaHandler", false)
    }

    @Test
    fun findPsiElementinSubFolderWithPackageJson() {
        val fileContent =
            """
            exports.lambdaHandler = async (event, context) => {
                return "Hello World";
            }
            """.trimIndent()

        projectRule.fixture.addFileToProject("foo/bar/app.js", fileContent)
        projectRule.fixture.addPackageJsonFile("foo")
        assertFindPsiElements(projectRule, "bar/app.lambdaHandler", true)
        assertFindPsiElements(projectRule, "foo/bar/app.lambdaHandler", false)
    }

    @Test
    fun findPsiElementinSubFolderButHandlerIsNotFullPath() {
        val fileContent =
            """
            exports.lambdaHandler = async (event, context) => {
                return "Hello World";
            }
            """.trimIndent()

        projectRule.fixture.addFileToProject("foo/app.js", fileContent)
        projectRule.fixture.addPackageJsonFile()
        assertFindPsiElements(projectRule, "app.lambdaHandler", false)
    }
}