serverless-application-model/tests/model/api/test_api_generator_security_policy.py at ec76244be5ce0c84df368e83d5b3674419750dcd · iph/serverless-application-model · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
from unittest import TestCase
from unittest.mock import Mock

from samtranslator.model.api.api_generator import ApiGenerator


class TestApiGeneratorSecurityPolicy(TestCase):
    def setUp(self):
        self.logical_id = "MyApi"
        self.default_args = {
            "logical_id": self.logical_id,
            "cache_cluster_enabled": None,
            "cache_cluster_size": None,
            "variables": None,
            "depends_on": None,
            "definition_body": {"swagger": "2.0"},
            "definition_uri": None,
            "name": None,
            "stage_name": "Prod",
            "shared_api_usage_plan": Mock(),
            "template_conditions": Mock(),
            "method_settings": None,
            "endpoint_configuration": {"Type": "REGIONAL"},
            "access_log_setting": None,
            "canary_setting": None,
            "tracing_enabled": None,
            "open_api_version": None,
            "always_deploy": None,
        }

    def test_security_policy_tls_1_3(self):
        api_generator = ApiGenerator(**self.default_args, security_policy="SecurityPolicy_TLS13_1_3_2025_09")

        rest_api = api_generator._construct_rest_api()

        self.assertEqual(rest_api.SecurityPolicy, "SecurityPolicy_TLS13_1_3_2025_09")

    def test_no_security_policy(self):
        api_generator = ApiGenerator(**self.default_args, security_policy=None)

        rest_api = api_generator._construct_rest_api()

        self.assertIsNone(rest_api.SecurityPolicy)