test_script_file_name.py

# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
import os

from tests.utils import testutils
from azure_functions_worker.constants import \
    PYTHON_SCRIPT_FILE_NAME, PYTHON_SCRIPT_FILE_NAME_DEFAULT

DEFAULT_SCRIPT_FILE_NAME_DIR = testutils.UNIT_TESTS_FOLDER / \
    'file_name_functions' / \
    'default_file_name'

NEW_SCRIPT_FILE_NAME_DIR = testutils.UNIT_TESTS_FOLDER / \
    'file_name_functions' / \
    'new_file_name'

INVALID_SCRIPT_FILE_NAME_DIR = testutils.UNIT_TESTS_FOLDER / \
    'file_name_functions' / \
    'invalid_file_name'


class TestDefaultScriptFileName(testutils.WebHostTestCase):
    """
    Tests for default file name
    """

    @classmethod
    def setUpClass(cls):
        os.environ["PYTHON_SCRIPT_FILE_NAME"] = "function_app.py"
        super().setUpClass()

    @classmethod
    def tearDownClass(cls):
        # Remove the PYTHON_SCRIPT_FILE_NAME environment variable
        os.environ.pop('PYTHON_SCRIPT_FILE_NAME')
        super().tearDownClass()

    @classmethod
    def get_script_dir(cls):
        return DEFAULT_SCRIPT_FILE_NAME_DIR

    def test_default_file_name(self):
        """
        Test the default file name
        """
        self.assertIsNotNone(os.environ.get(PYTHON_SCRIPT_FILE_NAME))
        self.assertEqual(os.environ.get(PYTHON_SCRIPT_FILE_NAME),
                         PYTHON_SCRIPT_FILE_NAME_DEFAULT)


class TestNewScriptFileName(testutils.WebHostTestCase):
    """
    Tests for changed file name
    """

    @classmethod
    def setUpClass(cls):
        os.environ["PYTHON_SCRIPT_FILE_NAME"] = "test.py"
        super().setUpClass()

    @classmethod
    def tearDownClass(cls):
        # Remove the PYTHON_SCRIPT_FILE_NAME environment variable
        os.environ.pop('PYTHON_SCRIPT_FILE_NAME')
        super().tearDownClass()

    @classmethod
    def get_script_dir(cls):
        return NEW_SCRIPT_FILE_NAME_DIR

    def test_new_file_name(self):
        """
        Test the new file name
        """
        self.assertIsNotNone(os.environ.get(PYTHON_SCRIPT_FILE_NAME))
        self.assertEqual(os.environ.get(PYTHON_SCRIPT_FILE_NAME),
                         'test.py')


class TestInvalidScriptFileName(testutils.WebHostTestCase):
    """
    Tests for invalid file name
    """

    @classmethod
    def setUpClass(cls):
        os.environ["PYTHON_SCRIPT_FILE_NAME"] = "main"
        super().setUpClass()

    @classmethod
    def tearDownClass(cls):
        # Remove the PYTHON_SCRIPT_FILE_NAME environment variable
        os.environ.pop('PYTHON_SCRIPT_FILE_NAME')
        super().tearDownClass()

    @classmethod
    def get_script_dir(cls):
        return INVALID_SCRIPT_FILE_NAME_DIR

    def test_invalid_file_name(self):
        """
        Test the invalid file name
        """
        self.assertIsNotNone(os.environ.get(PYTHON_SCRIPT_FILE_NAME))
        self.assertEqual(os.environ.get(PYTHON_SCRIPT_FILE_NAME),
                         'main')