fix(Enum): handle Enum classes correctly (#2702)

This commit is contained in:
Sergio Garcia
2023-08-10 13:21:24 +02:00
committed by GitHub
parent 3aba71ad2f
commit 552024c53e
3 changed files with 12 additions and 8 deletions

View File

@@ -33,6 +33,6 @@ class codeartifact_packages_external_public_publishing_disabled(Check):
report.status = "PASS" report.status = "PASS"
report.status_extended = f"Internal package {package.name} is not vulnerable to dependency confusion in repository {repository.arn}" report.status_extended = f"Internal package {package.name} is not vulnerable to dependency confusion in repository {repository.arn}"
findings.append(report) findings.append(report)
return findings return findings

View File

@@ -38,15 +38,17 @@ class DirectoryService(AWSService):
): ):
directory_id = directory["DirectoryId"] directory_id = directory["DirectoryId"]
directory_name = directory["Name"] directory_name = directory["Name"]
directory_type = directory["Type"] directory_type = DirectoryType(directory["Type"])
# Radius Configuration # Radius Configuration
radius_authentication_protocol = ( radius_authentication_protocol = (
directory["RadiusSettings"]["AuthenticationProtocol"] AuthenticationProtocol(
directory["RadiusSettings"]["AuthenticationProtocol"]
)
if "RadiusSettings" in directory if "RadiusSettings" in directory
else None else None
) )
radius_status = ( radius_status = (
directory["RadiusStatus"] RadiusStatus(directory["RadiusStatus"])
if "RadiusStatus" in directory if "RadiusStatus" in directory
else None else None
) )
@@ -114,7 +116,7 @@ class DirectoryService(AWSService):
EventTopics( EventTopics(
topic_arn=event_topic["TopicArn"], topic_arn=event_topic["TopicArn"],
topic_name=event_topic["TopicName"], topic_name=event_topic["TopicName"],
status=event_topic["Status"], status=EventTopicStatus(event_topic["Status"]),
created_date_time=event_topic["CreatedDateTime"], created_date_time=event_topic["CreatedDateTime"],
) )
) )
@@ -147,11 +149,13 @@ class DirectoryService(AWSService):
Certificate( Certificate(
id=certificate_info["CertificateId"], id=certificate_info["CertificateId"],
common_name=certificate_info["CommonName"], common_name=certificate_info["CommonName"],
state=certificate_info["State"], state=CertificateState(
certificate_info["State"]
),
expiry_date_time=certificate_info[ expiry_date_time=certificate_info[
"ExpiryDateTime" "ExpiryDateTime"
], ],
type=certificate_info["Type"], type=CertificateType(certificate_info["Type"]),
) )
) )
self.directories[directory.id].certificates = certificates self.directories[directory.id].certificates = certificates

View File

@@ -86,6 +86,6 @@ class emr_cluster_publicly_accesible(Check):
else "" else ""
) )
findings.append(report) findings.append(report)
return findings return findings