feat(#3341): Add remove_from_bucket test for aws Trusted Advisor inte…

…gration
wazuh · Jan 9, 2023 · f5eef7a · f5eef7a
1 parent d3cd160
commit f5eef7a
Show file tree

Hide file tree

Showing 3 changed files with 65 additions and 1 deletion.
diff --git a/deps/wazuh_testing/wazuh_testing/modules/aws/constants.py b/deps/wazuh_testing/wazuh_testing/modules/aws/constants.py
@@ -24,4 +24,5 @@
 CLOUD_TRAIL_TYPE = 'cloudtrail'
 VPC_FLOW_TYPE = 'vpcflow'
 KMS_TYPE = 'kms'
+TRUSTED_ADVISOR_TYPE = 'trusted'
 CUSTOM_TYPE = 'custom'
diff --git a/deps/wazuh_testing/wazuh_testing/modules/aws/data_generator.py b/deps/wazuh_testing/wazuh_testing/modules/aws/data_generator.py
@@ -228,11 +228,65 @@ def get_data_sample(self) -> str:
  )
 
 
+class TrustedAdvisorDataGenerator(DataGenerator):
+ BASE_PATH = ''
+ BASE_FILE_NAME = f'firehose_trustedadvisor-1-'
+
+ def get_filename(self, *args, **kwargs) -> str:
+ """Return the filename in the Trusted Advisor format.
+
+ Example:
+ <prefix>/<year>/<month>/<day>
+
+ Returns:
+ str: Syntetic filename.
+ """
+ now = datetime.now()
+ path = f"{self.BASE_PATH}{now.strftime(cons.PATH_DATE_FORMAT)}/"
+ name = f"{self.BASE_FILE_NAME}{now.strftime(cons.FILENAME_DATE_FORMAT)}{cons.JSON_EXT}"
+
+ return f'{path}{name}'
+
+ def get_data_sample(self) -> str:
+ """Returns a sample of data according to the KMS format.
+
+ Returns:
+ str: Syntetic data.
+ """
+ return json.dumps(
+ {
+ 'version': '0',
+ 'id': get_random_string(26),
+ 'detail-type': 'Trusted Advisor Check Item Refresh Notification',
+ 'source': 'aws.trustedadvisor',
+ 'account': cons.RANDOM_ACCOUNT_ID,
+ 'time': datetime.now().strftime(cons.FILENAME_DATE_FORMAT),
+ 'region': 'us-east-1',
+ 'resources': [],
+ 'detail': {
+ 'check-name': 'IAM Group',
+ 'check-item-detail': {
+ 'Status': 'Green',
+ 'Current Usage': '1',
+ 'Limit Name': 'Groups',
+ 'Region': '-',
+ 'Service': 'IAM',
+ 'Limit Amount': '300'
+ },
+ 'status': 'OK',
+ 'resource_id': '',
+ 'uuid': str(uuid4())
+ }
+ }
+ )
+
+
 # Maps bucket type with corresponding data generator
 buckets_data_mapping = {
  cons.CLOUD_TRAIL_TYPE: CloudTrailDataGenerator,
  cons.VPC_FLOW_TYPE: VPCDataGenerator,
- cons.KMS_TYPE: KMSDataGenerator
+ cons.KMS_TYPE: KMSDataGenerator,
+ cons.TRUSTED_ADVISOR_TYPE: TrustedAdvisorDataGenerator,
 }
 
 

diff --git a/...ion/test_aws/data/test_cases/remove_from_bucket_test_module/cases_remove_from_bucket.yaml b/...ion/test_aws/data/test_cases/remove_from_bucket_test_module/cases_remove_from_bucket.yaml
@@ -6,3 +6,12 @@
  metadata:
  bucket_type: cloudtrail
  bucket_name: wazuh-cloudtrail-integration-tests
+
+- name: trusted_advisor_remove_from_bucket
+ description: Trusted Advisor remove from bucket configurations
+ configuration_parameters:
+ BUCKET_TYPE: custom
+ BUCKET_NAME: wazuh-trusted-advisor-integration-tests
+ metadata:
+ bucket_type: custom
+ bucket_name: wazuh-trusted-advisor-integration-tests