fix(#3333): Improve some test findings

wazuh · Apr 26, 2023 · 582d62e · 582d62e
1 parent 837628b
commit 582d62e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/...ntegration/test_aws/data/test_cases/only_logs_after_test_module/cases_multiple_calls.yaml b/...ntegration/test_aws/data/test_cases/only_logs_after_test_module/cases_multiple_calls.yaml
@@ -55,20 +55,26 @@
  metadata:
  bucket_type: custom
  bucket_name: wazuh-macie-integration-tests
+ expected_skipped_logs_step_3: 3
+ expected_skipped_logs_step_4: 3
 
 - name: trusted_advisor_only_logs_after_multiple_calls
  description: Trusted Advisor only_logs_after multiple calls configurations
  configuration_parameters:
  metadata:
  bucket_type: custom
  bucket_name: wazuh-trusted-advisor-integration-tests
+ expected_skipped_logs_step_3: 3
+ expected_skipped_logs_step_4: 3
 
 - name: guardduty_only_logs_after_multiple_calls
  description: GuardDuty only_logs_after multiple calls configurations
  configuration_parameters:
  metadata:
  bucket_type: guardduty
  bucket_name: wazuh-guardduty-integration-tests
+ expected_skipped_logs_step_3: 3
+ expected_skipped_logs_step_4: 3
 
 - name: native_guardduty_only_logs_after_multiple_calls
  description: Native GuardDuty only_logs_after multiple calls configurations

diff --git a/tests/integration/test_aws/test_only_logs_after.py b/tests/integration/test_aws/test_only_logs_after.py
@@ -325,7 +325,7 @@ def test_multiple_calls(
 
  # Call the module without only_logs_after and check that no logs were processed
  last_marker_key = datetime.utcnow().strftime(cons.PATH_DATE_FORMAT)
- if bucket_type == cons.CUSTOM_TYPE:
+ if bucket_type == cons.CUSTOM_TYPE or (bucket_type == cons.GUARD_DUTY_TYPE and 'native' not in bucket_name):
  event_monitor.check_marker_from_output(
  command_output=call_aws_module(*base_parameters),
  file_key=last_marker_key