# coding=utf-8

import os
import tempfile

from subprocess import Popen, PIPE, TimeoutExpired
from unittest import skipIf

from parameterized import parameterized, param
import pytest

from tests.integration.local.invoke.invoke_integ_base import InvokeIntegBase
from pathlib import Path

from tests.testing_utils import SKIP_DOCKER_TESTS, SKIP_DOCKER_MESSAGE

TIMEOUT = 300


@skipIf(SKIP_DOCKER_TESTS, SKIP_DOCKER_MESSAGE)
class TestWithDifferentLambdaRuntimeZips(InvokeIntegBase):
    template = Path("runtimes", "template.yaml")

    def setUp(self):
        # Don't delete on close. Need the file to be present for tests to run.
        events_file = tempfile.NamedTemporaryFile(delete=False)
        events_file.write(b'"yolo"')  # Just empty event
        events_file.flush()
        events_file.close()

        self.events_file_path = events_file.name

    def tearDown(self):
        os.remove(self.events_file_path)

    @parameterized.expand([param("Go1xFunction"), param("Java8Function")])
    @pytest.mark.timeout(timeout=300, method="thread")
    def test_runtime_zip(self, function_name):
        command_list = InvokeIntegBase.get_command_list(
            function_name, template_path=self.template_path, event_path=self.events_file_path
        )

        process = Popen(command_list, stdout=PIPE)
        try:
            stdout, _ = process.communicate(timeout=TIMEOUT)
        except TimeoutExpired:
            process.kill()
            raise

        self.assertEqual(process.returncode, 0)
        process_stdout = stdout.strip()
        self.assertEqual(process_stdout.decode("utf-8"), '"Hello World"')

    @pytest.mark.timeout(timeout=300, method="thread")
    def test_custom_provided_runtime(self):
        command_list = InvokeIntegBase.get_command_list(
            "CustomBashFunction", template_path=self.template_path, event_path=self.events_file_path
        )

        command_list = command_list + ["--skip-pull-image"]

        process = Popen(command_list, stdout=PIPE)
        try:
            stdout, _ = process.communicate(timeout=TIMEOUT)
        except TimeoutExpired:
            process.kill()
            raise

        self.assertEqual(process.returncode, 0)
        process_stdout = stdout.strip()
        self.assertEqual(
            process_stdout.decode("utf-8"), '{"body": "hello 曰有冥 world 🐿", "statusCode": 200, "headers": {}}'
        )