Files
prowler/tests/lib/outputs/fixtures/fixtures.py
T

55 lines
1.8 KiB
Python

from datetime import datetime
from prowler.config.config import prowler_version
from prowler.lib.outputs.common_models import FindingOutput
from tests.providers.aws.utils import AWS_ACCOUNT_NUMBER
# TODO: customize it per provider
def generate_finding_output(status, severity, muted, region) -> FindingOutput:
# TODO: Include metadata from a valid file
return FindingOutput(
auth_method="profile: default",
timestamp=datetime.now(),
account_uid=AWS_ACCOUNT_NUMBER,
account_name=AWS_ACCOUNT_NUMBER,
account_email="",
account_organization_uid="test-organization-id",
account_organization_name="test-organization",
account_tags=["test-tag:test-value"],
finding_uid="test-unique-finding",
provider="aws",
check_id="test-check-id",
check_title="test-check-id",
check_type="test-type",
status=status,
status_extended="status extended",
muted=muted,
service_name="test-service",
subservice_name="",
severity=severity,
resource_type="test-resource",
resource_uid="resource-id",
resource_name="resource_name",
resource_details="resource_details",
resource_tags="",
partition="aws",
region=region,
description="check description",
risk="test-risk",
related_url="test-url",
remediation_recommendation_text="",
remediation_recommendation_url="",
remediation_code_nativeiac="",
remediation_code_terraform="",
remediation_code_cli="",
remediation_code_other="",
compliance={"test-compliance": "test-compliance"},
categories="test-category",
depends_on="test-dependency",
related_to="test-related-to",
notes="test-notes",
prowler_version=prowler_version,
)