tests.integration.test_minimum_tls_version

Complexity

Total Complexity

11

Size/Duplication

Total Lines	55
Duplicated Lines	0 %

Importance

Changes

0

8 Methods

Rating	Name	Duplication	Size	Complexity
A	TestMinimumTLSVersion._get_log_contents()	0	3	2
A	TestMinimumTLSVersionInvalid.test_minimum_tls_version_invalid()	0	4	1
A	TestMinimumTLSVersion._get_config_file_name()	0	21	3
A	TestMinimumTLSVersionValid.test_minimum_tls_version_valid()	0	3	1
A	TestMinimumTLSVersionValid._get_config_file_name()	0	2	1
A	TestMinimumTLSVersionInvalid._get_config_file_name()	0	2	1
A	TestMinimumTLSVersionNotSpecified._get_config_file_name()	0	2	1
A	TestMinimumTLSVersionNotSpecified.test_minimum_tls_version_not_specified()	0	3	1

from . import integ_test_base
import os

class TestMinimumTLSVersion(integ_test_base.IntegTestBase):
    def _get_log_contents(self):
        with open(self.log_file_path, 'r') as f:
            return f.read()

    def _get_config_file_name(self, tls_version: str) -> str:
        config_file = open(os.path.join(self.tmp_dir, "test.conf"), "w+")
        config_file.write(
            "[TabPy]\n"
            "TABPY_PORT = 9005\n"
            "TABPY_TRANSFER_PROTOCOL = https\n"
            "TABPY_CERTIFICATE_FILE = ./tests/integration/resources/2019_04_24_to_3018_08_25.crt\n"
            "TABPY_KEY_FILE = ./tests/integration/resources/2019_04_24_to_3018_08_25.key\n"
        )

        if tls_version is not None:
            config_file.write(f"TABPY_MINIMUM_TLS_VERSION = {tls_version}")

        pwd_file = self._get_pwd_file()
        if pwd_file is not None:
            pwd_file = os.path.abspath(pwd_file)
            config_file.write(f"TABPY_PWD_FILE = {pwd_file}\n")

        config_file.close()
        self.delete_config_file = True
        return config_file.name

class TestMinimumTLSVersionValid(TestMinimumTLSVersion):
    def _get_config_file_name(self) -> str:
        return super()._get_config_file_name("TLSv1_3")

    def test_minimum_tls_version_valid(self):
        log_contents = self._get_log_contents()
        self.assertIn("Setting minimum TLS version to TLSv1_3", log_contents)

class TestMinimumTLSVersionInvalid(TestMinimumTLSVersion):
    def _get_config_file_name(self) -> str:
        return super()._get_config_file_name("TLSv-1.3")

    def test_minimum_tls_version_invalid(self):
        log_contents = self._get_log_contents()
        self.assertIn("Unrecognized value for TABPY_MINIMUM_TLS_VERSION", log_contents)
        self.assertIn("Setting minimum TLS version to TLSv1_2", log_contents)

class TestMinimumTLSVersionNotSpecified(TestMinimumTLSVersion):
    def _get_config_file_name(self) -> str:
        return super()._get_config_file_name(None)

    def test_minimum_tls_version_not_specified(self):
        log_contents = self._get_log_contents()
        self.assertIn("Setting minimum TLS version to TLSv1_2", log_contents)