|
| 1 | +from unittest import TestCase |
| 2 | +from unittest.mock import Mock |
| 3 | + |
| 4 | +from samtranslator.model.api.api_generator import ApiGenerator |
| 5 | + |
| 6 | + |
| 7 | +class TestApiGeneratorSecurityPolicy(TestCase): |
| 8 | + def setUp(self): |
| 9 | + self.logical_id = "MyApi" |
| 10 | + self.default_args = { |
| 11 | + "logical_id": self.logical_id, |
| 12 | + "cache_cluster_enabled": None, |
| 13 | + "cache_cluster_size": None, |
| 14 | + "variables": None, |
| 15 | + "depends_on": None, |
| 16 | + "definition_body": {"swagger": "2.0"}, |
| 17 | + "definition_uri": None, |
| 18 | + "name": None, |
| 19 | + "stage_name": "Prod", |
| 20 | + "shared_api_usage_plan": Mock(), |
| 21 | + "template_conditions": Mock(), |
| 22 | + "method_settings": None, |
| 23 | + "endpoint_configuration": {"Type": "REGIONAL"}, |
| 24 | + "access_log_setting": None, |
| 25 | + "canary_setting": None, |
| 26 | + "tracing_enabled": None, |
| 27 | + "open_api_version": None, |
| 28 | + "always_deploy": None, |
| 29 | + } |
| 30 | + |
| 31 | + def test_security_policy_tls_1_3(self): |
| 32 | + api_generator = ApiGenerator(**self.default_args, security_policy="SecurityPolicy_TLS13_1_3_2025_09") |
| 33 | + |
| 34 | + rest_api = api_generator._construct_rest_api() |
| 35 | + |
| 36 | + self.assertEqual(rest_api.SecurityPolicy, "SecurityPolicy_TLS13_1_3_2025_09") |
| 37 | + |
| 38 | + def test_no_security_policy(self): |
| 39 | + api_generator = ApiGenerator(**self.default_args, security_policy=None) |
| 40 | + |
| 41 | + rest_api = api_generator._construct_rest_api() |
| 42 | + |
| 43 | + self.assertIsNone(rest_api.SecurityPolicy) |
0 commit comments