test_custom_make.py

import os
import shutil
import tempfile
from unittest import TestCase

from aws_lambda_builders.builder import LambdaBuilder
from aws_lambda_builders.exceptions import WorkflowFailedError


class TestCustomMakeWorkflow(TestCase):
    """
    Verifies that `custom_make` workflow works by building a Lambda that requires Numpy
    """

    MAKEFILE_DIRECTORY = "makefile-root"
    TEST_DATA_FOLDER = os.path.join(os.path.dirname(__file__), "testdata", MAKEFILE_DIRECTORY)

    def setUp(self):
        self.source_dir = self.TEST_DATA_FOLDER
        self.artifacts_dir = tempfile.mkdtemp()
        self.scratch_dir = tempfile.mkdtemp()

        self.manifest_path_valid = os.path.join(self.TEST_DATA_FOLDER, "Makefile")

        self.test_data_files = {"__init__.py", "main.py", "requirements-requests.txt"}

        self.builder = LambdaBuilder(language="provided", dependency_manager=None, application_framework=None)
        self.runtime = "provided"

    def tearDown(self):
        shutil.rmtree(self.artifacts_dir)
        shutil.rmtree(self.scratch_dir)

    def test_must_build_python_project_through_makefile(self):
        self.builder.build(
            self.source_dir,
            self.artifacts_dir,
            self.scratch_dir,
            self.manifest_path_valid,
            runtime=self.runtime,
            options={"build_logical_id": "HelloWorldFunction"},
        )
        dependencies_installed = {
            "chardet",
            "urllib3",
            "idna",
            "urllib3-1.25.9.dist-info",
            "chardet-3.0.4.dist-info",
            "certifi-2020.4.5.2.dist-info",
            "certifi",
            "idna-2.9.dist-info",
            "requests",
            "requests-2.23.0.dist-info",
        }

        expected_files = self.test_data_files.union(dependencies_installed)
        output_files = set(os.listdir(self.artifacts_dir))
        self.assertEquals(expected_files, output_files)

    def test_must_build_python_project_through_makefile_unknown_target(self):
        with self.assertRaises(WorkflowFailedError):
            self.builder.build(
                self.source_dir,
                self.artifacts_dir,
                self.scratch_dir,
                self.manifest_path_valid,
                runtime=self.runtime,
                options={"build_logical_id": "HelloWorldFunction2"},
            )