feat(opensearch): Add new attribute to Domain model in the opensearch service and a test for that change

This commit is contained in:
MarioRgzLpz
2024-10-10 19:55:01 +02:00
parent 2bd9f73c1b
commit 04993f4e76
2 changed files with 10 additions and 1 deletions

View File

@@ -131,6 +131,9 @@ class OpenSearchService(AWSService):
domain.zone_awareness_enabled = describe_domain["DomainStatus"][
"ClusterConfig"
].get("ZoneAwarenessEnabled", False)
domain.dedicated_master_count = describe_domain["DomainStatus"][
"ClusterConfig"
].get("DedicatedMasterCount", None)
except Exception as error:
logger.error(
f"{regional_client.region} -- {error.__class__.__name__}[{error.__traceback__.tb_lineno}]: {error}"
@@ -173,5 +176,6 @@ class OpenSearchDomain(BaseModel):
version: str = None
instance_count: Optional[int]
zone_awareness_enabled: Optional[bool]
dedicated_master_count: Optional[int]
tags: Optional[list] = []
advanced_settings_enabled: bool = None

View File

@@ -136,7 +136,11 @@ class Test_OpenSearchService_Service:
domain = opensearch_client.create_domain(
DomainName=test_domain_name,
EncryptionAtRestOptions={"Enabled": True},
ClusterConfig={"InstanceCount": 1, "ZoneAwarenessEnabled": True},
ClusterConfig={
"InstanceCount": 1,
"ZoneAwarenessEnabled": True,
"DedicatedMasterCount": 1,
},
NodeToNodeEncryptionOptions={"Enabled": True},
AdvancedSecurityOptions={
"Enabled": True,
@@ -179,6 +183,7 @@ class Test_OpenSearchService_Service:
)
assert opensearch.opensearch_domains[domain_arn].instance_count == 1
assert opensearch.opensearch_domains[domain_arn].zone_awareness_enabled
assert opensearch.opensearch_domains[domain_arn].dedicated_master_count == 1
assert opensearch.opensearch_domains[domain_arn].tags == [
{"Key": "test", "Value": "test"},
]