From a55cdf8d08a8c4576a47ecb61a557cc955064535 Mon Sep 17 00:00:00 2001 From: Pepe Fagoaga Date: Wed, 12 Mar 2025 22:09:05 +0545 Subject: [PATCH] Revert "chore(poetry): Upgrade to v2 (#7112)" This reverts commit 137365a6709731d8c6d3d97ff71dd80b002f8d62. --- .github/workflows/api-pull-request.yml | 4 +- .github/workflows/sdk-pull-request.yml | 4 +- .pre-commit-config.yaml | 6 +- Dockerfile | 24 +- README.md | 2 +- api/Dockerfile | 3 +- api/poetry.lock | 10 +- api/pyproject.toml | 60 ++--- api/src/backend/api/rls.py | 7 +- api/src/backend/api/tests/test_views.py | 9 +- api/src/backend/config/custom_logging.py | 9 +- poetry.lock | 9 +- pyproject.toml | 127 ++++----- tests/lib/check/check_test.py | 9 +- .../accessanalyzer_enabled_fixer_test.py | 15 +- .../accessanalyzer_enabled_test.py | 45 ++-- .../acm_certificates_expiration_check_test.py | 90 +++---- ...ificates_transparency_logs_enabled_test.py | 75 +++--- ...ficates_with_secure_key_algorithms_test.py | 60 ++--- .../apigateway_authorizers_enabled_test.py | 120 ++++----- ...gateway_client_certificate_enabled_test.py | 47 ++-- .../apigateway_endpoint_public_test.py | 45 ++-- ...endpoint_public_without_authorizer_test.py | 45 ++-- .../apigateway_logging_enabled_test.py | 45 ++-- ...apigateway_restapi_cache_encrypted_test.py | 60 ++--- ...apigateway_restapi_tracing_enabled_test.py | 45 ++-- .../apigateway_waf_acl_attached_test.py | 45 ++-- ...pigatewayv2_access_logging_enabled_test.py | 30 +-- .../apigatewayv2_authorizers_enabled_test.py | 30 +-- ...ppsync_field_level_logging_enabled_test.py | 45 ++-- ...phql_api_no_api_key_authentication_test.py | 60 ++--- .../athena_workgroup_encryption_test.py | 45 ++-- ...na_workgroup_enforce_configuration_test.py | 45 ++-- .../athena_workgroup_logging_enabled_test.py | 45 ++-- ...d_secrets_ec2_launch_configuration_test.py | 120 ++++----- ...g_group_capacity_rebalance_enabled_test.py | 45 ++-- ...ing_group_elb_health_check_enabled_test.py | 105 +++----- ..._launch_configuration_no_public_ip_test.py | 60 ++--- ...unch_configuration_requires_imdsv2_test.py | 60 ++--- .../autoscaling_group_multiple_az_test.py | 60 ++--- ...ling_group_multiple_instance_types_test.py | 90 +++---- ...ng_group_using_ec2_launch_template_test.py | 60 ++--- .../awslambda_function_inside_vpc_test.py | 45 ++-- ...lambda_function_no_secrets_in_code_test.py | 60 ++--- ...a_function_no_secrets_in_variables_test.py | 90 +++---- ...tion_not_publicly_accessible_fixer_test.py | 60 ++--- ...a_function_not_publicly_accessible_test.py | 135 ++++------ ...awslambda_function_url_cors_policy_test.py | 60 ++--- .../awslambda_function_url_public_test.py | 45 ++-- ..._function_using_supported_runtimes_test.py | 60 ++--- .../awslambda_function_vpc_multi_az_test.py | 147 +++++------ .../backup_plans_exist_test.py | 45 ++-- .../backup_reportplans_exist_test.py | 45 ++-- .../backup_vaults_encrypted_test.py | 45 ++-- .../backup_vaults_exist_test.py | 45 ++-- .../bedrock_agent_guardrail_enabled_test.py | 45 ++-- ...drail_prompt_attack_filter_enabled_test.py | 60 ++--- ...nsitive_information_filter_enabled_test.py | 45 ++-- ...k_model_invocation_logging_enabled_test.py | 60 ++--- ...invocation_logs_encryption_enabled_test.py | 203 ++++++--------- ...stack_cdktoolkit_bootstrap_version_test.py | 45 ++-- ...rmation_stack_outputs_find_secrets_test.py | 75 +++--- ...cks_termination_protection_enabled_test.py | 45 ++-- ...udtrail_bucket_requires_mfa_delete_test.py | 153 +++++------ .../cloudtrail_kms_encryption_enabled_test.py | 77 +++--- ...dtrail_log_file_validation_enabled_test.py | 77 +++--- ...s_s3_bucket_access_logging_enabled_test.py | 136 ++++------ ...t_is_not_publicly_accessible_fixer_test.py | 44 ++-- ..._bucket_is_not_publicly_accessible_test.py | 158 +++++------- ...udtrail_s3_dataevents_read_enabled_test.py | 198 ++++++--------- ...dtrail_s3_dataevents_write_enabled_test.py | 198 ++++++--------- ...arm_actions_alarm_state_configured_test.py | 45 ++-- .../cloudwatch_alarm_actions_enabled_test.py | 45 ++-- ...tch_cross_account_sharing_disabled_test.py | 47 ++-- ...h_log_group_kms_encryption_enabled_test.py | 62 ++--- ...l_public_publishing_disabled_fixer_test.py | 30 +-- ...xternal_public_publishing_disabled_test.py | 60 ++--- .../codebuild_project_logging_enabled_test.py | 75 +++--- ...ld_project_no_secrets_in_variables_test.py | 135 ++++------ .../codebuild_project_older_90_days_test.py | 45 ++-- ...odebuild_project_s3_logs_encrypted_test.py | 60 ++--- ..._repo_url_no_sensitive_credentials_test.py | 60 ++--- ..._project_user_controlled_buildspec_test.py | 60 ++--- ...uild_report_group_export_encrypted_test.py | 45 ++-- ...dentity_pool_guest_access_disabled_test.py | 45 ++-- .../services/cognito/cognito_service_test.py | 60 ++--- ...ser_pool_advanced_security_enabled_test.py | 60 ++--- ...mised_credentials_sign_in_attempts_test.py | 75 +++--- ...tential_malicious_sign_in_attempts_test.py | 75 +++--- ...ient_prevent_user_existence_errors_test.py | 45 ++-- ...ol_client_token_revocation_enabled_test.py | 45 ++-- ...r_pool_deletion_protection_enabled_test.py | 45 ++-- .../cognito_user_pool_mfa_enabled_test.py | 60 ++--- ...ser_pool_password_policy_lowercase_test.py | 60 ++--- ..._password_policy_minimum_length_14_test.py | 60 ++--- ...o_user_pool_password_policy_number_test.py | 60 ++--- ...o_user_pool_password_policy_symbol_test.py | 60 ++--- ...ser_pool_password_policy_uppercase_test.py | 60 ++--- ...er_pool_self_registration_disabled_test.py | 116 ++++----- ...pool_temporary_password_expiration_test.py | 45 ++-- ...cognito_user_pool_waf_acl_attached_test.py | 87 +++---- ...onfig_recorder_all_regions_enabled_test.py | 60 ++--- ...ig_recorder_using_aws_service_role_test.py | 60 ++--- .../datasync_task_logging_enabled_test.py | 45 ++-- ...irectconnect_connection_redundancy_test.py | 45 ++-- ...s_snapshot_lifecycle_policy_exists_test.py | 95 +++---- ...int_mongodb_authentication_enabled_test.py | 60 ++--- ..._neptune_iam_authorization_enabled_test.py | 60 ++--- ...edis_in_transit_encryption_enabled_test.py | 60 ++--- .../dms_endpoint_ssl_enabled_test.py | 75 +++--- ...ance_minor_version_upgrade_enabled_test.py | 45 ++-- .../dms_instance_multi_az_test.py | 45 ++-- .../dms_no_public_access_test.py | 89 +++---- ...cation_task_source_logging_enabled_test.py | 120 ++++----- ...cation_task_target_logging_enabled_test.py | 120 ++++----- .../documentdb_cluster_backup_enabled_test.py | 90 +++---- ...ntdb_cluster_cloudwatch_log_export_test.py | 75 +++--- ...mentdb_cluster_deletion_protection_test.py | 45 ++-- ...ocumentdb_cluster_multi_az_enabled_test.py | 45 ++-- ...ntdb_cluster_public_snapshot_fixer_test.py | 30 +-- ...documentdb_cluster_public_snapshot_test.py | 45 ++-- ...cumentdb_cluster_storage_encrypted_test.py | 45 ++-- ...lerator_cluster_encryption_enabled_test.py | 45 ++-- ...ster_in_transit_encryption_enabled_test.py | 45 ++-- ...amodb_accelerator_cluster_multi_az_test.py | 47 ++-- ...dynamodb_table_autoscaling_enabled_test.py | 118 ++++----- ..._table_deletion_protection_enabled_test.py | 45 ++-- ...odb_table_protected_by_backup_plan_test.py | 132 ++++------ ..._tables_kms_cmk_encryption_enabled_test.py | 45 ++-- .../dynamodb_tables_pitr_enabled_test.py | 45 ++-- .../ec2_ami_public_fixer_test.py | 45 ++-- .../ec2/ec2_ami_public/ec2_ami_public_test.py | 45 ++-- .../ec2_ebs_default_encryption_fixer_test.py | 15 +- .../ec2_ebs_default_encryption_test.py | 60 ++--- .../ec2_ebs_public_snapshot_fixer_test.py | 30 +-- .../ec2_ebs_public_snapshot_test.py | 45 ++-- ..._account_block_public_access_fixer_test.py | 15 +- ...apshot_account_block_public_access_test.py | 45 ++-- .../ec2_ebs_snapshots_encrypted_test.py | 45 ++-- .../ec2_ebs_volume_encryption_test.py | 45 ++-- ...bs_volume_protected_by_backup_plan_test.py | 132 ++++------ .../ec2_ebs_volume_snapshots_exists_test.py | 60 ++--- .../ec2_elastic_ip_shodan_test.py | 67 ++--- .../ec2_elastic_ip_unassigned_test.py | 45 ++-- ...tance_account_imdsv2_enabled_fixer_test.py | 15 +- ...c2_instance_account_imdsv2_enabled_test.py | 30 +-- ...stance_detailed_monitoring_enabled_test.py | 47 ++-- .../ec2_instance_imdsv2_enabled_test.py | 66 ++--- ...ernet_facing_with_instance_profile_test.py | 45 ++-- .../ec2_instance_managed_by_ssm_test.py | 174 ++++++------- ..._instance_older_than_specific_days_test.py | 47 ++-- .../ec2_instance_paravirtual_type_test.py | 45 ++-- ...assandra_exposed_to_internet_fixer_test.py | 75 +++--- ...port_cassandra_exposed_to_internet_test.py | 110 ++++---- ...ort_cifs_exposed_to_internet_fixer_test.py | 90 +++---- ...ance_port_cifs_exposed_to_internet_test.py | 110 ++++---- ...h_kibana_exposed_to_internet_fixer_test.py | 90 +++---- ...csearch_kibana_exposed_to_internet_test.py | 110 ++++---- ...port_ftp_exposed_to_internet_fixer_test.py | 75 +++--- ...tance_port_ftp_exposed_to_internet_test.py | 110 ++++---- ...rt_kafka_exposed_to_internet_fixer_test.py | 75 +++--- ...nce_port_kafka_exposed_to_internet_test.py | 110 ++++---- ...kerberos_exposed_to_internet_fixer_test.py | 75 +++--- ..._port_kerberos_exposed_to_internet_test.py | 110 ++++---- ...ort_ldap_exposed_to_internet_fixer_test.py | 75 +++--- ...ance_port_ldap_exposed_to_internet_test.py | 110 ++++---- ...emcached_exposed_to_internet_fixer_test.py | 75 +++--- ...port_memcached_exposed_to_internet_test.py | 110 ++++---- ..._mongodb_exposed_to_internet_fixer_test.py | 90 +++---- ...e_port_mongodb_exposed_to_internet_test.py | 110 ++++---- ...rt_mysql_exposed_to_internet_fixer_test.py | 75 +++--- ...nce_port_mysql_exposed_to_internet_test.py | 110 ++++---- ...t_oracle_exposed_to_internet_fixer_test.py | 75 +++--- ...ce_port_oracle_exposed_to_internet_test.py | 110 ++++---- ...stgresql_exposed_to_internet_fixer_test.py | 75 +++--- ...ort_postgresql_exposed_to_internet_test.py | 110 ++++---- ...port_rdp_exposed_to_internet_fixer_test.py | 75 +++--- ...tance_port_rdp_exposed_to_internet_test.py | 110 ++++---- ...rt_redis_exposed_to_internet_fixer_test.py | 75 +++--- ...nce_port_redis_exposed_to_internet_test.py | 110 ++++---- ...qlserver_exposed_to_internet_fixer_test.py | 75 +++--- ...port_sqlserver_exposed_to_internet_test.py | 110 ++++---- ...port_ssh_exposed_to_internet_fixer_test.py | 75 +++--- ...tance_port_ssh_exposed_to_internet_test.py | 110 ++++---- ...t_telnet_exposed_to_internet_fixer_test.py | 75 +++--- ...ce_port_telnet_exposed_to_internet_test.py | 110 ++++---- .../ec2_instance_profile_attached_test.py | 45 ++-- .../ec2_instance_public_ip_test.py | 45 ++-- .../ec2_instance_secrets_user_data_test.py | 120 ++++----- .../ec2_instance_uses_single_eni_test.py | 75 +++--- .../ec2_launch_template_no_public_ip_test.py | 75 +++--- .../ec2_launch_template_no_secrets_test.py | 135 ++++------ ..._networkacl_allow_ingress_any_port_test.py | 90 +++---- ...tworkacl_allow_ingress_tcp_port_22_test.py | 90 +++---- ...orkacl_allow_ingress_tcp_port_3389_test.py | 90 +++---- .../ec2_networkacl_unused_test.py | 45 ++-- ...ingress_from_internet_to_all_ports_test.py | 161 +++++------- ..._ingress_from_internet_to_any_port_test.py | 230 +++++++---------- ...ernet_to_high_risk_tcp_ports_fixer_test.py | 75 +++--- ...om_internet_to_high_risk_tcp_ports_test.py | 149 +++++------ ...ternet_to_port_mongodb_27017_27018_test.py | 149 +++++------ ...rom_internet_to_tcp_ftp_port_20_21_test.py | 149 +++++------ ...gress_from_internet_to_tcp_port_22_test.py | 149 +++++------ ...ess_from_internet_to_tcp_port_3389_test.py | 149 +++++------ ..._tcp_port_cassandra_7199_9160_8888_test.py | 149 +++++------ ...lasticsearch_kibana_9200_9300_5601_test.py | 149 +++++------ ...om_internet_to_tcp_port_kafka_9092_test.py | 149 +++++------ ...ternet_to_tcp_port_memcached_11211_test.py | 149 +++++------ ...om_internet_to_tcp_port_mysql_3306_test.py | 149 +++++------ ...ernet_to_tcp_port_oracle_1521_2483_test.py | 149 +++++------ ...internet_to_tcp_port_postgres_5432_test.py | 171 +++++-------- ...om_internet_to_tcp_port_redis_6379_test.py | 149 +++++------ ...t_to_tcp_port_sql_server_1433_1434_test.py | 149 +++++------ ...rom_internet_to_tcp_port_telnet_23_test.py | 149 +++++------ ...ygroup_allow_wide_open_public_ipv4_test.py | 66 ++--- ...ritygroup_default_restrict_traffic_test.py | 88 +++---- ...2_securitygroup_from_launch_wizard_test.py | 45 ++-- .../ec2_securitygroup_not_used_test.py | 110 ++++---- ...oup_with_many_ingress_egress_rules_test.py | 45 ++-- .../aws/services/ec2/ec2_service_test.py | 30 +-- ...ateway_auto_accept_vpc_attachments_test.py | 75 +++--- ...gistry_scan_images_on_push_enabled_test.py | 75 +++--- ...ositories_lifecycle_policy_enabled_test.py | 60 ++--- ...ries_not_publicly_accessible_fixer_test.py | 45 ++-- ...tories_scan_images_on_push_enabled_test.py | 60 ++--- ...an_vulnerabilities_in_latest_image_test.py | 180 ++++++------- .../ecr_repositories_tag_immutability_test.py | 60 ++--- ...cluster_container_insights_enabled_test.py | 60 ++--- ...nitions_containers_readonly_access_test.py | 60 ++--- ...initions_host_namespace_not_shared_test.py | 60 ++--- ...nitions_host_networking_mode_users_test.py | 75 +++--- ...ask_definitions_logging_block_mode_test.py | 60 ++--- ...s_task_definitions_logging_enabled_test.py | 45 ++-- ...definitions_no_environment_secrets_test.py | 106 +++----- ...finitions_no_privileged_containers_test.py | 45 ++-- ...ccess_point_enforce_root_directory_test.py | 60 ++--- ...access_point_enforce_user_identity_test.py | 60 ++--- .../efs_encryption_at_rest_enabled_test.py | 30 +-- .../efs_have_backup_enabled_test.py | 45 ++-- ...unt_target_not_publicly_accessible_test.py | 74 +++--- .../efs_not_publicly_accessible_test.py | 75 +++--- ...ticache_cluster_uses_public_subnet_test.py | 174 ++++++------- ...luster_auto_minor_version_upgrades_test.py | 59 ++--- ...cluster_automatic_failover_enabled_test.py | 59 ++--- ...cache_redis_cluster_backup_enabled_test.py | 90 +++---- ...ster_in_transit_encryption_enabled_test.py | 45 ++-- ...che_redis_cluster_multi_az_enabled_test.py | 45 ++-- ...is_cluster_rest_encryption_enabled_test.py | 45 ++-- ...dis_replication_group_auth_enabled_test.py | 116 ++++----- ...ronment_cloudwatch_logging_enabled_test.py | 45 ++-- ...ironment_enhanced_health_reporting_test.py | 45 ++-- ...nvironment_managed_updates_enabled_test.py | 45 ++-- .../elb_connection_draining_enabled_test.py | 45 ++-- ..._cross_zone_load_balancing_enabled_test.py | 60 ++--- .../elb_is_in_multiple_az_test.py | 45 ++-- ..._ssl_listeners_use_acm_certificate_test.py | 110 ++++---- .../elbv2_is_in_multiple_az_test.py | 45 ++-- ...entbridge_bus_cross_account_access_test.py | 60 ++--- .../eventbridge_bus_exposed_test.py | 60 ++--- ...endpoint_event_replication_enabled_test.py | 45 ++-- ...hema_registry_cross_account_access_test.py | 60 ++--- ...ystem_copy_tags_to_backups_enabled_test.py | 120 ++++----- ...ystem_copy_tags_to_volumes_enabled_test.py | 60 ++--- .../fsx_file_system_multi_az_enabled_test.py | 60 ++--- ..._vaults_policy_public_access_fixer_test.py | 30 +-- ...lacier_vaults_policy_public_access_test.py | 90 +++---- ...ction_passwords_encryption_enabled_test.py | 75 +++--- ...talogs_metadata_encryption_enabled_test.py | 75 +++--- ...a_catalogs_not_publicly_accessible_test.py | 45 ++-- ...e_database_connections_ssl_enabled_test.py | 45 ++-- ...cloudwatch_logs_encryption_enabled_test.py | 60 ++--- ...ts_job_bookmark_encryption_enabled_test.py | 60 ++--- ...nt_endpoints_s3_encryption_enabled_test.py | 60 ++--- ..._jobs_amazon_s3_encryption_enabled_test.py | 75 +++--- ...cloudwatch_logs_encryption_enabled_test.py | 60 ++--- ...bs_job_bookmark_encryption_enabled_test.py | 60 ++--- ...lue_ml_transform_encrypted_at_rest_test.py | 45 ++-- .../guardduty_centrally_managed_test.py | 60 ++--- ...uty_ec2_malware_protection_enabled_test.py | 60 ++--- .../guardduty_eks_audit_log_enabled_test.py | 60 ++--- ...uty_eks_runtime_monitoring_enabled_test.py | 60 ++--- .../guardduty_is_enabled_fixer_test.py | 30 +-- .../guardduty_is_enabled_test.py | 83 +++--- ...uardduty_lambda_protection_enabled_test.py | 60 ++--- ...uardduty_no_high_severity_findings_test.py | 45 ++-- .../guardduty_rds_protection_enabled_test.py | 60 ++--- .../guardduty_s3_protection_enabled_test.py | 60 ++--- ...olicy_no_administrative_privileges_test.py | 45 ++-- ...olicy_no_administrative_privileges_test.py | 45 ++-- ...olicy_no_administrative_privileges_test.py | 45 ++-- ..._group_administrator_access_policy_test.py | 45 ++-- ...olicy_no_administrative_privileges_test.py | 180 ++++++------- ...olicy_no_full_access_to_cloudtrail_test.py | 75 +++--- ...nline_policy_no_full_access_to_kms_test.py | 75 +++--- ...words_within_90_days_or_less_fixer_test.py | 17 +- ...am_password_policy_lowercase_fixer_test.py | 17 +- .../iam_password_policy_lowercase_test.py | 30 +-- ...ord_policy_minimum_length_14_fixer_test.py | 17 +- ..._password_policy_minimum_length_14_test.py | 45 ++-- .../iam_password_policy_number_fixer_test.py | 17 +- .../iam_password_policy_number_test.py | 30 +-- ...iam_password_policy_reuse_24_fixer_test.py | 17 +- .../iam_password_policy_reuse_24_test.py | 30 +-- .../iam_password_policy_symbol_fixer_test.py | 17 +- .../iam_password_policy_symbol_test.py | 30 +-- ...am_password_policy_uppercase_fixer_test.py | 17 +- .../iam_password_policy_uppercase_test.py | 30 +-- ...policy_allows_privilege_escalation_test.py | 240 +++++++----------- ...cy_attached_only_to_group_or_roles_test.py | 60 ++--- ...licy_cloudshell_admin_not_attached_test.py | 17 +- ...am_role_administratoraccess_policy_test.py | 105 +++----- ...ross_account_readonlyaccess_policy_test.py | 90 +++---- ...service_confused_deputy_prevention_test.py | 105 +++----- .../iam_root_hardware_mfa_enabled_test.py | 45 ++-- .../iam_root_mfa_enabled_test.py | 34 +-- .../iam_securityaudit_role_created_test.py | 17 +- .../iam_support_role_created_test.py | 47 ++-- ...m_user_administrator_access_policy_test.py | 45 ++-- .../iam_user_hardware_mfa_enabled_test.py | 51 ++-- ...am_user_mfa_enabled_console_access_test.py | 68 ++--- ...m_user_no_setup_initial_access_key_test.py | 68 ++--- .../iam_user_two_active_access_key_test.py | 67 ++--- ...am_user_with_temporary_credentials_test.py | 90 +++---- ...luster_encryption_at_rest_uses_cmk_test.py | 59 ++--- ...luster_enhanced_monitoring_enabled_test.py | 45 ++-- ...ster_in_transit_encryption_enabled_test.py | 60 ++--- .../kafka_cluster_is_public_test.py | 45 ++-- ..._mutual_tls_authentication_enabled_test.py | 45 ++-- ...uster_unrestricted_access_disabled_test.py | 45 ++-- .../kafka_cluster_uses_latest_version_test.py | 45 ++-- ...ctor_in_transit_encryption_enabled_test.py | 45 ++-- ...nesis_stream_data_retention_period_test.py | 45 ++-- .../kinesis_stream_encrypted_at_rest_test.py | 45 ++-- .../kms_cmk_are_used/kms_cmk_are_used_test.py | 84 +++--- ..._not_deleted_unintentionally_fixer_test.py | 45 ++-- ...ms_cmk_not_deleted_unintentionally_test.py | 75 +++--- .../kms_cmk_rotation_enabled_fixer_test.py | 15 +- .../kms_cmk_rotation_enabled_test.py | 69 ++--- .../lightsail_database_public_test.py | 45 ++-- ...tsail_instance_automated_snapshots_test.py | 45 ++-- .../lightsail_instance_public_test.py | 30 +-- .../lightsail_static_ip_unused_test.py | 45 ++-- ...luster_auto_minor_version_upgrades_test.py | 45 ++-- ...broker_auto_minor_version_upgrades_test.py | 45 ++-- .../mq_broker_logging_enabled_test.py | 75 +++--- .../mq_broker_not_publicly_accessible_test.py | 45 ++-- ...une_cluster_copy_tags_to_snapshots_test.py | 45 ++-- ...luster_integration_cloudwatch_logs_test.py | 15 +- ...tune_cluster_public_snapshot_fixer_test.py | 30 +-- .../neptune_cluster_public_snapshot_test.py | 45 ++-- ...neptune_cluster_snapshot_encrypted_test.py | 45 ++-- ...neptune_cluster_uses_public_subnet_test.py | 66 ++--- ...ice_domains_access_control_enabled_test.py | 45 ++-- ...vice_domains_audit_logging_enabled_test.py | 45 ++-- ...domains_cloudwatch_logging_enabled_test.py | 75 +++--- ...domains_encryption_at_rest_enabled_test.py | 45 ++-- ...ult_tolerant_data_nodes_data_nodes_test.py | 75 +++--- ...omains_fault_tolerant_master_nodes_test.py | 60 ++--- ...ains_https_communications_enforced_test.py | 45 ++-- ...ins_internal_user_database_enabled_test.py | 45 ++-- ...ns_node_to_node_encryption_enabled_test.py | 45 ++-- ...ains_not_publicly_accessible_fixer_test.py | 45 ++-- ...he_latest_service_software_version_test.py | 45 ++-- ..._cognito_authentication_for_kibana_test.py | 60 ++--- ...tags_policies_enabled_and_attached_test.py | 15 +- ...ster_minor_version_upgrade_enabled_test.py | 60 ++--- .../rds_cluster_multi_az_test.py | 45 ++-- ...s_cluster_protected_by_backup_plan_test.py | 160 +++++------- ...ds_instance_certificate_expiration_test.py | 165 +++++------- .../rds_instance_inside_vpc_test.py | 15 +- .../rds_instance_multi_az_test.py | 30 +-- ..._instance_protected_by_backup_plan_test.py | 146 +++++------ .../redshift_cluster_audit_logging_test.py | 45 ++-- ...edshift_cluster_automated_snapshot_test.py | 45 ++-- ...edshift_cluster_automatic_upgrades_test.py | 45 ++-- .../redshift_cluster_public_access_test.py | 132 ++++------ ...d_zones_cloudwatch_logging_enabled_test.py | 60 ++--- ...3_access_point_public_access_block_test.py | 75 +++--- ...t_level_public_access_blocks_fixer_test.py | 15 +- ...account_level_public_access_blocks_test.py | 66 ++--- .../s3_bucket_kms_encryption_test.py | 45 ++-- .../s3_bucket_lifecycle_enabled_test.py | 45 ++-- ...t_policy_public_write_access_fixer_test.py | 30 +-- ..._bucket_policy_public_write_access_test.py | 75 +++--- .../s3_bucket_public_access_test.py | 66 ++--- ...oint_config_prod_variant_instances_test.py | 45 ++-- ...r_models_network_isolation_enabled_test.py | 45 ++-- ...ker_models_vpc_settings_configured_test.py | 45 ++-- ...tebook_instance_encryption_enabled_test.py | 45 ++-- ...book_instance_root_access_disabled_test.py | 45 ++-- ...k_instance_vpc_settings_configured_test.py | 45 ++-- ..._direct_internet_access_configured_test.py | 45 ++-- ..._intercontainer_encryption_enabled_test.py | 45 ++-- ...ing_jobs_network_isolation_enabled_test.py | 45 ++-- ...lume_and_output_encryption_enabled_test.py | 45 ++-- ...ining_jobs_vpc_settings_configured_test.py | 45 ++-- ...manager_automatic_rotation_enabled_test.py | 45 ++-- ...etsmanager_not_publicly_accessible_test.py | 45 ++-- ...anager_secret_rotated_periodically_test.py | 60 ++--- .../secretsmanager_secret_unused_test.py | 60 ++--- .../securityhub_enabled_test.py | 15 +- ...s_identity_not_publicly_accessible_test.py | 45 ++-- ...ection_in_cloudfront_distributions_test.py | 60 ++--- ..._protection_in_global_accelerators_test.py | 60 ++--- ...protection_in_route53_hosted_zones_test.py | 88 +++---- ...eues_not_publicly_accessible_fixer_test.py | 45 ++-- ...sqs_queues_not_publicly_accessible_test.py | 105 +++----- ...ues_server_side_encryption_enabled_test.py | 45 ++-- .../ssm_managed_compliant_patching_test.py | 87 +++---- .../vpc_flow_logs_enabled_test.py | 75 +++--- .../vpc_vpn_connection_tunnels_up_test.py | 45 ++-- .../wafv2_webacl_logging_enabled_test.py | 105 +++----- .../wafv2_webacl_rule_logging_enabled_test.py | 90 +++---- ...d_workload_no_high_or_medium_risks_test.py | 60 ++--- ...rkspaces_volume_encryption_enabled_test.py | 75 +++--- tests/providers/azure/azure_provider_test.py | 162 +++++------- ...ce_public_access_level_is_disabled_test.py | 45 ++-- .../aks_cluster_rbac_enabled_test.py | 60 ++--- ...lusters_created_with_private_nodes_test.py | 75 +++--- ...ks_clusters_public_access_disabled_test.py | 75 +++--- .../aks_network_policy_enabled_test.py | 60 ++--- .../app_client_certificates_on_test.py | 60 ++--- .../app_ensure_auth_is_set_up_test.py | 60 ++--- ...ensure_http_is_redirected_to_https_test.py | 60 ++--- .../app_ensure_java_version_is_latest_test.py | 120 ++++----- .../app_ensure_php_version_is_latest_test.py | 75 +++--- ...pp_ensure_python_version_is_latest_test.py | 75 +++--- .../app_ensure_using_http20_test.py | 75 +++--- .../app_ftp_deployment_disabled_test.py | 75 +++--- ...pp_function_access_keys_configured_test.py | 60 ++--- ...ction_application_insights_enabled_test.py | 118 ++++----- ..._function_ftps_deployment_disabled_test.py | 75 +++--- ...pp_function_identity_is_configured_test.py | 60 ++--- ..._identity_without_admin_privileges_test.py | 89 +++---- ...pp_function_latest_runtime_version_test.py | 60 ++--- ...p_function_not_publicly_accessible_test.py | 60 ++--- ..._function_vnet_integration_enabled_test.py | 60 ++--- .../app_http_logs_enabled_test.py | 60 ++--- .../app_minimum_tls_version_12_test.py | 90 +++---- .../app_register_with_identity_test.py | 60 ++--- .../appinsights_ensure_is_configured_test.py | 45 ++-- ...tainerregistry_admin_user_disabled_test.py | 45 ++-- ...erregistry_not_publicly_accessible_test.py | 45 ++-- .../containerregistry_service_test.py | 15 +- ...ontainerregistry_uses_private_link_test.py | 45 ++-- ...unt_firewall_use_selected_networks_test.py | 45 ++-- .../cosmosdb_account_use_aad_and_rbac_test.py | 45 ++-- ...osdb_account_use_private_endpoints_test.py | 45 ++-- ...configured_with_a_security_contact_test.py | 105 +++----- ...s_vm_endpoint_protection_installed_test.py | 60 ++--- ...sioning_log_analytics_agent_vms_on_test.py | 60 ++--- ...lnerabilty_assessments_machines_on_test.py | 45 ++-- ...er_images_resolved_vulnerabilities_test.py | 90 +++---- ...nder_container_images_scan_enabled_test.py | 90 +++---- ...re_defender_for_app_services_is_on_test.py | 45 ++-- ...nder_ensure_defender_for_arm_is_on_test.py | 45 ++-- ...nder_for_azure_sql_databases_is_on_test.py | 45 ++-- ...sure_defender_for_containers_is_on_test.py | 45 ++-- ...ensure_defender_for_cosmosdb_is_on_test.py | 45 ++-- ...nsure_defender_for_databases_is_on_test.py | 105 +++----- ...nder_ensure_defender_for_dns_is_on_test.py | 45 ++-- ...ensure_defender_for_keyvault_is_on_test.py | 45 ++-- ..._for_os_relational_databases_is_on_test.py | 45 ++-- ...r_ensure_defender_for_server_is_on_test.py | 45 ++-- ...ure_defender_for_sql_servers_is_on_test.py | 45 ++-- ..._ensure_defender_for_storage_is_on_test.py | 45 ++-- .../defender_ensure_mcas_is_enabled_test.py | 60 ++--- ...der_ensure_notify_emails_to_owners_test.py | 75 +++--- ..._ensure_system_updates_are_applied_test.py | 75 +++--- .../defender_ensure_wdatp_is_enabled_test.py | 60 ++--- ...icy_require_mfa_for_management_api_test.py | 105 +++----- .../entra_security_defaults_enabled_test.py | 60 ++--- ...cannot_create_microsoft_365_groups_test.py | 75 +++--- ...sions_to_administer_resource_locks_test.py | 60 ++--- ...ion_roles_owner_custom_not_created_test.py | 45 ++-- ...ult_key_expiration_set_in_non_rbac_test.py | 60 ++--- .../keyvault_key_rotation_enabled_test.py | 60 ++--- .../keyvault_logging_enabled_test.py | 66 ++--- ...ult_non_rbac_secret_expiration_set_test.py | 75 +++--- .../keyvault_private_endpoints_test.py | 45 ++-- .../keyvault_rbac_enabled_test.py | 45 ++-- .../keyvault_rbac_key_expiration_set_test.py | 60 ++--- ...eyvault_rbac_secret_expiration_set_test.py | 75 +++--- .../keyvault_recoverable_test.py | 60 ++--- .../keyvault/keyvault_service_test.py | 15 +- ...tor_alert_create_policy_assignment_test.py | 45 ++-- .../monitor_alert_create_update_nsg_test.py | 45 ++-- ...eate_update_public_ip_address_rule_test.py | 45 ++-- ...rt_create_update_security_solution_test.py | 45 ++-- ...r_alert_create_update_sqlserver_fr_test.py | 45 ++-- .../monitor_alert_delete_nsg_test.py | 45 ++-- ...tor_alert_delete_policy_assignment_test.py | 45 ++-- ...lert_delete_public_ip_address_rule_test.py | 45 ++-- ...tor_alert_delete_security_solution_test.py | 45 ++-- .../monitor_alert_delete_sqlserver_fr_test.py | 45 ++-- ...etting_with_appropriate_categories_test.py | 45 ++-- ...monitor_diagnostic_settings_exists_test.py | 54 ++-- ...t_with_activity_logs_cmk_encrypted_test.py | 45 ++-- ...ount_with_activity_logs_is_private_test.py | 45 ++-- ...ver_audit_log_connection_activated_test.py | 75 +++--- ...ible_server_minimum_tls_version_12_test.py | 105 +++----- ...ible_server_ssl_connection_enabled_test.py | 90 +++---- .../network_bastion_host_exists_test.py | 44 ++-- .../network_flow_log_captured_sent_test.py | 60 ++--- ...network_flow_log_more_than_90_days_test.py | 110 ++++---- ...rk_http_internet_access_restricted_test.py | 110 ++++---- .../network_public_ip_shodan_test.py | 51 ++-- ...ork_rdp_internet_access_restricted_test.py | 88 +++---- ...ork_ssh_internet_access_restricted_test.py | 110 ++++---- ...ork_udp_internet_access_restricted_test.py | 88 +++---- .../network_watcher_enabled_test.py | 66 ++--- ...ver_allow_access_services_disabled_test.py | 45 ++-- ...le_server_connection_throttling_on_test.py | 45 ++-- ...lexible_server_enforce_ssl_enabled_test.py | 45 ++-- ...flexible_server_log_checkpoints_on_test.py | 45 ++-- ...flexible_server_log_connections_on_test.py | 45 ++-- ...xible_server_log_disconnections_on_test.py | 45 ++-- ...erver_log_retention_days_greater_3_test.py | 75 +++--- .../sqlserver_auditing_enabled_test.py | 45 ++-- ...lserver_auditing_retention_90_days_test.py | 90 +++---- ...rver_azuread_administrator_enabled_test.py | 60 ++--- ...lserver_microsoft_defender_enabled_test.py | 60 ++--- ...er_recommended_minimal_tls_version_test.py | 60 ++--- .../sqlserver_tde_encrypted_with_cmk_test.py | 75 +++--- .../sqlserver_tde_encryption_enabled_test.py | 75 +++--- ...server_unrestricted_inbound_access_test.py | 45 ++-- ...mails_notifications_admins_enabled_test.py | 75 +++--- ...a_periodic_recurring_scans_enabled_test.py | 75 +++--- ...lserver_va_scan_reports_configured_test.py | 90 +++---- ...r_vulnerability_assessment_enabled_test.py | 60 ++--- ...ob_public_access_level_is_disabled_test.py | 45 ++-- ...ault_network_access_rule_is_denied_test.py | 45 ++-- ...s_are_trusted_to_access_is_enabled_test.py | 45 ++-- ...ryption_with_customer_managed_keys_test.py | 45 ++-- ...rage_ensure_minimum_tls_version_12_test.py | 45 ++-- ...vate_endpoints_in_storage_accounts_test.py | 45 ++-- ...rage_ensure_soft_delete_is_enabled_test.py | 60 ++--- ...frastructure_encryption_is_enabled_test.py | 45 ++-- .../storage_key_rotation_90_days_test.py | 60 ++--- ...ecure_transfer_required_is_enabled_test.py | 45 ++-- ..._attached_disks_encrypted_with_cmk_test.py | 90 +++---- ...nattached_disks_encrypted_with_cmk_test.py | 90 +++---- .../vm_ensure_using_managed_disks_test.py | 75 +++--- ...pikeys_api_restrictions_configured_test.py | 60 ++--- .../apikeys_key_exists_test.py | 30 +-- .../apikeys_key_rotated_in_90_days_test.py | 45 ++-- .../services/apikeys/apikeys_service_test.py | 15 +- .../bigquery_dataset_cmk_encryption_test.py | 45 ++-- .../bigquery_dataset_public_access_test.py | 45 ++-- .../bigquery/bigquery_service_test.py | 15 +- .../bigquery_table_cmk_encryption_test.py | 45 ++-- .../cloudresourcemanager_service_test.py | 15 +- ...loudsql_instance_automated_backups_test.py | 45 ++-- ...l_instance_mysql_local_infile_flag_test.py | 75 +++--- ...ance_mysql_skip_show_database_flag_test.py | 75 +++--- ...tance_postgres_enable_pgaudit_flag_test.py | 75 +++--- ...ance_postgres_log_connections_flag_test.py | 75 +++--- ...e_postgres_log_disconnections_flag_test.py | 75 +++--- ..._postgres_log_error_verbosity_flag_test.py | 75 +++--- ...es_log_min_duration_statement_flag_test.py | 75 +++--- ...tgres_log_min_error_statement_flag_test.py | 75 +++--- ...nce_postgres_log_min_messages_flag_test.py | 75 +++--- ...stance_postgres_log_statement_flag_test.py | 75 +++--- .../cloudsql_instance_public_access_test.py | 45 ++-- .../cloudsql_instance_public_ip_test.py | 45 ++-- ...ained_database_authentication_flag_test.py | 75 +++--- ...r_cross_db_ownership_chaining_flag_test.py | 75 +++--- ...rver_external_scripts_enabled_flag_test.py | 75 +++--- ...tance_sqlserver_remote_access_flag_test.py | 75 +++--- ...dsql_instance_sqlserver_trace_flag_test.py | 75 +++--- ...ce_sqlserver_user_connections_flag_test.py | 75 +++--- ...stance_sqlserver_user_options_flag_test.py | 75 +++--- .../cloudsql_instance_ssl_connections_test.py | 90 +++---- .../cloudsql/cloudsql_service_test.py | 15 +- .../cloudstorage/cloudstorage_service_test.py | 15 +- ...p_access_from_the_internet_allowed_test.py | 180 ++++++------- ...h_access_from_the_internet_allowed_test.py | 180 ++++++------- ...ock_project_wide_ssh_keys_disabled_test.py | 60 ++--- ...nce_confidential_computing_enabled_test.py | 45 ++-- ...nce_default_service_account_in_use_test.py | 60 ++--- ...ccount_in_use_with_full_api_access_test.py | 60 ++--- ...tance_encryption_with_csek_enabled_test.py | 60 ++--- ..._instance_ip_forwarding_is_enabled_test.py | 60 ++--- .../compute_instance_public_ip_test.py | 45 ++-- ...mpute_loadbalancer_logging_enabled_test.py | 45 ++-- ...ompute_network_dns_logging_enabled_test.py | 45 ++-- .../compute_network_not_legacy_test.py | 45 ++-- .../compute_project_os_login_enabled_test.py | 45 ++-- .../compute_public_address_shodan_test.py | 37 ++- .../compute_serial_ports_in_use_test.py | 90 +++---- .../services/compute/compute_service_test.py | 15 +- .../compute_shielded_vm_enabled_test.py | 60 ++--- .../compute_subnet_flow_logs_enabled_test.py | 45 ++-- ...aproc_encrypted_with_cmks_disabled_test.py | 45 ++-- .../dataproc/dataproc_service_test.py | 29 +-- .../dns_dnssec_disabled_test.py | 45 ++-- ...asha1_in_use_to_key_sign_in_dnssec_test.py | 45 ++-- ...sha1_in_use_to_zone_sign_in_dnssec_test.py | 45 ++-- .../gcp/services/dns/dns_service_test.py | 15 +- ...cluster_no_default_service_account_test.py | 75 +++--- .../gcp/services/gke/gke_service_test.py | 15 +- .../iam_audit_logs_enabled_test.py | 45 ++-- ..._no_service_roles_at_project_level_test.py | 88 +++---- ...tion_essential_contacts_configured_test.py | 45 ++-- ...e_kms_enforce_separation_of_duties_test.py | 45 ++-- ...le_sa_enforce_separation_of_duties_test.py | 45 ++-- ...am_sa_no_administrative_privileges_test.py | 191 ++++++-------- .../iam_sa_no_user_managed_keys_test.py | 75 +++--- ...sa_user_managed_key_rotate_90_days_test.py | 60 ++--- .../gcp/services/iam/iamgcp_service_test.py | 75 +++--- ...ms_key_not_publicly_accessible_gcp_test.py | 60 ++--- .../kms_key_rotation_enabled_test.py | 165 +++++------- .../gcp/services/kms/kmsgcp_service_test.py | 15 +- ...udit_configuration_changes_enabled_test.py | 88 +++---- ..._bucket_permission_changes_enabled_test.py | 88 +++---- ...rt_for_custom_role_changes_enabled_test.py | 88 +++---- ..._project_ownership_changes_enabled_test.py | 88 +++---- ...ance_configuration_changes_enabled_test.py | 88 +++---- ..._vpc_firewall_rule_changes_enabled_test.py | 88 +++---- ...rt_for_vpc_network_changes_enabled_test.py | 88 +++---- ..._vpc_network_route_changes_enabled_test.py | 88 +++---- .../services/logging/logging_service_test.py | 15 +- .../logging_sink_created_test.py | 60 ++--- .../monitoring/monitoring_service_test.py | 15 +- .../serviceusage/serviceusage_test.py | 15 +- .../kubernetes/kubernetes_provider_test.py | 19 +- ...enter_groups_not_public_visibility_test.py | 45 ++-- ...between_two_and_four_global_admins_test.py | 60 ++--- ...tra_admin_consent_workflow_enabled_test.py | 60 ++--- ...a_dynamic_group_for_guests_created_test.py | 45 ++-- .../entra_password_hash_sync_enabled_test.py | 60 ++--- 631 files changed, 16569 insertions(+), 24501 deletions(-) diff --git a/.github/workflows/api-pull-request.yml b/.github/workflows/api-pull-request.yml index bbeddb1114..750ced6d87 100644 --- a/.github/workflows/api-pull-request.yml +++ b/.github/workflows/api-pull-request.yml @@ -90,7 +90,7 @@ jobs: if: steps.are-non-ignored-files-changed.outputs.any_changed == 'true' run: | python -m pip install --upgrade pip - pipx install poetry==2.1.1 + pipx install poetry==1.8.5 - name: Set up Python ${{ matrix.python-version }} if: steps.are-non-ignored-files-changed.outputs.any_changed == 'true' @@ -103,7 +103,7 @@ jobs: working-directory: ./api if: steps.are-non-ignored-files-changed.outputs.any_changed == 'true' run: | - poetry install --no-root + poetry install poetry run pip list VERSION=$(curl --silent "https://api.github.com/repos/hadolint/hadolint/releases/latest" | \ grep '"tag_name":' | \ diff --git a/.github/workflows/sdk-pull-request.yml b/.github/workflows/sdk-pull-request.yml index 646277dec5..0c41fed126 100644 --- a/.github/workflows/sdk-pull-request.yml +++ b/.github/workflows/sdk-pull-request.yml @@ -46,7 +46,7 @@ jobs: if: steps.are-non-ignored-files-changed.outputs.any_changed == 'true' run: | python -m pip install --upgrade pip - pipx install poetry==2.1.1 + pipx install poetry==1.8.5 - name: Set up Python ${{ matrix.python-version }} if: steps.are-non-ignored-files-changed.outputs.any_changed == 'true' @@ -58,7 +58,7 @@ jobs: - name: Install dependencies if: steps.are-non-ignored-files-changed.outputs.any_changed == 'true' run: | - poetry install --no-root + poetry install poetry run pip list VERSION=$(curl --silent "https://api.github.com/repos/hadolint/hadolint/releases/latest" | \ grep '"tag_name":' | \ diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index a5d011e23d..5032b12a85 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -59,7 +59,7 @@ repos: args: ["--ignore=E266,W503,E203,E501,W605"] - repo: https://github.com/python-poetry/poetry - rev: 2.1.1 + rev: 1.8.0 hooks: - id: poetry-check name: API - poetry-check @@ -68,7 +68,7 @@ repos: - id: poetry-lock name: API - poetry-lock - args: ["--directory=./api"] + args: ["--no-update", "--directory=./api"] pass_filenames: false - id: poetry-check @@ -78,7 +78,7 @@ repos: - id: poetry-lock name: SDK - poetry-lock - args: ["--directory=./"] + args: ["--no-update", "--directory=./"] pass_filenames: false diff --git a/Dockerfile b/Dockerfile index 02014d0203..1ed1140f44 100644 --- a/Dockerfile +++ b/Dockerfile @@ -4,7 +4,7 @@ LABEL maintainer="https://github.com/prowler-cloud/prowler" # Update system dependencies and install essential tools #hadolint ignore=DL3018 -RUN apk --no-cache upgrade && apk --no-cache add curl git gcc python3-dev musl-dev linux-headers +RUN apk --no-cache upgrade && apk --no-cache add curl git # Create non-root user RUN mkdir -p /home/prowler && \ @@ -18,25 +18,21 @@ WORKDIR /home/prowler COPY prowler/ /home/prowler/prowler/ COPY dashboard/ /home/prowler/dashboard/ COPY pyproject.toml /home/prowler -COPY README.md /home/prowler/ +COPY README.md /home/prowler # Install Python dependencies ENV HOME='/home/prowler' -ENV PATH="${HOME}/.local/bin:${PATH}" -#hadolint ignore=DL3013 -RUN pip install --no-cache-dir --upgrade pip && \ - pip install --no-cache-dir poetry - -# By default poetry does not compile Python source files to bytecode during installation. -# This speeds up the installation process, but the first execution may take a little more -# time because Python then compiles source files to bytecode automatically. If you want to -# compile source files to bytecode during installation, you can use the --compile option -RUN poetry install --compile && \ - rm -rf ~/.cache/pip +ENV PATH="$HOME/.local/bin:$PATH" +RUN pip install --no-cache-dir --upgrade pip setuptools wheel && \ + pip install --no-cache-dir . # Remove deprecated dash dependencies RUN pip uninstall dash-html-components -y && \ pip uninstall dash-core-components -y +# Remove Prowler directory and build files +USER 0 +RUN rm -rf /home/prowler/prowler /home/prowler/pyproject.toml /home/prowler/README.md /home/prowler/build /home/prowler/prowler.egg-info + USER prowler -ENTRYPOINT ["poetry", "run", "prowler"] +ENTRYPOINT ["prowler"] diff --git a/README.md b/README.md index cb40861994..c8f5d32009 100644 --- a/README.md +++ b/README.md @@ -109,7 +109,7 @@ docker compose up -d **Requirements** * `git` installed. -* `poetry` v2 installed: [poetry installation](https://python-poetry.org/docs/#installation). +* `poetry` installed: [poetry installation](https://python-poetry.org/docs/#installation). * `npm` installed: [npm installation](https://docs.npmjs.com/downloading-and-installing-node-js-and-npm). * `Docker Compose` installed: https://docs.docker.com/compose/install/. diff --git a/api/Dockerfile b/api/Dockerfile index 3569d003e7..11eaf32fda 100644 --- a/api/Dockerfile +++ b/api/Dockerfile @@ -21,8 +21,7 @@ COPY src/backend/ ./backend/ ENV PATH="/home/prowler/.local/bin:$PATH" -# Add `--no-root` to avoid installing the current project as a package -RUN poetry install --no-root && \ +RUN poetry install && \ rm -rf ~/.cache/pip COPY docker-entrypoint.sh ./docker-entrypoint.sh diff --git a/api/poetry.lock b/api/poetry.lock index 8e5d3173d9..d832d3ee33 100644 --- a/api/poetry.lock +++ b/api/poetry.lock @@ -3700,6 +3700,7 @@ files = [ {file = "psycopg2_binary-2.9.9-cp311-cp311-win32.whl", hash = "sha256:dc4926288b2a3e9fd7b50dc6a1909a13bbdadfc67d93f3374d984e56f885579d"}, {file = "psycopg2_binary-2.9.9-cp311-cp311-win_amd64.whl", hash = "sha256:b76bedd166805480ab069612119ea636f5ab8f8771e640ae103e05a4aae3e417"}, {file = "psycopg2_binary-2.9.9-cp312-cp312-macosx_10_9_x86_64.whl", hash = "sha256:8532fd6e6e2dc57bcb3bc90b079c60de896d2128c5d9d6f24a63875a95a088cf"}, + {file = "psycopg2_binary-2.9.9-cp312-cp312-macosx_11_0_arm64.whl", hash = "sha256:b0605eaed3eb239e87df0d5e3c6489daae3f7388d455d0c0b4df899519c6a38d"}, {file = "psycopg2_binary-2.9.9-cp312-cp312-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:8f8544b092a29a6ddd72f3556a9fcf249ec412e10ad28be6a0c0d948924f2212"}, {file = "psycopg2_binary-2.9.9-cp312-cp312-manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:2d423c8d8a3c82d08fe8af900ad5b613ce3632a1249fd6a223941d0735fce493"}, {file = "psycopg2_binary-2.9.9-cp312-cp312-manylinux_2_17_ppc64le.manylinux2014_ppc64le.whl", hash = "sha256:2e5afae772c00980525f6d6ecf7cbca55676296b580c0e6abb407f15f3706996"}, @@ -3708,6 +3709,8 @@ files = [ {file = "psycopg2_binary-2.9.9-cp312-cp312-musllinux_1_1_i686.whl", hash = "sha256:cb16c65dcb648d0a43a2521f2f0a2300f40639f6f8c1ecbc662141e4e3e1ee07"}, {file = "psycopg2_binary-2.9.9-cp312-cp312-musllinux_1_1_ppc64le.whl", hash = "sha256:911dda9c487075abd54e644ccdf5e5c16773470a6a5d3826fda76699410066fb"}, {file = "psycopg2_binary-2.9.9-cp312-cp312-musllinux_1_1_x86_64.whl", hash = "sha256:57fede879f08d23c85140a360c6a77709113efd1c993923c59fde17aa27599fe"}, + {file = "psycopg2_binary-2.9.9-cp312-cp312-win32.whl", hash = "sha256:64cf30263844fa208851ebb13b0732ce674d8ec6a0c86a4e160495d299ba3c93"}, + {file = "psycopg2_binary-2.9.9-cp312-cp312-win_amd64.whl", hash = "sha256:81ff62668af011f9a48787564ab7eded4e9fb17a4a6a74af5ffa6a457400d2ab"}, {file = "psycopg2_binary-2.9.9-cp37-cp37m-macosx_10_9_x86_64.whl", hash = "sha256:2293b001e319ab0d869d660a704942c9e2cce19745262a8aba2115ef41a0a42a"}, {file = "psycopg2_binary-2.9.9-cp37-cp37m-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:03ef7df18daf2c4c07e2695e8cfd5ee7f748a1d54d802330985a78d2a5a6dca9"}, {file = "psycopg2_binary-2.9.9-cp37-cp37m-manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:0a602ea5aff39bb9fac6308e9c9d82b9a35c2bf288e184a816002c9fae930b77"}, @@ -4602,6 +4605,7 @@ files = [ {file = "ruamel.yaml.clib-0.2.12-cp310-cp310-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:f66efbc1caa63c088dead1c4170d148eabc9b80d95fb75b6c92ac0aad2437d76"}, {file = "ruamel.yaml.clib-0.2.12-cp310-cp310-musllinux_1_1_i686.whl", hash = "sha256:22353049ba4181685023b25b5b51a574bce33e7f51c759371a7422dcae5402a6"}, {file = "ruamel.yaml.clib-0.2.12-cp310-cp310-musllinux_1_1_x86_64.whl", hash = "sha256:932205970b9f9991b34f55136be327501903f7c66830e9760a8ffb15b07f05cd"}, + {file = "ruamel.yaml.clib-0.2.12-cp310-cp310-musllinux_1_2_aarch64.whl", hash = "sha256:a52d48f4e7bf9005e8f0a89209bf9a73f7190ddf0489eee5eb51377385f59f2a"}, {file = "ruamel.yaml.clib-0.2.12-cp310-cp310-win32.whl", hash = "sha256:3eac5a91891ceb88138c113f9db04f3cebdae277f5d44eaa3651a4f573e6a5da"}, {file = "ruamel.yaml.clib-0.2.12-cp310-cp310-win_amd64.whl", hash = "sha256:ab007f2f5a87bd08ab1499bdf96f3d5c6ad4dcfa364884cb4549aa0154b13a28"}, {file = "ruamel.yaml.clib-0.2.12-cp311-cp311-macosx_13_0_arm64.whl", hash = "sha256:4a6679521a58256a90b0d89e03992c15144c5f3858f40d7c18886023d7943db6"}, @@ -4610,6 +4614,7 @@ files = [ {file = "ruamel.yaml.clib-0.2.12-cp311-cp311-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:811ea1594b8a0fb466172c384267a4e5e367298af6b228931f273b111f17ef52"}, {file = "ruamel.yaml.clib-0.2.12-cp311-cp311-musllinux_1_1_i686.whl", hash = "sha256:cf12567a7b565cbf65d438dec6cfbe2917d3c1bdddfce84a9930b7d35ea59642"}, {file = "ruamel.yaml.clib-0.2.12-cp311-cp311-musllinux_1_1_x86_64.whl", hash = "sha256:7dd5adc8b930b12c8fc5b99e2d535a09889941aa0d0bd06f4749e9a9397c71d2"}, + {file = "ruamel.yaml.clib-0.2.12-cp311-cp311-musllinux_1_2_aarch64.whl", hash = "sha256:1492a6051dab8d912fc2adeef0e8c72216b24d57bd896ea607cb90bb0c4981d3"}, {file = "ruamel.yaml.clib-0.2.12-cp311-cp311-win32.whl", hash = "sha256:bd0a08f0bab19093c54e18a14a10b4322e1eacc5217056f3c063bd2f59853ce4"}, {file = "ruamel.yaml.clib-0.2.12-cp311-cp311-win_amd64.whl", hash = "sha256:a274fb2cb086c7a3dea4322ec27f4cb5cc4b6298adb583ab0e211a4682f241eb"}, {file = "ruamel.yaml.clib-0.2.12-cp312-cp312-macosx_14_0_arm64.whl", hash = "sha256:20b0f8dc160ba83b6dcc0e256846e1a02d044e13f7ea74a3d1d56ede4e48c632"}, @@ -4618,6 +4623,7 @@ files = [ {file = "ruamel.yaml.clib-0.2.12-cp312-cp312-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:749c16fcc4a2b09f28843cda5a193e0283e47454b63ec4b81eaa2242f50e4ccd"}, {file = "ruamel.yaml.clib-0.2.12-cp312-cp312-musllinux_1_1_i686.whl", hash = "sha256:bf165fef1f223beae7333275156ab2022cffe255dcc51c27f066b4370da81e31"}, {file = "ruamel.yaml.clib-0.2.12-cp312-cp312-musllinux_1_1_x86_64.whl", hash = "sha256:32621c177bbf782ca5a18ba4d7af0f1082a3f6e517ac2a18b3974d4edf349680"}, + {file = "ruamel.yaml.clib-0.2.12-cp312-cp312-musllinux_1_2_aarch64.whl", hash = "sha256:b82a7c94a498853aa0b272fd5bc67f29008da798d4f93a2f9f289feb8426a58d"}, {file = "ruamel.yaml.clib-0.2.12-cp312-cp312-win32.whl", hash = "sha256:e8c4ebfcfd57177b572e2040777b8abc537cdef58a2120e830124946aa9b42c5"}, {file = "ruamel.yaml.clib-0.2.12-cp312-cp312-win_amd64.whl", hash = "sha256:0467c5965282c62203273b838ae77c0d29d7638c8a4e3a1c8bdd3602c10904e4"}, {file = "ruamel.yaml.clib-0.2.12-cp313-cp313-macosx_14_0_arm64.whl", hash = "sha256:4c8c5d82f50bb53986a5e02d1b3092b03622c02c2eb78e29bec33fd9593bae1a"}, @@ -4626,6 +4632,7 @@ files = [ {file = "ruamel.yaml.clib-0.2.12-cp313-cp313-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:96777d473c05ee3e5e3c3e999f5d23c6f4ec5b0c38c098b3a5229085f74236c6"}, {file = "ruamel.yaml.clib-0.2.12-cp313-cp313-musllinux_1_1_i686.whl", hash = "sha256:3bc2a80e6420ca8b7d3590791e2dfc709c88ab9152c00eeb511c9875ce5778bf"}, {file = "ruamel.yaml.clib-0.2.12-cp313-cp313-musllinux_1_1_x86_64.whl", hash = "sha256:e188d2699864c11c36cdfdada94d781fd5d6b0071cd9c427bceb08ad3d7c70e1"}, + {file = "ruamel.yaml.clib-0.2.12-cp313-cp313-musllinux_1_2_aarch64.whl", hash = "sha256:4f6f3eac23941b32afccc23081e1f50612bdbe4e982012ef4f5797986828cd01"}, {file = "ruamel.yaml.clib-0.2.12-cp313-cp313-win32.whl", hash = "sha256:6442cb36270b3afb1b4951f060eccca1ce49f3d087ca1ca4563a6eb479cb3de6"}, {file = "ruamel.yaml.clib-0.2.12-cp313-cp313-win_amd64.whl", hash = "sha256:e5b8daf27af0b90da7bb903a876477a9e6d7270be6146906b276605997c7e9a3"}, {file = "ruamel.yaml.clib-0.2.12-cp39-cp39-macosx_12_0_arm64.whl", hash = "sha256:fc4b630cd3fa2cf7fce38afa91d7cfe844a9f75d7f0f36393fa98815e911d987"}, @@ -4634,6 +4641,7 @@ files = [ {file = "ruamel.yaml.clib-0.2.12-cp39-cp39-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:e2f1c3765db32be59d18ab3953f43ab62a761327aafc1594a2a1fbe038b8b8a7"}, {file = "ruamel.yaml.clib-0.2.12-cp39-cp39-musllinux_1_1_i686.whl", hash = "sha256:d85252669dc32f98ebcd5d36768f5d4faeaeaa2d655ac0473be490ecdae3c285"}, {file = "ruamel.yaml.clib-0.2.12-cp39-cp39-musllinux_1_1_x86_64.whl", hash = "sha256:e143ada795c341b56de9418c58d028989093ee611aa27ffb9b7f609c00d813ed"}, + {file = "ruamel.yaml.clib-0.2.12-cp39-cp39-musllinux_1_2_aarch64.whl", hash = "sha256:2c59aa6170b990d8d2719323e628aaf36f3bfbc1c26279c0eeeb24d05d2d11c7"}, {file = "ruamel.yaml.clib-0.2.12-cp39-cp39-win32.whl", hash = "sha256:beffaed67936fbbeffd10966a4eb53c402fafd3d6833770516bf7314bc6ffa12"}, {file = "ruamel.yaml.clib-0.2.12-cp39-cp39-win_amd64.whl", hash = "sha256:040ae85536960525ea62868b642bdb0c2cc6021c9f9d507810c0c604e66f5a7b"}, {file = "ruamel.yaml.clib-0.2.12.tar.gz", hash = "sha256:6c8fbb13ec503f99a91901ab46e0b07ae7941cd527393187039aec586fdfd36f"}, @@ -5425,4 +5433,4 @@ type = ["pytest-mypy"] [metadata] lock-version = "2.1" python-versions = ">=3.11,<3.13" -content-hash = "5892eb6b702f29a7eebff3f9dc899ea8d78cac622aa8f299ef4c662df8f5b51d" +content-hash = "569b05d8ddc6d6479c19cc00e3313d194cd096ca68975d24595879c198d78bd3" diff --git a/api/pyproject.toml b/api/pyproject.toml index fa9c14148a..53eec3a4bc 100644 --- a/api/pyproject.toml +++ b/api/pyproject.toml @@ -2,42 +2,39 @@ build-backend = "poetry.core.masonry.api" requires = ["poetry-core"] -[project] -authors = [{name = "Prowler Engineering", email = "engineering@prowler.com"}] -dependencies = [ - "celery[pytest] (>=5.4.0,<6.0.0)", - "dj-rest-auth[with_social,jwt] (==7.0.1)", - "django==5.1.7", - "django-celery-beat (>=2.7.0,<3.0.0)", - "django-celery-results (>=2.5.1,<3.0.0)", - "django-cors-headers==4.4.0", - "django-environ==0.11.2", - "django-filter==24.3", - "django-guid==3.5.0", - "django-postgres-extra (>=2.0.8,<3.0.0)", - "djangorestframework==3.15.2", - "djangorestframework-jsonapi==7.0.2", - "djangorestframework-simplejwt (>=5.3.1,<6.0.0)", - "drf-nested-routers (>=0.94.1,<1.0.0)", - "drf-spectacular==0.27.2", - "drf-spectacular-jsonapi==0.5.1", - "gunicorn==23.0.0", - "prowler @ git+https://github.com/prowler-cloud/prowler.git@master", - "psycopg2-binary==2.9.9", - "pytest-celery[redis] (>=1.0.1,<2.0.0)", - "sentry-sdk[django] (>=2.20.0,<3.0.0)", - "uuid6==2024.7.10" -] +[tool.poetry] +authors = ["Prowler Team"] description = "Prowler's API (Django/DRF)" license = "Apache-2.0" name = "prowler-api" package-mode = false -# Needed for the SDK compatibility -requires-python = ">=3.11,<3.13" version = "1.5.0" -[project.scripts] -celery = "src.backend.config.settings.celery" +[tool.poetry.dependencies] +celery = {extras = ["pytest"], version = "^5.4.0"} +dj-rest-auth = {extras = ["with_social", "jwt"], version = "7.0.1"} +django = "5.1.7" +django-celery-beat = "^2.7.0" +django-celery-results = "^2.5.1" +django-cors-headers = "4.4.0" +django-environ = "0.11.2" +django-filter = "24.3" +django-guid = "3.5.0" +django-postgres-extra = "^2.0.8" +djangorestframework = "3.15.2" +djangorestframework-jsonapi = "7.0.2" +djangorestframework-simplejwt = "^5.3.1" +drf-nested-routers = "^0.94.1" +drf-spectacular = "0.27.2" +drf-spectacular-jsonapi = "0.5.1" +gunicorn = "23.0.0" +prowler = {git = "https://github.com/prowler-cloud/prowler.git", branch = "master"} +psycopg2-binary = "2.9.9" +pytest-celery = {extras = ["redis"], version = "^1.0.1"} +# Needed for prowler compatibility +python = ">=3.11,<3.13" +sentry-sdk = {extras = ["django"], version = "^2.20.0"} +uuid6 = "2024.7.10" [tool.poetry.group.dev.dependencies] bandit = "1.7.9" @@ -57,3 +54,6 @@ ruff = "0.5.0" safety = "3.2.9" tqdm = "4.67.1" vulture = "2.14" + +[tool.poetry.scripts] +celery = "src.backend.config.settings.celery" diff --git a/api/src/backend/api/rls.py b/api/src/backend/api/rls.py index 285b06a974..ce31493597 100644 --- a/api/src/backend/api/rls.py +++ b/api/src/backend/api/rls.py @@ -2,7 +2,8 @@ from typing import Any from uuid import uuid4 from django.core.exceptions import ValidationError -from django.db import DEFAULT_DB_ALIAS, models +from django.db import DEFAULT_DB_ALIAS +from django.db import models from django.db.backends.ddl_references import Statement, Table from api.db_utils import DB_USER, POSTGRES_TENANT_VAR @@ -87,7 +88,9 @@ class RowLevelSecurityConstraint(models.BaseConstraint): f"{grant_queries}{self.grant_sql_query.format(statement=statement)}" ) - full_create_sql_query = f"{self.rls_sql_query}{policy_queries}{grant_queries}" + full_create_sql_query = ( + f"{self.rls_sql_query}" f"{policy_queries}" f"{grant_queries}" + ) table_name = model._meta.db_table if self.partition_name: diff --git a/api/src/backend/api/tests/test_views.py b/api/src/backend/api/tests/test_views.py index 597561b310..9d30869ca7 100644 --- a/api/src/backend/api/tests/test_views.py +++ b/api/src/backend/api/tests/test_views.py @@ -2201,12 +2201,9 @@ class TestScanViewSet: dummy_task.id = "dummy-task-id" dummy_task_data = {"id": dummy_task.id, "state": StateChoices.EXECUTING} - with ( - patch("api.v1.views.Task.objects.get", return_value=dummy_task), - patch( - "api.v1.views.TaskSerializer", - return_value=type("DummySerializer", (), {"data": dummy_task_data}), - ), + with patch("api.v1.views.Task.objects.get", return_value=dummy_task), patch( + "api.v1.views.TaskSerializer", + return_value=type("DummySerializer", (), {"data": dummy_task_data}), ): url = reverse("scan-report", kwargs={"pk": scan.id}) response = authenticated_client.get(url) diff --git a/api/src/backend/config/custom_logging.py b/api/src/backend/config/custom_logging.py index 83bc94f023..04ef403b1d 100644 --- a/api/src/backend/config/custom_logging.py +++ b/api/src/backend/config/custom_logging.py @@ -2,9 +2,10 @@ import json import logging from enum import StrEnum -from config.env import env from django_guid.log_filters import CorrelationId +from config.env import env + class BackendLogger(StrEnum): GUNICORN = "gunicorn" @@ -38,9 +39,9 @@ class NDJSONFormatter(logging.Formatter): "funcName": record.funcName, "process": record.process, "thread": record.thread, - "transaction_id": ( - record.transaction_id if hasattr(record, "transaction_id") else None - ), + "transaction_id": record.transaction_id + if hasattr(record, "transaction_id") + else None, } # Add REST API extra fields diff --git a/poetry.lock b/poetry.lock index 7a4e9e4155..ad45405b52 100644 --- a/poetry.lock +++ b/poetry.lock @@ -2160,6 +2160,8 @@ python-versions = "*" groups = ["dev"] files = [ {file = "jsonpath-ng-1.7.0.tar.gz", hash = "sha256:f6f5f7fd4e5ff79c785f1573b394043b39849fb2bb47bcead935d12b00beab3c"}, + {file = "jsonpath_ng-1.7.0-py2-none-any.whl", hash = "sha256:898c93fc173f0c336784a3fa63d7434297544b7198124a68f9a3ef9597b0ae6e"}, + {file = "jsonpath_ng-1.7.0-py3-none-any.whl", hash = "sha256:f3d7f9e848cba1b6da28c55b1c26ff915dc9e0b1ba7e752a53d6da8d5cbd00b6"}, ] [package.dependencies] @@ -4519,6 +4521,7 @@ files = [ {file = "ruamel.yaml.clib-0.2.12-cp310-cp310-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:f66efbc1caa63c088dead1c4170d148eabc9b80d95fb75b6c92ac0aad2437d76"}, {file = "ruamel.yaml.clib-0.2.12-cp310-cp310-musllinux_1_1_i686.whl", hash = "sha256:22353049ba4181685023b25b5b51a574bce33e7f51c759371a7422dcae5402a6"}, {file = "ruamel.yaml.clib-0.2.12-cp310-cp310-musllinux_1_1_x86_64.whl", hash = "sha256:932205970b9f9991b34f55136be327501903f7c66830e9760a8ffb15b07f05cd"}, + {file = "ruamel.yaml.clib-0.2.12-cp310-cp310-musllinux_1_2_aarch64.whl", hash = "sha256:a52d48f4e7bf9005e8f0a89209bf9a73f7190ddf0489eee5eb51377385f59f2a"}, {file = "ruamel.yaml.clib-0.2.12-cp310-cp310-win32.whl", hash = "sha256:3eac5a91891ceb88138c113f9db04f3cebdae277f5d44eaa3651a4f573e6a5da"}, {file = "ruamel.yaml.clib-0.2.12-cp310-cp310-win_amd64.whl", hash = "sha256:ab007f2f5a87bd08ab1499bdf96f3d5c6ad4dcfa364884cb4549aa0154b13a28"}, {file = "ruamel.yaml.clib-0.2.12-cp311-cp311-macosx_13_0_arm64.whl", hash = "sha256:4a6679521a58256a90b0d89e03992c15144c5f3858f40d7c18886023d7943db6"}, @@ -4527,6 +4530,7 @@ files = [ {file = "ruamel.yaml.clib-0.2.12-cp311-cp311-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:811ea1594b8a0fb466172c384267a4e5e367298af6b228931f273b111f17ef52"}, {file = "ruamel.yaml.clib-0.2.12-cp311-cp311-musllinux_1_1_i686.whl", hash = "sha256:cf12567a7b565cbf65d438dec6cfbe2917d3c1bdddfce84a9930b7d35ea59642"}, {file = "ruamel.yaml.clib-0.2.12-cp311-cp311-musllinux_1_1_x86_64.whl", hash = "sha256:7dd5adc8b930b12c8fc5b99e2d535a09889941aa0d0bd06f4749e9a9397c71d2"}, + {file = "ruamel.yaml.clib-0.2.12-cp311-cp311-musllinux_1_2_aarch64.whl", hash = "sha256:1492a6051dab8d912fc2adeef0e8c72216b24d57bd896ea607cb90bb0c4981d3"}, {file = "ruamel.yaml.clib-0.2.12-cp311-cp311-win32.whl", hash = "sha256:bd0a08f0bab19093c54e18a14a10b4322e1eacc5217056f3c063bd2f59853ce4"}, {file = "ruamel.yaml.clib-0.2.12-cp311-cp311-win_amd64.whl", hash = "sha256:a274fb2cb086c7a3dea4322ec27f4cb5cc4b6298adb583ab0e211a4682f241eb"}, {file = "ruamel.yaml.clib-0.2.12-cp312-cp312-macosx_14_0_arm64.whl", hash = "sha256:20b0f8dc160ba83b6dcc0e256846e1a02d044e13f7ea74a3d1d56ede4e48c632"}, @@ -4535,6 +4539,7 @@ files = [ {file = "ruamel.yaml.clib-0.2.12-cp312-cp312-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:749c16fcc4a2b09f28843cda5a193e0283e47454b63ec4b81eaa2242f50e4ccd"}, {file = "ruamel.yaml.clib-0.2.12-cp312-cp312-musllinux_1_1_i686.whl", hash = "sha256:bf165fef1f223beae7333275156ab2022cffe255dcc51c27f066b4370da81e31"}, {file = "ruamel.yaml.clib-0.2.12-cp312-cp312-musllinux_1_1_x86_64.whl", hash = "sha256:32621c177bbf782ca5a18ba4d7af0f1082a3f6e517ac2a18b3974d4edf349680"}, + {file = "ruamel.yaml.clib-0.2.12-cp312-cp312-musllinux_1_2_aarch64.whl", hash = "sha256:b82a7c94a498853aa0b272fd5bc67f29008da798d4f93a2f9f289feb8426a58d"}, {file = "ruamel.yaml.clib-0.2.12-cp312-cp312-win32.whl", hash = "sha256:e8c4ebfcfd57177b572e2040777b8abc537cdef58a2120e830124946aa9b42c5"}, {file = "ruamel.yaml.clib-0.2.12-cp312-cp312-win_amd64.whl", hash = "sha256:0467c5965282c62203273b838ae77c0d29d7638c8a4e3a1c8bdd3602c10904e4"}, {file = "ruamel.yaml.clib-0.2.12-cp313-cp313-macosx_14_0_arm64.whl", hash = "sha256:4c8c5d82f50bb53986a5e02d1b3092b03622c02c2eb78e29bec33fd9593bae1a"}, @@ -4543,6 +4548,7 @@ files = [ {file = "ruamel.yaml.clib-0.2.12-cp313-cp313-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:96777d473c05ee3e5e3c3e999f5d23c6f4ec5b0c38c098b3a5229085f74236c6"}, {file = "ruamel.yaml.clib-0.2.12-cp313-cp313-musllinux_1_1_i686.whl", hash = "sha256:3bc2a80e6420ca8b7d3590791e2dfc709c88ab9152c00eeb511c9875ce5778bf"}, {file = "ruamel.yaml.clib-0.2.12-cp313-cp313-musllinux_1_1_x86_64.whl", hash = "sha256:e188d2699864c11c36cdfdada94d781fd5d6b0071cd9c427bceb08ad3d7c70e1"}, + {file = "ruamel.yaml.clib-0.2.12-cp313-cp313-musllinux_1_2_aarch64.whl", hash = "sha256:4f6f3eac23941b32afccc23081e1f50612bdbe4e982012ef4f5797986828cd01"}, {file = "ruamel.yaml.clib-0.2.12-cp313-cp313-win32.whl", hash = "sha256:6442cb36270b3afb1b4951f060eccca1ce49f3d087ca1ca4563a6eb479cb3de6"}, {file = "ruamel.yaml.clib-0.2.12-cp313-cp313-win_amd64.whl", hash = "sha256:e5b8daf27af0b90da7bb903a876477a9e6d7270be6146906b276605997c7e9a3"}, {file = "ruamel.yaml.clib-0.2.12-cp39-cp39-macosx_12_0_arm64.whl", hash = "sha256:fc4b630cd3fa2cf7fce38afa91d7cfe844a9f75d7f0f36393fa98815e911d987"}, @@ -4551,6 +4557,7 @@ files = [ {file = "ruamel.yaml.clib-0.2.12-cp39-cp39-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:e2f1c3765db32be59d18ab3953f43ab62a761327aafc1594a2a1fbe038b8b8a7"}, {file = "ruamel.yaml.clib-0.2.12-cp39-cp39-musllinux_1_1_i686.whl", hash = "sha256:d85252669dc32f98ebcd5d36768f5d4faeaeaa2d655ac0473be490ecdae3c285"}, {file = "ruamel.yaml.clib-0.2.12-cp39-cp39-musllinux_1_1_x86_64.whl", hash = "sha256:e143ada795c341b56de9418c58d028989093ee611aa27ffb9b7f609c00d813ed"}, + {file = "ruamel.yaml.clib-0.2.12-cp39-cp39-musllinux_1_2_aarch64.whl", hash = "sha256:2c59aa6170b990d8d2719323e628aaf36f3bfbc1c26279c0eeeb24d05d2d11c7"}, {file = "ruamel.yaml.clib-0.2.12-cp39-cp39-win32.whl", hash = "sha256:beffaed67936fbbeffd10966a4eb53c402fafd3d6833770516bf7314bc6ffa12"}, {file = "ruamel.yaml.clib-0.2.12-cp39-cp39-win_amd64.whl", hash = "sha256:040ae85536960525ea62868b642bdb0c2cc6021c9f9d507810c0c604e66f5a7b"}, {file = "ruamel.yaml.clib-0.2.12.tar.gz", hash = "sha256:6c8fbb13ec503f99a91901ab46e0b07ae7941cd527393187039aec586fdfd36f"}, @@ -5333,4 +5340,4 @@ type = ["pytest-mypy"] [metadata] lock-version = "2.1" python-versions = ">3.9.1,<3.13" -content-hash = "29ed097b3f41c777e0b28202291f80e0c7f932eb44b4d2ad3ef7dd8999ff42c6" +content-hash = "0bcd0c736f7e94914fd98530742dd6b7e3fa60bc547cd4badad48e72bcf5509e" diff --git a/pyproject.toml b/pyproject.toml index 8bfb4516bd..01197e2c85 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -3,84 +3,76 @@ build-backend = "poetry.core.masonry.api" requires = ["poetry-core"] # https://peps.python.org/pep-0621/ -[project] -authors = [{name = "Toni de la Fuente", email = "toni@blyx.com"}] +[tool.poetry] +authors = ["Toni de la Fuente "] classifiers = [ "Programming Language :: Python :: 3", "Programming Language :: Python :: 3.9", "License :: OSI Approved :: Apache Software License" ] -dependencies = [ - "awsipranges==0.3.3", - "alive-progress==3.2.0", - "azure-identity==1.19.0", - "azure-keyvault-keys==4.10.0", - "azure-mgmt-applicationinsights==4.0.0", - "azure-mgmt-authorization==4.0.0", - "azure-mgmt-compute==34.0.0", - "azure-mgmt-containerregistry==10.3.0", - "azure-mgmt-containerservice==34.0.0", - "azure-mgmt-cosmosdb==9.7.0", - "azure-mgmt-keyvault==10.3.1", - "azure-mgmt-monitor==6.0.2", - "azure-mgmt-network==28.1.0", - "azure-mgmt-rdbms==10.1.0", - "azure-mgmt-resource==23.2.0", - "azure-mgmt-search==9.1.0", - "azure-mgmt-security==7.0.0", - "azure-mgmt-sql==3.0.1", - "azure-mgmt-storage==21.2.1", - "azure-mgmt-subscription==3.1.1", - "azure-mgmt-web==8.0.0", - "azure-storage-blob==12.24.1", - "boto3==1.35.99", - "botocore==1.35.99", - "colorama==0.4.6", - "cryptography==44.0.1", - "dash==2.18.2", - "dash-bootstrap-components==1.6.0", - "detect-secrets==1.5.0", - "google-api-python-client==2.163.0", - "google-auth-httplib2>=0.1,<0.3", - "jsonschema==4.23.0", - "kubernetes==32.0.1", - "microsoft-kiota-abstractions==1.9.2", - "msgraph-sdk==1.18.0", - "numpy==2.0.2", - "pandas==2.2.3", - "py-ocsf-models==0.3.1", - "pydantic==1.10.21", - "python-dateutil (>=2.9.0.post0,<3.0.0)", - "pytz==2025.1", - "schema==0.7.7", - "shodan==1.31.0", - "slack-sdk==3.34.0", - "tabulate==0.9.0", - "tzlocal==5.3.1" -] description = "Prowler is an Open Source security tool to perform AWS, GCP and Azure security best practices assessments, audits, incident response, continuous monitoring, hardening and forensics readiness. It contains hundreds of controls covering CIS, NIST 800, NIST CSF, CISA, RBI, FedRAMP, PCI-DSS, GDPR, HIPAA, FFIEC, SOC2, GXP, AWS Well-Architected Framework Security Pillar, AWS Foundational Technical Review (FTR), ENS (Spanish National Security Scheme) and your custom security frameworks." license = "Apache-2.0" -maintainers = [{name = "Prowler Engineering", email = "engineering@prowler.com"}] +maintainers = [ + "Sergio Garcia ", + "Nacho Rivera ", + "Pepe Fagoaga " +] name = "prowler" packages = [ {include = "prowler"}, {include = "dashboard"} ] readme = "README.md" -requires-python = ">3.9.1,<3.13" version = "5.5.0" -[project.scripts] -prowler = "prowler.__main__:prowler" - -[project.urls] -"Changelog" = "https://github.com/prowler-cloud/prowler/releases" -"Documentation" = "https://docs.prowler.cloud" -"Homepage" = "https://github.com/prowler-cloud/prowler" -"Issue tracker" = "https://github.com/prowler-cloud/prowler/issues" - -[tool.poetry] -requires-poetry = ">=2.0" +[tool.poetry.dependencies] +alive-progress = "3.2.0" +awsipranges = "0.3.3" +azure-identity = "1.19.0" +azure-keyvault-keys = "4.10.0" +azure-mgmt-applicationinsights = "4.0.0" +azure-mgmt-authorization = "4.0.0" +azure-mgmt-compute = "34.0.0" +azure-mgmt-containerregistry = "10.3.0" +azure-mgmt-containerservice = "34.0.0" +azure-mgmt-cosmosdb = "9.7.0" +azure-mgmt-keyvault = "10.3.1" +azure-mgmt-monitor = "6.0.2" +azure-mgmt-network = "28.1.0" +azure-mgmt-rdbms = "10.1.0" +azure-mgmt-resource = "23.2.0" +azure-mgmt-search = "9.1.0" +azure-mgmt-security = "7.0.0" +azure-mgmt-sql = "3.0.1" +azure-mgmt-storage = "21.2.1" +azure-mgmt-subscription = "3.1.1" +azure-mgmt-web = "8.0.0" +azure-storage-blob = "12.24.1" +boto3 = "1.35.99" +botocore = "1.35.99" +colorama = "0.4.6" +cryptography = "44.0.1" +dash = "2.18.2" +dash-bootstrap-components = "1.6.0" +detect-secrets = "1.5.0" +google-api-python-client = "2.163.0" +google-auth-httplib2 = ">=0.1,<0.3" +jsonschema = "4.23.0" +kubernetes = "32.0.1" +microsoft-kiota-abstractions = "1.9.2" +msgraph-sdk = "1.18.0" +numpy = "2.0.2" +pandas = "2.2.3" +py-ocsf-models = "0.3.1" +pydantic = "1.10.21" +python = ">3.9.1,<3.13" +python-dateutil = "^2.9.0.post0" +pytz = "2025.1" +schema = "0.7.7" +shodan = "1.31.0" +slack-sdk = "3.34.0" +tabulate = "0.9.0" +tzlocal = "5.3.1" [tool.poetry.group.dev.dependencies] bandit = "1.8.3" @@ -111,6 +103,15 @@ mkdocs-git-revision-date-localized-plugin = "1.4.1" mkdocs-material = "9.6.5" mkdocs-material-extensions = "1.3.1" +[tool.poetry.scripts] +prowler = "prowler.__main__:prowler" + +[tool.poetry.urls] +"Changelog" = "https://github.com/prowler-cloud/prowler/releases" +"Documentation" = "https://docs.prowler.cloud" +"Homepage" = "https://github.com/prowler-cloud/prowler" +"Issue tracker" = "https://github.com/prowler-cloud/prowler/issues" + [tool.poetry-version-plugin] source = "init" diff --git a/tests/lib/check/check_test.py b/tests/lib/check/check_test.py index aea4fefdf9..f2603fb708 100644 --- a/tests/lib/check/check_test.py +++ b/tests/lib/check/check_test.py @@ -820,12 +820,9 @@ class TestCheck: check.CheckID = "test-check" check.execute = Mock(return_value=findings) - with ( - patch("prowler.lib.check.check.execute", return_value=findings), - patch( - "prowler.providers.aws.services.accessanalyzer.accessanalyzer_service.AccessAnalyzer", - accessanalyzer_client, - ), + with patch("prowler.lib.check.check.execute", return_value=findings), patch( + "prowler.providers.aws.services.accessanalyzer.accessanalyzer_service.AccessAnalyzer", + accessanalyzer_client, ): findings = execute( check=check, diff --git a/tests/providers/aws/services/accessanalyzer/accessanalyzer_enabled/accessanalyzer_enabled_fixer_test.py b/tests/providers/aws/services/accessanalyzer/accessanalyzer_enabled/accessanalyzer_enabled_fixer_test.py index 39f90095f1..ac600dc828 100644 --- a/tests/providers/aws/services/accessanalyzer/accessanalyzer_enabled/accessanalyzer_enabled_fixer_test.py +++ b/tests/providers/aws/services/accessanalyzer/accessanalyzer_enabled/accessanalyzer_enabled_fixer_test.py @@ -14,15 +14,12 @@ class Test_accessanalyzer_enabled_fixer: regional_client.create_analyzer.return_value = None accessanalyzer_client.regional_clients = {AWS_REGION_EU_WEST_1: regional_client} - with ( - mock.patch( - "prowler.providers.aws.services.accessanalyzer.accessanalyzer_service.AccessAnalyzer", - new=accessanalyzer_client, - ) as accessanalyzer_client, - mock.patch( - "prowler.providers.aws.services.accessanalyzer.accessanalyzer_client.accessanalyzer_client", - new=accessanalyzer_client, - ), + with mock.patch( + "prowler.providers.aws.services.accessanalyzer.accessanalyzer_service.AccessAnalyzer", + new=accessanalyzer_client, + ) as accessanalyzer_client, mock.patch( + "prowler.providers.aws.services.accessanalyzer.accessanalyzer_client.accessanalyzer_client", + new=accessanalyzer_client, ): # Test Check from prowler.providers.aws.services.accessanalyzer.accessanalyzer_enabled.accessanalyzer_enabled_fixer import ( diff --git a/tests/providers/aws/services/accessanalyzer/accessanalyzer_enabled/accessanalyzer_enabled_test.py b/tests/providers/aws/services/accessanalyzer/accessanalyzer_enabled/accessanalyzer_enabled_test.py index 3afe2f5db1..c6c4d29656 100644 --- a/tests/providers/aws/services/accessanalyzer/accessanalyzer_enabled/accessanalyzer_enabled_test.py +++ b/tests/providers/aws/services/accessanalyzer/accessanalyzer_enabled/accessanalyzer_enabled_test.py @@ -49,15 +49,12 @@ class Test_accessanalyzer_enabled: region=AWS_REGION_1, ) ] - with ( - mock.patch( - "prowler.providers.aws.services.accessanalyzer.accessanalyzer_service.AccessAnalyzer", - accessanalyzer_client, - ), - mock.patch( - "prowler.providers.aws.services.accessanalyzer.accessanalyzer_service.AccessAnalyzer.get_unknown_arn", - return_value="arn:aws:accessanalyzer:eu-west-1:123456789012:unknown", - ), + with mock.patch( + "prowler.providers.aws.services.accessanalyzer.accessanalyzer_service.AccessAnalyzer", + accessanalyzer_client, + ), mock.patch( + "prowler.providers.aws.services.accessanalyzer.accessanalyzer_service.AccessAnalyzer.get_unknown_arn", + return_value="arn:aws:accessanalyzer:eu-west-1:123456789012:unknown", ): from prowler.providers.aws.services.accessanalyzer.accessanalyzer_enabled.accessanalyzer_enabled import ( accessanalyzer_enabled, @@ -97,15 +94,12 @@ class Test_accessanalyzer_enabled: region=AWS_REGION_1, ) ] - with ( - mock.patch( - "prowler.providers.aws.services.accessanalyzer.accessanalyzer_service.AccessAnalyzer", - accessanalyzer_client, - ), - mock.patch( - "prowler.providers.aws.services.accessanalyzer.accessanalyzer_service.AccessAnalyzer.get_unknown_arn", - return_value="arn:aws:accessanalyzer:eu-west-1:123456789012:unknown", - ), + with mock.patch( + "prowler.providers.aws.services.accessanalyzer.accessanalyzer_service.AccessAnalyzer", + accessanalyzer_client, + ), mock.patch( + "prowler.providers.aws.services.accessanalyzer.accessanalyzer_service.AccessAnalyzer.get_unknown_arn", + return_value="arn:aws:accessanalyzer:eu-west-1:123456789012:unknown", ): from prowler.providers.aws.services.accessanalyzer.accessanalyzer_enabled.accessanalyzer_enabled import ( accessanalyzer_enabled, @@ -154,15 +148,12 @@ class Test_accessanalyzer_enabled: ] # Patch AccessAnalyzer Client - with ( - mock.patch( - "prowler.providers.aws.services.accessanalyzer.accessanalyzer_service.AccessAnalyzer", - new=accessanalyzer_client, - ), - mock.patch( - "prowler.providers.aws.services.accessanalyzer.accessanalyzer_service.AccessAnalyzer.get_unknown_arn", - return_value="arn:aws:accessanalyzer:eu-west-1:123456789012:analyzer/unknown", - ), + with mock.patch( + "prowler.providers.aws.services.accessanalyzer.accessanalyzer_service.AccessAnalyzer", + new=accessanalyzer_client, + ), mock.patch( + "prowler.providers.aws.services.accessanalyzer.accessanalyzer_service.AccessAnalyzer.get_unknown_arn", + return_value="arn:aws:accessanalyzer:eu-west-1:123456789012:analyzer/unknown", ): # Test Check from prowler.providers.aws.services.accessanalyzer.accessanalyzer_enabled.accessanalyzer_enabled import ( diff --git a/tests/providers/aws/services/acm/acm_certificates_expiration_check/acm_certificates_expiration_check_test.py b/tests/providers/aws/services/acm/acm_certificates_expiration_check/acm_certificates_expiration_check_test.py index 0c64befcbc..58b0597c9c 100644 --- a/tests/providers/aws/services/acm/acm_certificates_expiration_check/acm_certificates_expiration_check_test.py +++ b/tests/providers/aws/services/acm/acm_certificates_expiration_check/acm_certificates_expiration_check_test.py @@ -135,15 +135,12 @@ class Test_acm_certificates_expiration_check: aws_mocked_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_mocked_provider, - ), - mock.patch( - "prowler.providers.aws.services.acm.acm_certificates_expiration_check.acm_certificates_expiration_check.acm_client", - new=ACM(aws_mocked_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_mocked_provider, + ), mock.patch( + "prowler.providers.aws.services.acm.acm_certificates_expiration_check.acm_certificates_expiration_check.acm_client", + new=ACM(aws_mocked_provider), ): # Test Check from prowler.providers.aws.services.acm.acm_certificates_expiration_check.acm_certificates_expiration_check import ( @@ -162,15 +159,12 @@ class Test_acm_certificates_expiration_check: aws_mocked_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_mocked_provider, - ), - mock.patch( - "prowler.providers.aws.services.acm.acm_certificates_expiration_check.acm_certificates_expiration_check.acm_client", - new=ACM(aws_mocked_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_mocked_provider, + ), mock.patch( + "prowler.providers.aws.services.acm.acm_certificates_expiration_check.acm_certificates_expiration_check.acm_client", + new=ACM(aws_mocked_provider), ): from prowler.providers.aws.services.acm.acm_certificates_expiration_check.acm_certificates_expiration_check import ( acm_certificates_expiration_check, @@ -201,15 +195,12 @@ class Test_acm_certificates_expiration_check: aws_mocked_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_mocked_provider, - ), - mock.patch( - "prowler.providers.aws.services.acm.acm_certificates_expiration_check.acm_certificates_expiration_check.acm_client", - new=ACM(aws_mocked_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_mocked_provider, + ), mock.patch( + "prowler.providers.aws.services.acm.acm_certificates_expiration_check.acm_certificates_expiration_check.acm_client", + new=ACM(aws_mocked_provider), ): from prowler.providers.aws.services.acm.acm_certificates_expiration_check.acm_certificates_expiration_check import ( acm_certificates_expiration_check, @@ -249,15 +240,12 @@ class Test_acm_certificates_expiration_check: aws_mocked_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_mocked_provider, - ), - mock.patch( - "prowler.providers.aws.services.acm.acm_certificates_expiration_check.acm_certificates_expiration_check.acm_client", - new=ACM(aws_mocked_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_mocked_provider, + ), mock.patch( + "prowler.providers.aws.services.acm.acm_certificates_expiration_check.acm_certificates_expiration_check.acm_client", + new=ACM(aws_mocked_provider), ): # Test Check from prowler.providers.aws.services.acm.acm_certificates_expiration_check.acm_certificates_expiration_check import ( @@ -288,15 +276,12 @@ class Test_acm_certificates_expiration_check: aws_mocked_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) aws_mocked_provider._scan_unused_services = False - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_mocked_provider, - ), - mock.patch( - "prowler.providers.aws.services.acm.acm_certificates_expiration_check.acm_certificates_expiration_check.acm_client", - new=ACM(aws_mocked_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_mocked_provider, + ), mock.patch( + "prowler.providers.aws.services.acm.acm_certificates_expiration_check.acm_certificates_expiration_check.acm_client", + new=ACM(aws_mocked_provider), ): from prowler.providers.aws.services.acm.acm_certificates_expiration_check.acm_certificates_expiration_check import ( acm_certificates_expiration_check, @@ -314,15 +299,12 @@ class Test_acm_certificates_expiration_check: aws_mocked_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_mocked_provider, - ), - mock.patch( - "prowler.providers.aws.services.acm.acm_certificates_expiration_check.acm_certificates_expiration_check.acm_client", - new=ACM(aws_mocked_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_mocked_provider, + ), mock.patch( + "prowler.providers.aws.services.acm.acm_certificates_expiration_check.acm_certificates_expiration_check.acm_client", + new=ACM(aws_mocked_provider), ): from prowler.providers.aws.services.acm.acm_certificates_expiration_check.acm_certificates_expiration_check import ( acm_certificates_expiration_check, diff --git a/tests/providers/aws/services/acm/acm_certificates_transparency_logs_enabled/acm_certificates_transparency_logs_enabled_test.py b/tests/providers/aws/services/acm/acm_certificates_transparency_logs_enabled/acm_certificates_transparency_logs_enabled_test.py index c69dc496e8..2ac39a9740 100644 --- a/tests/providers/aws/services/acm/acm_certificates_transparency_logs_enabled/acm_certificates_transparency_logs_enabled_test.py +++ b/tests/providers/aws/services/acm/acm_certificates_transparency_logs_enabled/acm_certificates_transparency_logs_enabled_test.py @@ -128,15 +128,12 @@ class Test_acm_certificates_transparency_logs_enabled: aws_mocked_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_mocked_provider, - ), - mock.patch( - "prowler.providers.aws.services.acm.acm_certificates_transparency_logs_enabled.acm_certificates_transparency_logs_enabled.acm_client", - new=ACM(aws_mocked_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_mocked_provider, + ), mock.patch( + "prowler.providers.aws.services.acm.acm_certificates_transparency_logs_enabled.acm_certificates_transparency_logs_enabled.acm_client", + new=ACM(aws_mocked_provider), ): # Test Check from prowler.providers.aws.services.acm.acm_certificates_transparency_logs_enabled.acm_certificates_transparency_logs_enabled import ( @@ -158,15 +155,12 @@ class Test_acm_certificates_transparency_logs_enabled: aws_mocked_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) aws_mocked_provider._scan_unused_services = False - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_mocked_provider, - ), - mock.patch( - "prowler.providers.aws.services.acm.acm_certificates_transparency_logs_enabled.acm_certificates_transparency_logs_enabled.acm_client", - new=ACM(aws_mocked_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_mocked_provider, + ), mock.patch( + "prowler.providers.aws.services.acm.acm_certificates_transparency_logs_enabled.acm_certificates_transparency_logs_enabled.acm_client", + new=ACM(aws_mocked_provider), ): # Test Check from prowler.providers.aws.services.acm.acm_certificates_transparency_logs_enabled.acm_certificates_transparency_logs_enabled import ( @@ -185,15 +179,12 @@ class Test_acm_certificates_transparency_logs_enabled: aws_mocked_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_mocked_provider, - ), - mock.patch( - "prowler.providers.aws.services.acm.acm_certificates_transparency_logs_enabled.acm_certificates_transparency_logs_enabled.acm_client", - new=ACM(aws_mocked_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_mocked_provider, + ), mock.patch( + "prowler.providers.aws.services.acm.acm_certificates_transparency_logs_enabled.acm_certificates_transparency_logs_enabled.acm_client", + new=ACM(aws_mocked_provider), ): # Test Check from prowler.providers.aws.services.acm.acm_certificates_transparency_logs_enabled.acm_certificates_transparency_logs_enabled import ( @@ -224,15 +215,12 @@ class Test_acm_certificates_transparency_logs_enabled: aws_mocked_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_mocked_provider, - ), - mock.patch( - "prowler.providers.aws.services.acm.acm_certificates_transparency_logs_enabled.acm_certificates_transparency_logs_enabled.acm_client", - new=ACM(aws_mocked_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_mocked_provider, + ), mock.patch( + "prowler.providers.aws.services.acm.acm_certificates_transparency_logs_enabled.acm_certificates_transparency_logs_enabled.acm_client", + new=ACM(aws_mocked_provider), ): # Test Check from prowler.providers.aws.services.acm.acm_certificates_transparency_logs_enabled.acm_certificates_transparency_logs_enabled import ( @@ -268,15 +256,12 @@ class Test_acm_certificates_transparency_logs_enabled: aws_mocked_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_mocked_provider, - ), - mock.patch( - "prowler.providers.aws.services.acm.acm_certificates_transparency_logs_enabled.acm_certificates_transparency_logs_enabled.acm_client", - new=ACM(aws_mocked_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_mocked_provider, + ), mock.patch( + "prowler.providers.aws.services.acm.acm_certificates_transparency_logs_enabled.acm_certificates_transparency_logs_enabled.acm_client", + new=ACM(aws_mocked_provider), ): # Test Check from prowler.providers.aws.services.acm.acm_certificates_transparency_logs_enabled.acm_certificates_transparency_logs_enabled import ( diff --git a/tests/providers/aws/services/acm/acm_certificates_with_secure_key_algorithms/acm_certificates_with_secure_key_algorithms_test.py b/tests/providers/aws/services/acm/acm_certificates_with_secure_key_algorithms/acm_certificates_with_secure_key_algorithms_test.py index 0d8f6f79d7..65db20a311 100644 --- a/tests/providers/aws/services/acm/acm_certificates_with_secure_key_algorithms/acm_certificates_with_secure_key_algorithms_test.py +++ b/tests/providers/aws/services/acm/acm_certificates_with_secure_key_algorithms/acm_certificates_with_secure_key_algorithms_test.py @@ -97,15 +97,12 @@ class Test_acm_certificates_with_secure_key_algorithms: aws_mocked_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_mocked_provider, - ), - mock.patch( - "prowler.providers.aws.services.acm.acm_certificates_with_secure_key_algorithms.acm_certificates_with_secure_key_algorithms.acm_client", - new=ACM(aws_mocked_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_mocked_provider, + ), mock.patch( + "prowler.providers.aws.services.acm.acm_certificates_with_secure_key_algorithms.acm_certificates_with_secure_key_algorithms.acm_client", + new=ACM(aws_mocked_provider), ): # Test Check from prowler.providers.aws.services.acm.acm_certificates_with_secure_key_algorithms.acm_certificates_with_secure_key_algorithms import ( @@ -129,15 +126,12 @@ class Test_acm_certificates_with_secure_key_algorithms: aws_mocked_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_mocked_provider, - ), - mock.patch( - "prowler.providers.aws.services.acm.acm_certificates_with_secure_key_algorithms.acm_certificates_with_secure_key_algorithms.acm_client", - new=ACM(aws_mocked_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_mocked_provider, + ), mock.patch( + "prowler.providers.aws.services.acm.acm_certificates_with_secure_key_algorithms.acm_certificates_with_secure_key_algorithms.acm_client", + new=ACM(aws_mocked_provider), ): # Test Check from prowler.providers.aws.services.acm.acm_certificates_with_secure_key_algorithms.acm_certificates_with_secure_key_algorithms import ( @@ -165,15 +159,12 @@ class Test_acm_certificates_with_secure_key_algorithms: aws_mocked_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_mocked_provider, - ), - mock.patch( - "prowler.providers.aws.services.acm.acm_certificates_with_secure_key_algorithms.acm_certificates_with_secure_key_algorithms.acm_client", - new=ACM(aws_mocked_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_mocked_provider, + ), mock.patch( + "prowler.providers.aws.services.acm.acm_certificates_with_secure_key_algorithms.acm_certificates_with_secure_key_algorithms.acm_client", + new=ACM(aws_mocked_provider), ): # Test Check from prowler.providers.aws.services.acm.acm_certificates_with_secure_key_algorithms.acm_certificates_with_secure_key_algorithms import ( @@ -207,15 +198,12 @@ class Test_acm_certificates_with_secure_key_algorithms: aws_mocked_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) aws_mocked_provider._scan_unused_services = False - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_mocked_provider, - ), - mock.patch( - "prowler.providers.aws.services.acm.acm_certificates_with_secure_key_algorithms.acm_certificates_with_secure_key_algorithms.acm_client", - new=ACM(aws_mocked_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_mocked_provider, + ), mock.patch( + "prowler.providers.aws.services.acm.acm_certificates_with_secure_key_algorithms.acm_certificates_with_secure_key_algorithms.acm_client", + new=ACM(aws_mocked_provider), ): # Test Check from prowler.providers.aws.services.acm.acm_certificates_with_secure_key_algorithms.acm_certificates_with_secure_key_algorithms import ( diff --git a/tests/providers/aws/services/apigateway/apigateway_authorizers_enabled/apigateway_authorizers_enabled_test.py b/tests/providers/aws/services/apigateway/apigateway_authorizers_enabled/apigateway_authorizers_enabled_test.py index 916285efd9..21b92a5bd2 100644 --- a/tests/providers/aws/services/apigateway/apigateway_authorizers_enabled/apigateway_authorizers_enabled_test.py +++ b/tests/providers/aws/services/apigateway/apigateway_authorizers_enabled/apigateway_authorizers_enabled_test.py @@ -22,15 +22,12 @@ class Test_apigateway_restapi_authorizers_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled import ( @@ -79,15 +76,12 @@ class Test_apigateway_restapi_authorizers_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled import ( @@ -127,15 +121,12 @@ class Test_apigateway_restapi_authorizers_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled import ( @@ -191,15 +182,12 @@ class Test_apigateway_restapi_authorizers_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled import ( @@ -255,15 +243,12 @@ class Test_apigateway_restapi_authorizers_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled import ( @@ -326,15 +311,12 @@ class Test_apigateway_restapi_authorizers_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled import ( @@ -399,15 +381,12 @@ class Test_apigateway_restapi_authorizers_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled import ( @@ -462,15 +441,12 @@ class Test_apigateway_restapi_authorizers_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_authorizers_enabled.apigateway_restapi_authorizers_enabled import ( diff --git a/tests/providers/aws/services/apigateway/apigateway_client_certificate_enabled/apigateway_client_certificate_enabled_test.py b/tests/providers/aws/services/apigateway/apigateway_client_certificate_enabled/apigateway_client_certificate_enabled_test.py index 8801be89a5..77a0bca8ca 100644 --- a/tests/providers/aws/services/apigateway/apigateway_client_certificate_enabled/apigateway_client_certificate_enabled_test.py +++ b/tests/providers/aws/services/apigateway/apigateway_client_certificate_enabled/apigateway_client_certificate_enabled_test.py @@ -28,15 +28,12 @@ class Test_apigateway_restapi_client_certificate_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_client_certificate_enabled.apigateway_restapi_client_certificate_enabled.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_client_certificate_enabled.apigateway_restapi_client_certificate_enabled.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_client_certificate_enabled.apigateway_restapi_client_certificate_enabled import ( @@ -91,15 +88,12 @@ class Test_apigateway_restapi_client_certificate_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_client_certificate_enabled.apigateway_restapi_client_certificate_enabled.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_client_certificate_enabled.apigateway_restapi_client_certificate_enabled.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_client_certificate_enabled.apigateway_restapi_client_certificate_enabled import ( @@ -139,16 +133,13 @@ class Test_apigateway_restapi_client_certificate_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_client_certificate_enabled.apigateway_restapi_client_certificate_enabled.apigateway_client", - new=APIGateway(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_client_certificate_enabled.apigateway_restapi_client_certificate_enabled.apigateway_client", + new=APIGateway(aws_provider), + ) as service_client: # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_client_certificate_enabled.apigateway_restapi_client_certificate_enabled import ( apigateway_restapi_client_certificate_enabled, diff --git a/tests/providers/aws/services/apigateway/apigateway_endpoint_public/apigateway_endpoint_public_test.py b/tests/providers/aws/services/apigateway/apigateway_endpoint_public/apigateway_endpoint_public_test.py index ca118e9d7d..2e09d41d8a 100644 --- a/tests/providers/aws/services/apigateway/apigateway_endpoint_public/apigateway_endpoint_public_test.py +++ b/tests/providers/aws/services/apigateway/apigateway_endpoint_public/apigateway_endpoint_public_test.py @@ -21,15 +21,12 @@ class Test_apigateway_restapi_public: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_public.apigateway_restapi_public.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_public.apigateway_restapi_public.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_public.apigateway_restapi_public import ( @@ -62,15 +59,12 @@ class Test_apigateway_restapi_public: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_public.apigateway_restapi_public.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_public.apigateway_restapi_public.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_public.apigateway_restapi_public import ( @@ -115,15 +109,12 @@ class Test_apigateway_restapi_public: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_public.apigateway_restapi_public.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_public.apigateway_restapi_public.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_public.apigateway_restapi_public import ( diff --git a/tests/providers/aws/services/apigateway/apigateway_endpoint_public_without_authorizer/apigateway_endpoint_public_without_authorizer_test.py b/tests/providers/aws/services/apigateway/apigateway_endpoint_public_without_authorizer/apigateway_endpoint_public_without_authorizer_test.py index 97521f1ab7..b89df4a3d3 100644 --- a/tests/providers/aws/services/apigateway/apigateway_endpoint_public_without_authorizer/apigateway_endpoint_public_without_authorizer_test.py +++ b/tests/providers/aws/services/apigateway/apigateway_endpoint_public_without_authorizer/apigateway_endpoint_public_without_authorizer_test.py @@ -23,15 +23,12 @@ class Test_apigateway_restapi_public_with_authorizer: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_public_with_authorizer.apigateway_restapi_public_with_authorizer.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_public_with_authorizer.apigateway_restapi_public_with_authorizer.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_public_with_authorizer.apigateway_restapi_public_with_authorizer import ( @@ -64,15 +61,12 @@ class Test_apigateway_restapi_public_with_authorizer: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_public_with_authorizer.apigateway_restapi_public_with_authorizer.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_public_with_authorizer.apigateway_restapi_public_with_authorizer.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_public_with_authorizer.apigateway_restapi_public_with_authorizer import ( @@ -120,15 +114,12 @@ class Test_apigateway_restapi_public_with_authorizer: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_public_with_authorizer.apigateway_restapi_public_with_authorizer.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_public_with_authorizer.apigateway_restapi_public_with_authorizer.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_public_with_authorizer.apigateway_restapi_public_with_authorizer import ( diff --git a/tests/providers/aws/services/apigateway/apigateway_logging_enabled/apigateway_logging_enabled_test.py b/tests/providers/aws/services/apigateway/apigateway_logging_enabled/apigateway_logging_enabled_test.py index 6c2a00f455..22ec854c8b 100644 --- a/tests/providers/aws/services/apigateway/apigateway_logging_enabled/apigateway_logging_enabled_test.py +++ b/tests/providers/aws/services/apigateway/apigateway_logging_enabled/apigateway_logging_enabled_test.py @@ -21,15 +21,12 @@ class Test_apigateway_restapi_logging_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_logging_enabled.apigateway_restapi_logging_enabled.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_logging_enabled.apigateway_restapi_logging_enabled.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_logging_enabled.apigateway_restapi_logging_enabled import ( @@ -94,15 +91,12 @@ class Test_apigateway_restapi_logging_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_logging_enabled.apigateway_restapi_logging_enabled.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_logging_enabled.apigateway_restapi_logging_enabled.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_logging_enabled.apigateway_restapi_logging_enabled import ( @@ -170,15 +164,12 @@ class Test_apigateway_restapi_logging_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_logging_enabled.apigateway_restapi_logging_enabled.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_logging_enabled.apigateway_restapi_logging_enabled.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_logging_enabled.apigateway_restapi_logging_enabled import ( diff --git a/tests/providers/aws/services/apigateway/apigateway_restapi_cache_encrypted/apigateway_restapi_cache_encrypted_test.py b/tests/providers/aws/services/apigateway/apigateway_restapi_cache_encrypted/apigateway_restapi_cache_encrypted_test.py index 6cc6842c91..4da331fa9d 100644 --- a/tests/providers/aws/services/apigateway/apigateway_restapi_cache_encrypted/apigateway_restapi_cache_encrypted_test.py +++ b/tests/providers/aws/services/apigateway/apigateway_restapi_cache_encrypted/apigateway_restapi_cache_encrypted_test.py @@ -21,15 +21,12 @@ class Test_apigateway_restapi_cache_encrypted: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_cache_encrypted.apigateway_restapi_cache_encrypted.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_cache_encrypted.apigateway_restapi_cache_encrypted.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_cache_encrypted.apigateway_restapi_cache_encrypted import ( @@ -99,15 +96,12 @@ class Test_apigateway_restapi_cache_encrypted: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_cache_encrypted.apigateway_restapi_cache_encrypted.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_cache_encrypted.apigateway_restapi_cache_encrypted.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_cache_encrypted.apigateway_restapi_cache_encrypted import ( @@ -191,15 +185,12 @@ class Test_apigateway_restapi_cache_encrypted: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_cache_encrypted.apigateway_restapi_cache_encrypted.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_cache_encrypted.apigateway_restapi_cache_encrypted.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_cache_encrypted.apigateway_restapi_cache_encrypted import ( @@ -283,15 +274,12 @@ class Test_apigateway_restapi_cache_encrypted: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_cache_encrypted.apigateway_restapi_cache_encrypted.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_cache_encrypted.apigateway_restapi_cache_encrypted.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_cache_encrypted.apigateway_restapi_cache_encrypted import ( diff --git a/tests/providers/aws/services/apigateway/apigateway_restapi_tracing_enabled/apigateway_restapi_tracing_enabled_test.py b/tests/providers/aws/services/apigateway/apigateway_restapi_tracing_enabled/apigateway_restapi_tracing_enabled_test.py index 6efdb56040..3f6b1cfb80 100644 --- a/tests/providers/aws/services/apigateway/apigateway_restapi_tracing_enabled/apigateway_restapi_tracing_enabled_test.py +++ b/tests/providers/aws/services/apigateway/apigateway_restapi_tracing_enabled/apigateway_restapi_tracing_enabled_test.py @@ -21,15 +21,12 @@ class Test_apigateway_restapi_tracing_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_tracing_enabled.apigateway_restapi_tracing_enabled.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_tracing_enabled.apigateway_restapi_tracing_enabled.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_tracing_enabled.apigateway_restapi_tracing_enabled import ( @@ -94,15 +91,12 @@ class Test_apigateway_restapi_tracing_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_tracing_enabled.apigateway_restapi_tracing_enabled.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_tracing_enabled.apigateway_restapi_tracing_enabled.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_tracing_enabled.apigateway_restapi_tracing_enabled import ( @@ -181,15 +175,12 @@ class Test_apigateway_restapi_tracing_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_tracing_enabled.apigateway_restapi_tracing_enabled.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_tracing_enabled.apigateway_restapi_tracing_enabled.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_tracing_enabled.apigateway_restapi_tracing_enabled import ( diff --git a/tests/providers/aws/services/apigateway/apigateway_waf_acl_attached/apigateway_waf_acl_attached_test.py b/tests/providers/aws/services/apigateway/apigateway_waf_acl_attached/apigateway_waf_acl_attached_test.py index 9e05367b2a..0468804bf9 100644 --- a/tests/providers/aws/services/apigateway/apigateway_waf_acl_attached/apigateway_waf_acl_attached_test.py +++ b/tests/providers/aws/services/apigateway/apigateway_waf_acl_attached/apigateway_waf_acl_attached_test.py @@ -21,15 +21,12 @@ class Test_apigateway_restapi_waf_acl_attached: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_waf_acl_attached.apigateway_restapi_waf_acl_attached.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_waf_acl_attached.apigateway_restapi_waf_acl_attached.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_waf_acl_attached.apigateway_restapi_waf_acl_attached import ( @@ -99,15 +96,12 @@ class Test_apigateway_restapi_waf_acl_attached: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_waf_acl_attached.apigateway_restapi_waf_acl_attached.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_waf_acl_attached.apigateway_restapi_waf_acl_attached.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_waf_acl_attached.apigateway_restapi_waf_acl_attached import ( @@ -175,15 +169,12 @@ class Test_apigateway_restapi_waf_acl_attached: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigateway.apigateway_restapi_waf_acl_attached.apigateway_restapi_waf_acl_attached.apigateway_client", - new=APIGateway(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigateway.apigateway_restapi_waf_acl_attached.apigateway_restapi_waf_acl_attached.apigateway_client", + new=APIGateway(aws_provider), ): # Test Check from prowler.providers.aws.services.apigateway.apigateway_restapi_waf_acl_attached.apigateway_restapi_waf_acl_attached import ( diff --git a/tests/providers/aws/services/apigatewayv2/apigatewayv2_access_logging_enabled/apigatewayv2_access_logging_enabled_test.py b/tests/providers/aws/services/apigatewayv2/apigatewayv2_access_logging_enabled/apigatewayv2_access_logging_enabled_test.py index 5604b2b498..a4b29f1f89 100644 --- a/tests/providers/aws/services/apigatewayv2/apigatewayv2_access_logging_enabled/apigatewayv2_access_logging_enabled_test.py +++ b/tests/providers/aws/services/apigatewayv2/apigatewayv2_access_logging_enabled/apigatewayv2_access_logging_enabled_test.py @@ -44,15 +44,12 @@ class Test_apigatewayv2_api_access_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigatewayv2.apigatewayv2_api_access_logging_enabled.apigatewayv2_api_access_logging_enabled.apigatewayv2_client", - new=ApiGatewayV2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigatewayv2.apigatewayv2_api_access_logging_enabled.apigatewayv2_api_access_logging_enabled.apigatewayv2_client", + new=ApiGatewayV2(aws_provider), ): # Test Check from prowler.providers.aws.services.apigatewayv2.apigatewayv2_api_access_logging_enabled.apigatewayv2_api_access_logging_enabled import ( @@ -78,15 +75,12 @@ class Test_apigatewayv2_api_access_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigatewayv2.apigatewayv2_api_access_logging_enabled.apigatewayv2_api_access_logging_enabled.apigatewayv2_client", - new=ApiGatewayV2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigatewayv2.apigatewayv2_api_access_logging_enabled.apigatewayv2_api_access_logging_enabled.apigatewayv2_client", + new=ApiGatewayV2(aws_provider), ): # Test Check from prowler.providers.aws.services.apigatewayv2.apigatewayv2_api_access_logging_enabled.apigatewayv2_api_access_logging_enabled import ( diff --git a/tests/providers/aws/services/apigatewayv2/apigatewayv2_authorizers_enabled/apigatewayv2_authorizers_enabled_test.py b/tests/providers/aws/services/apigatewayv2/apigatewayv2_authorizers_enabled/apigatewayv2_authorizers_enabled_test.py index 83acc7880a..d1e7a07fd2 100644 --- a/tests/providers/aws/services/apigatewayv2/apigatewayv2_authorizers_enabled/apigatewayv2_authorizers_enabled_test.py +++ b/tests/providers/aws/services/apigatewayv2/apigatewayv2_authorizers_enabled/apigatewayv2_authorizers_enabled_test.py @@ -44,15 +44,12 @@ class Test_apigatewayv2_api_authorizers_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigatewayv2.apigatewayv2_api_authorizers_enabled.apigatewayv2_api_authorizers_enabled.apigatewayv2_client", - new=ApiGatewayV2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigatewayv2.apigatewayv2_api_authorizers_enabled.apigatewayv2_api_authorizers_enabled.apigatewayv2_client", + new=ApiGatewayV2(aws_provider), ): # Test Check from prowler.providers.aws.services.apigatewayv2.apigatewayv2_api_authorizers_enabled.apigatewayv2_api_authorizers_enabled import ( @@ -83,15 +80,12 @@ class Test_apigatewayv2_api_authorizers_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.apigatewayv2.apigatewayv2_api_authorizers_enabled.apigatewayv2_api_authorizers_enabled.apigatewayv2_client", - new=ApiGatewayV2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.apigatewayv2.apigatewayv2_api_authorizers_enabled.apigatewayv2_api_authorizers_enabled.apigatewayv2_client", + new=ApiGatewayV2(aws_provider), ): # Test Check from prowler.providers.aws.services.apigatewayv2.apigatewayv2_api_authorizers_enabled.apigatewayv2_api_authorizers_enabled import ( diff --git a/tests/providers/aws/services/appsync/appsync_field_level_logging_enabled/appsync_field_level_logging_enabled_test.py b/tests/providers/aws/services/appsync/appsync_field_level_logging_enabled/appsync_field_level_logging_enabled_test.py index d9ec0f63f1..f39e23c8ff 100644 --- a/tests/providers/aws/services/appsync/appsync_field_level_logging_enabled/appsync_field_level_logging_enabled_test.py +++ b/tests/providers/aws/services/appsync/appsync_field_level_logging_enabled/appsync_field_level_logging_enabled_test.py @@ -59,15 +59,12 @@ class Test_appsync_field_level_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.appsync.appsync_field_level_logging_enabled.appsync_field_level_logging_enabled.appsync_client", - new=AppSync(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.appsync.appsync_field_level_logging_enabled.appsync_field_level_logging_enabled.appsync_client", + new=AppSync(aws_provider), ): # Test Check from prowler.providers.aws.services.appsync.appsync_field_level_logging_enabled.appsync_field_level_logging_enabled import ( @@ -84,15 +81,12 @@ class Test_appsync_field_level_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.appsync.appsync_field_level_logging_enabled.appsync_field_level_logging_enabled.appsync_client", - new=AppSync(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.appsync.appsync_field_level_logging_enabled.appsync_field_level_logging_enabled.appsync_client", + new=AppSync(aws_provider), ): # Test Check from prowler.providers.aws.services.appsync.appsync_field_level_logging_enabled.appsync_field_level_logging_enabled import ( @@ -121,15 +115,12 @@ class Test_appsync_field_level_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.appsync.appsync_field_level_logging_enabled.appsync_field_level_logging_enabled.appsync_client", - new=AppSync(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.appsync.appsync_field_level_logging_enabled.appsync_field_level_logging_enabled.appsync_client", + new=AppSync(aws_provider), ): # Test Check from prowler.providers.aws.services.appsync.appsync_field_level_logging_enabled.appsync_field_level_logging_enabled import ( diff --git a/tests/providers/aws/services/appsync/appsync_graphql_api_no_api_key_authentication/appsync_graphql_api_no_api_key_authentication_test.py b/tests/providers/aws/services/appsync/appsync_graphql_api_no_api_key_authentication/appsync_graphql_api_no_api_key_authentication_test.py index b46fdbf8a8..caf7dbf74b 100644 --- a/tests/providers/aws/services/appsync/appsync_graphql_api_no_api_key_authentication/appsync_graphql_api_no_api_key_authentication_test.py +++ b/tests/providers/aws/services/appsync/appsync_graphql_api_no_api_key_authentication/appsync_graphql_api_no_api_key_authentication_test.py @@ -75,15 +75,12 @@ class Test_appsync_graphql_api_no_api_key_authentication: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.appsync.appsync_graphql_api_no_api_key_authentication.appsync_graphql_api_no_api_key_authentication.appsync_client", - new=AppSync(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.appsync.appsync_graphql_api_no_api_key_authentication.appsync_graphql_api_no_api_key_authentication.appsync_client", + new=AppSync(aws_provider), ): # Test Check from prowler.providers.aws.services.appsync.appsync_graphql_api_no_api_key_authentication.appsync_graphql_api_no_api_key_authentication import ( @@ -101,15 +98,12 @@ class Test_appsync_graphql_api_no_api_key_authentication: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.appsync.appsync_graphql_api_no_api_key_authentication.appsync_graphql_api_no_api_key_authentication.appsync_client", - new=AppSync(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.appsync.appsync_graphql_api_no_api_key_authentication.appsync_graphql_api_no_api_key_authentication.appsync_client", + new=AppSync(aws_provider), ): # Test Check from prowler.providers.aws.services.appsync.appsync_graphql_api_no_api_key_authentication.appsync_graphql_api_no_api_key_authentication import ( @@ -127,15 +121,12 @@ class Test_appsync_graphql_api_no_api_key_authentication: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.appsync.appsync_graphql_api_no_api_key_authentication.appsync_graphql_api_no_api_key_authentication.appsync_client", - new=AppSync(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.appsync.appsync_graphql_api_no_api_key_authentication.appsync_graphql_api_no_api_key_authentication.appsync_client", + new=AppSync(aws_provider), ): # Test Check from prowler.providers.aws.services.appsync.appsync_graphql_api_no_api_key_authentication.appsync_graphql_api_no_api_key_authentication import ( @@ -165,15 +156,12 @@ class Test_appsync_graphql_api_no_api_key_authentication: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.appsync.appsync_graphql_api_no_api_key_authentication.appsync_graphql_api_no_api_key_authentication.appsync_client", - new=AppSync(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.appsync.appsync_graphql_api_no_api_key_authentication.appsync_graphql_api_no_api_key_authentication.appsync_client", + new=AppSync(aws_provider), ): # Test Check from prowler.providers.aws.services.appsync.appsync_graphql_api_no_api_key_authentication.appsync_graphql_api_no_api_key_authentication import ( diff --git a/tests/providers/aws/services/athena/athena_workgroup_encryption/athena_workgroup_encryption_test.py b/tests/providers/aws/services/athena/athena_workgroup_encryption/athena_workgroup_encryption_test.py index e3817a25ab..cc04f6d2a9 100644 --- a/tests/providers/aws/services/athena/athena_workgroup_encryption/athena_workgroup_encryption_test.py +++ b/tests/providers/aws/services/athena/athena_workgroup_encryption/athena_workgroup_encryption_test.py @@ -21,15 +21,12 @@ class Test_athena_workgroup_encryption: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.athena.athena_workgroup_encryption.athena_workgroup_encryption.athena_client", - new=Athena(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.athena.athena_workgroup_encryption.athena_workgroup_encryption.athena_client", + new=Athena(aws_provider), ): from prowler.providers.aws.services.athena.athena_workgroup_encryption.athena_workgroup_encryption import ( athena_workgroup_encryption, @@ -56,15 +53,12 @@ class Test_athena_workgroup_encryption: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) aws_provider._scan_unused_services = False - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.athena.athena_workgroup_encryption.athena_workgroup_encryption.athena_client", - new=Athena(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.athena.athena_workgroup_encryption.athena_workgroup_encryption.athena_client", + new=Athena(aws_provider), ): from prowler.providers.aws.services.athena.athena_workgroup_encryption.athena_workgroup_encryption import ( athena_workgroup_encryption, @@ -83,15 +77,12 @@ class Test_athena_workgroup_encryption: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.athena.athena_workgroup_encryption.athena_workgroup_encryption.athena_client", - new=Athena(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.athena.athena_workgroup_encryption.athena_workgroup_encryption.athena_client", + new=Athena(aws_provider), ): from prowler.providers.aws.services.athena.athena_workgroup_encryption.athena_workgroup_encryption import ( athena_workgroup_encryption, diff --git a/tests/providers/aws/services/athena/athena_workgroup_enforce_configuration/athena_workgroup_enforce_configuration_test.py b/tests/providers/aws/services/athena/athena_workgroup_enforce_configuration/athena_workgroup_enforce_configuration_test.py index 6e4fc7a995..3b45596ff5 100644 --- a/tests/providers/aws/services/athena/athena_workgroup_enforce_configuration/athena_workgroup_enforce_configuration_test.py +++ b/tests/providers/aws/services/athena/athena_workgroup_enforce_configuration/athena_workgroup_enforce_configuration_test.py @@ -21,15 +21,12 @@ class Test_athena_workgroup_enforce_configuration: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.athena.athena_workgroup_enforce_configuration.athena_workgroup_enforce_configuration.athena_client", - new=Athena(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.athena.athena_workgroup_enforce_configuration.athena_workgroup_enforce_configuration.athena_client", + new=Athena(aws_provider), ): from prowler.providers.aws.services.athena.athena_workgroup_enforce_configuration.athena_workgroup_enforce_configuration import ( athena_workgroup_enforce_configuration, @@ -56,15 +53,12 @@ class Test_athena_workgroup_enforce_configuration: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) aws_provider._scan_unused_services = False - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.athena.athena_workgroup_enforce_configuration.athena_workgroup_enforce_configuration.athena_client", - new=Athena(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.athena.athena_workgroup_enforce_configuration.athena_workgroup_enforce_configuration.athena_client", + new=Athena(aws_provider), ): from prowler.providers.aws.services.athena.athena_workgroup_enforce_configuration.athena_workgroup_enforce_configuration import ( athena_workgroup_enforce_configuration, @@ -83,15 +77,12 @@ class Test_athena_workgroup_enforce_configuration: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.athena.athena_workgroup_enforce_configuration.athena_workgroup_enforce_configuration.athena_client", - new=Athena(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.athena.athena_workgroup_enforce_configuration.athena_workgroup_enforce_configuration.athena_client", + new=Athena(aws_provider), ): from prowler.providers.aws.services.athena.athena_workgroup_enforce_configuration.athena_workgroup_enforce_configuration import ( athena_workgroup_enforce_configuration, diff --git a/tests/providers/aws/services/athena/athena_workgroup_logging_enabled/athena_workgroup_logging_enabled_test.py b/tests/providers/aws/services/athena/athena_workgroup_logging_enabled/athena_workgroup_logging_enabled_test.py index feff3a8405..0c735f6b1c 100644 --- a/tests/providers/aws/services/athena/athena_workgroup_logging_enabled/athena_workgroup_logging_enabled_test.py +++ b/tests/providers/aws/services/athena/athena_workgroup_logging_enabled/athena_workgroup_logging_enabled_test.py @@ -20,15 +20,12 @@ class Test_athena_workgroup_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.athena.athena_workgroup_logging_enabled.athena_workgroup_logging_enabled.athena_client", - new=Athena(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.athena.athena_workgroup_logging_enabled.athena_workgroup_logging_enabled.athena_client", + new=Athena(aws_provider), ): from prowler.providers.aws.services.athena.athena_workgroup_logging_enabled.athena_workgroup_logging_enabled import ( athena_workgroup_logging_enabled, @@ -55,15 +52,12 @@ class Test_athena_workgroup_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) aws_provider._scan_unused_services = False - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.athena.athena_workgroup_logging_enabled.athena_workgroup_logging_enabled.athena_client", - new=Athena(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.athena.athena_workgroup_logging_enabled.athena_workgroup_logging_enabled.athena_client", + new=Athena(aws_provider), ): from prowler.providers.aws.services.athena.athena_workgroup_logging_enabled.athena_workgroup_logging_enabled import ( athena_workgroup_logging_enabled, @@ -104,15 +98,12 @@ class Test_athena_workgroup_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.athena.athena_workgroup_logging_enabled.athena_workgroup_logging_enabled.athena_client", - new=Athena(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.athena.athena_workgroup_logging_enabled.athena_workgroup_logging_enabled.athena_client", + new=Athena(aws_provider), ): from prowler.providers.aws.services.athena.athena_workgroup_logging_enabled.athena_workgroup_logging_enabled import ( athena_workgroup_logging_enabled, diff --git a/tests/providers/aws/services/autoscaling/autoscaling_find_secrets_ec2_launch_configuration/autoscaling_find_secrets_ec2_launch_configuration_test.py b/tests/providers/aws/services/autoscaling/autoscaling_find_secrets_ec2_launch_configuration/autoscaling_find_secrets_ec2_launch_configuration_test.py index 1005761834..1ff84a0678 100644 --- a/tests/providers/aws/services/autoscaling/autoscaling_find_secrets_ec2_launch_configuration/autoscaling_find_secrets_ec2_launch_configuration_test.py +++ b/tests/providers/aws/services/autoscaling/autoscaling_find_secrets_ec2_launch_configuration/autoscaling_find_secrets_ec2_launch_configuration_test.py @@ -23,15 +23,12 @@ class Test_autoscaling_find_secrets_ec2_launch_configuration: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration import ( @@ -66,15 +63,12 @@ class Test_autoscaling_find_secrets_ec2_launch_configuration: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_client", + new=AutoScaling(aws_provider), ): from prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration import ( autoscaling_find_secrets_ec2_launch_configuration, @@ -116,15 +110,12 @@ class Test_autoscaling_find_secrets_ec2_launch_configuration: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_client", + new=AutoScaling(aws_provider), ): from prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration import ( autoscaling_find_secrets_ec2_launch_configuration, @@ -171,15 +162,12 @@ class Test_autoscaling_find_secrets_ec2_launch_configuration: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_client", + new=AutoScaling(aws_provider), ): from prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration import ( autoscaling_find_secrets_ec2_launch_configuration, @@ -220,15 +208,12 @@ class Test_autoscaling_find_secrets_ec2_launch_configuration: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_client", + new=AutoScaling(aws_provider), ): from prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration import ( autoscaling_find_secrets_ec2_launch_configuration, @@ -276,15 +261,12 @@ class Test_autoscaling_find_secrets_ec2_launch_configuration: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_client", + new=AutoScaling(aws_provider), ): from prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration import ( autoscaling_find_secrets_ec2_launch_configuration, @@ -324,15 +306,12 @@ class Test_autoscaling_find_secrets_ec2_launch_configuration: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_client", + new=AutoScaling(aws_provider), ): from prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration import ( autoscaling_find_secrets_ec2_launch_configuration, @@ -364,15 +343,12 @@ class Test_autoscaling_find_secrets_ec2_launch_configuration: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_client", + new=AutoScaling(aws_provider), ): from prowler.providers.aws.services.autoscaling.autoscaling_find_secrets_ec2_launch_configuration.autoscaling_find_secrets_ec2_launch_configuration import ( autoscaling_find_secrets_ec2_launch_configuration, diff --git a/tests/providers/aws/services/autoscaling/autoscaling_group_capacity_rebalance_enabled/autoscaling_group_capacity_rebalance_enabled_test.py b/tests/providers/aws/services/autoscaling/autoscaling_group_capacity_rebalance_enabled/autoscaling_group_capacity_rebalance_enabled_test.py index 59dc1b7dc7..fc06c75dea 100644 --- a/tests/providers/aws/services/autoscaling/autoscaling_group_capacity_rebalance_enabled/autoscaling_group_capacity_rebalance_enabled_test.py +++ b/tests/providers/aws/services/autoscaling/autoscaling_group_capacity_rebalance_enabled/autoscaling_group_capacity_rebalance_enabled_test.py @@ -18,15 +18,12 @@ class Test_autoscaling_group_capacity_rebalance_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_capacity_rebalance_enabled.autoscaling_group_capacity_rebalance_enabled.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_capacity_rebalance_enabled.autoscaling_group_capacity_rebalance_enabled.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_capacity_rebalance_enabled.autoscaling_group_capacity_rebalance_enabled import ( @@ -92,15 +89,12 @@ class Test_autoscaling_group_capacity_rebalance_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_capacity_rebalance_enabled.autoscaling_group_capacity_rebalance_enabled.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_capacity_rebalance_enabled.autoscaling_group_capacity_rebalance_enabled.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_capacity_rebalance_enabled.autoscaling_group_capacity_rebalance_enabled import ( @@ -174,15 +168,12 @@ class Test_autoscaling_group_capacity_rebalance_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_capacity_rebalance_enabled.autoscaling_group_capacity_rebalance_enabled.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_capacity_rebalance_enabled.autoscaling_group_capacity_rebalance_enabled.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_capacity_rebalance_enabled.autoscaling_group_capacity_rebalance_enabled import ( diff --git a/tests/providers/aws/services/autoscaling/autoscaling_group_elb_health_check_enabled/autoscaling_group_elb_health_check_enabled_test.py b/tests/providers/aws/services/autoscaling/autoscaling_group_elb_health_check_enabled/autoscaling_group_elb_health_check_enabled_test.py index b046822e6d..df72d88660 100644 --- a/tests/providers/aws/services/autoscaling/autoscaling_group_elb_health_check_enabled/autoscaling_group_elb_health_check_enabled_test.py +++ b/tests/providers/aws/services/autoscaling/autoscaling_group_elb_health_check_enabled/autoscaling_group_elb_health_check_enabled_test.py @@ -18,15 +18,12 @@ class Test_autoscaling_group_elb_health_check_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled import ( @@ -91,15 +88,12 @@ class Test_autoscaling_group_elb_health_check_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled import ( @@ -173,15 +167,12 @@ class Test_autoscaling_group_elb_health_check_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled import ( @@ -254,15 +245,12 @@ class Test_autoscaling_group_elb_health_check_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled import ( @@ -330,15 +318,12 @@ class Test_autoscaling_group_elb_health_check_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled import ( @@ -379,15 +364,12 @@ class Test_autoscaling_group_elb_health_check_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled import ( @@ -427,15 +409,12 @@ class Test_autoscaling_group_elb_health_check_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_elb_health_check_enabled.autoscaling_group_elb_health_check_enabled import ( diff --git a/tests/providers/aws/services/autoscaling/autoscaling_group_launch_configuration_no_public_ip/autoscaling_group_launch_configuration_no_public_ip_test.py b/tests/providers/aws/services/autoscaling/autoscaling_group_launch_configuration_no_public_ip/autoscaling_group_launch_configuration_no_public_ip_test.py index a6d0e28260..a911e0169d 100644 --- a/tests/providers/aws/services/autoscaling/autoscaling_group_launch_configuration_no_public_ip/autoscaling_group_launch_configuration_no_public_ip_test.py +++ b/tests/providers/aws/services/autoscaling/autoscaling_group_launch_configuration_no_public_ip/autoscaling_group_launch_configuration_no_public_ip_test.py @@ -18,15 +18,12 @@ class Test_autoscaling_group_launch_configuration_no_public_ip: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_no_public_ip.autoscaling_group_launch_configuration_no_public_ip.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_no_public_ip.autoscaling_group_launch_configuration_no_public_ip.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_no_public_ip.autoscaling_group_launch_configuration_no_public_ip import ( @@ -69,15 +66,12 @@ class Test_autoscaling_group_launch_configuration_no_public_ip: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_no_public_ip.autoscaling_group_launch_configuration_no_public_ip.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_no_public_ip.autoscaling_group_launch_configuration_no_public_ip.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_no_public_ip.autoscaling_group_launch_configuration_no_public_ip import ( @@ -129,15 +123,12 @@ class Test_autoscaling_group_launch_configuration_no_public_ip: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_no_public_ip.autoscaling_group_launch_configuration_no_public_ip.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_no_public_ip.autoscaling_group_launch_configuration_no_public_ip.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_no_public_ip.autoscaling_group_launch_configuration_no_public_ip import ( @@ -186,15 +177,12 @@ class Test_autoscaling_group_launch_configuration_no_public_ip: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_no_public_ip.autoscaling_group_launch_configuration_no_public_ip.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_no_public_ip.autoscaling_group_launch_configuration_no_public_ip.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_no_public_ip.autoscaling_group_launch_configuration_no_public_ip import ( diff --git a/tests/providers/aws/services/autoscaling/autoscaling_group_launch_configuration_requires_imdsv2/autoscaling_group_launch_configuration_requires_imdsv2_test.py b/tests/providers/aws/services/autoscaling/autoscaling_group_launch_configuration_requires_imdsv2/autoscaling_group_launch_configuration_requires_imdsv2_test.py index 6a8e3b54a9..a9239a3ed1 100644 --- a/tests/providers/aws/services/autoscaling/autoscaling_group_launch_configuration_requires_imdsv2/autoscaling_group_launch_configuration_requires_imdsv2_test.py +++ b/tests/providers/aws/services/autoscaling/autoscaling_group_launch_configuration_requires_imdsv2/autoscaling_group_launch_configuration_requires_imdsv2_test.py @@ -18,15 +18,12 @@ class Test_autoscaling_group_launch_configuration_requires_imdsv2: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_requires_imdsv2.autoscaling_group_launch_configuration_requires_imdsv2.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_requires_imdsv2.autoscaling_group_launch_configuration_requires_imdsv2.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_requires_imdsv2.autoscaling_group_launch_configuration_requires_imdsv2 import ( @@ -73,15 +70,12 @@ class Test_autoscaling_group_launch_configuration_requires_imdsv2: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_requires_imdsv2.autoscaling_group_launch_configuration_requires_imdsv2.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_requires_imdsv2.autoscaling_group_launch_configuration_requires_imdsv2.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_requires_imdsv2.autoscaling_group_launch_configuration_requires_imdsv2 import ( @@ -131,15 +125,12 @@ class Test_autoscaling_group_launch_configuration_requires_imdsv2: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_requires_imdsv2.autoscaling_group_launch_configuration_requires_imdsv2.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_requires_imdsv2.autoscaling_group_launch_configuration_requires_imdsv2.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_requires_imdsv2.autoscaling_group_launch_configuration_requires_imdsv2 import ( @@ -186,15 +177,12 @@ class Test_autoscaling_group_launch_configuration_requires_imdsv2: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_requires_imdsv2.autoscaling_group_launch_configuration_requires_imdsv2.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_requires_imdsv2.autoscaling_group_launch_configuration_requires_imdsv2.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_launch_configuration_requires_imdsv2.autoscaling_group_launch_configuration_requires_imdsv2 import ( diff --git a/tests/providers/aws/services/autoscaling/autoscaling_group_multiple_az/autoscaling_group_multiple_az_test.py b/tests/providers/aws/services/autoscaling/autoscaling_group_multiple_az/autoscaling_group_multiple_az_test.py index b39d67ef18..cb28479249 100644 --- a/tests/providers/aws/services/autoscaling/autoscaling_group_multiple_az/autoscaling_group_multiple_az_test.py +++ b/tests/providers/aws/services/autoscaling/autoscaling_group_multiple_az/autoscaling_group_multiple_az_test.py @@ -18,15 +18,12 @@ class Test_autoscaling_group_multiple_az: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_az.autoscaling_group_multiple_az.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_az.autoscaling_group_multiple_az.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_az.autoscaling_group_multiple_az import ( @@ -68,15 +65,12 @@ class Test_autoscaling_group_multiple_az: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_az.autoscaling_group_multiple_az.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_az.autoscaling_group_multiple_az.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_az.autoscaling_group_multiple_az import ( @@ -127,15 +121,12 @@ class Test_autoscaling_group_multiple_az: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_az.autoscaling_group_multiple_az.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_az.autoscaling_group_multiple_az.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_az.autoscaling_group_multiple_az import ( @@ -197,15 +188,12 @@ class Test_autoscaling_group_multiple_az: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_az.autoscaling_group_multiple_az.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_az.autoscaling_group_multiple_az.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_az.autoscaling_group_multiple_az import ( diff --git a/tests/providers/aws/services/autoscaling/autoscaling_group_multiple_instance_types/autoscaling_group_multiple_instance_types_test.py b/tests/providers/aws/services/autoscaling/autoscaling_group_multiple_instance_types/autoscaling_group_multiple_instance_types_test.py index 37e9653fdb..9dc2a24976 100644 --- a/tests/providers/aws/services/autoscaling/autoscaling_group_multiple_instance_types/autoscaling_group_multiple_instance_types_test.py +++ b/tests/providers/aws/services/autoscaling/autoscaling_group_multiple_instance_types/autoscaling_group_multiple_instance_types_test.py @@ -112,15 +112,12 @@ class Test_autoscaling_group_multiple_instance_types: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_instance_types.autoscaling_group_multiple_instance_types.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_instance_types.autoscaling_group_multiple_instance_types.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_instance_types.autoscaling_group_multiple_instance_types import ( @@ -162,15 +159,12 @@ class Test_autoscaling_group_multiple_instance_types: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_instance_types.autoscaling_group_multiple_instance_types.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_instance_types.autoscaling_group_multiple_instance_types.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_instance_types.autoscaling_group_multiple_instance_types import ( @@ -221,15 +215,12 @@ class Test_autoscaling_group_multiple_instance_types: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_instance_types.autoscaling_group_multiple_instance_types.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_instance_types.autoscaling_group_multiple_instance_types.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_instance_types.autoscaling_group_multiple_instance_types import ( @@ -282,15 +273,12 @@ class Test_autoscaling_group_multiple_instance_types: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_instance_types.autoscaling_group_multiple_instance_types.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_instance_types.autoscaling_group_multiple_instance_types.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_instance_types.autoscaling_group_multiple_instance_types import ( @@ -345,15 +333,12 @@ class Test_autoscaling_group_multiple_instance_types: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_instance_types.autoscaling_group_multiple_instance_types.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_instance_types.autoscaling_group_multiple_instance_types.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_instance_types.autoscaling_group_multiple_instance_types import ( @@ -406,15 +391,12 @@ class Test_autoscaling_group_multiple_instance_types: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_instance_types.autoscaling_group_multiple_instance_types.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_instance_types.autoscaling_group_multiple_instance_types.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_multiple_instance_types.autoscaling_group_multiple_instance_types import ( diff --git a/tests/providers/aws/services/autoscaling/autoscaling_group_using_ec2_launch_template/autoscaling_group_using_ec2_launch_template_test.py b/tests/providers/aws/services/autoscaling/autoscaling_group_using_ec2_launch_template/autoscaling_group_using_ec2_launch_template_test.py index 2f19556842..34b7d7cec4 100644 --- a/tests/providers/aws/services/autoscaling/autoscaling_group_using_ec2_launch_template/autoscaling_group_using_ec2_launch_template_test.py +++ b/tests/providers/aws/services/autoscaling/autoscaling_group_using_ec2_launch_template/autoscaling_group_using_ec2_launch_template_test.py @@ -18,15 +18,12 @@ class Test_autoscaling_group_using_ec2_launch_template: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_using_ec2_launch_template.autoscaling_group_using_ec2_launch_template.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_using_ec2_launch_template.autoscaling_group_using_ec2_launch_template.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_using_ec2_launch_template.autoscaling_group_using_ec2_launch_template import ( @@ -72,15 +69,12 @@ class Test_autoscaling_group_using_ec2_launch_template: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_using_ec2_launch_template.autoscaling_group_using_ec2_launch_template.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_using_ec2_launch_template.autoscaling_group_using_ec2_launch_template.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_using_ec2_launch_template.autoscaling_group_using_ec2_launch_template import ( @@ -148,15 +142,12 @@ class Test_autoscaling_group_using_ec2_launch_template: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_using_ec2_launch_template.autoscaling_group_using_ec2_launch_template.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_using_ec2_launch_template.autoscaling_group_using_ec2_launch_template.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_using_ec2_launch_template.autoscaling_group_using_ec2_launch_template import ( @@ -206,15 +197,12 @@ class Test_autoscaling_group_using_ec2_launch_template: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.autoscaling.autoscaling_group_using_ec2_launch_template.autoscaling_group_using_ec2_launch_template.autoscaling_client", - new=AutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.autoscaling.autoscaling_group_using_ec2_launch_template.autoscaling_group_using_ec2_launch_template.autoscaling_client", + new=AutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.autoscaling.autoscaling_group_using_ec2_launch_template.autoscaling_group_using_ec2_launch_template import ( diff --git a/tests/providers/aws/services/awslambda/awslambda_function_inside_vpc/awslambda_function_inside_vpc_test.py b/tests/providers/aws/services/awslambda/awslambda_function_inside_vpc/awslambda_function_inside_vpc_test.py index e4f6ab1b97..05a2d89f4e 100644 --- a/tests/providers/aws/services/awslambda/awslambda_function_inside_vpc/awslambda_function_inside_vpc_test.py +++ b/tests/providers/aws/services/awslambda/awslambda_function_inside_vpc/awslambda_function_inside_vpc_test.py @@ -13,15 +13,12 @@ class Test_awslambda_function_inside_vpc: aws_provider = set_mocked_aws_provider(AWS_REGION_EU_WEST_1) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_inside_vpc.awslambda_function_inside_vpc.awslambda_client", - new=Lambda(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_inside_vpc.awslambda_function_inside_vpc.awslambda_client", + new=Lambda(aws_provider), ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_inside_vpc.awslambda_function_inside_vpc import ( @@ -72,15 +69,12 @@ class Test_awslambda_function_inside_vpc: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_inside_vpc.awslambda_function_inside_vpc.awslambda_client", - new=Lambda(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_inside_vpc.awslambda_function_inside_vpc.awslambda_client", + new=Lambda(aws_provider), ): from prowler.providers.aws.services.awslambda.awslambda_function_inside_vpc.awslambda_function_inside_vpc import ( awslambda_function_inside_vpc, @@ -135,15 +129,12 @@ class Test_awslambda_function_inside_vpc: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_inside_vpc.awslambda_function_inside_vpc.awslambda_client", - new=Lambda(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_inside_vpc.awslambda_function_inside_vpc.awslambda_client", + new=Lambda(aws_provider), ): from prowler.providers.aws.services.awslambda.awslambda_function_inside_vpc.awslambda_function_inside_vpc import ( awslambda_function_inside_vpc, diff --git a/tests/providers/aws/services/awslambda/awslambda_function_no_secrets_in_code/awslambda_function_no_secrets_in_code_test.py b/tests/providers/aws/services/awslambda/awslambda_function_no_secrets_in_code/awslambda_function_no_secrets_in_code_test.py index 5f97082c1b..8256f13ad3 100644 --- a/tests/providers/aws/services/awslambda/awslambda_function_no_secrets_in_code/awslambda_function_no_secrets_in_code_test.py +++ b/tests/providers/aws/services/awslambda/awslambda_function_no_secrets_in_code/awslambda_function_no_secrets_in_code_test.py @@ -76,15 +76,12 @@ class Test_awslambda_function_no_secrets_in_code: lambda_client = mock.MagicMock lambda_client.functions = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_code.awslambda_function_no_secrets_in_code.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_code.awslambda_function_no_secrets_in_code.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_code.awslambda_function_no_secrets_in_code import ( @@ -101,15 +98,12 @@ class Test_awslambda_function_no_secrets_in_code: lambda_client.functions = {LAMBDA_FUNCTION_ARN: create_lambda_function()} lambda_client._get_function_code = mock_get_function_codewith_secrets lambda_client.audit_config = {"secrets_ignore_patterns": []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_code.awslambda_function_no_secrets_in_code.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_code.awslambda_function_no_secrets_in_code.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_code.awslambda_function_no_secrets_in_code import ( @@ -137,15 +131,12 @@ class Test_awslambda_function_no_secrets_in_code: lambda_client._get_function_code = mock_get_function_codewithout_secrets lambda_client.audit_config = {"secrets_ignore_patterns": []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_code.awslambda_function_no_secrets_in_code.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_code.awslambda_function_no_secrets_in_code.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_code.awslambda_function_no_secrets_in_code import ( @@ -173,15 +164,12 @@ class Test_awslambda_function_no_secrets_in_code: lambda_client._get_function_code = mock_get_function_codewith_metadata_api lambda_client.audit_config = {"secrets_ignore_patterns": []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_code.awslambda_function_no_secrets_in_code.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_code.awslambda_function_no_secrets_in_code.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_code.awslambda_function_no_secrets_in_code import ( diff --git a/tests/providers/aws/services/awslambda/awslambda_function_no_secrets_in_variables/awslambda_function_no_secrets_in_variables_test.py b/tests/providers/aws/services/awslambda/awslambda_function_no_secrets_in_variables/awslambda_function_no_secrets_in_variables_test.py index 6ae517dfe2..bd38a225eb 100644 --- a/tests/providers/aws/services/awslambda/awslambda_function_no_secrets_in_variables/awslambda_function_no_secrets_in_variables_test.py +++ b/tests/providers/aws/services/awslambda/awslambda_function_no_secrets_in_variables/awslambda_function_no_secrets_in_variables_test.py @@ -14,15 +14,12 @@ class Test_awslambda_function_no_secrets_in_variables: lambda_client.functions = {} lambda_client.audit_config = {"secrets_ignore_patterns": []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_variables.awslambda_function_no_secrets_in_variables.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_variables.awslambda_function_no_secrets_in_variables.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_variables.awslambda_function_no_secrets_in_variables import ( @@ -53,15 +50,12 @@ class Test_awslambda_function_no_secrets_in_variables: lambda_client.audit_config = {"secrets_ignore_patterns": []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_variables.awslambda_function_no_secrets_in_variables.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_variables.awslambda_function_no_secrets_in_variables.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_variables.awslambda_function_no_secrets_in_variables import ( @@ -101,15 +95,12 @@ class Test_awslambda_function_no_secrets_in_variables: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_variables.awslambda_function_no_secrets_in_variables.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_variables.awslambda_function_no_secrets_in_variables.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_variables.awslambda_function_no_secrets_in_variables import ( @@ -149,15 +140,12 @@ class Test_awslambda_function_no_secrets_in_variables: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_variables.awslambda_function_no_secrets_in_variables.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_variables.awslambda_function_no_secrets_in_variables.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_variables.awslambda_function_no_secrets_in_variables import ( @@ -195,15 +183,12 @@ class Test_awslambda_function_no_secrets_in_variables: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_variables.awslambda_function_no_secrets_in_variables.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_variables.awslambda_function_no_secrets_in_variables.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_variables.awslambda_function_no_secrets_in_variables import ( @@ -242,15 +227,12 @@ class Test_awslambda_function_no_secrets_in_variables: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_variables.awslambda_function_no_secrets_in_variables.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_variables.awslambda_function_no_secrets_in_variables.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_no_secrets_in_variables.awslambda_function_no_secrets_in_variables import ( diff --git a/tests/providers/aws/services/awslambda/awslambda_function_not_publicly_accessible/awslambda_function_not_publicly_accessible_fixer_test.py b/tests/providers/aws/services/awslambda/awslambda_function_not_publicly_accessible/awslambda_function_not_publicly_accessible_fixer_test.py index 0535856c15..958a77c00e 100644 --- a/tests/providers/aws/services/awslambda/awslambda_function_not_publicly_accessible/awslambda_function_not_publicly_accessible_fixer_test.py +++ b/tests/providers/aws/services/awslambda/awslambda_function_not_publicly_accessible/awslambda_function_not_publicly_accessible_fixer_test.py @@ -61,15 +61,12 @@ class Test_awslambda_function_not_publicly_accessible_fixer: from prowler.providers.aws.services.awslambda.awslambda_service import Lambda - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible_fixer.awslambda_client", - new=Lambda(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible_fixer.awslambda_client", + new=Lambda(aws_provider), ): # Test Fixer from prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible_fixer import ( @@ -129,15 +126,12 @@ class Test_awslambda_function_not_publicly_accessible_fixer: from prowler.providers.aws.services.awslambda.awslambda_service import Lambda - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible_fixer.awslambda_client", - new=Lambda(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible_fixer.awslambda_client", + new=Lambda(aws_provider), ): # Test Fixer from prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible_fixer import ( @@ -196,15 +190,12 @@ class Test_awslambda_function_not_publicly_accessible_fixer: from prowler.providers.aws.services.awslambda.awslambda_service import Lambda - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible_fixer.awslambda_client", - new=Lambda(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible_fixer.awslambda_client", + new=Lambda(aws_provider), ): # Test Fixer from prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible_fixer import ( @@ -262,15 +253,12 @@ class Test_awslambda_function_not_publicly_accessible_fixer: from prowler.providers.aws.services.awslambda.awslambda_service import Lambda - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible_fixer.awslambda_client", - new=Lambda(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible_fixer.awslambda_client", + new=Lambda(aws_provider), ): # Test Fixer from prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible_fixer import ( diff --git a/tests/providers/aws/services/awslambda/awslambda_function_not_publicly_accessible/awslambda_function_not_publicly_accessible_test.py b/tests/providers/aws/services/awslambda/awslambda_function_not_publicly_accessible/awslambda_function_not_publicly_accessible_test.py index ef3168433b..99aaec831a 100644 --- a/tests/providers/aws/services/awslambda/awslambda_function_not_publicly_accessible/awslambda_function_not_publicly_accessible_test.py +++ b/tests/providers/aws/services/awslambda/awslambda_function_not_publicly_accessible/awslambda_function_not_publicly_accessible_test.py @@ -19,15 +19,12 @@ class Test_awslambda_function_not_publicly_accessible: from prowler.providers.aws.services.awslambda.awslambda_service import Lambda - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible.awslambda_client", - new=Lambda(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible.awslambda_client", + new=Lambda(aws_provider), ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible import ( @@ -88,15 +85,12 @@ class Test_awslambda_function_not_publicly_accessible: from prowler.providers.aws.services.awslambda.awslambda_service import Lambda - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible.awslambda_client", - new=Lambda(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible.awslambda_client", + new=Lambda(aws_provider), ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible import ( @@ -168,15 +162,12 @@ class Test_awslambda_function_not_publicly_accessible: from prowler.providers.aws.services.awslambda.awslambda_service import Lambda - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible.awslambda_client", - new=Lambda(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible.awslambda_client", + new=Lambda(aws_provider), ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible import ( @@ -247,15 +238,12 @@ class Test_awslambda_function_not_publicly_accessible: from prowler.providers.aws.services.awslambda.awslambda_service import Lambda - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible.awslambda_client", - new=Lambda(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible.awslambda_client", + new=Lambda(aws_provider), ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible import ( @@ -309,15 +297,12 @@ class Test_awslambda_function_not_publicly_accessible: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible import ( @@ -485,15 +470,12 @@ class Test_awslambda_function_not_publicly_accessible: from prowler.providers.aws.services.awslambda.awslambda_service import Lambda - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible.awslambda_client", - new=Lambda(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible.awslambda_client", + new=Lambda(aws_provider), ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible import ( @@ -549,15 +531,12 @@ class Test_awslambda_function_not_publicly_accessible: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible.awslambda_client", + new=lambda_client, ): from prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible import ( awslambda_function_not_publicly_accessible, @@ -612,15 +591,12 @@ class Test_awslambda_function_not_publicly_accessible: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible.awslambda_client", + new=lambda_client, ): from prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible import ( awslambda_function_not_publicly_accessible, @@ -687,15 +663,12 @@ class Test_awslambda_function_not_publicly_accessible: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible.awslambda_client", + new=lambda_client, ): from prowler.providers.aws.services.awslambda.awslambda_function_not_publicly_accessible.awslambda_function_not_publicly_accessible import ( awslambda_function_not_publicly_accessible, diff --git a/tests/providers/aws/services/awslambda/awslambda_function_url_cors_policy/awslambda_function_url_cors_policy_test.py b/tests/providers/aws/services/awslambda/awslambda_function_url_cors_policy/awslambda_function_url_cors_policy_test.py index 4413000a6f..ffc15ec91b 100644 --- a/tests/providers/aws/services/awslambda/awslambda_function_url_cors_policy/awslambda_function_url_cors_policy_test.py +++ b/tests/providers/aws/services/awslambda/awslambda_function_url_cors_policy/awslambda_function_url_cors_policy_test.py @@ -21,15 +21,12 @@ class Test_awslambda_function_url_cors_policy: lambda_client = mock.MagicMock lambda_client.functions = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_url_cors_policy.awslambda_function_url_cors_policy.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_url_cors_policy.awslambda_function_url_cors_policy.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_url_cors_policy.awslambda_function_url_cors_policy import ( @@ -61,15 +58,12 @@ class Test_awslambda_function_url_cors_policy: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_url_cors_policy.awslambda_function_url_cors_policy.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_url_cors_policy.awslambda_function_url_cors_policy.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_url_cors_policy.awslambda_function_url_cors_policy import ( @@ -110,15 +104,12 @@ class Test_awslambda_function_url_cors_policy: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_url_cors_policy.awslambda_function_url_cors_policy.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_url_cors_policy.awslambda_function_url_cors_policy.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_url_cors_policy.awslambda_function_url_cors_policy import ( @@ -161,15 +152,12 @@ class Test_awslambda_function_url_cors_policy: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_url_cors_policy.awslambda_function_url_cors_policy.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_url_cors_policy.awslambda_function_url_cors_policy.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_url_cors_policy.awslambda_function_url_cors_policy import ( diff --git a/tests/providers/aws/services/awslambda/awslambda_function_url_public/awslambda_function_url_public_test.py b/tests/providers/aws/services/awslambda/awslambda_function_url_public/awslambda_function_url_public_test.py index b198f39fd2..2eaf43478d 100644 --- a/tests/providers/aws/services/awslambda/awslambda_function_url_public/awslambda_function_url_public_test.py +++ b/tests/providers/aws/services/awslambda/awslambda_function_url_public/awslambda_function_url_public_test.py @@ -18,15 +18,12 @@ class Test_awslambda_function_url_public: lambda_client = mock.MagicMock lambda_client.functions = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_url_public.awslambda_function_url_public.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_url_public.awslambda_function_url_public.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_url_public.awslambda_function_url_public import ( @@ -58,15 +55,12 @@ class Test_awslambda_function_url_public: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_url_public.awslambda_function_url_public.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_url_public.awslambda_function_url_public.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_url_public.awslambda_function_url_public import ( @@ -107,15 +101,12 @@ class Test_awslambda_function_url_public: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_url_public.awslambda_function_url_public.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_url_public.awslambda_function_url_public.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_url_public.awslambda_function_url_public import ( diff --git a/tests/providers/aws/services/awslambda/awslambda_function_using_supported_runtimes/awslambda_function_using_supported_runtimes_test.py b/tests/providers/aws/services/awslambda/awslambda_function_using_supported_runtimes/awslambda_function_using_supported_runtimes_test.py index db07987839..130587da3e 100644 --- a/tests/providers/aws/services/awslambda/awslambda_function_using_supported_runtimes/awslambda_function_using_supported_runtimes_test.py +++ b/tests/providers/aws/services/awslambda/awslambda_function_using_supported_runtimes/awslambda_function_using_supported_runtimes_test.py @@ -13,15 +13,12 @@ class Test_awslambda_function_using_supported_runtimes: lambda_client = mock.MagicMock lambda_client.functions = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_using_supported_runtimes.awslambda_function_using_supported_runtimes.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_using_supported_runtimes.awslambda_function_using_supported_runtimes.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_using_supported_runtimes.awslambda_function_using_supported_runtimes import ( @@ -75,15 +72,12 @@ class Test_awslambda_function_using_supported_runtimes: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_using_supported_runtimes.awslambda_function_using_supported_runtimes.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_using_supported_runtimes.awslambda_function_using_supported_runtimes.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_using_supported_runtimes.awslambda_function_using_supported_runtimes import ( @@ -146,15 +140,12 @@ class Test_awslambda_function_using_supported_runtimes: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_using_supported_runtimes.awslambda_function_using_supported_runtimes.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_using_supported_runtimes.awslambda_function_using_supported_runtimes.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_using_supported_runtimes.awslambda_function_using_supported_runtimes import ( @@ -215,15 +206,12 @@ class Test_awslambda_function_using_supported_runtimes: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_using_supported_runtimes.awslambda_function_using_supported_runtimes.awslambda_client", - new=lambda_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_using_supported_runtimes.awslambda_function_using_supported_runtimes.awslambda_client", + new=lambda_client, ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_using_supported_runtimes.awslambda_function_using_supported_runtimes import ( diff --git a/tests/providers/aws/services/awslambda/awslambda_function_vpc_multi_az/awslambda_function_vpc_multi_az_test.py b/tests/providers/aws/services/awslambda/awslambda_function_vpc_multi_az/awslambda_function_vpc_multi_az_test.py index 3e7bd50046..566124edba 100644 --- a/tests/providers/aws/services/awslambda/awslambda_function_vpc_multi_az/awslambda_function_vpc_multi_az_test.py +++ b/tests/providers/aws/services/awslambda/awslambda_function_vpc_multi_az/awslambda_function_vpc_multi_az_test.py @@ -19,15 +19,12 @@ class Test_awslambda_function_vpc_is_in_multi_azs: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.awslambda_client", - new=Lambda(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.awslambda_client", + new=Lambda(aws_provider), ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az import ( @@ -74,15 +71,12 @@ class Test_awslambda_function_vpc_is_in_multi_azs: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.awslambda_client", - new=Lambda(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.awslambda_client", + new=Lambda(aws_provider), ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az import ( @@ -162,19 +156,15 @@ class Test_awslambda_function_vpc_is_in_multi_azs: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.awslambda_client", - new=Lambda(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.awslambda_client", + new=Lambda(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az import ( @@ -259,19 +249,15 @@ class Test_awslambda_function_vpc_is_in_multi_azs: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.awslambda_client", - new=Lambda(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.awslambda_client", + new=Lambda(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az import ( @@ -359,19 +345,15 @@ class Test_awslambda_function_vpc_is_in_multi_azs: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.awslambda_client", - new=Lambda(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.awslambda_client", + new=Lambda(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az import ( @@ -428,23 +410,18 @@ class Test_awslambda_function_vpc_is_in_multi_azs: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.awslambda_client", - new=Lambda(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.vpc_client", - new=VPC(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_inside_vpc.awslambda_function_inside_vpc.awslambda_client", - new=Lambda(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.awslambda_client", + new=Lambda(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.vpc_client", + new=VPC(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_inside_vpc.awslambda_function_inside_vpc.awslambda_client", + new=Lambda(aws_provider), ): # Test check inside_vpc first @@ -539,19 +516,15 @@ class Test_awslambda_function_vpc_is_in_multi_azs: # Filter the resource in the provider, this is like the --resource-arn option aws_provider._audit_resources = [function["FunctionArn"]] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.awslambda_client", - new=Lambda(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.awslambda_client", + new=Lambda(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.awslambda.awslambda_function_vpc_multi_az.awslambda_function_vpc_multi_az import ( diff --git a/tests/providers/aws/services/backup/backup_plans_exist/backup_plans_exist_test.py b/tests/providers/aws/services/backup/backup_plans_exist/backup_plans_exist_test.py index e7b07cd8d9..f4fb055a94 100644 --- a/tests/providers/aws/services/backup/backup_plans_exist/backup_plans_exist_test.py +++ b/tests/providers/aws/services/backup/backup_plans_exist/backup_plans_exist_test.py @@ -21,15 +21,12 @@ class Test_backup_plans_exist: ) backup_client.backup_plans = [] backup_client.backup_vaults = ["vault"] - with ( - mock.patch( - "prowler.providers.aws.services.backup.backup_service.Backup", - new=backup_client, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_plans_exist.backup_plans_exist.backup_client", - new=backup_client, - ), + with mock.patch( + "prowler.providers.aws.services.backup.backup_service.Backup", + new=backup_client, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_plans_exist.backup_plans_exist.backup_client", + new=backup_client, ): # Test Check from prowler.providers.aws.services.backup.backup_plans_exist.backup_plans_exist import ( @@ -57,15 +54,12 @@ class Test_backup_plans_exist: backup_client.region = AWS_REGION backup_client.backup_plans = [] backup_client.backup_vaults = [] - with ( - mock.patch( - "prowler.providers.aws.services.backup.backup_service.Backup", - new=backup_client, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_plans_exist.backup_plans_exist.backup_client", - new=backup_client, - ), + with mock.patch( + "prowler.providers.aws.services.backup.backup_service.Backup", + new=backup_client, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_plans_exist.backup_plans_exist.backup_client", + new=backup_client, ): # Test Check from prowler.providers.aws.services.backup.backup_plans_exist.backup_plans_exist import ( @@ -98,15 +92,12 @@ class Test_backup_plans_exist: tags=[], ) ] - with ( - mock.patch( - "prowler.providers.aws.services.backup.backup_service.Backup", - new=backup_client, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_plans_exist.backup_plans_exist.backup_client", - new=backup_client, - ), + with mock.patch( + "prowler.providers.aws.services.backup.backup_service.Backup", + new=backup_client, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_plans_exist.backup_plans_exist.backup_client", + new=backup_client, ): # Test Check from prowler.providers.aws.services.backup.backup_plans_exist.backup_plans_exist import ( diff --git a/tests/providers/aws/services/backup/backup_reportplans_exist/backup_reportplans_exist_test.py b/tests/providers/aws/services/backup/backup_reportplans_exist/backup_reportplans_exist_test.py index 88be2e32eb..3787ac3874 100644 --- a/tests/providers/aws/services/backup/backup_reportplans_exist/backup_reportplans_exist_test.py +++ b/tests/providers/aws/services/backup/backup_reportplans_exist/backup_reportplans_exist_test.py @@ -16,15 +16,12 @@ class Test_backup_reportplans_exist: backup_client = mock.MagicMock backup_client.region = AWS_REGION backup_client.backup_plans = [] - with ( - mock.patch( - "prowler.providers.aws.services.backup.backup_service.Backup", - new=backup_client, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_reportplans_exist.backup_reportplans_exist.backup_client", - new=backup_client, - ), + with mock.patch( + "prowler.providers.aws.services.backup.backup_service.Backup", + new=backup_client, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_reportplans_exist.backup_reportplans_exist.backup_client", + new=backup_client, ): # Test Check from prowler.providers.aws.services.backup.backup_reportplans_exist.backup_reportplans_exist import ( @@ -62,15 +59,12 @@ class Test_backup_reportplans_exist: ) ] backup_client.backup_report_plans = [] - with ( - mock.patch( - "prowler.providers.aws.services.backup.backup_service.Backup", - new=backup_client, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_reportplans_exist.backup_reportplans_exist.backup_client", - new=backup_client, - ), + with mock.patch( + "prowler.providers.aws.services.backup.backup_service.Backup", + new=backup_client, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_reportplans_exist.backup_reportplans_exist.backup_client", + new=backup_client, ): # Test Check from prowler.providers.aws.services.backup.backup_reportplans_exist.backup_reportplans_exist import ( @@ -127,15 +121,12 @@ class Test_backup_reportplans_exist: ) ] - with ( - mock.patch( - "prowler.providers.aws.services.backup.backup_service.Backup", - new=backup_client, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_reportplans_exist.backup_reportplans_exist.backup_client", - new=backup_client, - ), + with mock.patch( + "prowler.providers.aws.services.backup.backup_service.Backup", + new=backup_client, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_reportplans_exist.backup_reportplans_exist.backup_client", + new=backup_client, ): # Test Check from prowler.providers.aws.services.backup.backup_reportplans_exist.backup_reportplans_exist import ( diff --git a/tests/providers/aws/services/backup/backup_vaults_encrypted/backup_vaults_encrypted_test.py b/tests/providers/aws/services/backup/backup_vaults_encrypted/backup_vaults_encrypted_test.py index 05904a24d0..7aac56123f 100644 --- a/tests/providers/aws/services/backup/backup_vaults_encrypted/backup_vaults_encrypted_test.py +++ b/tests/providers/aws/services/backup/backup_vaults_encrypted/backup_vaults_encrypted_test.py @@ -10,15 +10,12 @@ class Test_backup_vaults_encrypted: def test_no_backup_vaults(self): backup_client = mock.MagicMock backup_client.backup_vaults = [] - with ( - mock.patch( - "prowler.providers.aws.services.backup.backup_service.Backup", - new=backup_client, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_vaults_encrypted.backup_vaults_encrypted.backup_client", - new=backup_client, - ), + with mock.patch( + "prowler.providers.aws.services.backup.backup_service.Backup", + new=backup_client, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_vaults_encrypted.backup_vaults_encrypted.backup_client", + new=backup_client, ): # Test Check from prowler.providers.aws.services.backup.backup_vaults_encrypted.backup_vaults_encrypted import ( @@ -47,15 +44,12 @@ class Test_backup_vaults_encrypted: ) ] - with ( - mock.patch( - "prowler.providers.aws.services.backup.backup_service.Backup", - new=backup_client, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_vaults_encrypted.backup_vaults_encrypted.backup_client", - new=backup_client, - ), + with mock.patch( + "prowler.providers.aws.services.backup.backup_service.Backup", + new=backup_client, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_vaults_encrypted.backup_vaults_encrypted.backup_client", + new=backup_client, ): # Test Check from prowler.providers.aws.services.backup.backup_vaults_encrypted.backup_vaults_encrypted import ( @@ -93,15 +87,12 @@ class Test_backup_vaults_encrypted: ) ] - with ( - mock.patch( - "prowler.providers.aws.services.backup.backup_service.Backup", - new=backup_client, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_vaults_encrypted.backup_vaults_encrypted.backup_client", - new=backup_client, - ), + with mock.patch( + "prowler.providers.aws.services.backup.backup_service.Backup", + new=backup_client, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_vaults_encrypted.backup_vaults_encrypted.backup_client", + new=backup_client, ): # Test Check from prowler.providers.aws.services.backup.backup_vaults_encrypted.backup_vaults_encrypted import ( diff --git a/tests/providers/aws/services/backup/backup_vaults_exist/backup_vaults_exist_test.py b/tests/providers/aws/services/backup/backup_vaults_exist/backup_vaults_exist_test.py index 002bad0f13..5267ecb33a 100644 --- a/tests/providers/aws/services/backup/backup_vaults_exist/backup_vaults_exist_test.py +++ b/tests/providers/aws/services/backup/backup_vaults_exist/backup_vaults_exist_test.py @@ -18,15 +18,12 @@ class Test_backup_vaults_exist: return_value=backup_client.backup_vault_arn_template ) backup_client.backup_vaults = [] - with ( - mock.patch( - "prowler.providers.aws.services.backup.backup_service.Backup", - new=backup_client, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_vaults_exist.backup_vaults_exist.backup_client", - new=backup_client, - ), + with mock.patch( + "prowler.providers.aws.services.backup.backup_service.Backup", + new=backup_client, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_vaults_exist.backup_vaults_exist.backup_client", + new=backup_client, ): # Test Check from prowler.providers.aws.services.backup.backup_vaults_exist.backup_vaults_exist import ( @@ -72,15 +69,12 @@ class Test_backup_vaults_exist: ) ] - with ( - mock.patch( - "prowler.providers.aws.services.backup.backup_service.Backup", - new=backup_client, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_vaults_exist.backup_vaults_exist.backup_client", - new=backup_client, - ), + with mock.patch( + "prowler.providers.aws.services.backup.backup_service.Backup", + new=backup_client, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_vaults_exist.backup_vaults_exist.backup_client", + new=backup_client, ): # Test Check from prowler.providers.aws.services.backup.backup_vaults_exist.backup_vaults_exist import ( @@ -112,15 +106,12 @@ class Test_backup_vaults_exist: return_value=backup_client.backup_vault_arn_template ) backup_client.backup_vaults = None - with ( - mock.patch( - "prowler.providers.aws.services.backup.backup_service.Backup", - new=backup_client, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_vaults_exist.backup_vaults_exist.backup_client", - new=backup_client, - ), + with mock.patch( + "prowler.providers.aws.services.backup.backup_service.Backup", + new=backup_client, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_vaults_exist.backup_vaults_exist.backup_client", + new=backup_client, ): # Test Check from prowler.providers.aws.services.backup.backup_vaults_exist.backup_vaults_exist import ( diff --git a/tests/providers/aws/services/bedrock/bedrock_agent_guardrail_enabled/bedrock_agent_guardrail_enabled_test.py b/tests/providers/aws/services/bedrock/bedrock_agent_guardrail_enabled/bedrock_agent_guardrail_enabled_test.py index d333963947..24ef7473a8 100644 --- a/tests/providers/aws/services/bedrock/bedrock_agent_guardrail_enabled/bedrock_agent_guardrail_enabled_test.py +++ b/tests/providers/aws/services/bedrock/bedrock_agent_guardrail_enabled/bedrock_agent_guardrail_enabled_test.py @@ -40,15 +40,12 @@ class Test_bedrock_agent_guardrail_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_agent_guardrail_enabled.bedrock_agent_guardrail_enabled.bedrock_agent_client", - new=BedrockAgent(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_agent_guardrail_enabled.bedrock_agent_guardrail_enabled.bedrock_agent_client", + new=BedrockAgent(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_agent_guardrail_enabled.bedrock_agent_guardrail_enabled import ( bedrock_agent_guardrail_enabled, @@ -76,15 +73,12 @@ class Test_bedrock_agent_guardrail_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_agent_guardrail_enabled.bedrock_agent_guardrail_enabled.bedrock_agent_client", - new=BedrockAgent(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_agent_guardrail_enabled.bedrock_agent_guardrail_enabled.bedrock_agent_client", + new=BedrockAgent(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_agent_guardrail_enabled.bedrock_agent_guardrail_enabled import ( bedrock_agent_guardrail_enabled, @@ -113,15 +107,12 @@ class Test_bedrock_agent_guardrail_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_agent_guardrail_enabled.bedrock_agent_guardrail_enabled.bedrock_agent_client", - new=BedrockAgent(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_agent_guardrail_enabled.bedrock_agent_guardrail_enabled.bedrock_agent_client", + new=BedrockAgent(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_agent_guardrail_enabled.bedrock_agent_guardrail_enabled import ( bedrock_agent_guardrail_enabled, diff --git a/tests/providers/aws/services/bedrock/bedrock_guardrail_prompt_attack_filter_enabled/bedrock_guardrail_prompt_attack_filter_enabled_test.py b/tests/providers/aws/services/bedrock/bedrock_guardrail_prompt_attack_filter_enabled/bedrock_guardrail_prompt_attack_filter_enabled_test.py index 010f27247e..4f5ed8a2a9 100644 --- a/tests/providers/aws/services/bedrock/bedrock_guardrail_prompt_attack_filter_enabled/bedrock_guardrail_prompt_attack_filter_enabled_test.py +++ b/tests/providers/aws/services/bedrock/bedrock_guardrail_prompt_attack_filter_enabled/bedrock_guardrail_prompt_attack_filter_enabled_test.py @@ -117,15 +117,12 @@ class Test_bedrock_guardrail_prompt_attack_filter_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_guardrail_prompt_attack_filter_enabled.bedrock_guardrail_prompt_attack_filter_enabled.bedrock_client", - new=Bedrock(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_guardrail_prompt_attack_filter_enabled.bedrock_guardrail_prompt_attack_filter_enabled.bedrock_client", + new=Bedrock(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_guardrail_prompt_attack_filter_enabled.bedrock_guardrail_prompt_attack_filter_enabled import ( bedrock_guardrail_prompt_attack_filter_enabled, @@ -145,15 +142,12 @@ class Test_bedrock_guardrail_prompt_attack_filter_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_guardrail_prompt_attack_filter_enabled.bedrock_guardrail_prompt_attack_filter_enabled.bedrock_client", - new=Bedrock(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_guardrail_prompt_attack_filter_enabled.bedrock_guardrail_prompt_attack_filter_enabled.bedrock_client", + new=Bedrock(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_guardrail_prompt_attack_filter_enabled.bedrock_guardrail_prompt_attack_filter_enabled import ( bedrock_guardrail_prompt_attack_filter_enabled, @@ -182,15 +176,12 @@ class Test_bedrock_guardrail_prompt_attack_filter_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_guardrail_prompt_attack_filter_enabled.bedrock_guardrail_prompt_attack_filter_enabled.bedrock_client", - new=Bedrock(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_guardrail_prompt_attack_filter_enabled.bedrock_guardrail_prompt_attack_filter_enabled.bedrock_client", + new=Bedrock(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_guardrail_prompt_attack_filter_enabled.bedrock_guardrail_prompt_attack_filter_enabled import ( bedrock_guardrail_prompt_attack_filter_enabled, @@ -219,15 +210,12 @@ class Test_bedrock_guardrail_prompt_attack_filter_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_guardrail_prompt_attack_filter_enabled.bedrock_guardrail_prompt_attack_filter_enabled.bedrock_client", - new=Bedrock(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_guardrail_prompt_attack_filter_enabled.bedrock_guardrail_prompt_attack_filter_enabled.bedrock_client", + new=Bedrock(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_guardrail_prompt_attack_filter_enabled.bedrock_guardrail_prompt_attack_filter_enabled import ( bedrock_guardrail_prompt_attack_filter_enabled, diff --git a/tests/providers/aws/services/bedrock/bedrock_guardrail_sensitive_information_filter_enabled/bedrock_guardrail_sensitive_information_filter_enabled_test.py b/tests/providers/aws/services/bedrock/bedrock_guardrail_sensitive_information_filter_enabled/bedrock_guardrail_sensitive_information_filter_enabled_test.py index 88c80ce034..b33eb6ffbc 100644 --- a/tests/providers/aws/services/bedrock/bedrock_guardrail_sensitive_information_filter_enabled/bedrock_guardrail_sensitive_information_filter_enabled_test.py +++ b/tests/providers/aws/services/bedrock/bedrock_guardrail_sensitive_information_filter_enabled/bedrock_guardrail_sensitive_information_filter_enabled_test.py @@ -76,15 +76,12 @@ class Test_bedrock_guardrail_sensitive_information_filter_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_guardrail_sensitive_information_filter_enabled.bedrock_guardrail_sensitive_information_filter_enabled.bedrock_client", - new=Bedrock(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_guardrail_sensitive_information_filter_enabled.bedrock_guardrail_sensitive_information_filter_enabled.bedrock_client", + new=Bedrock(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_guardrail_sensitive_information_filter_enabled.bedrock_guardrail_sensitive_information_filter_enabled import ( bedrock_guardrail_sensitive_information_filter_enabled, @@ -104,15 +101,12 @@ class Test_bedrock_guardrail_sensitive_information_filter_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_guardrail_sensitive_information_filter_enabled.bedrock_guardrail_sensitive_information_filter_enabled.bedrock_client", - new=Bedrock(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_guardrail_sensitive_information_filter_enabled.bedrock_guardrail_sensitive_information_filter_enabled.bedrock_client", + new=Bedrock(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_guardrail_sensitive_information_filter_enabled.bedrock_guardrail_sensitive_information_filter_enabled import ( bedrock_guardrail_sensitive_information_filter_enabled, @@ -141,15 +135,12 @@ class Test_bedrock_guardrail_sensitive_information_filter_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_guardrail_sensitive_information_filter_enabled.bedrock_guardrail_sensitive_information_filter_enabled.bedrock_client", - new=Bedrock(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_guardrail_sensitive_information_filter_enabled.bedrock_guardrail_sensitive_information_filter_enabled.bedrock_client", + new=Bedrock(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_guardrail_sensitive_information_filter_enabled.bedrock_guardrail_sensitive_information_filter_enabled import ( bedrock_guardrail_sensitive_information_filter_enabled, diff --git a/tests/providers/aws/services/bedrock/bedrock_model_invocation_logging_enabled/bedrock_model_invocation_logging_enabled_test.py b/tests/providers/aws/services/bedrock/bedrock_model_invocation_logging_enabled/bedrock_model_invocation_logging_enabled_test.py index b87ca77e21..bcf819a7ec 100644 --- a/tests/providers/aws/services/bedrock/bedrock_model_invocation_logging_enabled/bedrock_model_invocation_logging_enabled_test.py +++ b/tests/providers/aws/services/bedrock/bedrock_model_invocation_logging_enabled/bedrock_model_invocation_logging_enabled_test.py @@ -19,15 +19,12 @@ class Test_bedrock_model_invocation_logging_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logging_enabled.bedrock_model_invocation_logging_enabled.bedrock_client", - new=Bedrock(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logging_enabled.bedrock_model_invocation_logging_enabled.bedrock_client", + new=Bedrock(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_model_invocation_logging_enabled.bedrock_model_invocation_logging_enabled import ( bedrock_model_invocation_logging_enabled, @@ -82,15 +79,12 @@ class Test_bedrock_model_invocation_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logging_enabled.bedrock_model_invocation_logging_enabled.bedrock_client", - new=Bedrock(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logging_enabled.bedrock_model_invocation_logging_enabled.bedrock_client", + new=Bedrock(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_model_invocation_logging_enabled.bedrock_model_invocation_logging_enabled import ( bedrock_model_invocation_logging_enabled, @@ -128,15 +122,12 @@ class Test_bedrock_model_invocation_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logging_enabled.bedrock_model_invocation_logging_enabled.bedrock_client", - new=Bedrock(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logging_enabled.bedrock_model_invocation_logging_enabled.bedrock_client", + new=Bedrock(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_model_invocation_logging_enabled.bedrock_model_invocation_logging_enabled import ( bedrock_model_invocation_logging_enabled, @@ -175,15 +166,12 @@ class Test_bedrock_model_invocation_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logging_enabled.bedrock_model_invocation_logging_enabled.bedrock_client", - new=Bedrock(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logging_enabled.bedrock_model_invocation_logging_enabled.bedrock_client", + new=Bedrock(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_model_invocation_logging_enabled.bedrock_model_invocation_logging_enabled import ( bedrock_model_invocation_logging_enabled, diff --git a/tests/providers/aws/services/bedrock/bedrock_model_invocation_logs_encryption_enabled/bedrock_model_invocation_logs_encryption_enabled_test.py b/tests/providers/aws/services/bedrock/bedrock_model_invocation_logs_encryption_enabled/bedrock_model_invocation_logs_encryption_enabled_test.py index 4630ad25a0..74db1e4635 100644 --- a/tests/providers/aws/services/bedrock/bedrock_model_invocation_logs_encryption_enabled/bedrock_model_invocation_logs_encryption_enabled_test.py +++ b/tests/providers/aws/services/bedrock/bedrock_model_invocation_logs_encryption_enabled/bedrock_model_invocation_logs_encryption_enabled_test.py @@ -15,23 +15,18 @@ class Test_bedrock_model_invocation_logs_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.bedrock_client", - new=Bedrock(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.logs_client", - new=Logs(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.bedrock_client", + new=Bedrock(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.logs_client", + new=Logs(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.s3_client", + new=S3(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled import ( bedrock_model_invocation_logs_encryption_enabled, @@ -70,23 +65,18 @@ class Test_bedrock_model_invocation_logs_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.bedrock_client", - new=Bedrock(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.logs_client", - new=Logs(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.bedrock_client", + new=Bedrock(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.logs_client", + new=Logs(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.s3_client", + new=S3(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled import ( bedrock_model_invocation_logs_encryption_enabled, @@ -128,23 +118,18 @@ class Test_bedrock_model_invocation_logs_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.bedrock_client", - new=Bedrock(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.logs_client", - new=Logs(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.bedrock_client", + new=Bedrock(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.logs_client", + new=Logs(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.s3_client", + new=S3(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled import ( bedrock_model_invocation_logs_encryption_enabled, @@ -187,23 +172,18 @@ class Test_bedrock_model_invocation_logs_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.bedrock_client", - new=Bedrock(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.logs_client", - new=Logs(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.bedrock_client", + new=Bedrock(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.logs_client", + new=Logs(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.s3_client", + new=S3(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled import ( @@ -268,23 +248,18 @@ class Test_bedrock_model_invocation_logs_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.bedrock_client", - new=Bedrock(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.logs_client", - new=Logs(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.bedrock_client", + new=Bedrock(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.logs_client", + new=Logs(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.s3_client", + new=S3(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled import ( bedrock_model_invocation_logs_encryption_enabled, @@ -339,23 +314,18 @@ class Test_bedrock_model_invocation_logs_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.bedrock_client", - new=Bedrock(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.logs_client", - new=Logs(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.bedrock_client", + new=Bedrock(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.logs_client", + new=Logs(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.s3_client", + new=S3(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled import ( bedrock_model_invocation_logs_encryption_enabled, @@ -398,23 +368,18 @@ class Test_bedrock_model_invocation_logs_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.bedrock_client", - new=Bedrock(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.logs_client", - new=Logs(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.bedrock_client", + new=Bedrock(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.logs_client", + new=Logs(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled.s3_client", + new=S3(aws_provider), ): from prowler.providers.aws.services.bedrock.bedrock_model_invocation_logs_encryption_enabled.bedrock_model_invocation_logs_encryption_enabled import ( diff --git a/tests/providers/aws/services/cloudformation/cloudformation_stack_cdktoolkit_bootstrap_version/cloudformation_stack_cdktoolkit_bootstrap_version_test.py b/tests/providers/aws/services/cloudformation/cloudformation_stack_cdktoolkit_bootstrap_version/cloudformation_stack_cdktoolkit_bootstrap_version_test.py index d481300bd6..8ba9aabb78 100644 --- a/tests/providers/aws/services/cloudformation/cloudformation_stack_cdktoolkit_bootstrap_version/cloudformation_stack_cdktoolkit_bootstrap_version_test.py +++ b/tests/providers/aws/services/cloudformation/cloudformation_stack_cdktoolkit_bootstrap_version/cloudformation_stack_cdktoolkit_bootstrap_version_test.py @@ -11,15 +11,12 @@ class Test_cloudformation_stack_cdktoolkit_bootstrap_version: cloudformation_client = mock.MagicMock cloudformation_client.stacks = [] cloudformation_client.audit_config = {"recommended_cdk_bootstrap_version": 21} - with ( - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", - new=cloudformation_client, - ), - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", - new=cloudformation_client, - ), + with mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", + new=cloudformation_client, + ), mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", + new=cloudformation_client, ): from prowler.providers.aws.services.cloudformation.cloudformation_stack_cdktoolkit_bootstrap_version.cloudformation_stack_cdktoolkit_bootstrap_version import ( cloudformation_stack_cdktoolkit_bootstrap_version, @@ -42,15 +39,12 @@ class Test_cloudformation_stack_cdktoolkit_bootstrap_version: ] cloudformation_client.audit_config = {"recommended_cdk_bootstrap_version": 21} - with ( - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", - new=cloudformation_client, - ), - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", - new=cloudformation_client, - ), + with mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", + new=cloudformation_client, + ), mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", + new=cloudformation_client, ): from prowler.providers.aws.services.cloudformation.cloudformation_stack_cdktoolkit_bootstrap_version.cloudformation_stack_cdktoolkit_bootstrap_version import ( cloudformation_stack_cdktoolkit_bootstrap_version, @@ -84,15 +78,12 @@ class Test_cloudformation_stack_cdktoolkit_bootstrap_version: ] cloudformation_client.audit_config = {"recommended_cdk_bootstrap_version": 21} - with ( - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", - new=cloudformation_client, - ), - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", - new=cloudformation_client, - ), + with mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", + new=cloudformation_client, + ), mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", + new=cloudformation_client, ): from prowler.providers.aws.services.cloudformation.cloudformation_stack_cdktoolkit_bootstrap_version.cloudformation_stack_cdktoolkit_bootstrap_version import ( cloudformation_stack_cdktoolkit_bootstrap_version, diff --git a/tests/providers/aws/services/cloudformation/cloudformation_stack_outputs_find_secrets/cloudformation_stack_outputs_find_secrets_test.py b/tests/providers/aws/services/cloudformation/cloudformation_stack_outputs_find_secrets/cloudformation_stack_outputs_find_secrets_test.py index 6b8d0d9b15..c985c838f8 100644 --- a/tests/providers/aws/services/cloudformation/cloudformation_stack_outputs_find_secrets/cloudformation_stack_outputs_find_secrets_test.py +++ b/tests/providers/aws/services/cloudformation/cloudformation_stack_outputs_find_secrets/cloudformation_stack_outputs_find_secrets_test.py @@ -11,15 +11,12 @@ class Test_cloudformation_stack_outputs_find_secrets: cloudformation_client = mock.MagicMock cloudformation_client.stacks = [] cloudformation_client.audit_config = {"secrets_ignore_patterns": []} - with ( - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", - new=cloudformation_client, - ), - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", - new=cloudformation_client, - ), + with mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", + new=cloudformation_client, + ), mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", + new=cloudformation_client, ): # Test Check from prowler.providers.aws.services.cloudformation.cloudformation_stack_outputs_find_secrets.cloudformation_stack_outputs_find_secrets import ( @@ -45,15 +42,12 @@ class Test_cloudformation_stack_outputs_find_secrets: cloudformation_client.audit_config = {"secrets_ignore_patterns": []} - with ( - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", - cloudformation_client, - ), - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", - new=cloudformation_client, - ), + with mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", + cloudformation_client, + ), mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", + new=cloudformation_client, ): from prowler.providers.aws.services.cloudformation.cloudformation_stack_outputs_find_secrets.cloudformation_stack_outputs_find_secrets import ( cloudformation_stack_outputs_find_secrets, @@ -92,15 +86,12 @@ class Test_cloudformation_stack_outputs_find_secrets: cloudformation_client.audit_config = {"secrets_ignore_patterns": []} - with ( - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", - cloudformation_client, - ), - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", - new=cloudformation_client, - ), + with mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", + cloudformation_client, + ), mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", + new=cloudformation_client, ): from prowler.providers.aws.services.cloudformation.cloudformation_stack_outputs_find_secrets.cloudformation_stack_outputs_find_secrets import ( cloudformation_stack_outputs_find_secrets, @@ -137,15 +128,12 @@ class Test_cloudformation_stack_outputs_find_secrets: cloudformation_client.audit_config = {"secrets_ignore_patterns": []} - with ( - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", - cloudformation_client, - ), - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", - new=cloudformation_client, - ), + with mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", + cloudformation_client, + ), mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", + new=cloudformation_client, ): from prowler.providers.aws.services.cloudformation.cloudformation_stack_outputs_find_secrets.cloudformation_stack_outputs_find_secrets import ( cloudformation_stack_outputs_find_secrets, @@ -182,15 +170,12 @@ class Test_cloudformation_stack_outputs_find_secrets: cloudformation_client.audit_config = {"secrets_ignore_patterns": []} - with ( - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", - cloudformation_client, - ), - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", - new=cloudformation_client, - ), + with mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", + cloudformation_client, + ), mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", + new=cloudformation_client, ): from prowler.providers.aws.services.cloudformation.cloudformation_stack_outputs_find_secrets.cloudformation_stack_outputs_find_secrets import ( cloudformation_stack_outputs_find_secrets, diff --git a/tests/providers/aws/services/cloudformation/cloudformation_stacks_termination_protection_enabled/cloudformation_stacks_termination_protection_enabled_test.py b/tests/providers/aws/services/cloudformation/cloudformation_stacks_termination_protection_enabled/cloudformation_stacks_termination_protection_enabled_test.py index 59c6d69e79..5940c7bf22 100644 --- a/tests/providers/aws/services/cloudformation/cloudformation_stacks_termination_protection_enabled/cloudformation_stacks_termination_protection_enabled_test.py +++ b/tests/providers/aws/services/cloudformation/cloudformation_stacks_termination_protection_enabled/cloudformation_stacks_termination_protection_enabled_test.py @@ -10,15 +10,12 @@ class Test_cloudformation_stacks_termination_protection_enabled: def test_no_stacks(self): cloudformation_client = mock.MagicMock cloudformation_client.stacks = [] - with ( - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", - new=cloudformation_client, - ), - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", - new=cloudformation_client, - ), + with mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", + new=cloudformation_client, + ), mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", + new=cloudformation_client, ): # Test Check from prowler.providers.aws.services.cloudformation.cloudformation_stacks_termination_protection_enabled.cloudformation_stacks_termination_protection_enabled import ( @@ -43,15 +40,12 @@ class Test_cloudformation_stacks_termination_protection_enabled: ] cloudformation_client.stacks[0].enable_termination_protection = True - with ( - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", - cloudformation_client, - ), - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", - new=cloudformation_client, - ), + with mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", + cloudformation_client, + ), mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", + new=cloudformation_client, ): from prowler.providers.aws.services.cloudformation.cloudformation_stacks_termination_protection_enabled.cloudformation_stacks_termination_protection_enabled import ( cloudformation_stacks_termination_protection_enabled, @@ -87,15 +81,12 @@ class Test_cloudformation_stacks_termination_protection_enabled: ] cloudformation_client.stacks[0].enable_termination_protection = False - with ( - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", - cloudformation_client, - ), - mock.patch( - "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", - new=cloudformation_client, - ), + with mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_service.CloudFormation", + cloudformation_client, + ), mock.patch( + "prowler.providers.aws.services.cloudformation.cloudformation_client.cloudformation_client", + new=cloudformation_client, ): from prowler.providers.aws.services.cloudformation.cloudformation_stacks_termination_protection_enabled.cloudformation_stacks_termination_protection_enabled import ( cloudformation_stacks_termination_protection_enabled, diff --git a/tests/providers/aws/services/cloudtrail/cloudtrail_bucket_requires_mfa_delete/cloudtrail_bucket_requires_mfa_delete_test.py b/tests/providers/aws/services/cloudtrail/cloudtrail_bucket_requires_mfa_delete/cloudtrail_bucket_requires_mfa_delete_test.py index 5db914fd28..b5925ed189 100644 --- a/tests/providers/aws/services/cloudtrail/cloudtrail_bucket_requires_mfa_delete/cloudtrail_bucket_requires_mfa_delete_test.py +++ b/tests/providers/aws/services/cloudtrail/cloudtrail_bucket_requires_mfa_delete/cloudtrail_bucket_requires_mfa_delete_test.py @@ -24,15 +24,12 @@ class Test_cloudtrail_bucket_requires_mfa_delete: [AWS_REGION_US_EAST_1, AWS_REGION_EU_WEST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.cloudtrail_client", + new=Cloudtrail(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete import ( @@ -62,19 +59,15 @@ class Test_cloudtrail_bucket_requires_mfa_delete: cloudtrail_client_us_east_1.start_logging(Name=trail_name_us) cloudtrail_client_us_east_1.get_trail_status(Name=trail_name_us) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete import ( @@ -129,19 +122,15 @@ class Test_cloudtrail_bucket_requires_mfa_delete: cloudtrail_client_us_east_1.start_logging(Name=trail_name_us) cloudtrail_client_us_east_1.get_trail_status(Name=trail_name_us) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete import ( @@ -180,20 +169,16 @@ class Test_cloudtrail_bucket_requires_mfa_delete: cloudtrail_client_us_east_1.start_logging(Name=trail_name_us) cloudtrail_client_us_east_1.get_trail_status(Name=trail_name_us) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.s3_client", - new=S3(aws_provider), - ) as s3_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.s3_client", + new=S3(aws_provider), + ) as s3_client: # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete import ( cloudtrail_bucket_requires_mfa_delete, @@ -239,20 +224,16 @@ class Test_cloudtrail_bucket_requires_mfa_delete: cloudtrail_client_us_east_1.start_logging(Name=trail_name_us) cloudtrail_client_us_east_1.get_trail_status(Name=trail_name_us) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.s3_client", - new=S3(aws_provider), - ) as s3_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.s3_client", + new=S3(aws_provider), + ) as s3_client: # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete import ( cloudtrail_bucket_requires_mfa_delete, @@ -284,20 +265,16 @@ class Test_cloudtrail_bucket_requires_mfa_delete: [AWS_REGION_US_EAST_1, AWS_REGION_EU_WEST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.cloudtrail_client", - new=Cloudtrail(aws_provider), - ) as cloudtrail_client, - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.s3_client", - new=S3(aws_provider), - ) as s3_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.cloudtrail_client", + new=Cloudtrail(aws_provider), + ) as cloudtrail_client, mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.s3_client", + new=S3(aws_provider), + ) as s3_client: # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete import ( cloudtrail_bucket_requires_mfa_delete, @@ -327,19 +304,15 @@ class Test_cloudtrail_bucket_requires_mfa_delete: cloudtrail_client_us_east_1.start_logging(Name=trail_name_us) cloudtrail_client_us_east_1.get_trail_status(Name=trail_name_us) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_bucket_requires_mfa_delete.cloudtrail_bucket_requires_mfa_delete import ( diff --git a/tests/providers/aws/services/cloudtrail/cloudtrail_kms_encryption_enabled/cloudtrail_kms_encryption_enabled_test.py b/tests/providers/aws/services/cloudtrail/cloudtrail_kms_encryption_enabled/cloudtrail_kms_encryption_enabled_test.py index 9387fa1fe6..a49f178dc7 100644 --- a/tests/providers/aws/services/cloudtrail/cloudtrail_kms_encryption_enabled/cloudtrail_kms_encryption_enabled_test.py +++ b/tests/providers/aws/services/cloudtrail/cloudtrail_kms_encryption_enabled/cloudtrail_kms_encryption_enabled_test.py @@ -18,15 +18,12 @@ class Test_cloudtrail_kms_encryption_enabled: Cloudtrail, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_kms_encryption_enabled.cloudtrail_kms_encryption_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_kms_encryption_enabled.cloudtrail_kms_encryption_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_kms_encryption_enabled.cloudtrail_kms_encryption_enabled import ( @@ -56,15 +53,12 @@ class Test_cloudtrail_kms_encryption_enabled: Cloudtrail, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_kms_encryption_enabled.cloudtrail_kms_encryption_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_kms_encryption_enabled.cloudtrail_kms_encryption_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_kms_encryption_enabled.cloudtrail_kms_encryption_enabled import ( @@ -108,15 +102,12 @@ class Test_cloudtrail_kms_encryption_enabled: Cloudtrail, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_kms_encryption_enabled.cloudtrail_kms_encryption_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_kms_encryption_enabled.cloudtrail_kms_encryption_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_kms_encryption_enabled.cloudtrail_kms_encryption_enabled import ( @@ -144,16 +135,13 @@ class Test_cloudtrail_kms_encryption_enabled: Cloudtrail, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_kms_encryption_enabled.cloudtrail_kms_encryption_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_kms_encryption_enabled.cloudtrail_kms_encryption_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ) as service_client: # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_kms_encryption_enabled.cloudtrail_kms_encryption_enabled import ( cloudtrail_kms_encryption_enabled, @@ -186,15 +174,12 @@ class Test_cloudtrail_kms_encryption_enabled: Cloudtrail, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_kms_encryption_enabled.cloudtrail_kms_encryption_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_kms_encryption_enabled.cloudtrail_kms_encryption_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_kms_encryption_enabled.cloudtrail_kms_encryption_enabled import ( diff --git a/tests/providers/aws/services/cloudtrail/cloudtrail_log_file_validation_enabled/cloudtrail_log_file_validation_enabled_test.py b/tests/providers/aws/services/cloudtrail/cloudtrail_log_file_validation_enabled/cloudtrail_log_file_validation_enabled_test.py index 3d7edfaec4..cb30843728 100644 --- a/tests/providers/aws/services/cloudtrail/cloudtrail_log_file_validation_enabled/cloudtrail_log_file_validation_enabled_test.py +++ b/tests/providers/aws/services/cloudtrail/cloudtrail_log_file_validation_enabled/cloudtrail_log_file_validation_enabled_test.py @@ -18,15 +18,12 @@ class Test_cloudtrail_log_file_validation_enabled: Cloudtrail, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_log_file_validation_enabled.cloudtrail_log_file_validation_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_log_file_validation_enabled.cloudtrail_log_file_validation_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_log_file_validation_enabled.cloudtrail_log_file_validation_enabled import ( @@ -55,15 +52,12 @@ class Test_cloudtrail_log_file_validation_enabled: Cloudtrail, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_log_file_validation_enabled.cloudtrail_log_file_validation_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_log_file_validation_enabled.cloudtrail_log_file_validation_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_log_file_validation_enabled.cloudtrail_log_file_validation_enabled import ( @@ -120,15 +114,12 @@ class Test_cloudtrail_log_file_validation_enabled: Cloudtrail, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_log_file_validation_enabled.cloudtrail_log_file_validation_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_log_file_validation_enabled.cloudtrail_log_file_validation_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_log_file_validation_enabled.cloudtrail_log_file_validation_enabled import ( @@ -167,16 +158,13 @@ class Test_cloudtrail_log_file_validation_enabled: Cloudtrail, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_log_file_validation_enabled.cloudtrail_log_file_validation_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_log_file_validation_enabled.cloudtrail_log_file_validation_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ) as service_client: # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_log_file_validation_enabled.cloudtrail_log_file_validation_enabled import ( cloudtrail_log_file_validation_enabled, @@ -209,15 +197,12 @@ class Test_cloudtrail_log_file_validation_enabled: Cloudtrail, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_log_file_validation_enabled.cloudtrail_log_file_validation_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_log_file_validation_enabled.cloudtrail_log_file_validation_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_log_file_validation_enabled.cloudtrail_log_file_validation_enabled import ( diff --git a/tests/providers/aws/services/cloudtrail/cloudtrail_logs_s3_bucket_access_logging_enabled/cloudtrail_logs_s3_bucket_access_logging_enabled_test.py b/tests/providers/aws/services/cloudtrail/cloudtrail_logs_s3_bucket_access_logging_enabled/cloudtrail_logs_s3_bucket_access_logging_enabled_test.py index e472554b79..0d92f84f00 100644 --- a/tests/providers/aws/services/cloudtrail/cloudtrail_logs_s3_bucket_access_logging_enabled/cloudtrail_logs_s3_bucket_access_logging_enabled_test.py +++ b/tests/providers/aws/services/cloudtrail/cloudtrail_logs_s3_bucket_access_logging_enabled/cloudtrail_logs_s3_bucket_access_logging_enabled_test.py @@ -22,19 +22,15 @@ class Test_cloudtrail_logs_s3_bucket_access_logging_enabled: ) from prowler.providers.aws.services.s3.s3_service import S3 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled import ( @@ -67,19 +63,15 @@ class Test_cloudtrail_logs_s3_bucket_access_logging_enabled: ) from prowler.providers.aws.services.s3.s3_service import S3 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled import ( @@ -141,19 +133,15 @@ class Test_cloudtrail_logs_s3_bucket_access_logging_enabled: ) from prowler.providers.aws.services.s3.s3_service import S3 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled import ( @@ -195,20 +183,16 @@ class Test_cloudtrail_logs_s3_bucket_access_logging_enabled: ) from prowler.providers.aws.services.s3.s3_service import S3 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.s3_client", - new=S3(aws_provider), - ) as s3_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.s3_client", + new=S3(aws_provider), + ) as s3_client: # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled import ( cloudtrail_logs_s3_bucket_access_logging_enabled, @@ -242,20 +226,16 @@ class Test_cloudtrail_logs_s3_bucket_access_logging_enabled: ) from prowler.providers.aws.services.s3.s3_service import S3 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ) as cloudtrail_client, - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.s3_client", - new=S3(aws_provider), - ) as s3_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ) as cloudtrail_client, mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.s3_client", + new=S3(aws_provider), + ) as s3_client: # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled import ( cloudtrail_logs_s3_bucket_access_logging_enabled, @@ -291,19 +271,15 @@ class Test_cloudtrail_logs_s3_bucket_access_logging_enabled: ) from prowler.providers.aws.services.s3.s3_service import S3 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_access_logging_enabled.cloudtrail_logs_s3_bucket_access_logging_enabled import ( diff --git a/tests/providers/aws/services/cloudtrail/cloudtrail_logs_s3_bucket_is_not_publicly_accessible/cloudtrail_logs_s3_bucket_is_not_publicly_accessible_fixer_test.py b/tests/providers/aws/services/cloudtrail/cloudtrail_logs_s3_bucket_is_not_publicly_accessible/cloudtrail_logs_s3_bucket_is_not_publicly_accessible_fixer_test.py index a3eea32968..bd384748c2 100644 --- a/tests/providers/aws/services/cloudtrail/cloudtrail_logs_s3_bucket_is_not_publicly_accessible/cloudtrail_logs_s3_bucket_is_not_publicly_accessible_fixer_test.py +++ b/tests/providers/aws/services/cloudtrail/cloudtrail_logs_s3_bucket_is_not_publicly_accessible/cloudtrail_logs_s3_bucket_is_not_publicly_accessible_fixer_test.py @@ -67,19 +67,15 @@ class Test_cloudtrail_logs_s3_bucket_is_not_publicly_accessible_fixer: ) from prowler.providers.aws.services.s3.s3_service import S3 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible_fixer.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible_fixer.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible_fixer.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible_fixer.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible_fixer import ( @@ -131,19 +127,15 @@ class Test_cloudtrail_logs_s3_bucket_is_not_publicly_accessible_fixer: ) from prowler.providers.aws.services.s3.s3_service import S3 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible_fixer.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible_fixer.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible_fixer.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible_fixer.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible_fixer import ( diff --git a/tests/providers/aws/services/cloudtrail/cloudtrail_logs_s3_bucket_is_not_publicly_accessible/cloudtrail_logs_s3_bucket_is_not_publicly_accessible_test.py b/tests/providers/aws/services/cloudtrail/cloudtrail_logs_s3_bucket_is_not_publicly_accessible/cloudtrail_logs_s3_bucket_is_not_publicly_accessible_test.py index a051516ea1..52809f8f39 100644 --- a/tests/providers/aws/services/cloudtrail/cloudtrail_logs_s3_bucket_is_not_publicly_accessible/cloudtrail_logs_s3_bucket_is_not_publicly_accessible_test.py +++ b/tests/providers/aws/services/cloudtrail/cloudtrail_logs_s3_bucket_is_not_publicly_accessible/cloudtrail_logs_s3_bucket_is_not_publicly_accessible_test.py @@ -21,19 +21,15 @@ class Test_cloudtrail_logs_s3_bucket_is_not_publicly_accessible: ) from prowler.providers.aws.services.s3.s3_service import S3 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible import ( @@ -64,19 +60,15 @@ class Test_cloudtrail_logs_s3_bucket_is_not_publicly_accessible: ) from prowler.providers.aws.services.s3.s3_service import S3 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible import ( @@ -136,19 +128,15 @@ class Test_cloudtrail_logs_s3_bucket_is_not_publicly_accessible: ) from prowler.providers.aws.services.s3.s3_service import S3 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible import ( @@ -206,19 +194,15 @@ class Test_cloudtrail_logs_s3_bucket_is_not_publicly_accessible: ) from prowler.providers.aws.services.s3.s3_service import S3 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible import ( @@ -258,20 +242,16 @@ class Test_cloudtrail_logs_s3_bucket_is_not_publicly_accessible: ) from prowler.providers.aws.services.s3.s3_service import S3 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.s3_client", - new=S3(aws_provider), - ) as s3_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.s3_client", + new=S3(aws_provider), + ) as s3_client: # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible import ( cloudtrail_logs_s3_bucket_is_not_publicly_accessible, @@ -304,20 +284,16 @@ class Test_cloudtrail_logs_s3_bucket_is_not_publicly_accessible: ) from prowler.providers.aws.services.s3.s3_service import S3 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_client", - new=Cloudtrail(aws_provider), - ) as cloudtrail_client, - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.s3_client", - new=S3(aws_provider), - ) as s3_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_client", + new=Cloudtrail(aws_provider), + ) as cloudtrail_client, mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.s3_client", + new=S3(aws_provider), + ) as s3_client: # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible import ( cloudtrail_logs_s3_bucket_is_not_publicly_accessible, @@ -353,19 +329,15 @@ class Test_cloudtrail_logs_s3_bucket_is_not_publicly_accessible: ) from prowler.providers.aws.services.s3.s3_service import S3 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_logs_s3_bucket_is_not_publicly_accessible.cloudtrail_logs_s3_bucket_is_not_publicly_accessible import ( diff --git a/tests/providers/aws/services/cloudtrail/cloudtrail_s3_dataevents_read_enabled/cloudtrail_s3_dataevents_read_enabled_test.py b/tests/providers/aws/services/cloudtrail/cloudtrail_s3_dataevents_read_enabled/cloudtrail_s3_dataevents_read_enabled_test.py index 83eab92856..8fbe998a29 100644 --- a/tests/providers/aws/services/cloudtrail/cloudtrail_s3_dataevents_read_enabled/cloudtrail_s3_dataevents_read_enabled_test.py +++ b/tests/providers/aws/services/cloudtrail/cloudtrail_s3_dataevents_read_enabled/cloudtrail_s3_dataevents_read_enabled_test.py @@ -32,19 +32,15 @@ class Test_cloudtrail_s3_dataevents_read_enabled: aws_provider = set_mocked_aws_provider() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled import ( @@ -78,19 +74,15 @@ class Test_cloudtrail_s3_dataevents_read_enabled: aws_provider = set_mocked_aws_provider() aws_provider._scan_unused_services = False - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled import ( @@ -116,19 +108,15 @@ class Test_cloudtrail_s3_dataevents_read_enabled: aws_provider = set_mocked_aws_provider() aws_provider._scan_unused_services = False - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled import ( @@ -184,19 +172,15 @@ class Test_cloudtrail_s3_dataevents_read_enabled: aws_provider = set_mocked_aws_provider() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled import ( @@ -252,19 +236,15 @@ class Test_cloudtrail_s3_dataevents_read_enabled: aws_provider = set_mocked_aws_provider() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled import ( @@ -317,19 +297,15 @@ class Test_cloudtrail_s3_dataevents_read_enabled: aws_provider = set_mocked_aws_provider() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled import ( @@ -388,19 +364,15 @@ class Test_cloudtrail_s3_dataevents_read_enabled: aws_provider = set_mocked_aws_provider() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled import ( @@ -430,19 +402,15 @@ class Test_cloudtrail_s3_dataevents_read_enabled: aws_provider = set_mocked_aws_provider() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ) as cloudtrail_client, - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ) as cloudtrail_client, mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled import ( @@ -489,19 +457,15 @@ class Test_cloudtrail_s3_dataevents_read_enabled: ) from prowler.providers.aws.services.s3.s3_service import S3 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_read_enabled.cloudtrail_s3_dataevents_read_enabled import ( diff --git a/tests/providers/aws/services/cloudtrail/cloudtrail_s3_dataevents_write_enabled/cloudtrail_s3_dataevents_write_enabled_test.py b/tests/providers/aws/services/cloudtrail/cloudtrail_s3_dataevents_write_enabled/cloudtrail_s3_dataevents_write_enabled_test.py index 0fc3c2e4b5..93813a7e27 100644 --- a/tests/providers/aws/services/cloudtrail/cloudtrail_s3_dataevents_write_enabled/cloudtrail_s3_dataevents_write_enabled_test.py +++ b/tests/providers/aws/services/cloudtrail/cloudtrail_s3_dataevents_write_enabled/cloudtrail_s3_dataevents_write_enabled_test.py @@ -32,19 +32,15 @@ class Test_cloudtrail_s3_dataevents_write_enabled: aws_provider = set_mocked_aws_provider() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled import ( @@ -100,19 +96,15 @@ class Test_cloudtrail_s3_dataevents_write_enabled: aws_provider = set_mocked_aws_provider() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled import ( @@ -146,19 +138,15 @@ class Test_cloudtrail_s3_dataevents_write_enabled: aws_provider = set_mocked_aws_provider() aws_provider._scan_unused_services = False - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled import ( @@ -183,19 +171,15 @@ class Test_cloudtrail_s3_dataevents_write_enabled: aws_provider = set_mocked_aws_provider() aws_provider._scan_unused_services = False - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled import ( @@ -251,19 +235,15 @@ class Test_cloudtrail_s3_dataevents_write_enabled: aws_provider = set_mocked_aws_provider() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled import ( @@ -315,19 +295,15 @@ class Test_cloudtrail_s3_dataevents_write_enabled: aws_provider = set_mocked_aws_provider() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled import ( @@ -386,19 +362,15 @@ class Test_cloudtrail_s3_dataevents_write_enabled: aws_provider = set_mocked_aws_provider() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled import ( @@ -428,19 +400,15 @@ class Test_cloudtrail_s3_dataevents_write_enabled: aws_provider = set_mocked_aws_provider() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ) as cloudtrail_service, - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ) as cloudtrail_service, mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled import ( @@ -493,19 +461,15 @@ class Test_cloudtrail_s3_dataevents_write_enabled: ) from prowler.providers.aws.services.s3.s3_service import S3 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.cloudtrail_client", - new=Cloudtrail(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.cloudtrail_client", + new=Cloudtrail(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudtrail.cloudtrail_s3_dataevents_write_enabled.cloudtrail_s3_dataevents_write_enabled import ( diff --git a/tests/providers/aws/services/cloudwatch/cloudwatch_alarm_actions_alarm_state_configured/cloudwatch_alarm_actions_alarm_state_configured_test.py b/tests/providers/aws/services/cloudwatch/cloudwatch_alarm_actions_alarm_state_configured/cloudwatch_alarm_actions_alarm_state_configured_test.py index ae84723476..ef4aea0ae9 100644 --- a/tests/providers/aws/services/cloudwatch/cloudwatch_alarm_actions_alarm_state_configured/cloudwatch_alarm_actions_alarm_state_configured_test.py +++ b/tests/providers/aws/services/cloudwatch/cloudwatch_alarm_actions_alarm_state_configured/cloudwatch_alarm_actions_alarm_state_configured_test.py @@ -18,15 +18,12 @@ class Test_cloudwatch_alarm_actions_alarm_state_configured: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudwatch.cloudwatch_alarm_actions_alarm_state_configured.cloudwatch_alarm_actions_alarm_state_configured.cloudwatch_client", - new=CloudWatch(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudwatch.cloudwatch_alarm_actions_alarm_state_configured.cloudwatch_alarm_actions_alarm_state_configured.cloudwatch_client", + new=CloudWatch(aws_provider), ): from prowler.providers.aws.services.cloudwatch.cloudwatch_alarm_actions_alarm_state_configured.cloudwatch_alarm_actions_alarm_state_configured import ( cloudwatch_alarm_actions_alarm_state_configured, @@ -87,15 +84,12 @@ class Test_cloudwatch_alarm_actions_alarm_state_configured: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudwatch.cloudwatch_alarm_actions_alarm_state_configured.cloudwatch_alarm_actions_alarm_state_configured.cloudwatch_client", - new=CloudWatch(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudwatch.cloudwatch_alarm_actions_alarm_state_configured.cloudwatch_alarm_actions_alarm_state_configured.cloudwatch_client", + new=CloudWatch(aws_provider), ): from prowler.providers.aws.services.cloudwatch.cloudwatch_alarm_actions_alarm_state_configured.cloudwatch_alarm_actions_alarm_state_configured import ( cloudwatch_alarm_actions_alarm_state_configured, @@ -136,15 +130,12 @@ class Test_cloudwatch_alarm_actions_alarm_state_configured: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudwatch.cloudwatch_alarm_actions_alarm_state_configured.cloudwatch_alarm_actions_alarm_state_configured.cloudwatch_client", - new=CloudWatch(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudwatch.cloudwatch_alarm_actions_alarm_state_configured.cloudwatch_alarm_actions_alarm_state_configured.cloudwatch_client", + new=CloudWatch(aws_provider), ): from prowler.providers.aws.services.cloudwatch.cloudwatch_alarm_actions_alarm_state_configured.cloudwatch_alarm_actions_alarm_state_configured import ( cloudwatch_alarm_actions_alarm_state_configured, diff --git a/tests/providers/aws/services/cloudwatch/cloudwatch_alarm_actions_enabled/cloudwatch_alarm_actions_enabled_test.py b/tests/providers/aws/services/cloudwatch/cloudwatch_alarm_actions_enabled/cloudwatch_alarm_actions_enabled_test.py index 178e2f2f86..94af6b0d50 100644 --- a/tests/providers/aws/services/cloudwatch/cloudwatch_alarm_actions_enabled/cloudwatch_alarm_actions_enabled_test.py +++ b/tests/providers/aws/services/cloudwatch/cloudwatch_alarm_actions_enabled/cloudwatch_alarm_actions_enabled_test.py @@ -18,15 +18,12 @@ class Test_cloudwatch_alarm_actions_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudwatch.cloudwatch_alarm_actions_enabled.cloudwatch_alarm_actions_enabled.cloudwatch_client", - new=CloudWatch(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudwatch.cloudwatch_alarm_actions_enabled.cloudwatch_alarm_actions_enabled.cloudwatch_client", + new=CloudWatch(aws_provider), ): from prowler.providers.aws.services.cloudwatch.cloudwatch_alarm_actions_enabled.cloudwatch_alarm_actions_enabled import ( cloudwatch_alarm_actions_enabled, @@ -86,15 +83,12 @@ class Test_cloudwatch_alarm_actions_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudwatch.cloudwatch_alarm_actions_enabled.cloudwatch_alarm_actions_enabled.cloudwatch_client", - new=CloudWatch(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudwatch.cloudwatch_alarm_actions_enabled.cloudwatch_alarm_actions_enabled.cloudwatch_client", + new=CloudWatch(aws_provider), ): from prowler.providers.aws.services.cloudwatch.cloudwatch_alarm_actions_enabled.cloudwatch_alarm_actions_enabled import ( cloudwatch_alarm_actions_enabled, @@ -135,15 +129,12 @@ class Test_cloudwatch_alarm_actions_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudwatch.cloudwatch_alarm_actions_enabled.cloudwatch_alarm_actions_enabled.cloudwatch_client", - new=CloudWatch(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudwatch.cloudwatch_alarm_actions_enabled.cloudwatch_alarm_actions_enabled.cloudwatch_client", + new=CloudWatch(aws_provider), ): from prowler.providers.aws.services.cloudwatch.cloudwatch_alarm_actions_enabled.cloudwatch_alarm_actions_enabled import ( cloudwatch_alarm_actions_enabled, diff --git a/tests/providers/aws/services/cloudwatch/cloudwatch_cross_account_sharing_disabled/cloudwatch_cross_account_sharing_disabled_test.py b/tests/providers/aws/services/cloudwatch/cloudwatch_cross_account_sharing_disabled/cloudwatch_cross_account_sharing_disabled_test.py index 9e464b23cc..7a70c1f515 100644 --- a/tests/providers/aws/services/cloudwatch/cloudwatch_cross_account_sharing_disabled/cloudwatch_cross_account_sharing_disabled_test.py +++ b/tests/providers/aws/services/cloudwatch/cloudwatch_cross_account_sharing_disabled/cloudwatch_cross_account_sharing_disabled_test.py @@ -30,15 +30,12 @@ class Test_cloudwatch_cross_account_sharing_disabled: audit_progress=0, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudwatch.cloudwatch_cross_account_sharing_disabled.cloudwatch_cross_account_sharing_disabled.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudwatch.cloudwatch_cross_account_sharing_disabled.cloudwatch_cross_account_sharing_disabled.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudwatch.cloudwatch_cross_account_sharing_disabled.cloudwatch_cross_account_sharing_disabled import ( @@ -90,15 +87,12 @@ class Test_cloudwatch_cross_account_sharing_disabled: audit_progress=0, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudwatch.cloudwatch_cross_account_sharing_disabled.cloudwatch_cross_account_sharing_disabled.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudwatch.cloudwatch_cross_account_sharing_disabled.cloudwatch_cross_account_sharing_disabled.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudwatch.cloudwatch_cross_account_sharing_disabled.cloudwatch_cross_account_sharing_disabled import ( @@ -145,16 +139,13 @@ class Test_cloudwatch_cross_account_sharing_disabled: audit_progress=0, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudwatch.cloudwatch_cross_account_sharing_disabled.cloudwatch_cross_account_sharing_disabled.iam_client", - new=IAM(aws_provider), - ) as iam_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudwatch.cloudwatch_cross_account_sharing_disabled.cloudwatch_cross_account_sharing_disabled.iam_client", + new=IAM(aws_provider), + ) as iam_client: # Test Check from prowler.providers.aws.services.cloudwatch.cloudwatch_cross_account_sharing_disabled.cloudwatch_cross_account_sharing_disabled import ( cloudwatch_cross_account_sharing_disabled, diff --git a/tests/providers/aws/services/cloudwatch/cloudwatch_log_group_kms_encryption_enabled/cloudwatch_log_group_kms_encryption_enabled_test.py b/tests/providers/aws/services/cloudwatch/cloudwatch_log_group_kms_encryption_enabled/cloudwatch_log_group_kms_encryption_enabled_test.py index dfac03fe83..24310e279b 100644 --- a/tests/providers/aws/services/cloudwatch/cloudwatch_log_group_kms_encryption_enabled/cloudwatch_log_group_kms_encryption_enabled_test.py +++ b/tests/providers/aws/services/cloudwatch/cloudwatch_log_group_kms_encryption_enabled/cloudwatch_log_group_kms_encryption_enabled_test.py @@ -28,15 +28,12 @@ class Test_cloudwatch_log_group_kms_encryption_enabled: audit_progress=0, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudwatch.cloudwatch_log_group_kms_encryption_enabled.cloudwatch_log_group_kms_encryption_enabled.logs_client", - new=Logs(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudwatch.cloudwatch_log_group_kms_encryption_enabled.cloudwatch_log_group_kms_encryption_enabled.logs_client", + new=Logs(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudwatch.cloudwatch_log_group_kms_encryption_enabled.cloudwatch_log_group_kms_encryption_enabled import ( @@ -73,15 +70,12 @@ class Test_cloudwatch_log_group_kms_encryption_enabled: audit_progress=0, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudwatch.cloudwatch_log_group_kms_encryption_enabled.cloudwatch_log_group_kms_encryption_enabled.logs_client", - new=Logs(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudwatch.cloudwatch_log_group_kms_encryption_enabled.cloudwatch_log_group_kms_encryption_enabled.logs_client", + new=Logs(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudwatch.cloudwatch_log_group_kms_encryption_enabled.cloudwatch_log_group_kms_encryption_enabled import ( @@ -122,15 +116,12 @@ class Test_cloudwatch_log_group_kms_encryption_enabled: audit_progress=0, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudwatch.cloudwatch_log_group_kms_encryption_enabled.cloudwatch_log_group_kms_encryption_enabled.logs_client", - new=Logs(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudwatch.cloudwatch_log_group_kms_encryption_enabled.cloudwatch_log_group_kms_encryption_enabled.logs_client", + new=Logs(aws_provider), ): # Test Check from prowler.providers.aws.services.cloudwatch.cloudwatch_log_group_kms_encryption_enabled.cloudwatch_log_group_kms_encryption_enabled import ( @@ -166,16 +157,13 @@ class Test_cloudwatch_log_group_kms_encryption_enabled: audit_progress=0, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.cloudwatch.cloudwatch_log_group_kms_encryption_enabled.cloudwatch_log_group_kms_encryption_enabled.logs_client", - new=Logs(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.cloudwatch.cloudwatch_log_group_kms_encryption_enabled.cloudwatch_log_group_kms_encryption_enabled.logs_client", + new=Logs(aws_provider), + ) as service_client: # Test Check from prowler.providers.aws.services.cloudwatch.cloudwatch_log_group_kms_encryption_enabled.cloudwatch_log_group_kms_encryption_enabled import ( cloudwatch_log_group_kms_encryption_enabled, diff --git a/tests/providers/aws/services/codeartifact/codeartifact_packages_external_public_publishing_disabled/codeartifact_packages_external_public_publishing_disabled_fixer_test.py b/tests/providers/aws/services/codeartifact/codeartifact_packages_external_public_publishing_disabled/codeartifact_packages_external_public_publishing_disabled_fixer_test.py index f776406e9c..9b1d67c811 100644 --- a/tests/providers/aws/services/codeartifact/codeartifact_packages_external_public_publishing_disabled/codeartifact_packages_external_public_publishing_disabled_fixer_test.py +++ b/tests/providers/aws/services/codeartifact/codeartifact_packages_external_public_publishing_disabled/codeartifact_packages_external_public_publishing_disabled_fixer_test.py @@ -47,15 +47,12 @@ class Test_codeartifact_packages_external_public_publishing_disabled_fixer: ): aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.codeartifact.codeartifact_client.codeartifact_client", - new=CodeArtifact(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.codeartifact.codeartifact_client.codeartifact_client", + new=CodeArtifact(aws_provider), ): # Test Fixer from prowler.providers.aws.services.codeartifact.codeartifact_packages_external_public_publishing_disabled.codeartifact_packages_external_public_publishing_disabled_fixer import ( @@ -73,15 +70,12 @@ class Test_codeartifact_packages_external_public_publishing_disabled_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.codeartifact.codeartifact_client.codeartifact_client", - new=CodeArtifact(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.codeartifact.codeartifact_client.codeartifact_client", + new=CodeArtifact(aws_provider), ): # Test Fixer from prowler.providers.aws.services.codeartifact.codeartifact_packages_external_public_publishing_disabled.codeartifact_packages_external_public_publishing_disabled_fixer import ( diff --git a/tests/providers/aws/services/codeartifact/codeartifact_packages_external_public_publishing_disabled/codeartifact_packages_external_public_publishing_disabled_test.py b/tests/providers/aws/services/codeartifact/codeartifact_packages_external_public_publishing_disabled/codeartifact_packages_external_public_publishing_disabled_test.py index 342a8595bc..a8d1d526ea 100644 --- a/tests/providers/aws/services/codeartifact/codeartifact_packages_external_public_publishing_disabled/codeartifact_packages_external_public_publishing_disabled_test.py +++ b/tests/providers/aws/services/codeartifact/codeartifact_packages_external_public_publishing_disabled/codeartifact_packages_external_public_publishing_disabled_test.py @@ -20,15 +20,12 @@ class Test_codeartifact_packages_external_public_publishing_disabled: def test_no_repositories(self): codeartifact_client = mock.MagicMock codeartifact_client.repositories = {} - with ( - mock.patch( - "prowler.providers.aws.services.codeartifact.codeartifact_service.CodeArtifact", - new=codeartifact_client, - ), - mock.patch( - "prowler.providers.aws.services.codeartifact.codeartifact_client.codeartifact_client", - new=codeartifact_client, - ), + with mock.patch( + "prowler.providers.aws.services.codeartifact.codeartifact_service.CodeArtifact", + new=codeartifact_client, + ), mock.patch( + "prowler.providers.aws.services.codeartifact.codeartifact_client.codeartifact_client", + new=codeartifact_client, ): # Test Check from prowler.providers.aws.services.codeartifact.codeartifact_packages_external_public_publishing_disabled.codeartifact_packages_external_public_publishing_disabled import ( @@ -52,15 +49,12 @@ class Test_codeartifact_packages_external_public_publishing_disabled: packages=[], ) } - with ( - mock.patch( - "prowler.providers.aws.services.codeartifact.codeartifact_service.CodeArtifact", - new=codeartifact_client, - ), - mock.patch( - "prowler.providers.aws.services.codeartifact.codeartifact_client.codeartifact_client", - new=codeartifact_client, - ), + with mock.patch( + "prowler.providers.aws.services.codeartifact.codeartifact_service.CodeArtifact", + new=codeartifact_client, + ), mock.patch( + "prowler.providers.aws.services.codeartifact.codeartifact_client.codeartifact_client", + new=codeartifact_client, ): # Test Check from prowler.providers.aws.services.codeartifact.codeartifact_packages_external_public_publishing_disabled.codeartifact_packages_external_public_publishing_disabled import ( @@ -106,15 +100,12 @@ class Test_codeartifact_packages_external_public_publishing_disabled: ], ) } - with ( - mock.patch( - "prowler.providers.aws.services.codeartifact.codeartifact_service.CodeArtifact", - new=codeartifact_client, - ), - mock.patch( - "prowler.providers.aws.services.codeartifact.codeartifact_client.codeartifact_client", - new=codeartifact_client, - ), + with mock.patch( + "prowler.providers.aws.services.codeartifact.codeartifact_service.CodeArtifact", + new=codeartifact_client, + ), mock.patch( + "prowler.providers.aws.services.codeartifact.codeartifact_client.codeartifact_client", + new=codeartifact_client, ): # Test Check from prowler.providers.aws.services.codeartifact.codeartifact_packages_external_public_publishing_disabled.codeartifact_packages_external_public_publishing_disabled import ( @@ -172,15 +163,12 @@ class Test_codeartifact_packages_external_public_publishing_disabled: ], ) } - with ( - mock.patch( - "prowler.providers.aws.services.codeartifact.codeartifact_service.CodeArtifact", - new=codeartifact_client, - ), - mock.patch( - "prowler.providers.aws.services.codeartifact.codeartifact_client.codeartifact_client", - new=codeartifact_client, - ), + with mock.patch( + "prowler.providers.aws.services.codeartifact.codeartifact_service.CodeArtifact", + new=codeartifact_client, + ), mock.patch( + "prowler.providers.aws.services.codeartifact.codeartifact_client.codeartifact_client", + new=codeartifact_client, ): # Test Check from prowler.providers.aws.services.codeartifact.codeartifact_packages_external_public_publishing_disabled.codeartifact_packages_external_public_publishing_disabled import ( diff --git a/tests/providers/aws/services/codebuild/codebuild_project_logging_enabled/codebuild_project_logging_enabled_test.py b/tests/providers/aws/services/codebuild/codebuild_project_logging_enabled/codebuild_project_logging_enabled_test.py index 8428b1ece3..df1bf4936b 100644 --- a/tests/providers/aws/services/codebuild/codebuild_project_logging_enabled/codebuild_project_logging_enabled_test.py +++ b/tests/providers/aws/services/codebuild/codebuild_project_logging_enabled/codebuild_project_logging_enabled_test.py @@ -13,15 +13,12 @@ class Test_codebuild_project_logging_enabled: from prowler.providers.aws.services.codebuild.codebuild_service import Codebuild - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.codebuild.codebuild_project_logging_enabled.codebuild_project_logging_enabled.codebuild_client", - new=Codebuild(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.codebuild.codebuild_project_logging_enabled.codebuild_project_logging_enabled.codebuild_client", + new=Codebuild(aws_provider), ): from prowler.providers.aws.services.codebuild.codebuild_project_logging_enabled.codebuild_project_logging_enabled import ( codebuild_project_logging_enabled, @@ -69,15 +66,12 @@ class Test_codebuild_project_logging_enabled: from prowler.providers.aws.services.codebuild.codebuild_service import Codebuild - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.codebuild.codebuild_project_logging_enabled.codebuild_project_logging_enabled.codebuild_client", - new=Codebuild(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.codebuild.codebuild_project_logging_enabled.codebuild_project_logging_enabled.codebuild_client", + new=Codebuild(aws_provider), ): from prowler.providers.aws.services.codebuild.codebuild_project_logging_enabled.codebuild_project_logging_enabled import ( codebuild_project_logging_enabled, @@ -134,15 +128,12 @@ class Test_codebuild_project_logging_enabled: from prowler.providers.aws.services.codebuild.codebuild_service import Codebuild - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.codebuild.codebuild_project_logging_enabled.codebuild_project_logging_enabled.codebuild_client", - new=Codebuild(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.codebuild.codebuild_project_logging_enabled.codebuild_project_logging_enabled.codebuild_client", + new=Codebuild(aws_provider), ): from prowler.providers.aws.services.codebuild.codebuild_project_logging_enabled.codebuild_project_logging_enabled import ( codebuild_project_logging_enabled, @@ -203,15 +194,12 @@ class Test_codebuild_project_logging_enabled: from prowler.providers.aws.services.codebuild.codebuild_service import Codebuild - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.codebuild.codebuild_project_logging_enabled.codebuild_project_logging_enabled.codebuild_client", - new=Codebuild(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.codebuild.codebuild_project_logging_enabled.codebuild_project_logging_enabled.codebuild_client", + new=Codebuild(aws_provider), ): from prowler.providers.aws.services.codebuild.codebuild_project_logging_enabled.codebuild_project_logging_enabled import ( codebuild_project_logging_enabled, @@ -267,15 +255,12 @@ class Test_codebuild_project_logging_enabled: from prowler.providers.aws.services.codebuild.codebuild_service import Codebuild - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.codebuild.codebuild_project_logging_enabled.codebuild_project_logging_enabled.codebuild_client", - new=Codebuild(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.codebuild.codebuild_project_logging_enabled.codebuild_project_logging_enabled.codebuild_client", + new=Codebuild(aws_provider), ): from prowler.providers.aws.services.codebuild.codebuild_project_logging_enabled.codebuild_project_logging_enabled import ( codebuild_project_logging_enabled, diff --git a/tests/providers/aws/services/codebuild/codebuild_project_no_secrets_in_variables/codebuild_project_no_secrets_in_variables_test.py b/tests/providers/aws/services/codebuild/codebuild_project_no_secrets_in_variables/codebuild_project_no_secrets_in_variables_test.py index c08d1c8bb3..c719626296 100644 --- a/tests/providers/aws/services/codebuild/codebuild_project_no_secrets_in_variables/codebuild_project_no_secrets_in_variables_test.py +++ b/tests/providers/aws/services/codebuild/codebuild_project_no_secrets_in_variables/codebuild_project_no_secrets_in_variables_test.py @@ -11,15 +11,12 @@ class Test_codebuild_project_no_secrets_in_variables: codebuild_client.audit_config = {"excluded_sensitive_environment_variables": []} - with ( - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", - codebuild_client, - ), - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables.codebuild_client", - codebuild_client, - ), + with mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", + codebuild_client, + ), mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables.codebuild_client", + codebuild_client, ): from prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables import ( codebuild_project_no_secrets_in_variables, @@ -50,15 +47,12 @@ class Test_codebuild_project_no_secrets_in_variables: codebuild_client.audit_config = {"excluded_sensitive_environment_variables": []} - with ( - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", - codebuild_client, - ), - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables.codebuild_client", - codebuild_client, - ), + with mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", + codebuild_client, + ), mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables.codebuild_client", + codebuild_client, ): from prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables import ( codebuild_project_no_secrets_in_variables, @@ -104,15 +98,12 @@ class Test_codebuild_project_no_secrets_in_variables: codebuild_client.audit_config = {"excluded_sensitive_environment_variables": []} - with ( - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", - codebuild_client, - ), - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables.codebuild_client", - codebuild_client, - ), + with mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", + codebuild_client, + ), mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables.codebuild_client", + codebuild_client, ): from prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables import ( codebuild_project_no_secrets_in_variables, @@ -158,15 +149,12 @@ class Test_codebuild_project_no_secrets_in_variables: codebuild_client.audit_config = {"excluded_sensitive_environment_variables": []} - with ( - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", - codebuild_client, - ), - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables.codebuild_client", - codebuild_client, - ), + with mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", + codebuild_client, + ), mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables.codebuild_client", + codebuild_client, ): from prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables import ( codebuild_project_no_secrets_in_variables, @@ -212,15 +200,12 @@ class Test_codebuild_project_no_secrets_in_variables: codebuild_client.audit_config = {"excluded_sensitive_environment_variables": []} - with ( - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", - codebuild_client, - ), - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables.codebuild_client", - codebuild_client, - ), + with mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", + codebuild_client, + ), mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables.codebuild_client", + codebuild_client, ): from prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables import ( codebuild_project_no_secrets_in_variables, @@ -268,15 +253,12 @@ class Test_codebuild_project_no_secrets_in_variables: "excluded_sensitive_environment_variables": ["AWS_DUMB_ACCESS_KEY"] } - with ( - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", - codebuild_client, - ), - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables.codebuild_client", - codebuild_client, - ), + with mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", + codebuild_client, + ), mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables.codebuild_client", + codebuild_client, ): from prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables import ( codebuild_project_no_secrets_in_variables, @@ -329,15 +311,12 @@ class Test_codebuild_project_no_secrets_in_variables: "excluded_sensitive_environment_variables": ["AWS_DUMB_ACCESS_KEY"] } - with ( - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", - codebuild_client, - ), - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables.codebuild_client", - codebuild_client, - ), + with mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", + codebuild_client, + ), mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables.codebuild_client", + codebuild_client, ): from prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables import ( codebuild_project_no_secrets_in_variables, @@ -390,15 +369,12 @@ class Test_codebuild_project_no_secrets_in_variables: "excluded_sensitive_environment_variables": ["AWS_DUMB_ACCESS_KEY"] } - with ( - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", - codebuild_client, - ), - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables.codebuild_client", - codebuild_client, - ), + with mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", + codebuild_client, + ), mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables.codebuild_client", + codebuild_client, ): from prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables import ( codebuild_project_no_secrets_in_variables, @@ -449,15 +425,12 @@ class Test_codebuild_project_no_secrets_in_variables: codebuild_client.audit_config = {"excluded_sensitive_environment_variables": []} - with ( - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", - codebuild_client, - ), - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables.codebuild_client", - codebuild_client, - ), + with mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", + codebuild_client, + ), mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables.codebuild_client", + codebuild_client, ): from prowler.providers.aws.services.codebuild.codebuild_project_no_secrets_in_variables.codebuild_project_no_secrets_in_variables import ( codebuild_project_no_secrets_in_variables, diff --git a/tests/providers/aws/services/codebuild/codebuild_project_older_90_days/codebuild_project_older_90_days_test.py b/tests/providers/aws/services/codebuild/codebuild_project_older_90_days/codebuild_project_older_90_days_test.py index 338ed52f90..0f269280cc 100644 --- a/tests/providers/aws/services/codebuild/codebuild_project_older_90_days/codebuild_project_older_90_days_test.py +++ b/tests/providers/aws/services/codebuild/codebuild_project_older_90_days/codebuild_project_older_90_days_test.py @@ -23,15 +23,12 @@ class Test_codebuild_project_older_90_days: ) } - with ( - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", - codebuild_client, - ), - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_project_older_90_days.codebuild_project_older_90_days.codebuild_client", - codebuild_client, - ), + with mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", + codebuild_client, + ), mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_project_older_90_days.codebuild_project_older_90_days.codebuild_client", + codebuild_client, ): from prowler.providers.aws.services.codebuild.codebuild_project_older_90_days.codebuild_project_older_90_days import ( codebuild_project_older_90_days, @@ -66,15 +63,12 @@ class Test_codebuild_project_older_90_days: ) } - with ( - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", - codebuild_client, - ), - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_project_older_90_days.codebuild_project_older_90_days.codebuild_client", - codebuild_client, - ), + with mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", + codebuild_client, + ), mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_project_older_90_days.codebuild_project_older_90_days.codebuild_client", + codebuild_client, ): from prowler.providers.aws.services.codebuild.codebuild_project_older_90_days.codebuild_project_older_90_days import ( codebuild_project_older_90_days, @@ -109,15 +103,12 @@ class Test_codebuild_project_older_90_days: ) } - with ( - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", - codebuild_client, - ), - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_project_older_90_days.codebuild_project_older_90_days.codebuild_client", - codebuild_client, - ), + with mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", + codebuild_client, + ), mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_project_older_90_days.codebuild_project_older_90_days.codebuild_client", + codebuild_client, ): from prowler.providers.aws.services.codebuild.codebuild_project_older_90_days.codebuild_project_older_90_days import ( codebuild_project_older_90_days, diff --git a/tests/providers/aws/services/codebuild/codebuild_project_s3_logs_encrypted/codebuild_project_s3_logs_encrypted_test.py b/tests/providers/aws/services/codebuild/codebuild_project_s3_logs_encrypted/codebuild_project_s3_logs_encrypted_test.py index c6fd8f4849..36808c38a5 100644 --- a/tests/providers/aws/services/codebuild/codebuild_project_s3_logs_encrypted/codebuild_project_s3_logs_encrypted_test.py +++ b/tests/providers/aws/services/codebuild/codebuild_project_s3_logs_encrypted/codebuild_project_s3_logs_encrypted_test.py @@ -13,15 +13,12 @@ class Test_codebuild_project_s3_logs_encrypted: from prowler.providers.aws.services.codebuild.codebuild_service import Codebuild - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.codebuild.codebuild_project_s3_logs_encrypted.codebuild_project_s3_logs_encrypted.codebuild_client", - new=Codebuild(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.codebuild.codebuild_project_s3_logs_encrypted.codebuild_project_s3_logs_encrypted.codebuild_client", + new=Codebuild(aws_provider), ): from prowler.providers.aws.services.codebuild.codebuild_project_s3_logs_encrypted.codebuild_project_s3_logs_encrypted import ( codebuild_project_s3_logs_encrypted, @@ -63,15 +60,12 @@ class Test_codebuild_project_s3_logs_encrypted: from prowler.providers.aws.services.codebuild.codebuild_service import Codebuild - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.codebuild.codebuild_project_s3_logs_encrypted.codebuild_project_s3_logs_encrypted.codebuild_client", - new=Codebuild(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.codebuild.codebuild_project_s3_logs_encrypted.codebuild_project_s3_logs_encrypted.codebuild_client", + new=Codebuild(aws_provider), ): from prowler.providers.aws.services.codebuild.codebuild_project_s3_logs_encrypted.codebuild_project_s3_logs_encrypted import ( codebuild_project_s3_logs_encrypted, @@ -120,15 +114,12 @@ class Test_codebuild_project_s3_logs_encrypted: from prowler.providers.aws.services.codebuild.codebuild_service import Codebuild - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.codebuild.codebuild_project_s3_logs_encrypted.codebuild_project_s3_logs_encrypted.codebuild_client", - new=Codebuild(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.codebuild.codebuild_project_s3_logs_encrypted.codebuild_project_s3_logs_encrypted.codebuild_client", + new=Codebuild(aws_provider), ): from prowler.providers.aws.services.codebuild.codebuild_project_s3_logs_encrypted.codebuild_project_s3_logs_encrypted import ( codebuild_project_s3_logs_encrypted, @@ -186,15 +177,12 @@ class Test_codebuild_project_s3_logs_encrypted: from prowler.providers.aws.services.codebuild.codebuild_service import Codebuild - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.codebuild.codebuild_project_s3_logs_encrypted.codebuild_project_s3_logs_encrypted.codebuild_client", - new=Codebuild(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.codebuild.codebuild_project_s3_logs_encrypted.codebuild_project_s3_logs_encrypted.codebuild_client", + new=Codebuild(aws_provider), ): from prowler.providers.aws.services.codebuild.codebuild_project_s3_logs_encrypted.codebuild_project_s3_logs_encrypted import ( codebuild_project_s3_logs_encrypted, diff --git a/tests/providers/aws/services/codebuild/codebuild_project_source_repo_url_no_sensitive_credentials/codebuild_project_source_repo_url_no_sensitive_credentials_test.py b/tests/providers/aws/services/codebuild/codebuild_project_source_repo_url_no_sensitive_credentials/codebuild_project_source_repo_url_no_sensitive_credentials_test.py index dfc13d08ad..a4bd37aac3 100644 --- a/tests/providers/aws/services/codebuild/codebuild_project_source_repo_url_no_sensitive_credentials/codebuild_project_source_repo_url_no_sensitive_credentials_test.py +++ b/tests/providers/aws/services/codebuild/codebuild_project_source_repo_url_no_sensitive_credentials/codebuild_project_source_repo_url_no_sensitive_credentials_test.py @@ -23,15 +23,12 @@ class Test_codebuild_project_source_repo_url_no_sensitive_credentials: tags=[], ) } - with ( - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", - codebuild_client, - ), - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_project_source_repo_url_no_sensitive_credentials.codebuild_project_source_repo_url_no_sensitive_credentials.codebuild_client", - codebuild_client, - ), + with mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", + codebuild_client, + ), mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_project_source_repo_url_no_sensitive_credentials.codebuild_project_source_repo_url_no_sensitive_credentials.codebuild_client", + codebuild_client, ): from prowler.providers.aws.services.codebuild.codebuild_project_source_repo_url_no_sensitive_credentials.codebuild_project_source_repo_url_no_sensitive_credentials import ( codebuild_project_source_repo_url_no_sensitive_credentials, @@ -70,15 +67,12 @@ class Test_codebuild_project_source_repo_url_no_sensitive_credentials: tags=[], ) } - with ( - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", - codebuild_client, - ), - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_project_source_repo_url_no_sensitive_credentials.codebuild_project_source_repo_url_no_sensitive_credentials.codebuild_client", - codebuild_client, - ), + with mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", + codebuild_client, + ), mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_project_source_repo_url_no_sensitive_credentials.codebuild_project_source_repo_url_no_sensitive_credentials.codebuild_client", + codebuild_client, ): from prowler.providers.aws.services.codebuild.codebuild_project_source_repo_url_no_sensitive_credentials.codebuild_project_source_repo_url_no_sensitive_credentials import ( codebuild_project_source_repo_url_no_sensitive_credentials, @@ -117,15 +111,12 @@ class Test_codebuild_project_source_repo_url_no_sensitive_credentials: tags=[], ) } - with ( - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", - codebuild_client, - ), - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_project_source_repo_url_no_sensitive_credentials.codebuild_project_source_repo_url_no_sensitive_credentials.codebuild_client", - codebuild_client, - ), + with mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", + codebuild_client, + ), mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_project_source_repo_url_no_sensitive_credentials.codebuild_project_source_repo_url_no_sensitive_credentials.codebuild_client", + codebuild_client, ): from prowler.providers.aws.services.codebuild.codebuild_project_source_repo_url_no_sensitive_credentials.codebuild_project_source_repo_url_no_sensitive_credentials import ( codebuild_project_source_repo_url_no_sensitive_credentials, @@ -164,15 +155,12 @@ class Test_codebuild_project_source_repo_url_no_sensitive_credentials: tags=[], ) } - with ( - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", - codebuild_client, - ), - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_project_source_repo_url_no_sensitive_credentials.codebuild_project_source_repo_url_no_sensitive_credentials.codebuild_client", - codebuild_client, - ), + with mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", + codebuild_client, + ), mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_project_source_repo_url_no_sensitive_credentials.codebuild_project_source_repo_url_no_sensitive_credentials.codebuild_client", + codebuild_client, ): from prowler.providers.aws.services.codebuild.codebuild_project_source_repo_url_no_sensitive_credentials.codebuild_project_source_repo_url_no_sensitive_credentials import ( codebuild_project_source_repo_url_no_sensitive_credentials, diff --git a/tests/providers/aws/services/codebuild/codebuild_project_user_controlled_buildspec/codebuild_project_user_controlled_buildspec_test.py b/tests/providers/aws/services/codebuild/codebuild_project_user_controlled_buildspec/codebuild_project_user_controlled_buildspec_test.py index 3b394c0c59..331bc5f3f5 100644 --- a/tests/providers/aws/services/codebuild/codebuild_project_user_controlled_buildspec/codebuild_project_user_controlled_buildspec_test.py +++ b/tests/providers/aws/services/codebuild/codebuild_project_user_controlled_buildspec/codebuild_project_user_controlled_buildspec_test.py @@ -21,15 +21,12 @@ class Test_codebuild_project_user_controlled_buildspec: tags=[], ) } - with ( - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", - codebuild_client, - ), - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_project_user_controlled_buildspec.codebuild_project_user_controlled_buildspec.codebuild_client", - codebuild_client, - ), + with mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", + codebuild_client, + ), mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_project_user_controlled_buildspec.codebuild_project_user_controlled_buildspec.codebuild_client", + codebuild_client, ): from prowler.providers.aws.services.codebuild.codebuild_project_user_controlled_buildspec.codebuild_project_user_controlled_buildspec import ( codebuild_project_user_controlled_buildspec, @@ -64,15 +61,12 @@ class Test_codebuild_project_user_controlled_buildspec: ) } - with ( - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", - codebuild_client, - ), - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_project_user_controlled_buildspec.codebuild_project_user_controlled_buildspec.codebuild_client", - codebuild_client, - ), + with mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", + codebuild_client, + ), mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_project_user_controlled_buildspec.codebuild_project_user_controlled_buildspec.codebuild_client", + codebuild_client, ): from prowler.providers.aws.services.codebuild.codebuild_project_user_controlled_buildspec.codebuild_project_user_controlled_buildspec import ( codebuild_project_user_controlled_buildspec, @@ -106,15 +100,12 @@ class Test_codebuild_project_user_controlled_buildspec: tags=[], ) } - with ( - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", - codebuild_client, - ), - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_project_user_controlled_buildspec.codebuild_project_user_controlled_buildspec.codebuild_client", - codebuild_client, - ), + with mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", + codebuild_client, + ), mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_project_user_controlled_buildspec.codebuild_project_user_controlled_buildspec.codebuild_client", + codebuild_client, ): from prowler.providers.aws.services.codebuild.codebuild_project_user_controlled_buildspec.codebuild_project_user_controlled_buildspec import ( codebuild_project_user_controlled_buildspec, @@ -148,15 +139,12 @@ class Test_codebuild_project_user_controlled_buildspec: tags=[], ) } - with ( - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", - codebuild_client, - ), - mock.patch( - "prowler.providers.aws.services.codebuild.codebuild_project_user_controlled_buildspec.codebuild_project_user_controlled_buildspec.codebuild_client", - codebuild_client, - ), + with mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_service.Codebuild", + codebuild_client, + ), mock.patch( + "prowler.providers.aws.services.codebuild.codebuild_project_user_controlled_buildspec.codebuild_project_user_controlled_buildspec.codebuild_client", + codebuild_client, ): from prowler.providers.aws.services.codebuild.codebuild_project_user_controlled_buildspec.codebuild_project_user_controlled_buildspec import ( codebuild_project_user_controlled_buildspec, diff --git a/tests/providers/aws/services/codebuild/codebuild_report_group_export_encrypted/codebuild_report_group_export_encrypted_test.py b/tests/providers/aws/services/codebuild/codebuild_report_group_export_encrypted/codebuild_report_group_export_encrypted_test.py index b852d51ee2..49babc3477 100644 --- a/tests/providers/aws/services/codebuild/codebuild_report_group_export_encrypted/codebuild_report_group_export_encrypted_test.py +++ b/tests/providers/aws/services/codebuild/codebuild_report_group_export_encrypted/codebuild_report_group_export_encrypted_test.py @@ -83,15 +83,12 @@ class Test_codebuild_report_group_export_encrypted: from prowler.providers.aws.services.codebuild.codebuild_service import Codebuild - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.codebuild.codebuild_report_group_export_encrypted.codebuild_report_group_export_encrypted.codebuild_client", - new=Codebuild(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.codebuild.codebuild_report_group_export_encrypted.codebuild_report_group_export_encrypted.codebuild_client", + new=Codebuild(aws_provider), ): from prowler.providers.aws.services.codebuild.codebuild_report_group_export_encrypted.codebuild_report_group_export_encrypted import ( codebuild_report_group_export_encrypted, @@ -111,15 +108,12 @@ class Test_codebuild_report_group_export_encrypted: from prowler.providers.aws.services.codebuild.codebuild_service import Codebuild - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.codebuild.codebuild_report_group_export_encrypted.codebuild_report_group_export_encrypted.codebuild_client", - new=Codebuild(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.codebuild.codebuild_report_group_export_encrypted.codebuild_report_group_export_encrypted.codebuild_client", + new=Codebuild(aws_provider), ): from prowler.providers.aws.services.codebuild.codebuild_report_group_export_encrypted.codebuild_report_group_export_encrypted import ( codebuild_report_group_export_encrypted, @@ -154,15 +148,12 @@ class Test_codebuild_report_group_export_encrypted: from prowler.providers.aws.services.codebuild.codebuild_service import Codebuild - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.codebuild.codebuild_report_group_export_encrypted.codebuild_report_group_export_encrypted.codebuild_client", - new=Codebuild(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.codebuild.codebuild_report_group_export_encrypted.codebuild_report_group_export_encrypted.codebuild_client", + new=Codebuild(aws_provider), ): from prowler.providers.aws.services.codebuild.codebuild_report_group_export_encrypted.codebuild_report_group_export_encrypted import ( codebuild_report_group_export_encrypted, diff --git a/tests/providers/aws/services/cognito/cognito_identity_pool_guest_access_disabled/cognito_identity_pool_guest_access_disabled_test.py b/tests/providers/aws/services/cognito/cognito_identity_pool_guest_access_disabled/cognito_identity_pool_guest_access_disabled_test.py index 03daff3f95..a6802bb312 100644 --- a/tests/providers/aws/services/cognito/cognito_identity_pool_guest_access_disabled/cognito_identity_pool_guest_access_disabled_test.py +++ b/tests/providers/aws/services/cognito/cognito_identity_pool_guest_access_disabled/cognito_identity_pool_guest_access_disabled_test.py @@ -11,15 +11,12 @@ class Test_cognito_identity_pool_guest_access_disabled: def test_cognito_no_identity_pools(self): cognito_identity_client = mock.MagicMock cognito_identity_client.identity_pools = {} - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIdentity", - cognito_identity_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_identity_client.cognito_identity_client", - new=cognito_identity_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIdentity", + cognito_identity_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_identity_client.cognito_identity_client", + new=cognito_identity_client, ): from prowler.providers.aws.services.cognito.cognito_identity_pool_guest_access_disabled.cognito_identity_pool_guest_access_disabled import ( cognito_identity_pool_guest_access_disabled, @@ -44,15 +41,12 @@ class Test_cognito_identity_pool_guest_access_disabled: name=identity_pool_name, ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIdentity", - cognito_identity_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_identity_client.cognito_identity_client", - new=cognito_identity_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIdentity", + cognito_identity_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_identity_client.cognito_identity_client", + new=cognito_identity_client, ): from prowler.providers.aws.services.cognito.cognito_identity_pool_guest_access_disabled.cognito_identity_pool_guest_access_disabled import ( cognito_identity_pool_guest_access_disabled, @@ -87,15 +81,12 @@ class Test_cognito_identity_pool_guest_access_disabled: roles=IdentityPoolRoles(unauthenticated=unauthenticated_role), ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIdentity", - cognito_identity_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_identity_client.cognito_identity_client", - new=cognito_identity_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIdentity", + cognito_identity_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_identity_client.cognito_identity_client", + new=cognito_identity_client, ): from prowler.providers.aws.services.cognito.cognito_identity_pool_guest_access_disabled.cognito_identity_pool_guest_access_disabled import ( cognito_identity_pool_guest_access_disabled, diff --git a/tests/providers/aws/services/cognito/cognito_service_test.py b/tests/providers/aws/services/cognito/cognito_service_test.py index 9a9fbd3c2a..f94b1bf0af 100644 --- a/tests/providers/aws/services/cognito/cognito_service_test.py +++ b/tests/providers/aws/services/cognito/cognito_service_test.py @@ -116,15 +116,12 @@ class Test_Cognito_Service: ) ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): for user_pool in cognito_client.user_pools.values(): assert user_pool.region == "eu-west-1" @@ -166,15 +163,12 @@ class Test_Cognito_Service: ) ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): for user_pool in cognito_client.user_pools.values(): assert user_pool.region == "eu-west-1" @@ -256,15 +250,12 @@ class Test_Cognito_Service: ), ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): for user_pool in cognito_client.user_pools.values(): assert user_pool.region == "eu-west-1" @@ -390,15 +381,12 @@ class Test_Cognito_Service: unauthenticated="unauthenticated_role", ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_identity_client.cognito_identity_client", - new=cognito_identity_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_identity_client.cognito_identity_client", + new=cognito_identity_client, ): for identity_pool in cognito_identity_client.identity_pools.values(): assert identity_pool.region == "eu-west-1" diff --git a/tests/providers/aws/services/cognito/cognito_user_pool_advanced_security_enabled/cognito_user_pool_advanced_security_enabled_test.py b/tests/providers/aws/services/cognito/cognito_user_pool_advanced_security_enabled/cognito_user_pool_advanced_security_enabled_test.py index a06d17cc42..11a93d3997 100644 --- a/tests/providers/aws/services/cognito/cognito_user_pool_advanced_security_enabled/cognito_user_pool_advanced_security_enabled_test.py +++ b/tests/providers/aws/services/cognito/cognito_user_pool_advanced_security_enabled/cognito_user_pool_advanced_security_enabled_test.py @@ -9,15 +9,12 @@ class Test_cognito_user_pool_advanced_security_enabled: def test_cognito_no_user_pools(self): cognito_client = mock.MagicMock cognito_client.user_pools = {} - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_advanced_security_enabled.cognito_user_pool_advanced_security_enabled import ( cognito_user_pool_advanced_security_enabled, @@ -45,15 +42,12 @@ class Test_cognito_user_pool_advanced_security_enabled: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_advanced_security_enabled.cognito_user_pool_advanced_security_enabled import ( cognito_user_pool_advanced_security_enabled, @@ -88,15 +82,12 @@ class Test_cognito_user_pool_advanced_security_enabled: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_advanced_security_enabled.cognito_user_pool_advanced_security_enabled import ( cognito_user_pool_advanced_security_enabled, @@ -131,15 +122,12 @@ class Test_cognito_user_pool_advanced_security_enabled: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_advanced_security_enabled.cognito_user_pool_advanced_security_enabled import ( cognito_user_pool_advanced_security_enabled, diff --git a/tests/providers/aws/services/cognito/cognito_user_pool_blocks_compromised_credentials_sign_in_attempts/cognito_user_pool_blocks_compromised_credentials_sign_in_attempts_test.py b/tests/providers/aws/services/cognito/cognito_user_pool_blocks_compromised_credentials_sign_in_attempts/cognito_user_pool_blocks_compromised_credentials_sign_in_attempts_test.py index 6e37b67059..c8d226395d 100644 --- a/tests/providers/aws/services/cognito/cognito_user_pool_blocks_compromised_credentials_sign_in_attempts/cognito_user_pool_blocks_compromised_credentials_sign_in_attempts_test.py +++ b/tests/providers/aws/services/cognito/cognito_user_pool_blocks_compromised_credentials_sign_in_attempts/cognito_user_pool_blocks_compromised_credentials_sign_in_attempts_test.py @@ -13,15 +13,12 @@ class Test_cognito_user_pool_blocks_compromised_credentials_sign_in_attempts: def test_cognito_no_user_pools(self): cognito_client = mock.MagicMock cognito_client.user_pools = {} - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_blocks_compromised_credentials_sign_in_attempts.cognito_user_pool_blocks_compromised_credentials_sign_in_attempts import ( cognito_user_pool_blocks_compromised_credentials_sign_in_attempts, @@ -55,15 +52,12 @@ class Test_cognito_user_pool_blocks_compromised_credentials_sign_in_attempts: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_blocks_compromised_credentials_sign_in_attempts.cognito_user_pool_blocks_compromised_credentials_sign_in_attempts import ( cognito_user_pool_blocks_compromised_credentials_sign_in_attempts, @@ -103,15 +97,12 @@ class Test_cognito_user_pool_blocks_compromised_credentials_sign_in_attempts: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_blocks_compromised_credentials_sign_in_attempts.cognito_user_pool_blocks_compromised_credentials_sign_in_attempts import ( cognito_user_pool_blocks_compromised_credentials_sign_in_attempts, @@ -151,15 +142,12 @@ class Test_cognito_user_pool_blocks_compromised_credentials_sign_in_attempts: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_blocks_compromised_credentials_sign_in_attempts.cognito_user_pool_blocks_compromised_credentials_sign_in_attempts import ( cognito_user_pool_blocks_compromised_credentials_sign_in_attempts, @@ -199,15 +187,12 @@ class Test_cognito_user_pool_blocks_compromised_credentials_sign_in_attempts: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_blocks_compromised_credentials_sign_in_attempts.cognito_user_pool_blocks_compromised_credentials_sign_in_attempts import ( cognito_user_pool_blocks_compromised_credentials_sign_in_attempts, diff --git a/tests/providers/aws/services/cognito/cognito_user_pool_blocks_potential_malicious_sign_in_attempts/cognito_user_pool_blocks_potential_malicious_sign_in_attempts_test.py b/tests/providers/aws/services/cognito/cognito_user_pool_blocks_potential_malicious_sign_in_attempts/cognito_user_pool_blocks_potential_malicious_sign_in_attempts_test.py index ae233b2d17..416b6b4170 100644 --- a/tests/providers/aws/services/cognito/cognito_user_pool_blocks_potential_malicious_sign_in_attempts/cognito_user_pool_blocks_potential_malicious_sign_in_attempts_test.py +++ b/tests/providers/aws/services/cognito/cognito_user_pool_blocks_potential_malicious_sign_in_attempts/cognito_user_pool_blocks_potential_malicious_sign_in_attempts_test.py @@ -14,15 +14,12 @@ class Test_cognito_user_pool_blocks_potential_malicious_sign_in_attempts: cognito_client = mock.MagicMock cognito_client.user_pools = {} - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_blocks_potential_malicious_sign_in_attempts.cognito_user_pool_blocks_potential_malicious_sign_in_attempts import ( cognito_user_pool_blocks_potential_malicious_sign_in_attempts, @@ -57,15 +54,12 @@ class Test_cognito_user_pool_blocks_potential_malicious_sign_in_attempts: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_blocks_potential_malicious_sign_in_attempts.cognito_user_pool_blocks_potential_malicious_sign_in_attempts import ( cognito_user_pool_blocks_potential_malicious_sign_in_attempts, @@ -106,15 +100,12 @@ class Test_cognito_user_pool_blocks_potential_malicious_sign_in_attempts: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_blocks_potential_malicious_sign_in_attempts.cognito_user_pool_blocks_potential_malicious_sign_in_attempts import ( cognito_user_pool_blocks_potential_malicious_sign_in_attempts, @@ -155,15 +146,12 @@ class Test_cognito_user_pool_blocks_potential_malicious_sign_in_attempts: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_blocks_potential_malicious_sign_in_attempts.cognito_user_pool_blocks_potential_malicious_sign_in_attempts import ( cognito_user_pool_blocks_potential_malicious_sign_in_attempts, @@ -203,15 +191,12 @@ class Test_cognito_user_pool_blocks_potential_malicious_sign_in_attempts: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_blocks_potential_malicious_sign_in_attempts.cognito_user_pool_blocks_potential_malicious_sign_in_attempts import ( cognito_user_pool_blocks_potential_malicious_sign_in_attempts, diff --git a/tests/providers/aws/services/cognito/cognito_user_pool_client_prevent_user_existence_errors/cognito_user_pool_client_prevent_user_existence_errors_test.py b/tests/providers/aws/services/cognito/cognito_user_pool_client_prevent_user_existence_errors/cognito_user_pool_client_prevent_user_existence_errors_test.py index c73cc9db9a..723f5b4d06 100644 --- a/tests/providers/aws/services/cognito/cognito_user_pool_client_prevent_user_existence_errors/cognito_user_pool_client_prevent_user_existence_errors_test.py +++ b/tests/providers/aws/services/cognito/cognito_user_pool_client_prevent_user_existence_errors/cognito_user_pool_client_prevent_user_existence_errors_test.py @@ -12,15 +12,12 @@ class Test_cognito_user_pool_client_prevent_user_existence_errors: def test_cognito_no_user_pools(self): cognito_client = mock.MagicMock cognito_client.user_pools = {} - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_client_prevent_user_existence_errors.cognito_user_pool_client_prevent_user_existence_errors import ( cognito_user_pool_client_prevent_user_existence_errors, @@ -59,15 +56,12 @@ class Test_cognito_user_pool_client_prevent_user_existence_errors: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_client_prevent_user_existence_errors.cognito_user_pool_client_prevent_user_existence_errors import ( cognito_user_pool_client_prevent_user_existence_errors, @@ -112,15 +106,12 @@ class Test_cognito_user_pool_client_prevent_user_existence_errors: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_client_prevent_user_existence_errors.cognito_user_pool_client_prevent_user_existence_errors import ( cognito_user_pool_client_prevent_user_existence_errors, diff --git a/tests/providers/aws/services/cognito/cognito_user_pool_client_token_revocation_enabled/cognito_user_pool_client_token_revocation_enabled_test.py b/tests/providers/aws/services/cognito/cognito_user_pool_client_token_revocation_enabled/cognito_user_pool_client_token_revocation_enabled_test.py index 9c6d1ac4ea..e56eff411d 100644 --- a/tests/providers/aws/services/cognito/cognito_user_pool_client_token_revocation_enabled/cognito_user_pool_client_token_revocation_enabled_test.py +++ b/tests/providers/aws/services/cognito/cognito_user_pool_client_token_revocation_enabled/cognito_user_pool_client_token_revocation_enabled_test.py @@ -12,15 +12,12 @@ class Test_cognito_user_pool_client_token_revocation_enabled: def test_cognito_no_user_pools(self): cognito_client = mock.MagicMock cognito_client.user_pools = {} - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_client_token_revocation_enabled.cognito_user_pool_client_token_revocation_enabled import ( cognito_user_pool_client_token_revocation_enabled, @@ -59,15 +56,12 @@ class Test_cognito_user_pool_client_token_revocation_enabled: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_client_token_revocation_enabled.cognito_user_pool_client_token_revocation_enabled import ( cognito_user_pool_client_token_revocation_enabled, @@ -113,15 +107,12 @@ class Test_cognito_user_pool_client_token_revocation_enabled: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_client_token_revocation_enabled.cognito_user_pool_client_token_revocation_enabled import ( cognito_user_pool_client_token_revocation_enabled, diff --git a/tests/providers/aws/services/cognito/cognito_user_pool_deletion_protection_enabled/cognito_user_pool_deletion_protection_enabled_test.py b/tests/providers/aws/services/cognito/cognito_user_pool_deletion_protection_enabled/cognito_user_pool_deletion_protection_enabled_test.py index 5f45f14dc6..427f89b777 100644 --- a/tests/providers/aws/services/cognito/cognito_user_pool_deletion_protection_enabled/cognito_user_pool_deletion_protection_enabled_test.py +++ b/tests/providers/aws/services/cognito/cognito_user_pool_deletion_protection_enabled/cognito_user_pool_deletion_protection_enabled_test.py @@ -9,15 +9,12 @@ class Test_cognito_user_pool_deletion_protection_enabled: def test_cognito_no_user_pools(self): cognito_client = mock.MagicMock cognito_client.user_pools = {} - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_deletion_protection_enabled.cognito_user_pool_deletion_protection_enabled import ( cognito_user_pool_deletion_protection_enabled, @@ -45,15 +42,12 @@ class Test_cognito_user_pool_deletion_protection_enabled: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_deletion_protection_enabled.cognito_user_pool_deletion_protection_enabled import ( cognito_user_pool_deletion_protection_enabled, @@ -87,15 +81,12 @@ class Test_cognito_user_pool_deletion_protection_enabled: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_deletion_protection_enabled.cognito_user_pool_deletion_protection_enabled import ( cognito_user_pool_deletion_protection_enabled, diff --git a/tests/providers/aws/services/cognito/cognito_user_pool_mfa_enabled/cognito_user_pool_mfa_enabled_test.py b/tests/providers/aws/services/cognito/cognito_user_pool_mfa_enabled/cognito_user_pool_mfa_enabled_test.py index c5793d6d1f..70c3225144 100644 --- a/tests/providers/aws/services/cognito/cognito_user_pool_mfa_enabled/cognito_user_pool_mfa_enabled_test.py +++ b/tests/providers/aws/services/cognito/cognito_user_pool_mfa_enabled/cognito_user_pool_mfa_enabled_test.py @@ -9,15 +9,12 @@ class Test_cognito_user_pool_mfa_enabled: def test_cognito_no_user_pools(self): cognito_client = mock.MagicMock cognito_client.user_pools = {} - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_mfa_enabled.cognito_user_pool_mfa_enabled import ( cognito_user_pool_mfa_enabled, @@ -45,15 +42,12 @@ class Test_cognito_user_pool_mfa_enabled: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_mfa_enabled.cognito_user_pool_mfa_enabled import ( cognito_user_pool_mfa_enabled, @@ -88,15 +82,12 @@ class Test_cognito_user_pool_mfa_enabled: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_mfa_enabled.cognito_user_pool_mfa_enabled import ( cognito_user_pool_mfa_enabled, @@ -131,15 +122,12 @@ class Test_cognito_user_pool_mfa_enabled: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_mfa_enabled.cognito_user_pool_mfa_enabled import ( cognito_user_pool_mfa_enabled, diff --git a/tests/providers/aws/services/cognito/cognito_user_pool_password_policy_lowercase/cognito_user_pool_password_policy_lowercase_test.py b/tests/providers/aws/services/cognito/cognito_user_pool_password_policy_lowercase/cognito_user_pool_password_policy_lowercase_test.py index be36bc3fcc..eacfbfb11a 100644 --- a/tests/providers/aws/services/cognito/cognito_user_pool_password_policy_lowercase/cognito_user_pool_password_policy_lowercase_test.py +++ b/tests/providers/aws/services/cognito/cognito_user_pool_password_policy_lowercase/cognito_user_pool_password_policy_lowercase_test.py @@ -12,15 +12,12 @@ class Test_cognito_user_pool_password_policy_lowercase: def test_cognito_no_user_pools(self): cognito_client = mock.MagicMock cognito_client.user_pools = {} - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_password_policy_lowercase.cognito_user_pool_password_policy_lowercase import ( cognito_user_pool_password_policy_lowercase, @@ -50,15 +47,12 @@ class Test_cognito_user_pool_password_policy_lowercase: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_password_policy_lowercase.cognito_user_pool_password_policy_lowercase import ( cognito_user_pool_password_policy_lowercase, @@ -95,15 +89,12 @@ class Test_cognito_user_pool_password_policy_lowercase: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_password_policy_lowercase.cognito_user_pool_password_policy_lowercase import ( cognito_user_pool_password_policy_lowercase, @@ -138,15 +129,12 @@ class Test_cognito_user_pool_password_policy_lowercase: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_password_policy_lowercase.cognito_user_pool_password_policy_lowercase import ( cognito_user_pool_password_policy_lowercase, diff --git a/tests/providers/aws/services/cognito/cognito_user_pool_password_policy_minimum_length_14/cognito_user_pool_password_policy_minimum_length_14_test.py b/tests/providers/aws/services/cognito/cognito_user_pool_password_policy_minimum_length_14/cognito_user_pool_password_policy_minimum_length_14_test.py index 4a1b3e8a3c..e7a0204f95 100644 --- a/tests/providers/aws/services/cognito/cognito_user_pool_password_policy_minimum_length_14/cognito_user_pool_password_policy_minimum_length_14_test.py +++ b/tests/providers/aws/services/cognito/cognito_user_pool_password_policy_minimum_length_14/cognito_user_pool_password_policy_minimum_length_14_test.py @@ -12,15 +12,12 @@ class Test_cognito_user_pool_password_policy_minimum_length_14: def test_cognito_no_user_pools(self): cognito_client = mock.MagicMock cognito_client.user_pools = {} - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_password_policy_minimum_length_14.cognito_user_pool_password_policy_minimum_length_14 import ( cognito_user_pool_password_policy_minimum_length_14, @@ -50,15 +47,12 @@ class Test_cognito_user_pool_password_policy_minimum_length_14: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_password_policy_minimum_length_14.cognito_user_pool_password_policy_minimum_length_14 import ( cognito_user_pool_password_policy_minimum_length_14, @@ -95,15 +89,12 @@ class Test_cognito_user_pool_password_policy_minimum_length_14: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_password_policy_minimum_length_14.cognito_user_pool_password_policy_minimum_length_14 import ( cognito_user_pool_password_policy_minimum_length_14, @@ -138,15 +129,12 @@ class Test_cognito_user_pool_password_policy_minimum_length_14: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_password_policy_minimum_length_14.cognito_user_pool_password_policy_minimum_length_14 import ( cognito_user_pool_password_policy_minimum_length_14, diff --git a/tests/providers/aws/services/cognito/cognito_user_pool_password_policy_number/cognito_user_pool_password_policy_number_test.py b/tests/providers/aws/services/cognito/cognito_user_pool_password_policy_number/cognito_user_pool_password_policy_number_test.py index e018764d1f..f674d9d25b 100644 --- a/tests/providers/aws/services/cognito/cognito_user_pool_password_policy_number/cognito_user_pool_password_policy_number_test.py +++ b/tests/providers/aws/services/cognito/cognito_user_pool_password_policy_number/cognito_user_pool_password_policy_number_test.py @@ -12,15 +12,12 @@ class Test_cognito_user_pool_password_policy_number: def test_cognito_no_user_pools(self): cognito_client = mock.MagicMock cognito_client.user_pools = {} - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_password_policy_number.cognito_user_pool_password_policy_number import ( cognito_user_pool_password_policy_number, @@ -50,15 +47,12 @@ class Test_cognito_user_pool_password_policy_number: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_password_policy_number.cognito_user_pool_password_policy_number import ( cognito_user_pool_password_policy_number, @@ -95,15 +89,12 @@ class Test_cognito_user_pool_password_policy_number: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_password_policy_number.cognito_user_pool_password_policy_number import ( cognito_user_pool_password_policy_number, @@ -138,15 +129,12 @@ class Test_cognito_user_pool_password_policy_number: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_password_policy_number.cognito_user_pool_password_policy_number import ( cognito_user_pool_password_policy_number, diff --git a/tests/providers/aws/services/cognito/cognito_user_pool_password_policy_symbol/cognito_user_pool_password_policy_symbol_test.py b/tests/providers/aws/services/cognito/cognito_user_pool_password_policy_symbol/cognito_user_pool_password_policy_symbol_test.py index bc775fcfa1..69bcbb055c 100644 --- a/tests/providers/aws/services/cognito/cognito_user_pool_password_policy_symbol/cognito_user_pool_password_policy_symbol_test.py +++ b/tests/providers/aws/services/cognito/cognito_user_pool_password_policy_symbol/cognito_user_pool_password_policy_symbol_test.py @@ -12,15 +12,12 @@ class Test_cognito_user_pool_password_policy_symbol: def test_cognito_no_user_pools(self): cognito_client = mock.MagicMock cognito_client.user_pools = {} - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_password_policy_symbol.cognito_user_pool_password_policy_symbol import ( cognito_user_pool_password_policy_symbol, @@ -50,15 +47,12 @@ class Test_cognito_user_pool_password_policy_symbol: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_password_policy_symbol.cognito_user_pool_password_policy_symbol import ( cognito_user_pool_password_policy_symbol, @@ -95,15 +89,12 @@ class Test_cognito_user_pool_password_policy_symbol: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_password_policy_symbol.cognito_user_pool_password_policy_symbol import ( cognito_user_pool_password_policy_symbol, @@ -138,15 +129,12 @@ class Test_cognito_user_pool_password_policy_symbol: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_password_policy_symbol.cognito_user_pool_password_policy_symbol import ( cognito_user_pool_password_policy_symbol, diff --git a/tests/providers/aws/services/cognito/cognito_user_pool_password_policy_uppercase/cognito_user_pool_password_policy_uppercase_test.py b/tests/providers/aws/services/cognito/cognito_user_pool_password_policy_uppercase/cognito_user_pool_password_policy_uppercase_test.py index e458882aa4..c3ac13cfde 100644 --- a/tests/providers/aws/services/cognito/cognito_user_pool_password_policy_uppercase/cognito_user_pool_password_policy_uppercase_test.py +++ b/tests/providers/aws/services/cognito/cognito_user_pool_password_policy_uppercase/cognito_user_pool_password_policy_uppercase_test.py @@ -12,15 +12,12 @@ class Test_cognito_user_pool_password_policy_uppercase: def test_cognito_no_user_pools(self): cognito_client = mock.MagicMock cognito_client.user_pools = {} - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_password_policy_uppercase.cognito_user_pool_password_policy_uppercase import ( cognito_user_pool_password_policy_uppercase, @@ -50,15 +47,12 @@ class Test_cognito_user_pool_password_policy_uppercase: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_password_policy_uppercase.cognito_user_pool_password_policy_uppercase import ( cognito_user_pool_password_policy_uppercase, @@ -95,15 +89,12 @@ class Test_cognito_user_pool_password_policy_uppercase: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_password_policy_uppercase.cognito_user_pool_password_policy_uppercase import ( cognito_user_pool_password_policy_uppercase, @@ -138,15 +129,12 @@ class Test_cognito_user_pool_password_policy_uppercase: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_password_policy_uppercase.cognito_user_pool_password_policy_uppercase import ( cognito_user_pool_password_policy_uppercase, diff --git a/tests/providers/aws/services/cognito/cognito_user_pool_self_registration_disabled/cognito_user_pool_self_registration_disabled_test.py b/tests/providers/aws/services/cognito/cognito_user_pool_self_registration_disabled/cognito_user_pool_self_registration_disabled_test.py index 55438c9ff6..6b81a7cc50 100644 --- a/tests/providers/aws/services/cognito/cognito_user_pool_self_registration_disabled/cognito_user_pool_self_registration_disabled_test.py +++ b/tests/providers/aws/services/cognito/cognito_user_pool_self_registration_disabled/cognito_user_pool_self_registration_disabled_test.py @@ -16,23 +16,18 @@ class Test_cognito_user_pool_self_registration_disabled: cognito_client.user_pools = {} cognito_identity_client = mock.MagicMock cognito_identity_client.identity_pools = {} - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIdentity", - cognito_identity_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_identity_client.cognito_identity_client", - cognito_identity_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIdentity", + cognito_identity_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_identity_client.cognito_identity_client", + cognito_identity_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_self_registration_disabled.cognito_user_pool_self_registration_disabled import ( cognito_user_pool_self_registration_disabled, @@ -64,23 +59,18 @@ class Test_cognito_user_pool_self_registration_disabled: } cognito_identity_client = mock.MagicMock cognito_identity_client.identity_pools = {} - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIdentity", - cognito_identity_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_identity_client.cognito_identity_client", - cognito_identity_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIdentity", + cognito_identity_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_identity_client.cognito_identity_client", + cognito_identity_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_self_registration_disabled.cognito_user_pool_self_registration_disabled import ( cognito_user_pool_self_registration_disabled, @@ -138,23 +128,18 @@ class Test_cognito_user_pool_self_registration_disabled: ), ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIdentity", - cognito_identity_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_identity_client.cognito_identity_client", - cognito_identity_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIdentity", + cognito_identity_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_identity_client.cognito_identity_client", + cognito_identity_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_self_registration_disabled.cognito_user_pool_self_registration_disabled import ( cognito_user_pool_self_registration_disabled, @@ -212,23 +197,18 @@ class Test_cognito_user_pool_self_registration_disabled: ), ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIdentity", - cognito_identity_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_identity_client.cognito_identity_client", - cognito_identity_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIdentity", + cognito_identity_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_identity_client.cognito_identity_client", + cognito_identity_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_self_registration_disabled.cognito_user_pool_self_registration_disabled import ( cognito_user_pool_self_registration_disabled, diff --git a/tests/providers/aws/services/cognito/cognito_user_pool_temporary_password_expiration/cognito_user_pool_temporary_password_expiration_test.py b/tests/providers/aws/services/cognito/cognito_user_pool_temporary_password_expiration/cognito_user_pool_temporary_password_expiration_test.py index 81b521735b..c6c2bf2661 100644 --- a/tests/providers/aws/services/cognito/cognito_user_pool_temporary_password_expiration/cognito_user_pool_temporary_password_expiration_test.py +++ b/tests/providers/aws/services/cognito/cognito_user_pool_temporary_password_expiration/cognito_user_pool_temporary_password_expiration_test.py @@ -12,15 +12,12 @@ class Test_cognito_user_pool_temporary_password_expiration: def test_cognito_no_user_pools(self): cognito_client = mock.MagicMock cognito_client.user_pools = {} - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_temporary_password_expiration.cognito_user_pool_temporary_password_expiration import ( cognito_user_pool_temporary_password_expiration, @@ -50,15 +47,12 @@ class Test_cognito_user_pool_temporary_password_expiration: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_temporary_password_expiration.cognito_user_pool_temporary_password_expiration import ( cognito_user_pool_temporary_password_expiration, @@ -95,15 +89,12 @@ class Test_cognito_user_pool_temporary_password_expiration: status="ACTIVE", ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_temporary_password_expiration.cognito_user_pool_temporary_password_expiration import ( cognito_user_pool_temporary_password_expiration, diff --git a/tests/providers/aws/services/cognito/cognito_user_pool_waf_acl_attached/cognito_user_pool_waf_acl_attached_test.py b/tests/providers/aws/services/cognito/cognito_user_pool_waf_acl_attached/cognito_user_pool_waf_acl_attached_test.py index 283219e661..5389fc05a9 100644 --- a/tests/providers/aws/services/cognito/cognito_user_pool_waf_acl_attached/cognito_user_pool_waf_acl_attached_test.py +++ b/tests/providers/aws/services/cognito/cognito_user_pool_waf_acl_attached/cognito_user_pool_waf_acl_attached_test.py @@ -13,23 +13,18 @@ class Test_cognito_user_pool_waf_acl_attached: cognito_client.audited_account = AWS_ACCOUNT_NUMBER wafv2_client = mock.MagicMock wafv2_client.web_acls = {} - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.wafv2.wafv2_service.WAFv2", - new=wafv2_client, - ), - mock.patch( - "prowler.providers.aws.services.wafv2.wafv2_client.wafv2_client", - new=wafv2_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.wafv2.wafv2_service.WAFv2", + new=wafv2_client, + ), mock.patch( + "prowler.providers.aws.services.wafv2.wafv2_client.wafv2_client", + new=wafv2_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_waf_acl_attached.cognito_user_pool_waf_acl_attached import ( cognito_user_pool_waf_acl_attached, @@ -60,23 +55,18 @@ class Test_cognito_user_pool_waf_acl_attached: wafv2_client = mock.MagicMock wafv2_client.web_acls = {} - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.wafv2.wafv2_service.WAFv2", - new=wafv2_client, - ), - mock.patch( - "prowler.providers.aws.services.wafv2.wafv2_client.wafv2_client", - new=wafv2_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.wafv2.wafv2_service.WAFv2", + new=wafv2_client, + ), mock.patch( + "prowler.providers.aws.services.wafv2.wafv2_client.wafv2_client", + new=wafv2_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_waf_acl_attached.cognito_user_pool_waf_acl_attached import ( cognito_user_pool_waf_acl_attached, @@ -126,23 +116,18 @@ class Test_cognito_user_pool_waf_acl_attached: ) } - with ( - mock.patch( - "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", - new=cognito_client, - ), - mock.patch( - "prowler.providers.aws.services.wafv2.wafv2_service.WAFv2", - new=wafv2_client, - ), - mock.patch( - "prowler.providers.aws.services.wafv2.wafv2_client.wafv2_client", - new=wafv2_client, - ), + with mock.patch( + "prowler.providers.aws.services.cognito.cognito_service.CognitoIDP", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.cognito.cognito_idp_client.cognito_idp_client", + new=cognito_client, + ), mock.patch( + "prowler.providers.aws.services.wafv2.wafv2_service.WAFv2", + new=wafv2_client, + ), mock.patch( + "prowler.providers.aws.services.wafv2.wafv2_client.wafv2_client", + new=wafv2_client, ): from prowler.providers.aws.services.cognito.cognito_user_pool_waf_acl_attached.cognito_user_pool_waf_acl_attached import ( cognito_user_pool_waf_acl_attached, diff --git a/tests/providers/aws/services/config/config_recorder_all_regions_enabled/config_recorder_all_regions_enabled_test.py b/tests/providers/aws/services/config/config_recorder_all_regions_enabled/config_recorder_all_regions_enabled_test.py index 6388c630f1..12a7c3e6e5 100644 --- a/tests/providers/aws/services/config/config_recorder_all_regions_enabled/config_recorder_all_regions_enabled_test.py +++ b/tests/providers/aws/services/config/config_recorder_all_regions_enabled/config_recorder_all_regions_enabled_test.py @@ -21,15 +21,12 @@ class Test_config_recorder_all_regions_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.config.config_recorder_all_regions_enabled.config_recorder_all_regions_enabled.config_client", - new=Config(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.config.config_recorder_all_regions_enabled.config_recorder_all_regions_enabled.config_client", + new=Config(aws_provider), ): # Test Check from prowler.providers.aws.services.config.config_recorder_all_regions_enabled.config_recorder_all_regions_enabled import ( @@ -77,15 +74,12 @@ class Test_config_recorder_all_regions_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.config.config_recorder_all_regions_enabled.config_recorder_all_regions_enabled.config_client", - new=Config(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.config.config_recorder_all_regions_enabled.config_recorder_all_regions_enabled.config_client", + new=Config(aws_provider), ): # Test Check from prowler.providers.aws.services.config.config_recorder_all_regions_enabled.config_recorder_all_regions_enabled import ( @@ -127,15 +121,12 @@ class Test_config_recorder_all_regions_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.config.config_recorder_all_regions_enabled.config_recorder_all_regions_enabled.config_client", - new=Config(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.config.config_recorder_all_regions_enabled.config_recorder_all_regions_enabled.config_client", + new=Config(aws_provider), ): # Test Check from prowler.providers.aws.services.config.config_recorder_all_regions_enabled.config_recorder_all_regions_enabled import ( @@ -176,15 +167,12 @@ class Test_config_recorder_all_regions_enabled: audit_config={"mute_non_default_regions": True}, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.config.config_recorder_all_regions_enabled.config_recorder_all_regions_enabled.config_client", - new=Config(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.config.config_recorder_all_regions_enabled.config_recorder_all_regions_enabled.config_client", + new=Config(aws_provider), ): # Test Check from prowler.providers.aws.services.config.config_recorder_all_regions_enabled.config_recorder_all_regions_enabled import ( diff --git a/tests/providers/aws/services/config/config_recorder_using_aws_service_role/config_recorder_using_aws_service_role_test.py b/tests/providers/aws/services/config/config_recorder_using_aws_service_role/config_recorder_using_aws_service_role_test.py index 3f05a0e343..e7b95b2ce6 100644 --- a/tests/providers/aws/services/config/config_recorder_using_aws_service_role/config_recorder_using_aws_service_role_test.py +++ b/tests/providers/aws/services/config/config_recorder_using_aws_service_role/config_recorder_using_aws_service_role_test.py @@ -20,15 +20,12 @@ class Test_config_recorder_using_aws_service_role: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.config.config_recorder_using_aws_service_role.config_recorder_using_aws_service_role.config_client", - new=Config(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.config.config_recorder_using_aws_service_role.config_recorder_using_aws_service_role.config_client", + new=Config(aws_provider), ): # Test Check from prowler.providers.aws.services.config.config_recorder_using_aws_service_role.config_recorder_using_aws_service_role import ( @@ -52,15 +49,12 @@ class Test_config_recorder_using_aws_service_role: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.config.config_recorder_using_aws_service_role.config_recorder_using_aws_service_role.config_client", - new=Config(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.config.config_recorder_using_aws_service_role.config_recorder_using_aws_service_role.config_client", + new=Config(aws_provider), ): # Test Check from prowler.providers.aws.services.config.config_recorder_using_aws_service_role.config_recorder_using_aws_service_role import ( @@ -91,15 +85,12 @@ class Test_config_recorder_using_aws_service_role: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.config.config_recorder_using_aws_service_role.config_recorder_using_aws_service_role.config_client", - new=Config(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.config.config_recorder_using_aws_service_role.config_recorder_using_aws_service_role.config_client", + new=Config(aws_provider), ): # Test Check from prowler.providers.aws.services.config.config_recorder_using_aws_service_role.config_recorder_using_aws_service_role import ( @@ -143,15 +134,12 @@ class Test_config_recorder_using_aws_service_role: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.config.config_recorder_using_aws_service_role.config_recorder_using_aws_service_role.config_client", - new=Config(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.config.config_recorder_using_aws_service_role.config_recorder_using_aws_service_role.config_client", + new=Config(aws_provider), ): # Test Check from prowler.providers.aws.services.config.config_recorder_using_aws_service_role.config_recorder_using_aws_service_role import ( diff --git a/tests/providers/aws/services/datasync/datasync_task_logging_enabled/datasync_task_logging_enabled_test.py b/tests/providers/aws/services/datasync/datasync_task_logging_enabled/datasync_task_logging_enabled_test.py index 5c2af70239..fd79aad29e 100644 --- a/tests/providers/aws/services/datasync/datasync_task_logging_enabled/datasync_task_logging_enabled_test.py +++ b/tests/providers/aws/services/datasync/datasync_task_logging_enabled/datasync_task_logging_enabled_test.py @@ -17,15 +17,12 @@ class Test_datasync_task_logging_enabled: datasync_client = DataSync(mocked_aws_provider) datasync_client.tasks = {} - with ( - patch( - "prowler.providers.aws.services.datasync.datasync_task_logging_enabled.datasync_task_logging_enabled.datasync_client", - new=datasync_client, - ), - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), + with patch( + "prowler.providers.aws.services.datasync.datasync_task_logging_enabled.datasync_task_logging_enabled.datasync_client", + new=datasync_client, + ), patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, ): from prowler.providers.aws.services.datasync.datasync_task_logging_enabled.datasync_task_logging_enabled import ( datasync_task_logging_enabled, @@ -58,15 +55,12 @@ class Test_datasync_task_logging_enabled: datasync_client = DataSync(mocked_aws_provider) datasync_client.tasks[TASK_ARN] = task - with ( - patch( - "prowler.providers.aws.services.datasync.datasync_task_logging_enabled.datasync_task_logging_enabled.datasync_client", - new=datasync_client, - ), - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), + with patch( + "prowler.providers.aws.services.datasync.datasync_task_logging_enabled.datasync_task_logging_enabled.datasync_client", + new=datasync_client, + ), patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, ): from prowler.providers.aws.services.datasync.datasync_task_logging_enabled.datasync_task_logging_enabled import ( datasync_task_logging_enabled, @@ -108,15 +102,12 @@ class Test_datasync_task_logging_enabled: datasync_client = DataSync(mocked_aws_provider) datasync_client.tasks[TASK_ARN] = task - with ( - patch( - "prowler.providers.aws.services.datasync.datasync_task_logging_enabled.datasync_task_logging_enabled.datasync_client", - new=datasync_client, - ), - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), + with patch( + "prowler.providers.aws.services.datasync.datasync_task_logging_enabled.datasync_task_logging_enabled.datasync_client", + new=datasync_client, + ), patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, ): from prowler.providers.aws.services.datasync.datasync_task_logging_enabled.datasync_task_logging_enabled import ( datasync_task_logging_enabled, diff --git a/tests/providers/aws/services/directconnect/directconnect_connection_redundancy/directconnect_connection_redundancy_test.py b/tests/providers/aws/services/directconnect/directconnect_connection_redundancy/directconnect_connection_redundancy_test.py index 99ad6718d2..faf43ba088 100644 --- a/tests/providers/aws/services/directconnect/directconnect_connection_redundancy/directconnect_connection_redundancy_test.py +++ b/tests/providers/aws/services/directconnect/directconnect_connection_redundancy/directconnect_connection_redundancy_test.py @@ -43,15 +43,12 @@ class Test_directconnect_connection_redundancy: region="eu-west-1", ) } - with ( - mock.patch( - "prowler.providers.aws.services.directconnect.directconnect_service.DirectConnect", - new=dx_client, - ), - mock.patch( - "prowler.providers.aws.services.directconnect.directconnect_service.DirectConnect._get_connection_arn_template", - return_value=f"arn:aws:directconnect:{AWS_REGION_EU_WEST_1}:{AWS_ACCOUNT_NUMBER}:connection", - ), + with mock.patch( + "prowler.providers.aws.services.directconnect.directconnect_service.DirectConnect", + new=dx_client, + ), mock.patch( + "prowler.providers.aws.services.directconnect.directconnect_service.DirectConnect._get_connection_arn_template", + return_value=f"arn:aws:directconnect:{AWS_REGION_EU_WEST_1}:{AWS_ACCOUNT_NUMBER}:connection", ): # Test Check from prowler.providers.aws.services.directconnect.directconnect_connection_redundancy.directconnect_connection_redundancy import ( @@ -99,15 +96,12 @@ class Test_directconnect_connection_redundancy: region="eu-west-1", ), } - with ( - mock.patch( - "prowler.providers.aws.services.directconnect.directconnect_service.DirectConnect", - new=dx_client, - ), - mock.patch( - "prowler.providers.aws.services.directconnect.directconnect_service.DirectConnect._get_connection_arn_template", - return_value=f"arn:aws:directconnect:{AWS_REGION_EU_WEST_1}:{AWS_ACCOUNT_NUMBER}:connection", - ), + with mock.patch( + "prowler.providers.aws.services.directconnect.directconnect_service.DirectConnect", + new=dx_client, + ), mock.patch( + "prowler.providers.aws.services.directconnect.directconnect_service.DirectConnect._get_connection_arn_template", + return_value=f"arn:aws:directconnect:{AWS_REGION_EU_WEST_1}:{AWS_ACCOUNT_NUMBER}:connection", ): # Test Check from prowler.providers.aws.services.directconnect.directconnect_connection_redundancy.directconnect_connection_redundancy import ( @@ -155,15 +149,12 @@ class Test_directconnect_connection_redundancy: region="eu-west-1", ), } - with ( - mock.patch( - "prowler.providers.aws.services.directconnect.directconnect_service.DirectConnect", - new=dx_client, - ), - mock.patch( - "prowler.providers.aws.services.directconnect.directconnect_service.DirectConnect._get_connection_arn_template", - return_value=f"arn:aws:directconnect:{AWS_REGION_EU_WEST_1}:{AWS_ACCOUNT_NUMBER}:connection", - ), + with mock.patch( + "prowler.providers.aws.services.directconnect.directconnect_service.DirectConnect", + new=dx_client, + ), mock.patch( + "prowler.providers.aws.services.directconnect.directconnect_service.DirectConnect._get_connection_arn_template", + return_value=f"arn:aws:directconnect:{AWS_REGION_EU_WEST_1}:{AWS_ACCOUNT_NUMBER}:connection", ): # Test Check from prowler.providers.aws.services.directconnect.directconnect_connection_redundancy.directconnect_connection_redundancy import ( diff --git a/tests/providers/aws/services/dlm/dlm_ebs_snapshot_lifecycle_policy_exists/dlm_ebs_snapshot_lifecycle_policy_exists_test.py b/tests/providers/aws/services/dlm/dlm_ebs_snapshot_lifecycle_policy_exists/dlm_ebs_snapshot_lifecycle_policy_exists_test.py index 3412aead68..a39e51b80e 100644 --- a/tests/providers/aws/services/dlm/dlm_ebs_snapshot_lifecycle_policy_exists/dlm_ebs_snapshot_lifecycle_policy_exists_test.py +++ b/tests/providers/aws/services/dlm/dlm_ebs_snapshot_lifecycle_policy_exists/dlm_ebs_snapshot_lifecycle_policy_exists_test.py @@ -27,23 +27,18 @@ class Test_dlm_ebs_snapshot_lifecycle_policy_exists: from prowler.providers.aws.services.ec2.ec2_service import EC2 - with ( - mock.patch( - "prowler.providers.aws.services.dlm.dlm_service.DLM", - new=dlm_client, - ), - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_service.EC2", - return_value=EC2(aws_provider), - ) as ec2_client, - mock.patch( - "prowler.providers.aws.services.ec2.ec2_client.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.aws.services.dlm.dlm_service.DLM", + new=dlm_client, + ), mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_service.EC2", + return_value=EC2(aws_provider), + ) as ec2_client, mock.patch( + "prowler.providers.aws.services.ec2.ec2_client.ec2_client", + new=ec2_client, ): from prowler.providers.aws.services.dlm.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_ebs_snapshot_lifecycle_policy_exists import ( dlm_ebs_snapshot_lifecycle_policy_exists, @@ -100,19 +95,15 @@ class Test_dlm_ebs_snapshot_lifecycle_policy_exists: from prowler.providers.aws.services.ec2.ec2_service import EC2 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dlm.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_ebs_snapshot_lifecycle_policy_exists.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.dlm.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_client", - new=dlm_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dlm.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_ebs_snapshot_lifecycle_policy_exists.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.dlm.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_client", + new=dlm_client, ): from prowler.providers.aws.services.dlm.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_ebs_snapshot_lifecycle_policy_exists import ( dlm_ebs_snapshot_lifecycle_policy_exists, @@ -165,19 +156,15 @@ class Test_dlm_ebs_snapshot_lifecycle_policy_exists: from prowler.providers.aws.services.ec2.ec2_service import EC2 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dlm.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_ebs_snapshot_lifecycle_policy_exists.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.dlm.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_client", - new=dlm_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dlm.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_ebs_snapshot_lifecycle_policy_exists.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.dlm.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_client", + new=dlm_client, ): from prowler.providers.aws.services.dlm.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_ebs_snapshot_lifecycle_policy_exists import ( dlm_ebs_snapshot_lifecycle_policy_exists, @@ -210,19 +197,15 @@ class Test_dlm_ebs_snapshot_lifecycle_policy_exists: from prowler.providers.aws.services.ec2.ec2_service import EC2 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dlm.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_ebs_snapshot_lifecycle_policy_exists.ec2_client", - new=EC2(aws_provider), - ) as ec2_client, - mock.patch( - "prowler.providers.aws.services.dlm.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_client", - new=dlm_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dlm.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_ebs_snapshot_lifecycle_policy_exists.ec2_client", + new=EC2(aws_provider), + ) as ec2_client, mock.patch( + "prowler.providers.aws.services.dlm.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_ebs_snapshot_lifecycle_policy_exists.dlm_client", + new=dlm_client, ): # Remove all snapshots ec2_client.regions_with_snapshots = {} diff --git a/tests/providers/aws/services/dms/dms_endpoint_mongodb_authentication_enabled/dms_endpoint_mongodb_authentication_enabled_test.py b/tests/providers/aws/services/dms/dms_endpoint_mongodb_authentication_enabled/dms_endpoint_mongodb_authentication_enabled_test.py index cbd104ba64..f7a2c8dea6 100644 --- a/tests/providers/aws/services/dms/dms_endpoint_mongodb_authentication_enabled/dms_endpoint_mongodb_authentication_enabled_test.py +++ b/tests/providers/aws/services/dms/dms_endpoint_mongodb_authentication_enabled/dms_endpoint_mongodb_authentication_enabled_test.py @@ -129,15 +129,12 @@ class Test_dms_endpoint_mongodb_authentication_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_endpoint_mongodb_authentication_enabled.dms_endpoint_mongodb_authentication_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_endpoint_mongodb_authentication_enabled.dms_endpoint_mongodb_authentication_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_endpoint_mongodb_authentication_enabled.dms_endpoint_mongodb_authentication_enabled import ( @@ -160,15 +157,12 @@ class Test_dms_endpoint_mongodb_authentication_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_endpoint_mongodb_authentication_enabled.dms_endpoint_mongodb_authentication_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_endpoint_mongodb_authentication_enabled.dms_endpoint_mongodb_authentication_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_endpoint_mongodb_authentication_enabled.dms_endpoint_mongodb_authentication_enabled import ( @@ -191,15 +185,12 @@ class Test_dms_endpoint_mongodb_authentication_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_endpoint_mongodb_authentication_enabled.dms_endpoint_mongodb_authentication_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_endpoint_mongodb_authentication_enabled.dms_endpoint_mongodb_authentication_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_endpoint_mongodb_authentication_enabled.dms_endpoint_mongodb_authentication_enabled import ( @@ -242,15 +233,12 @@ class Test_dms_endpoint_mongodb_authentication_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_endpoint_mongodb_authentication_enabled.dms_endpoint_mongodb_authentication_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_endpoint_mongodb_authentication_enabled.dms_endpoint_mongodb_authentication_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_endpoint_mongodb_authentication_enabled.dms_endpoint_mongodb_authentication_enabled import ( diff --git a/tests/providers/aws/services/dms/dms_endpoint_neptune_iam_authorization_enabled/dms_endpoint_neptune_iam_authorization_enabled_test.py b/tests/providers/aws/services/dms/dms_endpoint_neptune_iam_authorization_enabled/dms_endpoint_neptune_iam_authorization_enabled_test.py index d0aecc30a0..ee9316b0dd 100644 --- a/tests/providers/aws/services/dms/dms_endpoint_neptune_iam_authorization_enabled/dms_endpoint_neptune_iam_authorization_enabled_test.py +++ b/tests/providers/aws/services/dms/dms_endpoint_neptune_iam_authorization_enabled/dms_endpoint_neptune_iam_authorization_enabled_test.py @@ -129,15 +129,12 @@ class Test_dms_endpoint_neptune_iam_authorization_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_endpoint_neptune_iam_authorization_enabled.dms_endpoint_neptune_iam_authorization_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_endpoint_neptune_iam_authorization_enabled.dms_endpoint_neptune_iam_authorization_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_endpoint_neptune_iam_authorization_enabled.dms_endpoint_neptune_iam_authorization_enabled import ( @@ -160,15 +157,12 @@ class Test_dms_endpoint_neptune_iam_authorization_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_endpoint_neptune_iam_authorization_enabled.dms_endpoint_neptune_iam_authorization_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_endpoint_neptune_iam_authorization_enabled.dms_endpoint_neptune_iam_authorization_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_endpoint_neptune_iam_authorization_enabled.dms_endpoint_neptune_iam_authorization_enabled import ( @@ -191,15 +185,12 @@ class Test_dms_endpoint_neptune_iam_authorization_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_endpoint_neptune_iam_authorization_enabled.dms_endpoint_neptune_iam_authorization_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_endpoint_neptune_iam_authorization_enabled.dms_endpoint_neptune_iam_authorization_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_endpoint_neptune_iam_authorization_enabled.dms_endpoint_neptune_iam_authorization_enabled import ( @@ -242,15 +233,12 @@ class Test_dms_endpoint_neptune_iam_authorization_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_endpoint_neptune_iam_authorization_enabled.dms_endpoint_neptune_iam_authorization_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_endpoint_neptune_iam_authorization_enabled.dms_endpoint_neptune_iam_authorization_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_endpoint_neptune_iam_authorization_enabled.dms_endpoint_neptune_iam_authorization_enabled import ( diff --git a/tests/providers/aws/services/dms/dms_endpoint_redis_in_transit_encryption_enabled/dms_endpoint_redis_in_transit_encryption_enabled_test.py b/tests/providers/aws/services/dms/dms_endpoint_redis_in_transit_encryption_enabled/dms_endpoint_redis_in_transit_encryption_enabled_test.py index a478deda8c..639b5c5bcb 100644 --- a/tests/providers/aws/services/dms/dms_endpoint_redis_in_transit_encryption_enabled/dms_endpoint_redis_in_transit_encryption_enabled_test.py +++ b/tests/providers/aws/services/dms/dms_endpoint_redis_in_transit_encryption_enabled/dms_endpoint_redis_in_transit_encryption_enabled_test.py @@ -129,15 +129,12 @@ class Test_dms_endpoint_redis_in_transit_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_endpoint_redis_in_transit_encryption_enabled.dms_endpoint_redis_in_transit_encryption_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_endpoint_redis_in_transit_encryption_enabled.dms_endpoint_redis_in_transit_encryption_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_endpoint_redis_in_transit_encryption_enabled.dms_endpoint_redis_in_transit_encryption_enabled import ( @@ -160,15 +157,12 @@ class Test_dms_endpoint_redis_in_transit_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_endpoint_redis_in_transit_encryption_enabled.dms_endpoint_redis_in_transit_encryption_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_endpoint_redis_in_transit_encryption_enabled.dms_endpoint_redis_in_transit_encryption_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_endpoint_redis_in_transit_encryption_enabled.dms_endpoint_redis_in_transit_encryption_enabled import ( @@ -191,15 +185,12 @@ class Test_dms_endpoint_redis_in_transit_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_endpoint_redis_in_transit_encryption_enabled.dms_endpoint_redis_in_transit_encryption_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_endpoint_redis_in_transit_encryption_enabled.dms_endpoint_redis_in_transit_encryption_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_endpoint_redis_in_transit_encryption_enabled.dms_endpoint_redis_in_transit_encryption_enabled import ( @@ -242,15 +233,12 @@ class Test_dms_endpoint_redis_in_transit_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_endpoint_redis_in_transit_encryption_enabled.dms_endpoint_redis_in_transit_encryption_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_endpoint_redis_in_transit_encryption_enabled.dms_endpoint_redis_in_transit_encryption_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_endpoint_redis_in_transit_encryption_enabled.dms_endpoint_redis_in_transit_encryption_enabled import ( diff --git a/tests/providers/aws/services/dms/dms_endpoint_ssl_enabled_test/dms_endpoint_ssl_enabled_test.py b/tests/providers/aws/services/dms/dms_endpoint_ssl_enabled_test/dms_endpoint_ssl_enabled_test.py index 811c7f5aff..7dc11c9a25 100644 --- a/tests/providers/aws/services/dms/dms_endpoint_ssl_enabled_test/dms_endpoint_ssl_enabled_test.py +++ b/tests/providers/aws/services/dms/dms_endpoint_ssl_enabled_test/dms_endpoint_ssl_enabled_test.py @@ -9,15 +9,12 @@ class Test_dms_endpoint_ssl_enabled: dms_client = mock.MagicMock dms_client.endpoints = {} - with ( - mock.patch( - "prowler.providers.aws.services.dms.dms_service.DMS", - new=dms_client, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_client.dms_client", - new=dms_client, - ), + with mock.patch( + "prowler.providers.aws.services.dms.dms_service.DMS", + new=dms_client, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_client.dms_client", + new=dms_client, ): from prowler.providers.aws.services.dms.dms_endpoint_ssl_enabled.dms_endpoint_ssl_enabled import ( dms_endpoint_ssl_enabled, @@ -46,15 +43,12 @@ class Test_dms_endpoint_ssl_enabled: dms_client.audited_partition = "aws" dms_client.audited_region = AWS_REGION_US_EAST_1 - with ( - mock.patch( - "prowler.providers.aws.services.dms.dms_service.DMS", - new=dms_client, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_client.dms_client", - new=dms_client, - ), + with mock.patch( + "prowler.providers.aws.services.dms.dms_service.DMS", + new=dms_client, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_client.dms_client", + new=dms_client, ): from prowler.providers.aws.services.dms.dms_endpoint_ssl_enabled.dms_endpoint_ssl_enabled import ( dms_endpoint_ssl_enabled, @@ -98,15 +92,12 @@ class Test_dms_endpoint_ssl_enabled: dms_client.audited_partition = "aws" dms_client.audited_region = AWS_REGION_US_EAST_1 - with ( - mock.patch( - "prowler.providers.aws.services.dms.dms_service.DMS", - new=dms_client, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_client.dms_client", - new=dms_client, - ), + with mock.patch( + "prowler.providers.aws.services.dms.dms_service.DMS", + new=dms_client, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_client.dms_client", + new=dms_client, ): from prowler.providers.aws.services.dms.dms_endpoint_ssl_enabled.dms_endpoint_ssl_enabled import ( dms_endpoint_ssl_enabled, @@ -147,15 +138,12 @@ class Test_dms_endpoint_ssl_enabled: dms_client.audited_partition = "aws" dms_client.audited_region = AWS_REGION_US_EAST_1 - with ( - mock.patch( - "prowler.providers.aws.services.dms.dms_service.DMS", - new=dms_client, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_client.dms_client", - new=dms_client, - ), + with mock.patch( + "prowler.providers.aws.services.dms.dms_service.DMS", + new=dms_client, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_client.dms_client", + new=dms_client, ): from prowler.providers.aws.services.dms.dms_endpoint_ssl_enabled.dms_endpoint_ssl_enabled import ( dms_endpoint_ssl_enabled, @@ -196,15 +184,12 @@ class Test_dms_endpoint_ssl_enabled: dms_client.audited_partition = "aws" dms_client.audited_region = AWS_REGION_US_EAST_1 - with ( - mock.patch( - "prowler.providers.aws.services.dms.dms_service.DMS", - new=dms_client, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_client.dms_client", - new=dms_client, - ), + with mock.patch( + "prowler.providers.aws.services.dms.dms_service.DMS", + new=dms_client, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_client.dms_client", + new=dms_client, ): from prowler.providers.aws.services.dms.dms_endpoint_ssl_enabled.dms_endpoint_ssl_enabled import ( dms_endpoint_ssl_enabled, diff --git a/tests/providers/aws/services/dms/dms_instance_minor_version_upgrade_enabled/dms_instance_minor_version_upgrade_enabled_test.py b/tests/providers/aws/services/dms/dms_instance_minor_version_upgrade_enabled/dms_instance_minor_version_upgrade_enabled_test.py index fd6311590a..b5a9c98716 100644 --- a/tests/providers/aws/services/dms/dms_instance_minor_version_upgrade_enabled/dms_instance_minor_version_upgrade_enabled_test.py +++ b/tests/providers/aws/services/dms/dms_instance_minor_version_upgrade_enabled/dms_instance_minor_version_upgrade_enabled_test.py @@ -15,15 +15,12 @@ class Test_dms_instance_minor_version_upgrade_enabled: dms_client = mock.MagicMock dms_client.instances = [] - with ( - mock.patch( - "prowler.providers.aws.services.dms.dms_service.DMS", - new=dms_client, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_client.dms_client", - new=dms_client, - ), + with mock.patch( + "prowler.providers.aws.services.dms.dms_service.DMS", + new=dms_client, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_client.dms_client", + new=dms_client, ): from prowler.providers.aws.services.dms.dms_instance_minor_version_upgrade_enabled.dms_instance_minor_version_upgrade_enabled import ( dms_instance_minor_version_upgrade_enabled, @@ -50,15 +47,12 @@ class Test_dms_instance_minor_version_upgrade_enabled: ) ) - with ( - mock.patch( - "prowler.providers.aws.services.dms.dms_service.DMS", - new=dms_client, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_client.dms_client", - new=dms_client, - ), + with mock.patch( + "prowler.providers.aws.services.dms.dms_service.DMS", + new=dms_client, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_client.dms_client", + new=dms_client, ): from prowler.providers.aws.services.dms.dms_instance_minor_version_upgrade_enabled.dms_instance_minor_version_upgrade_enabled import ( dms_instance_minor_version_upgrade_enabled, @@ -96,15 +90,12 @@ class Test_dms_instance_minor_version_upgrade_enabled: ) ) - with ( - mock.patch( - "prowler.providers.aws.services.dms.dms_service.DMS", - new=dms_client, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_client.dms_client", - new=dms_client, - ), + with mock.patch( + "prowler.providers.aws.services.dms.dms_service.DMS", + new=dms_client, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_client.dms_client", + new=dms_client, ): from prowler.providers.aws.services.dms.dms_instance_minor_version_upgrade_enabled.dms_instance_minor_version_upgrade_enabled import ( dms_instance_minor_version_upgrade_enabled, diff --git a/tests/providers/aws/services/dms/dms_instance_multi_az/dms_instance_multi_az_test.py b/tests/providers/aws/services/dms/dms_instance_multi_az/dms_instance_multi_az_test.py index 3917476a48..86ce420427 100644 --- a/tests/providers/aws/services/dms/dms_instance_multi_az/dms_instance_multi_az_test.py +++ b/tests/providers/aws/services/dms/dms_instance_multi_az/dms_instance_multi_az_test.py @@ -15,15 +15,12 @@ class Test_dms_instance_multi_az: dms_client = mock.MagicMock dms_client.instances = [] - with ( - mock.patch( - "prowler.providers.aws.services.dms.dms_service.DMS", - new=dms_client, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_client.dms_client", - new=dms_client, - ), + with mock.patch( + "prowler.providers.aws.services.dms.dms_service.DMS", + new=dms_client, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_client.dms_client", + new=dms_client, ): from prowler.providers.aws.services.dms.dms_instance_multi_az_enabled.dms_instance_multi_az_enabled import ( dms_instance_multi_az_enabled, @@ -50,15 +47,12 @@ class Test_dms_instance_multi_az: ) ) - with ( - mock.patch( - "prowler.providers.aws.services.dms.dms_service.DMS", - new=dms_client, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_client.dms_client", - new=dms_client, - ), + with mock.patch( + "prowler.providers.aws.services.dms.dms_service.DMS", + new=dms_client, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_client.dms_client", + new=dms_client, ): from prowler.providers.aws.services.dms.dms_instance_multi_az_enabled.dms_instance_multi_az_enabled import ( dms_instance_multi_az_enabled, @@ -96,15 +90,12 @@ class Test_dms_instance_multi_az: ) ) - with ( - mock.patch( - "prowler.providers.aws.services.dms.dms_service.DMS", - new=dms_client, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_client.dms_client", - new=dms_client, - ), + with mock.patch( + "prowler.providers.aws.services.dms.dms_service.DMS", + new=dms_client, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_client.dms_client", + new=dms_client, ): from prowler.providers.aws.services.dms.dms_instance_multi_az_enabled.dms_instance_multi_az_enabled import ( dms_instance_multi_az_enabled, diff --git a/tests/providers/aws/services/dms/dms_instance_no_public_access/dms_no_public_access_test.py b/tests/providers/aws/services/dms/dms_instance_no_public_access/dms_no_public_access_test.py index c8a99a0a82..0b9a5209c5 100644 --- a/tests/providers/aws/services/dms/dms_instance_no_public_access/dms_no_public_access_test.py +++ b/tests/providers/aws/services/dms/dms_instance_no_public_access/dms_no_public_access_test.py @@ -70,15 +70,12 @@ class Test_dms_instance_no_public_access: from prowler.providers.aws.services.dms.dms_service import DMS - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_instance_no_public_access.dms_instance_no_public_access.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_instance_no_public_access.dms_instance_no_public_access.dms_client", + new=DMS(aws_provider), ): from prowler.providers.aws.services.dms.dms_instance_no_public_access.dms_instance_no_public_access import ( dms_instance_no_public_access, @@ -99,15 +96,12 @@ class Test_dms_instance_no_public_access: from prowler.providers.aws.services.dms.dms_service import DMS - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_instance_no_public_access.dms_instance_no_public_access.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_instance_no_public_access.dms_instance_no_public_access.dms_client", + new=DMS(aws_provider), ): from prowler.providers.aws.services.dms.dms_instance_no_public_access.dms_instance_no_public_access import ( dms_instance_no_public_access, @@ -137,15 +131,12 @@ class Test_dms_instance_no_public_access: from prowler.providers.aws.services.dms.dms_service import DMS - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_instance_no_public_access.dms_instance_no_public_access.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_instance_no_public_access.dms_instance_no_public_access.dms_client", + new=DMS(aws_provider), ): from prowler.providers.aws.services.dms.dms_instance_no_public_access.dms_instance_no_public_access import ( dms_instance_no_public_access, @@ -208,19 +199,15 @@ class Test_dms_instance_no_public_access: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.dms.dms_instance_no_public_access.dms_instance_no_public_access.dms_client", - new=dms_client, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_client.dms_client", - new=dms_client, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_instance_no_public_access.dms_instance_no_public_access.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.aws.services.dms.dms_instance_no_public_access.dms_instance_no_public_access.dms_client", + new=dms_client, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_client.dms_client", + new=dms_client, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_instance_no_public_access.dms_instance_no_public_access.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_instance_no_public_access.dms_instance_no_public_access import ( @@ -288,19 +275,15 @@ class Test_dms_instance_no_public_access: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.dms.dms_instance_no_public_access.dms_instance_no_public_access.dms_client", - new=dms_client, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_client.dms_client", - new=dms_client, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_instance_no_public_access.dms_instance_no_public_access.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.aws.services.dms.dms_instance_no_public_access.dms_instance_no_public_access.dms_client", + new=dms_client, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_client.dms_client", + new=dms_client, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_instance_no_public_access.dms_instance_no_public_access.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_instance_no_public_access.dms_instance_no_public_access import ( diff --git a/tests/providers/aws/services/dms/dms_replication_task_source_logging_enabled/dms_replication_task_source_logging_enabled_test.py b/tests/providers/aws/services/dms/dms_replication_task_source_logging_enabled/dms_replication_task_source_logging_enabled_test.py index bd02922804..885b86af10 100644 --- a/tests/providers/aws/services/dms/dms_replication_task_source_logging_enabled/dms_replication_task_source_logging_enabled_test.py +++ b/tests/providers/aws/services/dms/dms_replication_task_source_logging_enabled/dms_replication_task_source_logging_enabled_test.py @@ -27,15 +27,12 @@ class Test_dms_replication_task_source_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled import ( @@ -80,15 +77,12 @@ class Test_dms_replication_task_source_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled import ( @@ -141,15 +135,12 @@ class Test_dms_replication_task_source_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled import ( @@ -202,15 +193,12 @@ class Test_dms_replication_task_source_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled import ( @@ -269,15 +257,12 @@ class Test_dms_replication_task_source_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled import ( @@ -336,15 +321,12 @@ class Test_dms_replication_task_source_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled import ( @@ -403,15 +385,12 @@ class Test_dms_replication_task_source_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled import ( @@ -470,15 +449,12 @@ class Test_dms_replication_task_source_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_replication_task_source_logging_enabled.dms_replication_task_source_logging_enabled import ( diff --git a/tests/providers/aws/services/dms/dms_replication_task_target_logging_enabled/dms_replication_task_target_logging_enabled_test.py b/tests/providers/aws/services/dms/dms_replication_task_target_logging_enabled/dms_replication_task_target_logging_enabled_test.py index 37dd3d660d..507334c75f 100644 --- a/tests/providers/aws/services/dms/dms_replication_task_target_logging_enabled/dms_replication_task_target_logging_enabled_test.py +++ b/tests/providers/aws/services/dms/dms_replication_task_target_logging_enabled/dms_replication_task_target_logging_enabled_test.py @@ -27,15 +27,12 @@ class Test_dms_replication_task_target_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled import ( @@ -80,15 +77,12 @@ class Test_dms_replication_task_target_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled import ( @@ -141,15 +135,12 @@ class Test_dms_replication_task_target_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled import ( @@ -202,15 +193,12 @@ class Test_dms_replication_task_target_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled import ( @@ -269,15 +257,12 @@ class Test_dms_replication_task_target_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled import ( @@ -336,15 +321,12 @@ class Test_dms_replication_task_target_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled import ( @@ -403,15 +385,12 @@ class Test_dms_replication_task_target_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled import ( @@ -470,15 +449,12 @@ class Test_dms_replication_task_target_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled.dms_client", - new=DMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled.dms_client", + new=DMS(aws_provider), ): # Test Check from prowler.providers.aws.services.dms.dms_replication_task_target_logging_enabled.dms_replication_task_target_logging_enabled import ( diff --git a/tests/providers/aws/services/documentdb/documentdb_cluster_backup_enabled/documentdb_cluster_backup_enabled_test.py b/tests/providers/aws/services/documentdb/documentdb_cluster_backup_enabled/documentdb_cluster_backup_enabled_test.py index 1b66d6ca45..5e61250dc2 100644 --- a/tests/providers/aws/services/documentdb/documentdb_cluster_backup_enabled/documentdb_cluster_backup_enabled_test.py +++ b/tests/providers/aws/services/documentdb/documentdb_cluster_backup_enabled/documentdb_cluster_backup_enabled_test.py @@ -18,15 +18,12 @@ class Test_documentdb_cluster_backup_enabled: documentdb_client.db_clusters = {} documentdb_client.audit_config = {"minimum_backup_retention_period": 7} - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_backup_enabled.documentdb_cluster_backup_enabled import ( documentdb_cluster_backup_enabled, @@ -55,15 +52,12 @@ class Test_documentdb_cluster_backup_enabled: ) } documentdb_client.audit_config = {"minimum_backup_retention_period": 7} - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_backup_enabled.documentdb_cluster_backup_enabled import ( documentdb_cluster_backup_enabled, @@ -101,15 +95,12 @@ class Test_documentdb_cluster_backup_enabled: ) } documentdb_client.audit_config = {"minimum_backup_retention_period": 7} - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_backup_enabled.documentdb_cluster_backup_enabled import ( documentdb_cluster_backup_enabled, @@ -145,15 +136,12 @@ class Test_documentdb_cluster_backup_enabled: ) } documentdb_client.audit_config = {"minimum_backup_retention_period": 7} - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_backup_enabled.documentdb_cluster_backup_enabled import ( documentdb_cluster_backup_enabled, @@ -189,15 +177,12 @@ class Test_documentdb_cluster_backup_enabled: ) } documentdb_client.audit_config = {"minimum_backup_retention_period": 7} - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_backup_enabled.documentdb_cluster_backup_enabled import ( documentdb_cluster_backup_enabled, @@ -234,15 +219,12 @@ class Test_documentdb_cluster_backup_enabled: } documentdb_client.audit_config = {"minimum_backup_retention_period": 1} - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_backup_enabled.documentdb_cluster_backup_enabled import ( documentdb_cluster_backup_enabled, diff --git a/tests/providers/aws/services/documentdb/documentdb_cluster_cloudwatch_log_export/documentdb_cluster_cloudwatch_log_export_test.py b/tests/providers/aws/services/documentdb/documentdb_cluster_cloudwatch_log_export/documentdb_cluster_cloudwatch_log_export_test.py index 97eb51f674..3dc1cdb1c2 100644 --- a/tests/providers/aws/services/documentdb/documentdb_cluster_cloudwatch_log_export/documentdb_cluster_cloudwatch_log_export_test.py +++ b/tests/providers/aws/services/documentdb/documentdb_cluster_cloudwatch_log_export/documentdb_cluster_cloudwatch_log_export_test.py @@ -17,15 +17,12 @@ class Test_documentdb_cluster_cloudwatch_log_export: documentdb_client = mock.MagicMock documentdb_client.db_clusters = {} - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_cloudwatch_log_export.documentdb_cluster_cloudwatch_log_export import ( documentdb_cluster_cloudwatch_log_export, @@ -54,15 +51,12 @@ class Test_documentdb_cluster_cloudwatch_log_export: ) } - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_cloudwatch_log_export.documentdb_cluster_cloudwatch_log_export import ( documentdb_cluster_cloudwatch_log_export, @@ -99,15 +93,12 @@ class Test_documentdb_cluster_cloudwatch_log_export: tags=[], ) } - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_cloudwatch_log_export.documentdb_cluster_cloudwatch_log_export import ( documentdb_cluster_cloudwatch_log_export, @@ -142,15 +133,12 @@ class Test_documentdb_cluster_cloudwatch_log_export: tags=[], ) } - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_cloudwatch_log_export.documentdb_cluster_cloudwatch_log_export import ( documentdb_cluster_cloudwatch_log_export, @@ -185,15 +173,12 @@ class Test_documentdb_cluster_cloudwatch_log_export: tags=[], ) } - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_cloudwatch_log_export.documentdb_cluster_cloudwatch_log_export import ( documentdb_cluster_cloudwatch_log_export, diff --git a/tests/providers/aws/services/documentdb/documentdb_cluster_deletion_protection/documentdb_cluster_deletion_protection_test.py b/tests/providers/aws/services/documentdb/documentdb_cluster_deletion_protection/documentdb_cluster_deletion_protection_test.py index bc26f0091c..45e28afe18 100644 --- a/tests/providers/aws/services/documentdb/documentdb_cluster_deletion_protection/documentdb_cluster_deletion_protection_test.py +++ b/tests/providers/aws/services/documentdb/documentdb_cluster_deletion_protection/documentdb_cluster_deletion_protection_test.py @@ -17,15 +17,12 @@ class Test_documentdb_cluster_deletion_protection: documentdb_client = mock.MagicMock documentdb_client.db_clusters = {} - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_deletion_protection.documentdb_cluster_deletion_protection import ( documentdb_cluster_deletion_protection, @@ -54,15 +51,12 @@ class Test_documentdb_cluster_deletion_protection: ) } - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_deletion_protection.documentdb_cluster_deletion_protection import ( documentdb_cluster_deletion_protection, @@ -99,15 +93,12 @@ class Test_documentdb_cluster_deletion_protection: tags=[], ) } - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_deletion_protection.documentdb_cluster_deletion_protection import ( documentdb_cluster_deletion_protection, diff --git a/tests/providers/aws/services/documentdb/documentdb_cluster_multi_az_enabled/documentdb_cluster_multi_az_enabled_test.py b/tests/providers/aws/services/documentdb/documentdb_cluster_multi_az_enabled/documentdb_cluster_multi_az_enabled_test.py index 67bf7adf2f..92c4268f83 100644 --- a/tests/providers/aws/services/documentdb/documentdb_cluster_multi_az_enabled/documentdb_cluster_multi_az_enabled_test.py +++ b/tests/providers/aws/services/documentdb/documentdb_cluster_multi_az_enabled/documentdb_cluster_multi_az_enabled_test.py @@ -17,15 +17,12 @@ class Test_documentdb_cluster_multi_az_enabled: documentdb_client = mock.MagicMock documentdb_client.db_clusters = {} - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_multi_az_enabled.documentdb_cluster_multi_az_enabled import ( documentdb_cluster_multi_az_enabled, @@ -54,15 +51,12 @@ class Test_documentdb_cluster_multi_az_enabled: ) } - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_multi_az_enabled.documentdb_cluster_multi_az_enabled import ( documentdb_cluster_multi_az_enabled, @@ -98,15 +92,12 @@ class Test_documentdb_cluster_multi_az_enabled: tags=[], ) } - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_multi_az_enabled.documentdb_cluster_multi_az_enabled import ( documentdb_cluster_multi_az_enabled, diff --git a/tests/providers/aws/services/documentdb/documentdb_cluster_public_snapshot/documentdb_cluster_public_snapshot_fixer_test.py b/tests/providers/aws/services/documentdb/documentdb_cluster_public_snapshot/documentdb_cluster_public_snapshot_fixer_test.py index 6ac53f42cc..861230655c 100644 --- a/tests/providers/aws/services/documentdb/documentdb_cluster_public_snapshot/documentdb_cluster_public_snapshot_fixer_test.py +++ b/tests/providers/aws/services/documentdb/documentdb_cluster_public_snapshot/documentdb_cluster_public_snapshot_fixer_test.py @@ -52,15 +52,12 @@ class Test_documentdb_cluster_public_snapshot_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_cluster_public_snapshot.documentdb_cluster_public_snapshot_fixer.documentdb_client", - new=DocumentDB(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_cluster_public_snapshot.documentdb_cluster_public_snapshot_fixer.documentdb_client", + new=DocumentDB(aws_provider), ): from prowler.providers.aws.services.documentdb.documentdb_cluster_public_snapshot.documentdb_cluster_public_snapshot_fixer import ( fixer, @@ -80,15 +77,12 @@ class Test_documentdb_cluster_public_snapshot_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_cluster_public_snapshot.documentdb_cluster_public_snapshot_fixer.documentdb_client", - new=DocumentDB(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_cluster_public_snapshot.documentdb_cluster_public_snapshot_fixer.documentdb_client", + new=DocumentDB(aws_provider), ): from prowler.providers.aws.services.documentdb.documentdb_cluster_public_snapshot.documentdb_cluster_public_snapshot_fixer import ( fixer, diff --git a/tests/providers/aws/services/documentdb/documentdb_cluster_public_snapshot/documentdb_cluster_public_snapshot_test.py b/tests/providers/aws/services/documentdb/documentdb_cluster_public_snapshot/documentdb_cluster_public_snapshot_test.py index da9eb0d7e2..53c8276350 100644 --- a/tests/providers/aws/services/documentdb/documentdb_cluster_public_snapshot/documentdb_cluster_public_snapshot_test.py +++ b/tests/providers/aws/services/documentdb/documentdb_cluster_public_snapshot/documentdb_cluster_public_snapshot_test.py @@ -19,15 +19,12 @@ class Test_documentdb_cluster_public_snapshot: documentdb_client.db_clusters = {} documentdb_client.db_cluster_snapshots = [] - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_public_snapshot.documentdb_cluster_public_snapshot import ( documentdb_cluster_public_snapshot, @@ -66,15 +63,12 @@ class Test_documentdb_cluster_public_snapshot: ) ] - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_public_snapshot.documentdb_cluster_public_snapshot import ( documentdb_cluster_public_snapshot, @@ -125,15 +119,12 @@ class Test_documentdb_cluster_public_snapshot: tags=[], ) ] - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_public_snapshot.documentdb_cluster_public_snapshot import ( documentdb_cluster_public_snapshot, diff --git a/tests/providers/aws/services/documentdb/documentdb_cluster_storage_encrypted/documentdb_cluster_storage_encrypted_test.py b/tests/providers/aws/services/documentdb/documentdb_cluster_storage_encrypted/documentdb_cluster_storage_encrypted_test.py index 990fb1df51..cadb11d0ee 100644 --- a/tests/providers/aws/services/documentdb/documentdb_cluster_storage_encrypted/documentdb_cluster_storage_encrypted_test.py +++ b/tests/providers/aws/services/documentdb/documentdb_cluster_storage_encrypted/documentdb_cluster_storage_encrypted_test.py @@ -17,15 +17,12 @@ class Test_documentdb_cluster_storage_encrypted: documentdb_client = mock.MagicMock documentdb_client.db_clusters = {} - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_storage_encrypted.documentdb_cluster_storage_encrypted import ( documentdb_cluster_storage_encrypted, @@ -54,15 +51,12 @@ class Test_documentdb_cluster_storage_encrypted: ) } - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_storage_encrypted.documentdb_cluster_storage_encrypted import ( documentdb_cluster_storage_encrypted, @@ -98,15 +92,12 @@ class Test_documentdb_cluster_storage_encrypted: tags=[], ) } - with ( - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", - new=documentdb_client, - ), - mock.patch( - "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", - new=documentdb_client, - ), + with mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_service.DocumentDB", + new=documentdb_client, + ), mock.patch( + "prowler.providers.aws.services.documentdb.documentdb_client.documentdb_client", + new=documentdb_client, ): from prowler.providers.aws.services.documentdb.documentdb_cluster_storage_encrypted.documentdb_cluster_storage_encrypted import ( documentdb_cluster_storage_encrypted, diff --git a/tests/providers/aws/services/dynamodb/dynamodb_accelerator_cluster_encryption_enabled/dynamodb_accelerator_cluster_encryption_enabled_test.py b/tests/providers/aws/services/dynamodb/dynamodb_accelerator_cluster_encryption_enabled/dynamodb_accelerator_cluster_encryption_enabled_test.py index 3d2131e219..c8c096c669 100644 --- a/tests/providers/aws/services/dynamodb/dynamodb_accelerator_cluster_encryption_enabled/dynamodb_accelerator_cluster_encryption_enabled_test.py +++ b/tests/providers/aws/services/dynamodb/dynamodb_accelerator_cluster_encryption_enabled/dynamodb_accelerator_cluster_encryption_enabled_test.py @@ -20,15 +20,12 @@ class Test_dynamodb_accelerator_cluster_encryption_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_encryption_enabled.dynamodb_accelerator_cluster_encryption_enabled.dax_client", - new=DAX(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_encryption_enabled.dynamodb_accelerator_cluster_encryption_enabled.dax_client", + new=DAX(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_encryption_enabled.dynamodb_accelerator_cluster_encryption_enabled import ( @@ -57,15 +54,12 @@ class Test_dynamodb_accelerator_cluster_encryption_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_encryption_enabled.dynamodb_accelerator_cluster_encryption_enabled.dax_client", - new=DAX(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_encryption_enabled.dynamodb_accelerator_cluster_encryption_enabled.dax_client", + new=DAX(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_encryption_enabled.dynamodb_accelerator_cluster_encryption_enabled import ( @@ -104,15 +98,12 @@ class Test_dynamodb_accelerator_cluster_encryption_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_encryption_enabled.dynamodb_accelerator_cluster_encryption_enabled.dax_client", - new=DAX(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_encryption_enabled.dynamodb_accelerator_cluster_encryption_enabled.dax_client", + new=DAX(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_encryption_enabled.dynamodb_accelerator_cluster_encryption_enabled import ( diff --git a/tests/providers/aws/services/dynamodb/dynamodb_accelerator_cluster_in_transit_encryption_enabled/dynamodb_accelerator_cluster_in_transit_encryption_enabled_test.py b/tests/providers/aws/services/dynamodb/dynamodb_accelerator_cluster_in_transit_encryption_enabled/dynamodb_accelerator_cluster_in_transit_encryption_enabled_test.py index 3f7411ccc4..512ce2b93d 100644 --- a/tests/providers/aws/services/dynamodb/dynamodb_accelerator_cluster_in_transit_encryption_enabled/dynamodb_accelerator_cluster_in_transit_encryption_enabled_test.py +++ b/tests/providers/aws/services/dynamodb/dynamodb_accelerator_cluster_in_transit_encryption_enabled/dynamodb_accelerator_cluster_in_transit_encryption_enabled_test.py @@ -20,15 +20,12 @@ class Test_dynamodb_accelerator_cluster_in_transit_encryption_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_in_transit_encryption_enabled.dynamodb_accelerator_cluster_in_transit_encryption_enabled.dax_client", - new=DAX(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_in_transit_encryption_enabled.dynamodb_accelerator_cluster_in_transit_encryption_enabled.dax_client", + new=DAX(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_in_transit_encryption_enabled.dynamodb_accelerator_cluster_in_transit_encryption_enabled import ( @@ -56,15 +53,12 @@ class Test_dynamodb_accelerator_cluster_in_transit_encryption_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_in_transit_encryption_enabled.dynamodb_accelerator_cluster_in_transit_encryption_enabled.dax_client", - new=DAX(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_in_transit_encryption_enabled.dynamodb_accelerator_cluster_in_transit_encryption_enabled.dax_client", + new=DAX(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_in_transit_encryption_enabled.dynamodb_accelerator_cluster_in_transit_encryption_enabled import ( @@ -102,15 +96,12 @@ class Test_dynamodb_accelerator_cluster_in_transit_encryption_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_in_transit_encryption_enabled.dynamodb_accelerator_cluster_in_transit_encryption_enabled.dax_client", - new=DAX(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_in_transit_encryption_enabled.dynamodb_accelerator_cluster_in_transit_encryption_enabled.dax_client", + new=DAX(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_in_transit_encryption_enabled.dynamodb_accelerator_cluster_in_transit_encryption_enabled import ( diff --git a/tests/providers/aws/services/dynamodb/dynamodb_accelerator_cluster_multi_az/dynamodb_accelerator_cluster_multi_az_test.py b/tests/providers/aws/services/dynamodb/dynamodb_accelerator_cluster_multi_az/dynamodb_accelerator_cluster_multi_az_test.py index 099e17d1b0..edc581afee 100644 --- a/tests/providers/aws/services/dynamodb/dynamodb_accelerator_cluster_multi_az/dynamodb_accelerator_cluster_multi_az_test.py +++ b/tests/providers/aws/services/dynamodb/dynamodb_accelerator_cluster_multi_az/dynamodb_accelerator_cluster_multi_az_test.py @@ -20,15 +20,12 @@ class Test_dynamodb_accelerator_cluster_multi_az: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_multi_az.dynamodb_accelerator_cluster_multi_az.dax_client", - new=DAX(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_multi_az.dynamodb_accelerator_cluster_multi_az.dax_client", + new=DAX(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_multi_az.dynamodb_accelerator_cluster_multi_az import ( @@ -56,15 +53,12 @@ class Test_dynamodb_accelerator_cluster_multi_az: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_multi_az.dynamodb_accelerator_cluster_multi_az.dax_client", - new=DAX(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_multi_az.dynamodb_accelerator_cluster_multi_az.dax_client", + new=DAX(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_multi_az.dynamodb_accelerator_cluster_multi_az import ( @@ -102,16 +96,13 @@ class Test_dynamodb_accelerator_cluster_multi_az: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_multi_az.dynamodb_accelerator_cluster_multi_az.dax_client", - new=DAX(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_multi_az.dynamodb_accelerator_cluster_multi_az.dax_client", + new=DAX(aws_provider), + ) as service_client: # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_accelerator_cluster_multi_az.dynamodb_accelerator_cluster_multi_az import ( dynamodb_accelerator_cluster_multi_az, diff --git a/tests/providers/aws/services/dynamodb/dynamodb_table_autoscaling_enabled/dynamodb_table_autoscaling_enabled_test.py b/tests/providers/aws/services/dynamodb/dynamodb_table_autoscaling_enabled/dynamodb_table_autoscaling_enabled_test.py index 4c0b1b6187..9573762a59 100644 --- a/tests/providers/aws/services/dynamodb/dynamodb_table_autoscaling_enabled/dynamodb_table_autoscaling_enabled_test.py +++ b/tests/providers/aws/services/dynamodb/dynamodb_table_autoscaling_enabled/dynamodb_table_autoscaling_enabled_test.py @@ -19,15 +19,12 @@ class Test_dynamodb_table_autoscaling_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled.dynamodb_client", - new=DynamoDB(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled.dynamodb_client", + new=DynamoDB(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled import ( @@ -61,15 +58,12 @@ class Test_dynamodb_table_autoscaling_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled.dynamodb_client", - new=DynamoDB(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled.dynamodb_client", + new=DynamoDB(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled import ( @@ -133,19 +127,15 @@ class Test_dynamodb_table_autoscaling_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled.dynamodb_client", - new=DynamoDB(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled.applicationautoscaling_client", - new=ApplicationAutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled.dynamodb_client", + new=DynamoDB(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled.applicationautoscaling_client", + new=ApplicationAutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled import ( @@ -202,19 +192,15 @@ class Test_dynamodb_table_autoscaling_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled.dynamodb_client", - new=DynamoDB(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled.applicationautoscaling_client", - new=ApplicationAutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled.dynamodb_client", + new=DynamoDB(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled.applicationautoscaling_client", + new=ApplicationAutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled import ( @@ -271,19 +257,15 @@ class Test_dynamodb_table_autoscaling_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled.dynamodb_client", - new=DynamoDB(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled.applicationautoscaling_client", - new=ApplicationAutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled.dynamodb_client", + new=DynamoDB(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled.applicationautoscaling_client", + new=ApplicationAutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled import ( @@ -329,19 +311,15 @@ class Test_dynamodb_table_autoscaling_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled.dynamodb_client", - new=DynamoDB(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled.applicationautoscaling_client", - new=ApplicationAutoScaling(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled.dynamodb_client", + new=DynamoDB(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled.applicationautoscaling_client", + new=ApplicationAutoScaling(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_table_autoscaling_enabled.dynamodb_table_autoscaling_enabled import ( diff --git a/tests/providers/aws/services/dynamodb/dynamodb_table_deletion_protection_enabled/dynamodb_table_deletion_protection_enabled_test.py b/tests/providers/aws/services/dynamodb/dynamodb_table_deletion_protection_enabled/dynamodb_table_deletion_protection_enabled_test.py index fb8391ce7a..8020816aca 100644 --- a/tests/providers/aws/services/dynamodb/dynamodb_table_deletion_protection_enabled/dynamodb_table_deletion_protection_enabled_test.py +++ b/tests/providers/aws/services/dynamodb/dynamodb_table_deletion_protection_enabled/dynamodb_table_deletion_protection_enabled_test.py @@ -19,15 +19,12 @@ class Test_dynamodb_table_deletion_protection_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_deletion_protection_enabled.dynamodb_table_deletion_protection_enabled.dynamodb_client", - new=DynamoDB(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_deletion_protection_enabled.dynamodb_table_deletion_protection_enabled.dynamodb_client", + new=DynamoDB(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_table_deletion_protection_enabled.dynamodb_table_deletion_protection_enabled import ( @@ -62,15 +59,12 @@ class Test_dynamodb_table_deletion_protection_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_deletion_protection_enabled.dynamodb_table_deletion_protection_enabled.dynamodb_client", - new=DynamoDB(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_deletion_protection_enabled.dynamodb_table_deletion_protection_enabled.dynamodb_client", + new=DynamoDB(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_table_deletion_protection_enabled.dynamodb_table_deletion_protection_enabled import ( @@ -113,15 +107,12 @@ class Test_dynamodb_table_deletion_protection_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_deletion_protection_enabled.dynamodb_table_deletion_protection_enabled.dynamodb_client", - new=DynamoDB(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_deletion_protection_enabled.dynamodb_table_deletion_protection_enabled.dynamodb_client", + new=DynamoDB(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_table_deletion_protection_enabled.dynamodb_table_deletion_protection_enabled import ( diff --git a/tests/providers/aws/services/dynamodb/dynamodb_table_protected_by_backup_plan/dynamodb_table_protected_by_backup_plan_test.py b/tests/providers/aws/services/dynamodb/dynamodb_table_protected_by_backup_plan/dynamodb_table_protected_by_backup_plan_test.py index e07ba1e05c..870c0f81a1 100644 --- a/tests/providers/aws/services/dynamodb/dynamodb_table_protected_by_backup_plan/dynamodb_table_protected_by_backup_plan_test.py +++ b/tests/providers/aws/services/dynamodb/dynamodb_table_protected_by_backup_plan/dynamodb_table_protected_by_backup_plan_test.py @@ -22,15 +22,12 @@ class Test_dynamodb_table_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.dynamodb_client", - new=DynamoDB(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.backup_client", - new=Backup(aws_provider), - ), + with mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.dynamodb_client", + new=DynamoDB(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.backup_client", + new=Backup(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan import ( @@ -68,15 +65,12 @@ class Test_dynamodb_table_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.dynamodb_client", - new=DynamoDB(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.backup_client", - new=Backup(aws_provider), - ), + with mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.dynamodb_client", + new=DynamoDB(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.backup_client", + new=Backup(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan import ( @@ -141,15 +135,12 @@ class Test_dynamodb_table_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.dynamodb_client", - new=DynamoDB(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.backup_client", - new=Backup(aws_provider), - ), + with mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.dynamodb_client", + new=DynamoDB(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.backup_client", + new=Backup(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan import ( @@ -193,23 +184,18 @@ class Test_dynamodb_table_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.dynamodb_client", - new=dynamodb_client, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_client.dynamodb_client", - new=dynamodb_client, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.backup_client", - new=backup, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_client.backup_client", - new=backup, - ), + with mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.dynamodb_client", + new=dynamodb_client, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_client.dynamodb_client", + new=dynamodb_client, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.backup_client", + new=backup, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_client.backup_client", + new=backup, ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan import ( @@ -255,23 +241,18 @@ class Test_dynamodb_table_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.dynamodb_client", - new=dynamodb_client, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_client.dynamodb_client", - new=dynamodb_client, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.backup_client", - new=backup, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_client.backup_client", - new=backup, - ), + with mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.dynamodb_client", + new=dynamodb_client, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_client.dynamodb_client", + new=dynamodb_client, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.backup_client", + new=backup, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_client.backup_client", + new=backup, ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan import ( @@ -316,23 +297,18 @@ class Test_dynamodb_table_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.dynamodb_client", - new=dynamodb_client, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_client.dynamodb_client", - new=dynamodb_client, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.backup_client", - new=backup, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_client.backup_client", - new=backup, - ), + with mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.dynamodb_client", + new=dynamodb_client, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_client.dynamodb_client", + new=dynamodb_client, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan.backup_client", + new=backup, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_client.backup_client", + new=backup, ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_table_protected_by_backup_plan.dynamodb_table_protected_by_backup_plan import ( diff --git a/tests/providers/aws/services/dynamodb/dynamodb_tables_kms_cmk_encryption_enabled/dynamodb_tables_kms_cmk_encryption_enabled_test.py b/tests/providers/aws/services/dynamodb/dynamodb_tables_kms_cmk_encryption_enabled/dynamodb_tables_kms_cmk_encryption_enabled_test.py index 83d2f8d02d..45817c081a 100644 --- a/tests/providers/aws/services/dynamodb/dynamodb_tables_kms_cmk_encryption_enabled/dynamodb_tables_kms_cmk_encryption_enabled_test.py +++ b/tests/providers/aws/services/dynamodb/dynamodb_tables_kms_cmk_encryption_enabled/dynamodb_tables_kms_cmk_encryption_enabled_test.py @@ -19,15 +19,12 @@ class Test_dynamodb_tables_kms_cmk_encryption_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_tables_kms_cmk_encryption_enabled.dynamodb_tables_kms_cmk_encryption_enabled.dynamodb_client", - new=DynamoDB(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_tables_kms_cmk_encryption_enabled.dynamodb_tables_kms_cmk_encryption_enabled.dynamodb_client", + new=DynamoDB(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_tables_kms_cmk_encryption_enabled.dynamodb_tables_kms_cmk_encryption_enabled import ( @@ -61,15 +58,12 @@ class Test_dynamodb_tables_kms_cmk_encryption_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_tables_kms_cmk_encryption_enabled.dynamodb_tables_kms_cmk_encryption_enabled.dynamodb_client", - new=DynamoDB(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_tables_kms_cmk_encryption_enabled.dynamodb_tables_kms_cmk_encryption_enabled.dynamodb_client", + new=DynamoDB(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_tables_kms_cmk_encryption_enabled.dynamodb_tables_kms_cmk_encryption_enabled import ( @@ -111,15 +105,12 @@ class Test_dynamodb_tables_kms_cmk_encryption_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_tables_kms_cmk_encryption_enabled.dynamodb_tables_kms_cmk_encryption_enabled.dynamodb_client", - new=DynamoDB(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_tables_kms_cmk_encryption_enabled.dynamodb_tables_kms_cmk_encryption_enabled.dynamodb_client", + new=DynamoDB(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_tables_kms_cmk_encryption_enabled.dynamodb_tables_kms_cmk_encryption_enabled import ( diff --git a/tests/providers/aws/services/dynamodb/dynamodb_tables_pitr_enabled/dynamodb_tables_pitr_enabled_test.py b/tests/providers/aws/services/dynamodb/dynamodb_tables_pitr_enabled/dynamodb_tables_pitr_enabled_test.py index 97a2645233..1d32d93898 100644 --- a/tests/providers/aws/services/dynamodb/dynamodb_tables_pitr_enabled/dynamodb_tables_pitr_enabled_test.py +++ b/tests/providers/aws/services/dynamodb/dynamodb_tables_pitr_enabled/dynamodb_tables_pitr_enabled_test.py @@ -19,15 +19,12 @@ class Test_dynamodb_tables_pitr_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_tables_pitr_enabled.dynamodb_tables_pitr_enabled.dynamodb_client", - new=DynamoDB(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_tables_pitr_enabled.dynamodb_tables_pitr_enabled.dynamodb_client", + new=DynamoDB(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_tables_pitr_enabled.dynamodb_tables_pitr_enabled import ( @@ -60,15 +57,12 @@ class Test_dynamodb_tables_pitr_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_tables_pitr_enabled.dynamodb_tables_pitr_enabled.dynamodb_client", - new=DynamoDB(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_tables_pitr_enabled.dynamodb_tables_pitr_enabled.dynamodb_client", + new=DynamoDB(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_tables_pitr_enabled.dynamodb_tables_pitr_enabled import ( @@ -114,15 +108,12 @@ class Test_dynamodb_tables_pitr_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.dynamodb.dynamodb_tables_pitr_enabled.dynamodb_tables_pitr_enabled.dynamodb_client", - new=DynamoDB(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.dynamodb.dynamodb_tables_pitr_enabled.dynamodb_tables_pitr_enabled.dynamodb_client", + new=DynamoDB(aws_provider), ): # Test Check from prowler.providers.aws.services.dynamodb.dynamodb_tables_pitr_enabled.dynamodb_tables_pitr_enabled import ( diff --git a/tests/providers/aws/services/ec2/ec2_ami_public/ec2_ami_public_fixer_test.py b/tests/providers/aws/services/ec2/ec2_ami_public/ec2_ami_public_fixer_test.py index e87718efb8..f5b6c5c48c 100644 --- a/tests/providers/aws/services/ec2/ec2_ami_public/ec2_ami_public_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_ami_public/ec2_ami_public_fixer_test.py @@ -31,15 +31,12 @@ class Test_ec2_ami_public_fixer: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ami_public.ec2_ami_public_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ami_public.ec2_ami_public_fixer.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_ami_public.ec2_ami_public_fixer import ( fixer, @@ -74,15 +71,12 @@ class Test_ec2_ami_public_fixer: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ami_public.ec2_ami_public_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ami_public.ec2_ami_public_fixer.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_ami_public.ec2_ami_public_fixer import ( fixer, @@ -117,15 +111,12 @@ class Test_ec2_ami_public_fixer: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ami_public.ec2_ami_public_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ami_public.ec2_ami_public_fixer.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_ami_public.ec2_ami_public_fixer import ( fixer, diff --git a/tests/providers/aws/services/ec2/ec2_ami_public/ec2_ami_public_test.py b/tests/providers/aws/services/ec2/ec2_ami_public/ec2_ami_public_test.py index aace181328..593128f506 100644 --- a/tests/providers/aws/services/ec2/ec2_ami_public/ec2_ami_public_test.py +++ b/tests/providers/aws/services/ec2/ec2_ami_public/ec2_ami_public_test.py @@ -21,15 +21,12 @@ class Test_ec2_ami_public: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ami_public.ec2_ami_public.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ami_public.ec2_ami_public.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ami_public.ec2_ami_public import ( @@ -59,15 +56,12 @@ class Test_ec2_ami_public: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ami_public.ec2_ami_public.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ami_public.ec2_ami_public.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_ami_public.ec2_ami_public import ( ec2_ami_public, @@ -114,15 +108,12 @@ class Test_ec2_ami_public: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ami_public.ec2_ami_public.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ami_public.ec2_ami_public.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_ami_public.ec2_ami_public import ( ec2_ami_public, diff --git a/tests/providers/aws/services/ec2/ec2_ebs_default_encryption/ec2_ebs_default_encryption_fixer_test.py b/tests/providers/aws/services/ec2/ec2_ebs_default_encryption/ec2_ebs_default_encryption_fixer_test.py index 1fdf72c337..b0e029f7ad 100644 --- a/tests/providers/aws/services/ec2/ec2_ebs_default_encryption/ec2_ebs_default_encryption_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_ebs_default_encryption/ec2_ebs_default_encryption_fixer_test.py @@ -12,15 +12,12 @@ class Test_ec2_ebs_default_encryption_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_default_encryption.ec2_ebs_default_encryption_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_default_encryption.ec2_ebs_default_encryption_fixer.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_ebs_default_encryption.ec2_ebs_default_encryption_fixer import ( fixer, diff --git a/tests/providers/aws/services/ec2/ec2_ebs_default_encryption/ec2_ebs_default_encryption_test.py b/tests/providers/aws/services/ec2/ec2_ebs_default_encryption/ec2_ebs_default_encryption_test.py index 5bca87b485..95042b8d56 100644 --- a/tests/providers/aws/services/ec2/ec2_ebs_default_encryption/ec2_ebs_default_encryption_test.py +++ b/tests/providers/aws/services/ec2/ec2_ebs_default_encryption/ec2_ebs_default_encryption_test.py @@ -26,15 +26,12 @@ class Test_ec2_ebs_default_encryption: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_default_encryption.ec2_ebs_default_encryption.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_default_encryption.ec2_ebs_default_encryption.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_default_encryption.ec2_ebs_default_encryption import ( @@ -77,15 +74,12 @@ class Test_ec2_ebs_default_encryption: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_default_encryption.ec2_ebs_default_encryption.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_default_encryption.ec2_ebs_default_encryption.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_default_encryption.ec2_ebs_default_encryption import ( @@ -130,15 +124,12 @@ class Test_ec2_ebs_default_encryption: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_default_encryption.ec2_ebs_default_encryption.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_default_encryption.ec2_ebs_default_encryption.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_default_encryption.ec2_ebs_default_encryption import ( @@ -163,15 +154,12 @@ class Test_ec2_ebs_default_encryption: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_default_encryption.ec2_ebs_default_encryption.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_default_encryption.ec2_ebs_default_encryption.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_default_encryption.ec2_ebs_default_encryption import ( diff --git a/tests/providers/aws/services/ec2/ec2_ebs_public_snapshot/ec2_ebs_public_snapshot_fixer_test.py b/tests/providers/aws/services/ec2/ec2_ebs_public_snapshot/ec2_ebs_public_snapshot_fixer_test.py index 77e7e802c3..905d85956b 100644 --- a/tests/providers/aws/services/ec2/ec2_ebs_public_snapshot/ec2_ebs_public_snapshot_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_ebs_public_snapshot/ec2_ebs_public_snapshot_fixer_test.py @@ -52,15 +52,12 @@ class Test_ec2_ebs_public_snapshot_fixer_test: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_public_snapshot.ec2_ebs_public_snapshot_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_public_snapshot.ec2_ebs_public_snapshot_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_public_snapshot.ec2_ebs_public_snapshot_fixer import ( @@ -81,15 +78,12 @@ class Test_ec2_ebs_public_snapshot_fixer_test: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_public_snapshot.ec2_ebs_public_snapshot_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_public_snapshot.ec2_ebs_public_snapshot_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_public_snapshot.ec2_ebs_public_snapshot_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_ebs_public_snapshot/ec2_ebs_public_snapshot_test.py b/tests/providers/aws/services/ec2/ec2_ebs_public_snapshot/ec2_ebs_public_snapshot_test.py index c3dbdcc023..05e69ad0e8 100644 --- a/tests/providers/aws/services/ec2/ec2_ebs_public_snapshot/ec2_ebs_public_snapshot_test.py +++ b/tests/providers/aws/services/ec2/ec2_ebs_public_snapshot/ec2_ebs_public_snapshot_test.py @@ -32,15 +32,12 @@ class Test_ec2_ebs_public_snapshot: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_public_snapshot.ec2_ebs_public_snapshot.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_public_snapshot.ec2_ebs_public_snapshot.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_public_snapshot.ec2_ebs_public_snapshot import ( @@ -73,15 +70,12 @@ class Test_ec2_ebs_public_snapshot: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_public_snapshot.ec2_ebs_public_snapshot.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_public_snapshot.ec2_ebs_public_snapshot.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_public_snapshot.ec2_ebs_public_snapshot import ( @@ -123,15 +117,12 @@ class Test_ec2_ebs_public_snapshot: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_public_snapshot.ec2_ebs_public_snapshot.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_public_snapshot.ec2_ebs_public_snapshot.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_public_snapshot.ec2_ebs_public_snapshot import ( diff --git a/tests/providers/aws/services/ec2/ec2_ebs_snapshot_account_block_public_access/ec2_ebs_snapshot_account_block_public_access_fixer_test.py b/tests/providers/aws/services/ec2/ec2_ebs_snapshot_account_block_public_access/ec2_ebs_snapshot_account_block_public_access_fixer_test.py index be715ac5ed..e323db2501 100644 --- a/tests/providers/aws/services/ec2/ec2_ebs_snapshot_account_block_public_access/ec2_ebs_snapshot_account_block_public_access_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_ebs_snapshot_account_block_public_access/ec2_ebs_snapshot_account_block_public_access_fixer_test.py @@ -51,15 +51,12 @@ class Test_ec2_ebs_snapshot_account_block_public_access_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_snapshot_account_block_public_access.ec2_ebs_snapshot_account_block_public_access_fixer.ec2_client", - ec2_service, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_snapshot_account_block_public_access.ec2_ebs_snapshot_account_block_public_access_fixer.ec2_client", + ec2_service, ): from prowler.providers.aws.services.ec2.ec2_ebs_snapshot_account_block_public_access.ec2_ebs_snapshot_account_block_public_access_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_ebs_snapshot_account_block_public_access/ec2_ebs_snapshot_account_block_public_access_test.py b/tests/providers/aws/services/ec2/ec2_ebs_snapshot_account_block_public_access/ec2_ebs_snapshot_account_block_public_access_test.py index 8c0a1c2730..95c4a9aba1 100644 --- a/tests/providers/aws/services/ec2/ec2_ebs_snapshot_account_block_public_access/ec2_ebs_snapshot_account_block_public_access_test.py +++ b/tests/providers/aws/services/ec2/ec2_ebs_snapshot_account_block_public_access/ec2_ebs_snapshot_account_block_public_access_test.py @@ -28,15 +28,12 @@ class Test_ec2_ebs_snapshot_account_block_public_access: f"arn:aws:ec2:{AWS_REGION_US_EAST_1}:{AWS_ACCOUNT_NUMBER}:account" ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_snapshot_account_block_public_access.ec2_ebs_snapshot_account_block_public_access.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_snapshot_account_block_public_access.ec2_ebs_snapshot_account_block_public_access.ec2_client", + new=ec2_client, ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_snapshot_account_block_public_access.ec2_ebs_snapshot_account_block_public_access import ( @@ -76,15 +73,12 @@ class Test_ec2_ebs_snapshot_account_block_public_access: f"arn:aws:ec2:{AWS_REGION_US_EAST_1}:{AWS_ACCOUNT_NUMBER}:account" ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_snapshot_account_block_public_access.ec2_ebs_snapshot_account_block_public_access.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_snapshot_account_block_public_access.ec2_ebs_snapshot_account_block_public_access.ec2_client", + new=ec2_client, ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_snapshot_account_block_public_access.ec2_ebs_snapshot_account_block_public_access import ( @@ -124,15 +118,12 @@ class Test_ec2_ebs_snapshot_account_block_public_access: f"arn:aws:ec2:{AWS_REGION_US_EAST_1}:{AWS_ACCOUNT_NUMBER}:account" ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_snapshot_account_block_public_access.ec2_ebs_snapshot_account_block_public_access.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_snapshot_account_block_public_access.ec2_ebs_snapshot_account_block_public_access.ec2_client", + new=ec2_client, ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_snapshot_account_block_public_access.ec2_ebs_snapshot_account_block_public_access import ( diff --git a/tests/providers/aws/services/ec2/ec2_ebs_snapshots_encrypted/ec2_ebs_snapshots_encrypted_test.py b/tests/providers/aws/services/ec2/ec2_ebs_snapshots_encrypted/ec2_ebs_snapshots_encrypted_test.py index 1365ead212..dbce574804 100644 --- a/tests/providers/aws/services/ec2/ec2_ebs_snapshots_encrypted/ec2_ebs_snapshots_encrypted_test.py +++ b/tests/providers/aws/services/ec2/ec2_ebs_snapshots_encrypted/ec2_ebs_snapshots_encrypted_test.py @@ -32,15 +32,12 @@ class Test_ec2_ebs_snapshots_encrypted: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_snapshots_encrypted.ec2_ebs_snapshots_encrypted.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_snapshots_encrypted.ec2_ebs_snapshots_encrypted.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_snapshots_encrypted.ec2_ebs_snapshots_encrypted import ( @@ -66,15 +63,12 @@ class Test_ec2_ebs_snapshots_encrypted: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_snapshots_encrypted.ec2_ebs_snapshots_encrypted.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_snapshots_encrypted.ec2_ebs_snapshots_encrypted.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_snapshots_encrypted.ec2_ebs_snapshots_encrypted import ( @@ -116,15 +110,12 @@ class Test_ec2_ebs_snapshots_encrypted: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_snapshots_encrypted.ec2_ebs_snapshots_encrypted.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_snapshots_encrypted.ec2_ebs_snapshots_encrypted.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_snapshots_encrypted.ec2_ebs_snapshots_encrypted import ( diff --git a/tests/providers/aws/services/ec2/ec2_ebs_volume_encryption/ec2_ebs_volume_encryption_test.py b/tests/providers/aws/services/ec2/ec2_ebs_volume_encryption/ec2_ebs_volume_encryption_test.py index 6dda7e7423..568bb6dd24 100644 --- a/tests/providers/aws/services/ec2/ec2_ebs_volume_encryption/ec2_ebs_volume_encryption_test.py +++ b/tests/providers/aws/services/ec2/ec2_ebs_volume_encryption/ec2_ebs_volume_encryption_test.py @@ -19,15 +19,12 @@ class Test_ec2_ebs_volume_encryption: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_volume_encryption.ec2_ebs_volume_encryption.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_volume_encryption.ec2_ebs_volume_encryption.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_volume_encryption.ec2_ebs_volume_encryption import ( @@ -51,15 +48,12 @@ class Test_ec2_ebs_volume_encryption: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_volume_encryption.ec2_ebs_volume_encryption.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_volume_encryption.ec2_ebs_volume_encryption.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_volume_encryption.ec2_ebs_volume_encryption import ( @@ -97,15 +91,12 @@ class Test_ec2_ebs_volume_encryption: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_volume_encryption.ec2_ebs_volume_encryption.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_volume_encryption.ec2_ebs_volume_encryption.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_volume_encryption.ec2_ebs_volume_encryption import ( diff --git a/tests/providers/aws/services/ec2/ec2_ebs_volume_protected_by_backup_plan/ec2_ebs_volume_protected_by_backup_plan_test.py b/tests/providers/aws/services/ec2/ec2_ebs_volume_protected_by_backup_plan/ec2_ebs_volume_protected_by_backup_plan_test.py index 391b6233cd..36e9259fa6 100644 --- a/tests/providers/aws/services/ec2/ec2_ebs_volume_protected_by_backup_plan/ec2_ebs_volume_protected_by_backup_plan_test.py +++ b/tests/providers/aws/services/ec2/ec2_ebs_volume_protected_by_backup_plan/ec2_ebs_volume_protected_by_backup_plan_test.py @@ -22,15 +22,12 @@ class Test_ec2_ebs_volume_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.backup_client", - new=Backup(aws_provider), - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.backup_client", + new=Backup(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan import ( @@ -58,15 +55,12 @@ class Test_ec2_ebs_volume_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.backup_client", - new=Backup(aws_provider), - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.backup_client", + new=Backup(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan import ( @@ -123,15 +117,12 @@ class Test_ec2_ebs_volume_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.backup_client", - new=Backup(aws_provider), - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.backup_client", + new=Backup(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan import ( @@ -180,23 +171,18 @@ class Test_ec2_ebs_volume_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.ec2_client", - new=ec2_client, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_client.ec2_client", - new=ec2_client, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.backup_client", - new=backup_client, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_client.backup_client", - new=backup_client, - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.ec2_client", + new=ec2_client, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_client.ec2_client", + new=ec2_client, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.backup_client", + new=backup_client, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_client.backup_client", + new=backup_client, ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan import ( @@ -244,23 +230,18 @@ class Test_ec2_ebs_volume_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.ec2_client", - new=ec2_client, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_client.ec2_client", - new=ec2_client, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.backup_client", - new=backup_client, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_client.backup_client", - new=backup_client, - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.ec2_client", + new=ec2_client, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_client.ec2_client", + new=ec2_client, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.backup_client", + new=backup_client, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_client.backup_client", + new=backup_client, ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan import ( @@ -307,23 +288,18 @@ class Test_ec2_ebs_volume_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.ec2_client", - new=ec2_client, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_client.ec2_client", - new=ec2_client, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.backup_client", - new=backup_client, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_client.backup_client", - new=backup_client, - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.ec2_client", + new=ec2_client, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_client.ec2_client", + new=ec2_client, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan.backup_client", + new=backup_client, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_client.backup_client", + new=backup_client, ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_volume_protected_by_backup_plan.ec2_ebs_volume_protected_by_backup_plan import ( diff --git a/tests/providers/aws/services/ec2/ec2_ebs_volume_snapshots_exists/ec2_ebs_volume_snapshots_exists_test.py b/tests/providers/aws/services/ec2/ec2_ebs_volume_snapshots_exists/ec2_ebs_volume_snapshots_exists_test.py index 3c10fb93bd..a90924e047 100644 --- a/tests/providers/aws/services/ec2/ec2_ebs_volume_snapshots_exists/ec2_ebs_volume_snapshots_exists_test.py +++ b/tests/providers/aws/services/ec2/ec2_ebs_volume_snapshots_exists/ec2_ebs_volume_snapshots_exists_test.py @@ -34,15 +34,12 @@ class Test_ec2_ebs_volume_snapshots_exists: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_volume_snapshots_exists.ec2_ebs_volume_snapshots_exists.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_volume_snapshots_exists.ec2_ebs_volume_snapshots_exists.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_volume_snapshots_exists.ec2_ebs_volume_snapshots_exists import ( @@ -65,15 +62,12 @@ class Test_ec2_ebs_volume_snapshots_exists: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_volume_snapshots_exists.ec2_ebs_volume_snapshots_exists.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_volume_snapshots_exists.ec2_ebs_volume_snapshots_exists.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_volume_snapshots_exists.ec2_ebs_volume_snapshots_exists import ( @@ -108,15 +102,12 @@ class Test_ec2_ebs_volume_snapshots_exists: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_volume_snapshots_exists.ec2_ebs_volume_snapshots_exists.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_volume_snapshots_exists.ec2_ebs_volume_snapshots_exists.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_volume_snapshots_exists.ec2_ebs_volume_snapshots_exists import ( @@ -155,15 +146,12 @@ class Test_ec2_ebs_volume_snapshots_exists: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_ebs_volume_snapshots_exists.ec2_ebs_volume_snapshots_exists.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_ebs_volume_snapshots_exists.ec2_ebs_volume_snapshots_exists.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_ebs_volume_snapshots_exists.ec2_ebs_volume_snapshots_exists import ( diff --git a/tests/providers/aws/services/ec2/ec2_elastic_ip_shodan/ec2_elastic_ip_shodan_test.py b/tests/providers/aws/services/ec2/ec2_elastic_ip_shodan/ec2_elastic_ip_shodan_test.py index 535d7a3027..2440123f78 100644 --- a/tests/providers/aws/services/ec2/ec2_elastic_ip_shodan/ec2_elastic_ip_shodan_test.py +++ b/tests/providers/aws/services/ec2/ec2_elastic_ip_shodan/ec2_elastic_ip_shodan_test.py @@ -28,15 +28,12 @@ class Test_ec2_elastic_ip_shodan: audit_config={"shodan_api_key": ""}, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_elastic_ip_shodan.ec2_elastic_ip_shodan.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_elastic_ip_shodan.ec2_elastic_ip_shodan.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_elastic_ip_shodan.ec2_elastic_ip_shodan import ( @@ -62,15 +59,12 @@ class Test_ec2_elastic_ip_shodan: audit_config={"shodan_api_key": ""}, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_elastic_ip_shodan.ec2_elastic_ip_shodan.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_elastic_ip_shodan.ec2_elastic_ip_shodan.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_elastic_ip_shodan.ec2_elastic_ip_shodan import ( @@ -103,15 +97,12 @@ class Test_ec2_elastic_ip_shodan: audit_config={"shodan_api_key": ""}, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_elastic_ip_shodan.ec2_elastic_ip_shodan.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_elastic_ip_shodan.ec2_elastic_ip_shodan.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_elastic_ip_shodan.ec2_elastic_ip_shodan import ( @@ -151,19 +142,15 @@ class Test_ec2_elastic_ip_shodan: isp = "test-isp" country = "france" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_elastic_ip_shodan.ec2_elastic_ip_shodan.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_elastic_ip_shodan.ec2_elastic_ip_shodan.shodan.Shodan.host", - return_value={"ports": ports, "isp": isp, "country_name": country}, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_elastic_ip_shodan.ec2_elastic_ip_shodan.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_elastic_ip_shodan.ec2_elastic_ip_shodan.shodan.Shodan.host", + return_value={"ports": ports, "isp": isp, "country_name": country}, ): # Test Check from prowler.providers.aws.services.ec2.ec2_elastic_ip_shodan.ec2_elastic_ip_shodan import ( diff --git a/tests/providers/aws/services/ec2/ec2_elastic_ip_unassigned/ec2_elastic_ip_unassigned_test.py b/tests/providers/aws/services/ec2/ec2_elastic_ip_unassigned/ec2_elastic_ip_unassigned_test.py index 8117ee659d..b3fdbfd199 100644 --- a/tests/providers/aws/services/ec2/ec2_elastic_ip_unassigned/ec2_elastic_ip_unassigned_test.py +++ b/tests/providers/aws/services/ec2/ec2_elastic_ip_unassigned/ec2_elastic_ip_unassigned_test.py @@ -22,15 +22,12 @@ class Test_ec2_elastic_ip_unassigned: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_elastic_ip_unassigned.ec2_elastic_ip_unassigned.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_elastic_ip_unassigned.ec2_elastic_ip_unassigned.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_elastic_ip_unassigned.ec2_elastic_ip_unassigned import ( @@ -56,15 +53,12 @@ class Test_ec2_elastic_ip_unassigned: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_elastic_ip_unassigned.ec2_elastic_ip_unassigned.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_elastic_ip_unassigned.ec2_elastic_ip_unassigned.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_elastic_ip_unassigned.ec2_elastic_ip_unassigned import ( @@ -112,15 +106,12 @@ class Test_ec2_elastic_ip_unassigned: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_elastic_ip_unassigned.ec2_elastic_ip_unassigned.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_elastic_ip_unassigned.ec2_elastic_ip_unassigned.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_elastic_ip_unassigned.ec2_elastic_ip_unassigned import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_account_imdsv2_enabled/ec2_instance_account_imdsv2_enabled_fixer_test.py b/tests/providers/aws/services/ec2/ec2_instance_account_imdsv2_enabled/ec2_instance_account_imdsv2_enabled_fixer_test.py index 76fdd75c31..b8f0835fbf 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_account_imdsv2_enabled/ec2_instance_account_imdsv2_enabled_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_account_imdsv2_enabled/ec2_instance_account_imdsv2_enabled_fixer_test.py @@ -52,15 +52,12 @@ class Test_ec2_instance_account_imdsv2_enabled_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_account_imdsv2_enabled.ec2_instance_account_imdsv2_enabled_fixer.ec2_client", - ec2_service, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_account_imdsv2_enabled.ec2_instance_account_imdsv2_enabled_fixer.ec2_client", + ec2_service, ): from prowler.providers.aws.services.ec2.ec2_instance_account_imdsv2_enabled.ec2_instance_account_imdsv2_enabled_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_account_imdsv2_enabled/ec2_instance_account_imdsv2_enabled_test.py b/tests/providers/aws/services/ec2/ec2_instance_account_imdsv2_enabled/ec2_instance_account_imdsv2_enabled_test.py index 8a1f323409..3707476472 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_account_imdsv2_enabled/ec2_instance_account_imdsv2_enabled_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_account_imdsv2_enabled/ec2_instance_account_imdsv2_enabled_test.py @@ -28,15 +28,12 @@ class Test_ec2_instance_account_imdsv2_enabled: f"arn:aws:ec2:{AWS_REGION_US_EAST_1}:{AWS_ACCOUNT_NUMBER}:account" ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_account_imdsv2_enabled.ec2_instance_account_imdsv2_enabled.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_account_imdsv2_enabled.ec2_instance_account_imdsv2_enabled.ec2_client", + new=ec2_client, ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_account_imdsv2_enabled.ec2_instance_account_imdsv2_enabled import ( @@ -77,15 +74,12 @@ class Test_ec2_instance_account_imdsv2_enabled: f"arn:aws:ec2:{AWS_REGION_US_EAST_1}:{AWS_ACCOUNT_NUMBER}:account" ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_account_imdsv2_enabled.ec2_instance_account_imdsv2_enabled.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_account_imdsv2_enabled.ec2_instance_account_imdsv2_enabled.ec2_client", + new=ec2_client, ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_account_imdsv2_enabled.ec2_instance_account_imdsv2_enabled import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_detailed_monitoring_enabled/ec2_instance_detailed_monitoring_enabled_test.py b/tests/providers/aws/services/ec2/ec2_instance_detailed_monitoring_enabled/ec2_instance_detailed_monitoring_enabled_test.py index ad310250cb..2baae48d3f 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_detailed_monitoring_enabled/ec2_instance_detailed_monitoring_enabled_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_detailed_monitoring_enabled/ec2_instance_detailed_monitoring_enabled_test.py @@ -21,15 +21,12 @@ class Test_ec2_instance_detailed_monitoring_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_detailed_monitoring_enabled.ec2_instance_detailed_monitoring_enabled.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_detailed_monitoring_enabled.ec2_instance_detailed_monitoring_enabled.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_detailed_monitoring_enabled.ec2_instance_detailed_monitoring_enabled import ( @@ -57,15 +54,12 @@ class Test_ec2_instance_detailed_monitoring_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_detailed_monitoring_enabled.ec2_instance_detailed_monitoring_enabled.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_detailed_monitoring_enabled.ec2_instance_detailed_monitoring_enabled.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_detailed_monitoring_enabled.ec2_instance_detailed_monitoring_enabled import ( ec2_instance_detailed_monitoring_enabled, @@ -105,16 +99,13 @@ class Test_ec2_instance_detailed_monitoring_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_detailed_monitoring_enabled.ec2_instance_detailed_monitoring_enabled.ec2_client", - new=EC2(aws_provider), - ) as ec2_service, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_detailed_monitoring_enabled.ec2_instance_detailed_monitoring_enabled.ec2_client", + new=EC2(aws_provider), + ) as ec2_service: from prowler.providers.aws.services.ec2.ec2_instance_detailed_monitoring_enabled.ec2_instance_detailed_monitoring_enabled import ( ec2_instance_detailed_monitoring_enabled, ) diff --git a/tests/providers/aws/services/ec2/ec2_instance_imdsv2_enabled/ec2_instance_imdsv2_enabled_test.py b/tests/providers/aws/services/ec2/ec2_instance_imdsv2_enabled/ec2_instance_imdsv2_enabled_test.py index 7fe2d06fed..b431df0fda 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_imdsv2_enabled/ec2_instance_imdsv2_enabled_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_imdsv2_enabled/ec2_instance_imdsv2_enabled_test.py @@ -22,15 +22,12 @@ class Test_ec2_instance_imdsv2_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_imdsv2_enabled.ec2_instance_imdsv2_enabled.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_imdsv2_enabled.ec2_instance_imdsv2_enabled.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_imdsv2_enabled.ec2_instance_imdsv2_enabled import ( @@ -61,16 +58,13 @@ class Test_ec2_instance_imdsv2_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_imdsv2_enabled.ec2_instance_imdsv2_enabled.ec2_client", - new=EC2(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_imdsv2_enabled.ec2_instance_imdsv2_enabled.ec2_client", + new=EC2(aws_provider), + ) as service_client: from prowler.providers.aws.services.ec2.ec2_instance_imdsv2_enabled.ec2_instance_imdsv2_enabled import ( ec2_instance_imdsv2_enabled, ) @@ -115,16 +109,13 @@ class Test_ec2_instance_imdsv2_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_imdsv2_enabled.ec2_instance_imdsv2_enabled.ec2_client", - new=EC2(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_imdsv2_enabled.ec2_instance_imdsv2_enabled.ec2_client", + new=EC2(aws_provider), + ) as service_client: from prowler.providers.aws.services.ec2.ec2_instance_imdsv2_enabled.ec2_instance_imdsv2_enabled import ( ec2_instance_imdsv2_enabled, ) @@ -169,16 +160,13 @@ class Test_ec2_instance_imdsv2_enabled: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_imdsv2_enabled.ec2_instance_imdsv2_enabled.ec2_client", - new=EC2(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_imdsv2_enabled.ec2_instance_imdsv2_enabled.ec2_client", + new=EC2(aws_provider), + ) as service_client: from prowler.providers.aws.services.ec2.ec2_instance_imdsv2_enabled.ec2_instance_imdsv2_enabled import ( ec2_instance_imdsv2_enabled, ) diff --git a/tests/providers/aws/services/ec2/ec2_instance_internet_facing_with_instance_profile/ec2_instance_internet_facing_with_instance_profile_test.py b/tests/providers/aws/services/ec2/ec2_instance_internet_facing_with_instance_profile/ec2_instance_internet_facing_with_instance_profile_test.py index 40fab2dc31..15b1209782 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_internet_facing_with_instance_profile/ec2_instance_internet_facing_with_instance_profile_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_internet_facing_with_instance_profile/ec2_instance_internet_facing_with_instance_profile_test.py @@ -22,15 +22,12 @@ class Test_ec2_instance_internet_facing_with_instance_profile: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_internet_facing_with_instance_profile.ec2_instance_internet_facing_with_instance_profile.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_internet_facing_with_instance_profile.ec2_instance_internet_facing_with_instance_profile.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_internet_facing_with_instance_profile.ec2_instance_internet_facing_with_instance_profile import ( @@ -72,15 +69,12 @@ class Test_ec2_instance_internet_facing_with_instance_profile: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_internet_facing_with_instance_profile.ec2_instance_internet_facing_with_instance_profile.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_internet_facing_with_instance_profile.ec2_instance_internet_facing_with_instance_profile.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_internet_facing_with_instance_profile.ec2_instance_internet_facing_with_instance_profile import ( ec2_instance_internet_facing_with_instance_profile, @@ -132,15 +126,12 @@ class Test_ec2_instance_internet_facing_with_instance_profile: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_internet_facing_with_instance_profile.ec2_instance_internet_facing_with_instance_profile.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_internet_facing_with_instance_profile.ec2_instance_internet_facing_with_instance_profile.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_internet_facing_with_instance_profile.ec2_instance_internet_facing_with_instance_profile import ( ec2_instance_internet_facing_with_instance_profile, diff --git a/tests/providers/aws/services/ec2/ec2_instance_managed_by_ssm/ec2_instance_managed_by_ssm_test.py b/tests/providers/aws/services/ec2/ec2_instance_managed_by_ssm/ec2_instance_managed_by_ssm_test.py index 6e84fc8239..6325d064d7 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_managed_by_ssm/ec2_instance_managed_by_ssm_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_managed_by_ssm/ec2_instance_managed_by_ssm_test.py @@ -26,23 +26,18 @@ class Test_ec2_instance_managed_by_ssm_test: ssm_client = mock.MagicMock ssm_client.managed_instances = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ssm.ssm_service.SSM", - new=ssm_client, - ), - mock.patch( - "prowler.providers.aws.services.ssm.ssm_client.ssm_client", - new=ssm_client, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_managed_by_ssm.ec2_instance_managed_by_ssm.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ssm.ssm_service.SSM", + new=ssm_client, + ), mock.patch( + "prowler.providers.aws.services.ssm.ssm_client.ssm_client", + new=ssm_client, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_managed_by_ssm.ec2_instance_managed_by_ssm.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_managed_by_ssm.ec2_instance_managed_by_ssm import ( @@ -76,23 +71,18 @@ class Test_ec2_instance_managed_by_ssm_test: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ssm.ssm_service.SSM", - new=ssm_client, - ), - mock.patch( - "prowler.providers.aws.services.ssm.ssm_client.ssm_client", - new=ssm_client, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_managed_by_ssm.ec2_instance_managed_by_ssm.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ssm.ssm_service.SSM", + new=ssm_client, + ), mock.patch( + "prowler.providers.aws.services.ssm.ssm_client.ssm_client", + new=ssm_client, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_managed_by_ssm.ec2_instance_managed_by_ssm.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_managed_by_ssm.ec2_instance_managed_by_ssm import ( @@ -137,23 +127,18 @@ class Test_ec2_instance_managed_by_ssm_test: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ssm.ssm_service.SSM", - new=ssm_client, - ), - mock.patch( - "prowler.providers.aws.services.ssm.ssm_client.ssm_client", - new=ssm_client, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_managed_by_ssm.ec2_instance_managed_by_ssm.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ssm.ssm_service.SSM", + new=ssm_client, + ), mock.patch( + "prowler.providers.aws.services.ssm.ssm_client.ssm_client", + new=ssm_client, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_managed_by_ssm.ec2_instance_managed_by_ssm.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_managed_by_ssm.ec2_instance_managed_by_ssm import ( @@ -202,23 +187,18 @@ class Test_ec2_instance_managed_by_ssm_test: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ssm.ssm_service.SSM", - new=ssm_client, - ), - mock.patch( - "prowler.providers.aws.services.ssm.ssm_client.ssm_client", - new=ssm_client, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_managed_by_ssm.ec2_instance_managed_by_ssm.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ssm.ssm_service.SSM", + new=ssm_client, + ), mock.patch( + "prowler.providers.aws.services.ssm.ssm_client.ssm_client", + new=ssm_client, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_managed_by_ssm.ec2_instance_managed_by_ssm.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_managed_by_ssm.ec2_instance_managed_by_ssm import ( @@ -270,23 +250,18 @@ class Test_ec2_instance_managed_by_ssm_test: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ssm.ssm_service.SSM", - new=ssm_client, - ), - mock.patch( - "prowler.providers.aws.services.ssm.ssm_client.ssm_client", - new=ssm_client, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_managed_by_ssm.ec2_instance_managed_by_ssm.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ssm.ssm_service.SSM", + new=ssm_client, + ), mock.patch( + "prowler.providers.aws.services.ssm.ssm_client.ssm_client", + new=ssm_client, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_managed_by_ssm.ec2_instance_managed_by_ssm.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_managed_by_ssm.ec2_instance_managed_by_ssm import ( @@ -327,23 +302,18 @@ class Test_ec2_instance_managed_by_ssm_test: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ssm.ssm_service.SSM", - new=ssm_client, - ), - mock.patch( - "prowler.providers.aws.services.ssm.ssm_client.ssm_client", - new=ssm_client, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_managed_by_ssm.ec2_instance_managed_by_ssm.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ssm.ssm_service.SSM", + new=ssm_client, + ), mock.patch( + "prowler.providers.aws.services.ssm.ssm_client.ssm_client", + new=ssm_client, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_managed_by_ssm.ec2_instance_managed_by_ssm.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_managed_by_ssm.ec2_instance_managed_by_ssm import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_older_than_specific_days/ec2_instance_older_than_specific_days_test.py b/tests/providers/aws/services/ec2/ec2_instance_older_than_specific_days/ec2_instance_older_than_specific_days_test.py index 78bee43a6f..1af5aaafcd 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_older_than_specific_days/ec2_instance_older_than_specific_days_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_older_than_specific_days/ec2_instance_older_than_specific_days_test.py @@ -25,15 +25,12 @@ class Test_ec2_instance_older_than_specific_days: ) aws_provider._audit_config = {"max_ec2_instance_age_in_days": 180} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_older_than_specific_days.ec2_instance_older_than_specific_days.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_older_than_specific_days.ec2_instance_older_than_specific_days.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_older_than_specific_days.ec2_instance_older_than_specific_days import ( @@ -62,15 +59,12 @@ class Test_ec2_instance_older_than_specific_days: ) aws_provider._audit_config = {"max_ec2_instance_age_in_days": 180} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_older_than_specific_days.ec2_instance_older_than_specific_days.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_older_than_specific_days.ec2_instance_older_than_specific_days.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_older_than_specific_days.ec2_instance_older_than_specific_days import ( ec2_instance_older_than_specific_days, @@ -109,16 +103,13 @@ class Test_ec2_instance_older_than_specific_days: ) aws_provider._audit_config = {"max_ec2_instance_age_in_days": 180} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_older_than_specific_days.ec2_instance_older_than_specific_days.ec2_client", - new=EC2(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_older_than_specific_days.ec2_instance_older_than_specific_days.ec2_client", + new=EC2(aws_provider), + ) as service_client: from prowler.providers.aws.services.ec2.ec2_instance_older_than_specific_days.ec2_instance_older_than_specific_days import ( ec2_instance_older_than_specific_days, ) diff --git a/tests/providers/aws/services/ec2/ec2_instance_paravirtual_type/ec2_instance_paravirtual_type_test.py b/tests/providers/aws/services/ec2/ec2_instance_paravirtual_type/ec2_instance_paravirtual_type_test.py index dcdfb76601..0d71c1ebe8 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_paravirtual_type/ec2_instance_paravirtual_type_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_paravirtual_type/ec2_instance_paravirtual_type_test.py @@ -22,15 +22,12 @@ class Test_ec2_instance_paravirtual_type: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_paravirtual_type.ec2_instance_paravirtual_type.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_paravirtual_type.ec2_instance_paravirtual_type.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_paravirtual_type.ec2_instance_paravirtual_type import ( @@ -66,15 +63,12 @@ class Test_ec2_instance_paravirtual_type: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_paravirtual_type.ec2_instance_paravirtual_type.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_paravirtual_type.ec2_instance_paravirtual_type.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_paravirtual_type.ec2_instance_paravirtual_type import ( ec2_instance_paravirtual_type, @@ -118,15 +112,12 @@ class Test_ec2_instance_paravirtual_type: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_paravirtual_type.ec2_instance_paravirtual_type.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_paravirtual_type.ec2_instance_paravirtual_type.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_paravirtual_type.ec2_instance_paravirtual_type import ( ec2_instance_paravirtual_type, diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_cassandra_exposed_to_internet/ec2_instance_port_cassandra_exposed_to_internet_fixer_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_cassandra_exposed_to_internet/ec2_instance_port_cassandra_exposed_to_internet_fixer_test.py index 78f1e289d8..2e0c54e06d 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_cassandra_exposed_to_internet/ec2_instance_port_cassandra_exposed_to_internet_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_cassandra_exposed_to_internet/ec2_instance_port_cassandra_exposed_to_internet_fixer_test.py @@ -73,15 +73,12 @@ class Test_ec2_instance_port_cassandra_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet_fixer import ( @@ -156,15 +153,12 @@ class Test_ec2_instance_port_cassandra_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet_fixer import ( @@ -213,15 +207,12 @@ class Test_ec2_instance_port_cassandra_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet_fixer import ( @@ -270,15 +261,12 @@ class Test_ec2_instance_port_cassandra_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet_fixer import ( @@ -333,15 +321,12 @@ class Test_ec2_instance_port_cassandra_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_cassandra_exposed_to_internet/ec2_instance_port_cassandra_exposed_to_internet_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_cassandra_exposed_to_internet/ec2_instance_port_cassandra_exposed_to_internet_test.py index 9028bb89a8..d9ee15e932 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_cassandra_exposed_to_internet/ec2_instance_port_cassandra_exposed_to_internet_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_cassandra_exposed_to_internet/ec2_instance_port_cassandra_exposed_to_internet_test.py @@ -24,19 +24,15 @@ class Test_ec2_instance_port_cassandra_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet import ( @@ -91,19 +87,15 @@ class Test_ec2_instance_port_cassandra_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet import ( @@ -171,19 +163,15 @@ class Test_ec2_instance_port_cassandra_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet import ( @@ -257,19 +245,15 @@ class Test_ec2_instance_port_cassandra_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet import ( @@ -363,19 +347,15 @@ class Test_ec2_instance_port_cassandra_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_cassandra_exposed_to_internet.ec2_instance_port_cassandra_exposed_to_internet import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_cifs_exposed_to_internet/ec2_instance_port_cifs_exposed_to_internet_fixer_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_cifs_exposed_to_internet/ec2_instance_port_cifs_exposed_to_internet_fixer_test.py index 6cb8f5d598..ef7771637e 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_cifs_exposed_to_internet/ec2_instance_port_cifs_exposed_to_internet_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_cifs_exposed_to_internet/ec2_instance_port_cifs_exposed_to_internet_fixer_test.py @@ -73,15 +73,12 @@ class Test_ec2_instance_port_cifs_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet_fixer import ( @@ -156,15 +153,12 @@ class Test_ec2_instance_port_cifs_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet_fixer import ( @@ -219,15 +213,12 @@ class Test_ec2_instance_port_cifs_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet_fixer import ( @@ -282,15 +273,12 @@ class Test_ec2_instance_port_cifs_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet_fixer import ( @@ -345,15 +333,12 @@ class Test_ec2_instance_port_cifs_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet_fixer import ( @@ -430,15 +415,12 @@ class Test_ec2_instance_port_cifs_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_cifs_exposed_to_internet/ec2_instance_port_cifs_exposed_to_internet_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_cifs_exposed_to_internet/ec2_instance_port_cifs_exposed_to_internet_test.py index 2dda778fc4..0e6a838227 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_cifs_exposed_to_internet/ec2_instance_port_cifs_exposed_to_internet_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_cifs_exposed_to_internet/ec2_instance_port_cifs_exposed_to_internet_test.py @@ -24,19 +24,15 @@ class Test_ec2_instance_port_cifs_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet import ( @@ -91,19 +87,15 @@ class Test_ec2_instance_port_cifs_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet import ( @@ -171,19 +163,15 @@ class Test_ec2_instance_port_cifs_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet import ( @@ -257,19 +245,15 @@ class Test_ec2_instance_port_cifs_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet import ( @@ -363,19 +347,15 @@ class Test_ec2_instance_port_cifs_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_cifs_exposed_to_internet.ec2_instance_port_cifs_exposed_to_internet import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_elasticsearch_kibana_exposed_to_internet/ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_elasticsearch_kibana_exposed_to_internet/ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer_test.py index b05a0fe9b8..044f2def25 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_elasticsearch_kibana_exposed_to_internet/ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_elasticsearch_kibana_exposed_to_internet/ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer_test.py @@ -73,15 +73,12 @@ class Test_ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer import ( @@ -156,15 +153,12 @@ class Test_ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer import ( @@ -219,15 +213,12 @@ class Test_ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer import ( @@ -282,15 +273,12 @@ class Test_ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer import ( @@ -345,15 +333,12 @@ class Test_ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer import ( @@ -430,15 +415,12 @@ class Test_ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_elasticsearch_kibana_exposed_to_internet/ec2_instance_port_elasticsearch_kibana_exposed_to_internet_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_elasticsearch_kibana_exposed_to_internet/ec2_instance_port_elasticsearch_kibana_exposed_to_internet_test.py index 1290464e9a..5ba9aa593e 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_elasticsearch_kibana_exposed_to_internet/ec2_instance_port_elasticsearch_kibana_exposed_to_internet_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_elasticsearch_kibana_exposed_to_internet/ec2_instance_port_elasticsearch_kibana_exposed_to_internet_test.py @@ -24,19 +24,15 @@ class Test_ec2_instance_port_elasticsearch_kibana_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet import ( @@ -91,19 +87,15 @@ class Test_ec2_instance_port_elasticsearch_kibana_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet import ( @@ -171,19 +163,15 @@ class Test_ec2_instance_port_elasticsearch_kibana_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet import ( @@ -257,19 +245,15 @@ class Test_ec2_instance_port_elasticsearch_kibana_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet import ( @@ -363,19 +347,15 @@ class Test_ec2_instance_port_elasticsearch_kibana_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_elasticsearch_kibana_exposed_to_internet.ec2_instance_port_elasticsearch_kibana_exposed_to_internet import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_ftp_exposed_to_internet/ec2_instance_port_ftp_exposed_to_internet_fixer_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_ftp_exposed_to_internet/ec2_instance_port_ftp_exposed_to_internet_fixer_test.py index 383ef59af1..b3bf476cc5 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_ftp_exposed_to_internet/ec2_instance_port_ftp_exposed_to_internet_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_ftp_exposed_to_internet/ec2_instance_port_ftp_exposed_to_internet_fixer_test.py @@ -73,15 +73,12 @@ class Test_ec2_instance_port_ftp_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet_fixer import ( @@ -156,15 +153,12 @@ class Test_ec2_instance_port_ftp_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet_fixer import ( @@ -219,15 +213,12 @@ class Test_ec2_instance_port_ftp_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet_fixer import ( @@ -282,15 +273,12 @@ class Test_ec2_instance_port_ftp_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet_fixer import ( @@ -345,15 +333,12 @@ class Test_ec2_instance_port_ftp_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_ftp_exposed_to_internet/ec2_instance_port_ftp_exposed_to_internet_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_ftp_exposed_to_internet/ec2_instance_port_ftp_exposed_to_internet_test.py index 29aa587b88..63cf5f2417 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_ftp_exposed_to_internet/ec2_instance_port_ftp_exposed_to_internet_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_ftp_exposed_to_internet/ec2_instance_port_ftp_exposed_to_internet_test.py @@ -24,19 +24,15 @@ class Test_ec2_instance_port_ftp_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet import ( @@ -91,19 +87,15 @@ class Test_ec2_instance_port_ftp_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet import ( @@ -171,19 +163,15 @@ class Test_ec2_instance_port_ftp_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet import ( @@ -257,19 +245,15 @@ class Test_ec2_instance_port_ftp_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet import ( @@ -363,19 +347,15 @@ class Test_ec2_instance_port_ftp_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_ftp_exposed_to_internet.ec2_instance_port_ftp_exposed_to_internet import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_kafka_exposed_to_internet/ec2_instance_port_kafka_exposed_to_internet_fixer_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_kafka_exposed_to_internet/ec2_instance_port_kafka_exposed_to_internet_fixer_test.py index 07500f532e..dfa739ea37 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_kafka_exposed_to_internet/ec2_instance_port_kafka_exposed_to_internet_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_kafka_exposed_to_internet/ec2_instance_port_kafka_exposed_to_internet_fixer_test.py @@ -73,15 +73,12 @@ class Test_ec2_instance_port_kafka_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet_fixer import ( @@ -143,15 +140,12 @@ class Test_ec2_instance_port_kafka_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet_fixer import ( @@ -200,15 +194,12 @@ class Test_ec2_instance_port_kafka_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet_fixer import ( @@ -257,15 +248,12 @@ class Test_ec2_instance_port_kafka_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet_fixer import ( @@ -320,15 +308,12 @@ class Test_ec2_instance_port_kafka_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_kafka_exposed_to_internet/ec2_instance_port_kafka_exposed_to_internet_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_kafka_exposed_to_internet/ec2_instance_port_kafka_exposed_to_internet_test.py index 0e55046b2e..40ea7dea81 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_kafka_exposed_to_internet/ec2_instance_port_kafka_exposed_to_internet_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_kafka_exposed_to_internet/ec2_instance_port_kafka_exposed_to_internet_test.py @@ -24,19 +24,15 @@ class Test_ec2_instance_port_kafka_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet import ( @@ -91,19 +87,15 @@ class Test_ec2_instance_port_kafka_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet import ( @@ -171,19 +163,15 @@ class Test_ec2_instance_port_kafka_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet import ( @@ -257,19 +245,15 @@ class Test_ec2_instance_port_kafka_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet import ( @@ -363,19 +347,15 @@ class Test_ec2_instance_port_kafka_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_kafka_exposed_to_internet.ec2_instance_port_kafka_exposed_to_internet import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_kerberos_exposed_to_internet/ec2_instance_port_kerberos_exposed_to_internet_fixer_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_kerberos_exposed_to_internet/ec2_instance_port_kerberos_exposed_to_internet_fixer_test.py index 09ce71bb02..4b1a14997c 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_kerberos_exposed_to_internet/ec2_instance_port_kerberos_exposed_to_internet_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_kerberos_exposed_to_internet/ec2_instance_port_kerberos_exposed_to_internet_fixer_test.py @@ -73,15 +73,12 @@ class Test_ec2_instance_port_kerberos_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet_fixer import ( @@ -156,15 +153,12 @@ class Test_ec2_instance_port_kerberos_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet_fixer import ( @@ -219,15 +213,12 @@ class Test_ec2_instance_port_kerberos_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet_fixer import ( @@ -282,15 +273,12 @@ class Test_ec2_instance_port_kerberos_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet_fixer import ( @@ -345,15 +333,12 @@ class Test_ec2_instance_port_kerberos_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_kerberos_exposed_to_internet/ec2_instance_port_kerberos_exposed_to_internet_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_kerberos_exposed_to_internet/ec2_instance_port_kerberos_exposed_to_internet_test.py index cce3ca59e0..65ff232f65 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_kerberos_exposed_to_internet/ec2_instance_port_kerberos_exposed_to_internet_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_kerberos_exposed_to_internet/ec2_instance_port_kerberos_exposed_to_internet_test.py @@ -24,19 +24,15 @@ class Test_ec2_instance_port_kerberos_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet import ( @@ -91,19 +87,15 @@ class Test_ec2_instance_port_kerberos_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet import ( @@ -171,19 +163,15 @@ class Test_ec2_instance_port_kerberos_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet import ( @@ -257,19 +245,15 @@ class Test_ec2_instance_port_kerberos_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet import ( @@ -363,19 +347,15 @@ class Test_ec2_instance_port_kerberos_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_kerberos_exposed_to_internet.ec2_instance_port_kerberos_exposed_to_internet import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_ldap_exposed_to_internet/ec2_instance_port_ldap_exposed_to_internet_fixer_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_ldap_exposed_to_internet/ec2_instance_port_ldap_exposed_to_internet_fixer_test.py index a8c748c9e3..9a25c8be2b 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_ldap_exposed_to_internet/ec2_instance_port_ldap_exposed_to_internet_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_ldap_exposed_to_internet/ec2_instance_port_ldap_exposed_to_internet_fixer_test.py @@ -73,15 +73,12 @@ class Test_ec2_instance_port_ldap_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet_fixer import ( @@ -156,15 +153,12 @@ class Test_ec2_instance_port_ldap_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet_fixer import ( @@ -219,15 +213,12 @@ class Test_ec2_instance_port_ldap_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet_fixer import ( @@ -282,15 +273,12 @@ class Test_ec2_instance_port_ldap_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet_fixer import ( @@ -345,15 +333,12 @@ class Test_ec2_instance_port_ldap_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_ldap_exposed_to_internet/ec2_instance_port_ldap_exposed_to_internet_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_ldap_exposed_to_internet/ec2_instance_port_ldap_exposed_to_internet_test.py index 190bed6613..53c7cd05dd 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_ldap_exposed_to_internet/ec2_instance_port_ldap_exposed_to_internet_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_ldap_exposed_to_internet/ec2_instance_port_ldap_exposed_to_internet_test.py @@ -24,19 +24,15 @@ class Test_ec2_instance_port_ldap_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet import ( @@ -91,19 +87,15 @@ class Test_ec2_instance_port_ldap_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet import ( @@ -171,19 +163,15 @@ class Test_ec2_instance_port_ldap_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet import ( @@ -257,19 +245,15 @@ class Test_ec2_instance_port_ldap_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet import ( @@ -363,19 +347,15 @@ class Test_ec2_instance_port_ldap_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_ldap_exposed_to_internet.ec2_instance_port_ldap_exposed_to_internet import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_memcached_exposed_to_internet/ec2_instance_port_memcached_exposed_to_internet_fixer_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_memcached_exposed_to_internet/ec2_instance_port_memcached_exposed_to_internet_fixer_test.py index b60d6af421..f150df762f 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_memcached_exposed_to_internet/ec2_instance_port_memcached_exposed_to_internet_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_memcached_exposed_to_internet/ec2_instance_port_memcached_exposed_to_internet_fixer_test.py @@ -73,15 +73,12 @@ class Test_ec2_instance_port_memcached_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet_fixer import ( @@ -156,15 +153,12 @@ class Test_ec2_instance_port_memcached_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet_fixer import ( @@ -213,15 +207,12 @@ class Test_ec2_instance_port_memcached_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet_fixer import ( @@ -270,15 +261,12 @@ class Test_ec2_instance_port_memcached_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet_fixer import ( @@ -333,15 +321,12 @@ class Test_ec2_instance_port_memcached_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_memcached_exposed_to_internet/ec2_instance_port_memcached_exposed_to_internet_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_memcached_exposed_to_internet/ec2_instance_port_memcached_exposed_to_internet_test.py index d3d978e6ed..59c7d54d49 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_memcached_exposed_to_internet/ec2_instance_port_memcached_exposed_to_internet_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_memcached_exposed_to_internet/ec2_instance_port_memcached_exposed_to_internet_test.py @@ -24,19 +24,15 @@ class Test_ec2_instance_port_memcached_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet import ( @@ -91,19 +87,15 @@ class Test_ec2_instance_port_memcached_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet import ( @@ -171,19 +163,15 @@ class Test_ec2_instance_port_memcached_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet import ( @@ -257,19 +245,15 @@ class Test_ec2_instance_port_memcached_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet import ( @@ -363,19 +347,15 @@ class Test_ec2_instance_port_memcached_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_memcached_exposed_to_internet.ec2_instance_port_memcached_exposed_to_internet import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_mongodb_exposed_to_internet/ec2_instance_port_mongodb_exposed_to_internet_fixer_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_mongodb_exposed_to_internet/ec2_instance_port_mongodb_exposed_to_internet_fixer_test.py index deb684c092..8c305100c8 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_mongodb_exposed_to_internet/ec2_instance_port_mongodb_exposed_to_internet_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_mongodb_exposed_to_internet/ec2_instance_port_mongodb_exposed_to_internet_fixer_test.py @@ -73,15 +73,12 @@ class Test_ec2_instance_port_mongodb_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet_fixer import ( @@ -156,15 +153,12 @@ class Test_ec2_instance_port_mongodb_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet_fixer import ( @@ -219,15 +213,12 @@ class Test_ec2_instance_port_mongodb_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet_fixer import ( @@ -282,15 +273,12 @@ class Test_ec2_instance_port_mongodb_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet_fixer import ( @@ -345,15 +333,12 @@ class Test_ec2_instance_port_mongodb_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet_fixer import ( @@ -430,15 +415,12 @@ class Test_ec2_instance_port_mongodb_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_mongodb_exposed_to_internet/ec2_instance_port_mongodb_exposed_to_internet_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_mongodb_exposed_to_internet/ec2_instance_port_mongodb_exposed_to_internet_test.py index 1f95786c63..8a825c1548 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_mongodb_exposed_to_internet/ec2_instance_port_mongodb_exposed_to_internet_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_mongodb_exposed_to_internet/ec2_instance_port_mongodb_exposed_to_internet_test.py @@ -24,19 +24,15 @@ class Test_ec2_instance_port_mongodb_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet import ( @@ -91,19 +87,15 @@ class Test_ec2_instance_port_mongodb_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet import ( @@ -171,19 +163,15 @@ class Test_ec2_instance_port_mongodb_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet import ( @@ -257,19 +245,15 @@ class Test_ec2_instance_port_mongodb_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet import ( @@ -363,19 +347,15 @@ class Test_ec2_instance_port_mongodb_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_mongodb_exposed_to_internet.ec2_instance_port_mongodb_exposed_to_internet import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_mysql_exposed_to_internet/ec2_instance_port_mysql_exposed_to_internet_fixer_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_mysql_exposed_to_internet/ec2_instance_port_mysql_exposed_to_internet_fixer_test.py index aba595c0e9..195f56737f 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_mysql_exposed_to_internet/ec2_instance_port_mysql_exposed_to_internet_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_mysql_exposed_to_internet/ec2_instance_port_mysql_exposed_to_internet_fixer_test.py @@ -73,15 +73,12 @@ class Test_ec2_instance_port_mysql_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet_fixer import ( @@ -156,15 +153,12 @@ class Test_ec2_instance_port_mysql_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet_fixer import ( @@ -213,15 +207,12 @@ class Test_ec2_instance_port_mysql_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet_fixer import ( @@ -270,15 +261,12 @@ class Test_ec2_instance_port_mysql_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet_fixer import ( @@ -333,15 +321,12 @@ class Test_ec2_instance_port_mysql_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_mysql_exposed_to_internet/ec2_instance_port_mysql_exposed_to_internet_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_mysql_exposed_to_internet/ec2_instance_port_mysql_exposed_to_internet_test.py index af135eabc7..61c400df22 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_mysql_exposed_to_internet/ec2_instance_port_mysql_exposed_to_internet_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_mysql_exposed_to_internet/ec2_instance_port_mysql_exposed_to_internet_test.py @@ -24,19 +24,15 @@ class Test_ec2_instance_port_mysql_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet import ( @@ -91,19 +87,15 @@ class Test_ec2_instance_port_mysql_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet import ( @@ -171,19 +163,15 @@ class Test_ec2_instance_port_mysql_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet import ( @@ -257,19 +245,15 @@ class Test_ec2_instance_port_mysql_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet import ( @@ -363,19 +347,15 @@ class Test_ec2_instance_port_mysql_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_mysql_exposed_to_internet.ec2_instance_port_mysql_exposed_to_internet import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_oracle_exposed_to_internet/ec2_instance_port_oracle_exposed_to_internet_fixer_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_oracle_exposed_to_internet/ec2_instance_port_oracle_exposed_to_internet_fixer_test.py index 0510d1ed74..98de63431e 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_oracle_exposed_to_internet/ec2_instance_port_oracle_exposed_to_internet_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_oracle_exposed_to_internet/ec2_instance_port_oracle_exposed_to_internet_fixer_test.py @@ -73,15 +73,12 @@ class Test_ec2_instance_port_oracle_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet_fixer import ( @@ -156,15 +153,12 @@ class Test_ec2_instance_port_oracle_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet_fixer import ( @@ -219,15 +213,12 @@ class Test_ec2_instance_port_oracle_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet_fixer import ( @@ -282,15 +273,12 @@ class Test_ec2_instance_port_oracle_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet_fixer import ( @@ -345,15 +333,12 @@ class Test_ec2_instance_port_oracle_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_oracle_exposed_to_internet/ec2_instance_port_oracle_exposed_to_internet_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_oracle_exposed_to_internet/ec2_instance_port_oracle_exposed_to_internet_test.py index f8696fb87c..b4397077a5 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_oracle_exposed_to_internet/ec2_instance_port_oracle_exposed_to_internet_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_oracle_exposed_to_internet/ec2_instance_port_oracle_exposed_to_internet_test.py @@ -24,19 +24,15 @@ class Test_ec2_instance_port_oracle_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet import ( @@ -91,19 +87,15 @@ class Test_ec2_instance_port_oracle_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet import ( @@ -171,19 +163,15 @@ class Test_ec2_instance_port_oracle_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet import ( @@ -257,19 +245,15 @@ class Test_ec2_instance_port_oracle_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet import ( @@ -363,19 +347,15 @@ class Test_ec2_instance_port_oracle_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_oracle_exposed_to_internet.ec2_instance_port_oracle_exposed_to_internet import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_postgresql_exposed_to_internet/ec2_instance_port_postgresql_exposed_to_internet_fixer_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_postgresql_exposed_to_internet/ec2_instance_port_postgresql_exposed_to_internet_fixer_test.py index b61b5fab60..38b64208f7 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_postgresql_exposed_to_internet/ec2_instance_port_postgresql_exposed_to_internet_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_postgresql_exposed_to_internet/ec2_instance_port_postgresql_exposed_to_internet_fixer_test.py @@ -73,15 +73,12 @@ class Test_ec2_instance_port_postgresql_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet_fixer import ( @@ -156,15 +153,12 @@ class Test_ec2_instance_port_postgresql_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet_fixer import ( @@ -213,15 +207,12 @@ class Test_ec2_instance_port_postgresql_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet_fixer import ( @@ -270,15 +261,12 @@ class Test_ec2_instance_port_postgresql_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet_fixer import ( @@ -333,15 +321,12 @@ class Test_ec2_instance_port_postgresql_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_postgresql_exposed_to_internet/ec2_instance_port_postgresql_exposed_to_internet_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_postgresql_exposed_to_internet/ec2_instance_port_postgresql_exposed_to_internet_test.py index 64268f0892..20c4b7eb18 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_postgresql_exposed_to_internet/ec2_instance_port_postgresql_exposed_to_internet_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_postgresql_exposed_to_internet/ec2_instance_port_postgresql_exposed_to_internet_test.py @@ -24,19 +24,15 @@ class Test_ec2_instance_port_postgresql_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet import ( @@ -91,19 +87,15 @@ class Test_ec2_instance_port_postgresql_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet import ( @@ -171,19 +163,15 @@ class Test_ec2_instance_port_postgresql_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet import ( @@ -257,19 +245,15 @@ class Test_ec2_instance_port_postgresql_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet import ( @@ -363,19 +347,15 @@ class Test_ec2_instance_port_postgresql_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_postgresql_exposed_to_internet.ec2_instance_port_postgresql_exposed_to_internet import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_rdp_exposed_to_internet/ec2_instance_port_rdp_exposed_to_internet_fixer_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_rdp_exposed_to_internet/ec2_instance_port_rdp_exposed_to_internet_fixer_test.py index fd4945f37a..67581a5723 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_rdp_exposed_to_internet/ec2_instance_port_rdp_exposed_to_internet_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_rdp_exposed_to_internet/ec2_instance_port_rdp_exposed_to_internet_fixer_test.py @@ -73,15 +73,12 @@ class Test_ec2_instance_port_rdp_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet_fixer import ( @@ -156,15 +153,12 @@ class Test_ec2_instance_port_rdp_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet_fixer import ( @@ -213,15 +207,12 @@ class Test_ec2_instance_port_rdp_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet_fixer import ( @@ -270,15 +261,12 @@ class Test_ec2_instance_port_rdp_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet_fixer import ( @@ -333,15 +321,12 @@ class Test_ec2_instance_port_rdp_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_rdp_exposed_to_internet/ec2_instance_port_rdp_exposed_to_internet_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_rdp_exposed_to_internet/ec2_instance_port_rdp_exposed_to_internet_test.py index ec036d319c..fcf9521e2e 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_rdp_exposed_to_internet/ec2_instance_port_rdp_exposed_to_internet_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_rdp_exposed_to_internet/ec2_instance_port_rdp_exposed_to_internet_test.py @@ -24,19 +24,15 @@ class Test_ec2_instance_port_rdp_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet import ( @@ -91,19 +87,15 @@ class Test_ec2_instance_port_rdp_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet import ( @@ -171,19 +163,15 @@ class Test_ec2_instance_port_rdp_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet import ( @@ -257,19 +245,15 @@ class Test_ec2_instance_port_rdp_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet import ( @@ -363,19 +347,15 @@ class Test_ec2_instance_port_rdp_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_rdp_exposed_to_internet.ec2_instance_port_rdp_exposed_to_internet import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_redis_exposed_to_internet/ec2_instance_port_redis_exposed_to_internet_fixer_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_redis_exposed_to_internet/ec2_instance_port_redis_exposed_to_internet_fixer_test.py index 35714af2ee..8affd0e7b1 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_redis_exposed_to_internet/ec2_instance_port_redis_exposed_to_internet_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_redis_exposed_to_internet/ec2_instance_port_redis_exposed_to_internet_fixer_test.py @@ -73,15 +73,12 @@ class Test_ec2_instance_port_redis_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet_fixer import ( @@ -156,15 +153,12 @@ class Test_ec2_instance_port_redis_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet_fixer import ( @@ -213,15 +207,12 @@ class Test_ec2_instance_port_redis_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet_fixer import ( @@ -270,15 +261,12 @@ class Test_ec2_instance_port_redis_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet_fixer import ( @@ -333,15 +321,12 @@ class Test_ec2_instance_port_redis_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_redis_exposed_to_internet/ec2_instance_port_redis_exposed_to_internet_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_redis_exposed_to_internet/ec2_instance_port_redis_exposed_to_internet_test.py index 8bfe6fc676..b08a9d7fde 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_redis_exposed_to_internet/ec2_instance_port_redis_exposed_to_internet_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_redis_exposed_to_internet/ec2_instance_port_redis_exposed_to_internet_test.py @@ -24,19 +24,15 @@ class Test_ec2_instance_port_redis_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet import ( @@ -91,19 +87,15 @@ class Test_ec2_instance_port_redis_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet import ( @@ -171,19 +163,15 @@ class Test_ec2_instance_port_redis_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet import ( @@ -257,19 +245,15 @@ class Test_ec2_instance_port_redis_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet import ( @@ -363,19 +347,15 @@ class Test_ec2_instance_port_redis_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_redis_exposed_to_internet.ec2_instance_port_redis_exposed_to_internet import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_sqlserver_exposed_to_internet/ec2_instance_port_sqlserver_exposed_to_internet_fixer_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_sqlserver_exposed_to_internet/ec2_instance_port_sqlserver_exposed_to_internet_fixer_test.py index 77a51cab85..a24190c9b1 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_sqlserver_exposed_to_internet/ec2_instance_port_sqlserver_exposed_to_internet_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_sqlserver_exposed_to_internet/ec2_instance_port_sqlserver_exposed_to_internet_fixer_test.py @@ -73,15 +73,12 @@ class Test_ec2_instance_port_sqlserver_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet_fixer import ( @@ -156,15 +153,12 @@ class Test_ec2_instance_port_sqlserver_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet_fixer import ( @@ -219,15 +213,12 @@ class Test_ec2_instance_port_sqlserver_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet_fixer import ( @@ -282,15 +273,12 @@ class Test_ec2_instance_port_sqlserver_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet_fixer import ( @@ -345,15 +333,12 @@ class Test_ec2_instance_port_sqlserver_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_sqlserver_exposed_to_internet/ec2_instance_port_sqlserver_exposed_to_internet_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_sqlserver_exposed_to_internet/ec2_instance_port_sqlserver_exposed_to_internet_test.py index 6b3bd46f98..8ae6373b24 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_sqlserver_exposed_to_internet/ec2_instance_port_sqlserver_exposed_to_internet_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_sqlserver_exposed_to_internet/ec2_instance_port_sqlserver_exposed_to_internet_test.py @@ -24,19 +24,15 @@ class Test_ec2_instance_port_sqlserver_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet import ( @@ -91,19 +87,15 @@ class Test_ec2_instance_port_sqlserver_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet import ( @@ -171,19 +163,15 @@ class Test_ec2_instance_port_sqlserver_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet import ( @@ -257,19 +245,15 @@ class Test_ec2_instance_port_sqlserver_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet import ( @@ -363,19 +347,15 @@ class Test_ec2_instance_port_sqlserver_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_sqlserver_exposed_to_internet.ec2_instance_port_sqlserver_exposed_to_internet import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_ssh_exposed_to_internet/ec2_instance_port_ssh_exposed_to_internet_fixer_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_ssh_exposed_to_internet/ec2_instance_port_ssh_exposed_to_internet_fixer_test.py index 8c049b1a22..89e3bb552c 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_ssh_exposed_to_internet/ec2_instance_port_ssh_exposed_to_internet_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_ssh_exposed_to_internet/ec2_instance_port_ssh_exposed_to_internet_fixer_test.py @@ -73,15 +73,12 @@ class Test_ec2_instance_port_ssh_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet_fixer import ( @@ -156,15 +153,12 @@ class Test_ec2_instance_port_ssh_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet_fixer import ( @@ -213,15 +207,12 @@ class Test_ec2_instance_port_ssh_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet_fixer import ( @@ -270,15 +261,12 @@ class Test_ec2_instance_port_ssh_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet_fixer import ( @@ -333,15 +321,12 @@ class Test_ec2_instance_port_ssh_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_ssh_exposed_to_internet/ec2_instance_port_ssh_exposed_to_internet_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_ssh_exposed_to_internet/ec2_instance_port_ssh_exposed_to_internet_test.py index f4eff3dc3c..7536a3eec3 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_ssh_exposed_to_internet/ec2_instance_port_ssh_exposed_to_internet_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_ssh_exposed_to_internet/ec2_instance_port_ssh_exposed_to_internet_test.py @@ -24,19 +24,15 @@ class Test_ec2_instance_port_ssh_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet import ( @@ -91,19 +87,15 @@ class Test_ec2_instance_port_ssh_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet import ( @@ -171,19 +163,15 @@ class Test_ec2_instance_port_ssh_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet import ( @@ -257,19 +245,15 @@ class Test_ec2_instance_port_ssh_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet import ( @@ -363,19 +347,15 @@ class Test_ec2_instance_port_ssh_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_ssh_exposed_to_internet.ec2_instance_port_ssh_exposed_to_internet import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_telnet_exposed_to_internet/ec2_instance_port_telnet_exposed_to_internet_fixer_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_telnet_exposed_to_internet/ec2_instance_port_telnet_exposed_to_internet_fixer_test.py index 300da2276a..3e13565fbe 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_telnet_exposed_to_internet/ec2_instance_port_telnet_exposed_to_internet_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_telnet_exposed_to_internet/ec2_instance_port_telnet_exposed_to_internet_fixer_test.py @@ -73,15 +73,12 @@ class Test_ec2_instance_port_telnet_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet_fixer import ( @@ -156,15 +153,12 @@ class Test_ec2_instance_port_telnet_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet_fixer import ( @@ -213,15 +207,12 @@ class Test_ec2_instance_port_telnet_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet_fixer import ( @@ -270,15 +261,12 @@ class Test_ec2_instance_port_telnet_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet_fixer import ( @@ -333,15 +321,12 @@ class Test_ec2_instance_port_telnet_exposed_to_internet_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_port_telnet_exposed_to_internet/ec2_instance_port_telnet_exposed_to_internet_test.py b/tests/providers/aws/services/ec2/ec2_instance_port_telnet_exposed_to_internet/ec2_instance_port_telnet_exposed_to_internet_test.py index 27d218ab3f..b2b8fdeb81 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_port_telnet_exposed_to_internet/ec2_instance_port_telnet_exposed_to_internet_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_port_telnet_exposed_to_internet/ec2_instance_port_telnet_exposed_to_internet_test.py @@ -24,19 +24,15 @@ class Test_ec2_instance_port_telnet_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet import ( @@ -91,19 +87,15 @@ class Test_ec2_instance_port_telnet_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet import ( @@ -171,19 +163,15 @@ class Test_ec2_instance_port_telnet_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet import ( @@ -257,19 +245,15 @@ class Test_ec2_instance_port_telnet_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet import ( @@ -363,19 +347,15 @@ class Test_ec2_instance_port_telnet_exposed_to_internet: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_port_telnet_exposed_to_internet.ec2_instance_port_telnet_exposed_to_internet import ( diff --git a/tests/providers/aws/services/ec2/ec2_instance_profile_attached/ec2_instance_profile_attached_test.py b/tests/providers/aws/services/ec2/ec2_instance_profile_attached/ec2_instance_profile_attached_test.py index 1e33138d75..920908706c 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_profile_attached/ec2_instance_profile_attached_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_profile_attached/ec2_instance_profile_attached_test.py @@ -22,15 +22,12 @@ class Test_ec2_instance_profile_attached: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_profile_attached.ec2_instance_profile_attached.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_profile_attached.ec2_instance_profile_attached.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_profile_attached.ec2_instance_profile_attached import ( @@ -72,15 +69,12 @@ class Test_ec2_instance_profile_attached: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_profile_attached.ec2_instance_profile_attached.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_profile_attached.ec2_instance_profile_attached.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_profile_attached.ec2_instance_profile_attached import ( ec2_instance_profile_attached, @@ -127,15 +121,12 @@ class Test_ec2_instance_profile_attached: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_profile_attached.ec2_instance_profile_attached.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_profile_attached.ec2_instance_profile_attached.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_profile_attached.ec2_instance_profile_attached import ( ec2_instance_profile_attached, diff --git a/tests/providers/aws/services/ec2/ec2_instance_public_ip/ec2_instance_public_ip_test.py b/tests/providers/aws/services/ec2/ec2_instance_public_ip/ec2_instance_public_ip_test.py index 132a85508f..646fe94246 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_public_ip/ec2_instance_public_ip_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_public_ip/ec2_instance_public_ip_test.py @@ -22,15 +22,12 @@ class Test_ec2_instance_public_ip: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_public_ip.ec2_instance_public_ip.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_public_ip.ec2_instance_public_ip.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_public_ip.ec2_instance_public_ip import ( @@ -66,15 +63,12 @@ class Test_ec2_instance_public_ip: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_public_ip.ec2_instance_public_ip.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_public_ip.ec2_instance_public_ip.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_public_ip.ec2_instance_public_ip import ( ec2_instance_public_ip, @@ -121,15 +115,12 @@ class Test_ec2_instance_public_ip: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_public_ip.ec2_instance_public_ip.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_public_ip.ec2_instance_public_ip.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_public_ip.ec2_instance_public_ip import ( ec2_instance_public_ip, diff --git a/tests/providers/aws/services/ec2/ec2_instance_secrets_user_data/ec2_instance_secrets_user_data_test.py b/tests/providers/aws/services/ec2/ec2_instance_secrets_user_data/ec2_instance_secrets_user_data_test.py index 0e8d303fe0..8abb111a1d 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_secrets_user_data/ec2_instance_secrets_user_data_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_secrets_user_data/ec2_instance_secrets_user_data_test.py @@ -26,15 +26,12 @@ class Test_ec2_instance_secrets_user_data: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data import ( @@ -62,15 +59,12 @@ class Test_ec2_instance_secrets_user_data: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data import ( ec2_instance_secrets_user_data, @@ -109,15 +103,12 @@ class Test_ec2_instance_secrets_user_data: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data import ( ec2_instance_secrets_user_data, @@ -162,15 +153,12 @@ class Test_ec2_instance_secrets_user_data: "secrets_ignore_patterns": [".*_ALLOW_EMPTY_PASSWORD.*"] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data.ec2_client", + new=ec2_client, ): from prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data import ( ec2_instance_secrets_user_data, @@ -212,15 +200,12 @@ class Test_ec2_instance_secrets_user_data: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data import ( ec2_instance_secrets_user_data, @@ -256,15 +241,12 @@ class Test_ec2_instance_secrets_user_data: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data import ( ec2_instance_secrets_user_data, @@ -306,15 +288,12 @@ class Test_ec2_instance_secrets_user_data: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data import ( ec2_instance_secrets_user_data, @@ -351,15 +330,12 @@ class Test_ec2_instance_secrets_user_data: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_secrets_user_data.ec2_instance_secrets_user_data import ( ec2_instance_secrets_user_data, diff --git a/tests/providers/aws/services/ec2/ec2_instance_uses_single_eni/ec2_instance_uses_single_eni_test.py b/tests/providers/aws/services/ec2/ec2_instance_uses_single_eni/ec2_instance_uses_single_eni_test.py index 5460248454..863067a294 100644 --- a/tests/providers/aws/services/ec2/ec2_instance_uses_single_eni/ec2_instance_uses_single_eni_test.py +++ b/tests/providers/aws/services/ec2/ec2_instance_uses_single_eni/ec2_instance_uses_single_eni_test.py @@ -170,15 +170,12 @@ class Test_ec2_instance_uses_single_eni: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_uses_single_eni.ec2_instance_uses_single_eni.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_uses_single_eni.ec2_instance_uses_single_eni.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_uses_single_eni.ec2_instance_uses_single_eni import ( ec2_instance_uses_single_eni, @@ -196,15 +193,12 @@ class Test_ec2_instance_uses_single_eni: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_uses_single_eni.ec2_instance_uses_single_eni.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_uses_single_eni.ec2_instance_uses_single_eni.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_uses_single_eni.ec2_instance_uses_single_eni import ( ec2_instance_uses_single_eni, @@ -229,15 +223,12 @@ class Test_ec2_instance_uses_single_eni: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_uses_single_eni.ec2_instance_uses_single_eni.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_uses_single_eni.ec2_instance_uses_single_eni.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_uses_single_eni.ec2_instance_uses_single_eni import ( ec2_instance_uses_single_eni, @@ -269,15 +260,12 @@ class Test_ec2_instance_uses_single_eni: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_uses_single_eni.ec2_instance_uses_single_eni.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_uses_single_eni.ec2_instance_uses_single_eni.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_uses_single_eni.ec2_instance_uses_single_eni import ( ec2_instance_uses_single_eni, @@ -302,15 +290,12 @@ class Test_ec2_instance_uses_single_eni: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_instance_uses_single_eni.ec2_instance_uses_single_eni.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_instance_uses_single_eni.ec2_instance_uses_single_eni.ec2_client", + new=EC2(aws_provider), ): from prowler.providers.aws.services.ec2.ec2_instance_uses_single_eni.ec2_instance_uses_single_eni import ( ec2_instance_uses_single_eni, diff --git a/tests/providers/aws/services/ec2/ec2_launch_template_no_public_ip/ec2_launch_template_no_public_ip_test.py b/tests/providers/aws/services/ec2/ec2_launch_template_no_public_ip/ec2_launch_template_no_public_ip_test.py index f0c6eb13e7..d582b73698 100644 --- a/tests/providers/aws/services/ec2/ec2_launch_template_no_public_ip/ec2_launch_template_no_public_ip_test.py +++ b/tests/providers/aws/services/ec2/ec2_launch_template_no_public_ip/ec2_launch_template_no_public_ip_test.py @@ -49,15 +49,12 @@ class Test_ec2_launch_template_no_public_ip: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_launch_template_no_public_ip.ec2_launch_template_no_public_ip.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_launch_template_no_public_ip.ec2_launch_template_no_public_ip.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_launch_template_no_public_ip.ec2_launch_template_no_public_ip import ( @@ -91,15 +88,12 @@ class Test_ec2_launch_template_no_public_ip: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_launch_template_no_public_ip.ec2_launch_template_no_public_ip.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_launch_template_no_public_ip.ec2_launch_template_no_public_ip.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_launch_template_no_public_ip.ec2_launch_template_no_public_ip import ( @@ -155,15 +149,12 @@ class Test_ec2_launch_template_no_public_ip: ec2_client.launch_templates = [launch_template] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=ec2_client, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_launch_template_no_public_ip.ec2_launch_template_no_public_ip.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=ec2_client, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_launch_template_no_public_ip.ec2_launch_template_no_public_ip.ec2_client", + new=ec2_client, ): # Test Check from prowler.providers.aws.services.ec2.ec2_launch_template_no_public_ip.ec2_launch_template_no_public_ip import ( @@ -255,15 +246,12 @@ class Test_ec2_launch_template_no_public_ip: ec2_client.launch_templates = [launch_template] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=ec2_client, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_launch_template_no_public_ip.ec2_launch_template_no_public_ip.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=ec2_client, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_launch_template_no_public_ip.ec2_launch_template_no_public_ip.ec2_client", + new=ec2_client, ): # Test Check from prowler.providers.aws.services.ec2.ec2_launch_template_no_public_ip.ec2_launch_template_no_public_ip import ( @@ -345,15 +333,12 @@ class Test_ec2_launch_template_no_public_ip: ec2_client.launch_templates = [launch_template] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=ec2_client, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_launch_template_no_public_ip.ec2_launch_template_no_public_ip.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=ec2_client, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_launch_template_no_public_ip.ec2_launch_template_no_public_ip.ec2_client", + new=ec2_client, ): # Test Check from prowler.providers.aws.services.ec2.ec2_launch_template_no_public_ip.ec2_launch_template_no_public_ip import ( diff --git a/tests/providers/aws/services/ec2/ec2_launch_template_no_secrets/ec2_launch_template_no_secrets_test.py b/tests/providers/aws/services/ec2/ec2_launch_template_no_secrets/ec2_launch_template_no_secrets_test.py index 0430ca5cae..4659d88af7 100644 --- a/tests/providers/aws/services/ec2/ec2_launch_template_no_secrets/ec2_launch_template_no_secrets_test.py +++ b/tests/providers/aws/services/ec2/ec2_launch_template_no_secrets/ec2_launch_template_no_secrets_test.py @@ -61,15 +61,12 @@ class Test_ec2_launch_template_no_secrets: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets import ( @@ -105,15 +102,12 @@ class Test_ec2_launch_template_no_secrets: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets import ( @@ -142,15 +136,12 @@ class Test_ec2_launch_template_no_secrets: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets import ( @@ -214,15 +205,12 @@ class Test_ec2_launch_template_no_secrets: ec2_client.launch_templates = [launch_template] ec2_client.audit_config = {"detect_secrets_plugins": None} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=ec2_client, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=ec2_client, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets.ec2_client", + new=ec2_client, ): # Test Check from prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets import ( @@ -292,15 +280,12 @@ class Test_ec2_launch_template_no_secrets: ec2_client.launch_templates = [launch_template] ec2_client.audit_config = {"detect_secrets_plugins": None} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=ec2_client, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=ec2_client, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets.ec2_client", + new=ec2_client, ): # Test Check from prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets import ( @@ -360,15 +345,12 @@ class Test_ec2_launch_template_no_secrets: ec2_client.launch_templates = [launch_template] ec2_client.audit_config = {"detect_secrets_plugins": None} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=ec2_client, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=ec2_client, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets.ec2_client", + new=ec2_client, ): # Test Check from prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets import ( @@ -412,15 +394,12 @@ class Test_ec2_launch_template_no_secrets: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets import ( @@ -508,15 +487,12 @@ class Test_ec2_launch_template_no_secrets: ] ec2_client.audit_config = {"detect_secrets_plugins": None} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=ec2_client, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=ec2_client, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets.ec2_client", + new=ec2_client, ): # Test Check from prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets import ( @@ -574,15 +550,12 @@ class Test_ec2_launch_template_no_secrets: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_launch_template_no_secrets.ec2_launch_template_no_secrets import ( diff --git a/tests/providers/aws/services/ec2/ec2_networkacl_allow_ingress_any_port/ec2_networkacl_allow_ingress_any_port_test.py b/tests/providers/aws/services/ec2/ec2_networkacl_allow_ingress_any_port/ec2_networkacl_allow_ingress_any_port_test.py index 729f065b28..c89d19f680 100644 --- a/tests/providers/aws/services/ec2/ec2_networkacl_allow_ingress_any_port/ec2_networkacl_allow_ingress_any_port_test.py +++ b/tests/providers/aws/services/ec2/ec2_networkacl_allow_ingress_any_port/ec2_networkacl_allow_ingress_any_port_test.py @@ -19,15 +19,12 @@ class Test_ec2_networkacl_allow_ingress_any_port: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_any_port.ec2_networkacl_allow_ingress_any_port.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_any_port.ec2_networkacl_allow_ingress_any_port.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_any_port.ec2_networkacl_allow_ingress_any_port import ( @@ -48,15 +45,12 @@ class Test_ec2_networkacl_allow_ingress_any_port: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_any_port.ec2_networkacl_allow_ingress_any_port.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_any_port.ec2_networkacl_allow_ingress_any_port.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_any_port.ec2_networkacl_allow_ingress_any_port import ( @@ -101,15 +95,12 @@ class Test_ec2_networkacl_allow_ingress_any_port: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_any_port.ec2_networkacl_allow_ingress_any_port.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_any_port.ec2_networkacl_allow_ingress_any_port.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_any_port.ec2_networkacl_allow_ingress_any_port import ( @@ -159,15 +150,12 @@ class Test_ec2_networkacl_allow_ingress_any_port: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_any_port.ec2_networkacl_allow_ingress_any_port.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_any_port.ec2_networkacl_allow_ingress_any_port.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_any_port.ec2_networkacl_allow_ingress_any_port import ( @@ -218,15 +206,12 @@ class Test_ec2_networkacl_allow_ingress_any_port: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_any_port.ec2_networkacl_allow_ingress_any_port.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_any_port.ec2_networkacl_allow_ingress_any_port.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_any_port.ec2_networkacl_allow_ingress_any_port import ( @@ -263,15 +248,12 @@ class Test_ec2_networkacl_allow_ingress_any_port: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_any_port.ec2_networkacl_allow_ingress_any_port.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_any_port.ec2_networkacl_allow_ingress_any_port.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_any_port.ec2_networkacl_allow_ingress_any_port import ( diff --git a/tests/providers/aws/services/ec2/ec2_networkacl_allow_ingress_tcp_port_22/ec2_networkacl_allow_ingress_tcp_port_22_test.py b/tests/providers/aws/services/ec2/ec2_networkacl_allow_ingress_tcp_port_22/ec2_networkacl_allow_ingress_tcp_port_22_test.py index 496cc9092a..991fc9fbee 100644 --- a/tests/providers/aws/services/ec2/ec2_networkacl_allow_ingress_tcp_port_22/ec2_networkacl_allow_ingress_tcp_port_22_test.py +++ b/tests/providers/aws/services/ec2/ec2_networkacl_allow_ingress_tcp_port_22/ec2_networkacl_allow_ingress_tcp_port_22_test.py @@ -19,15 +19,12 @@ class Test_ec2_networkacl_allow_ingress_tcp_port_22: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_22.ec2_networkacl_allow_ingress_tcp_port_22.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_22.ec2_networkacl_allow_ingress_tcp_port_22.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_22.ec2_networkacl_allow_ingress_tcp_port_22 import ( @@ -48,15 +45,12 @@ class Test_ec2_networkacl_allow_ingress_tcp_port_22: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_22.ec2_networkacl_allow_ingress_tcp_port_22.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_22.ec2_networkacl_allow_ingress_tcp_port_22.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_22.ec2_networkacl_allow_ingress_tcp_port_22 import ( @@ -102,15 +96,12 @@ class Test_ec2_networkacl_allow_ingress_tcp_port_22: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_22.ec2_networkacl_allow_ingress_tcp_port_22.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_22.ec2_networkacl_allow_ingress_tcp_port_22.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_22.ec2_networkacl_allow_ingress_tcp_port_22 import ( @@ -161,15 +152,12 @@ class Test_ec2_networkacl_allow_ingress_tcp_port_22: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_22.ec2_networkacl_allow_ingress_tcp_port_22.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_22.ec2_networkacl_allow_ingress_tcp_port_22.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_22.ec2_networkacl_allow_ingress_tcp_port_22 import ( @@ -220,15 +208,12 @@ class Test_ec2_networkacl_allow_ingress_tcp_port_22: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_22.ec2_networkacl_allow_ingress_tcp_port_22.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_22.ec2_networkacl_allow_ingress_tcp_port_22.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_22.ec2_networkacl_allow_ingress_tcp_port_22 import ( @@ -265,15 +250,12 @@ class Test_ec2_networkacl_allow_ingress_tcp_port_22: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_22.ec2_networkacl_allow_ingress_tcp_port_22.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_22.ec2_networkacl_allow_ingress_tcp_port_22.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_22.ec2_networkacl_allow_ingress_tcp_port_22 import ( diff --git a/tests/providers/aws/services/ec2/ec2_networkacl_allow_ingress_tcp_port_3389/ec2_networkacl_allow_ingress_tcp_port_3389_test.py b/tests/providers/aws/services/ec2/ec2_networkacl_allow_ingress_tcp_port_3389/ec2_networkacl_allow_ingress_tcp_port_3389_test.py index 397804bde5..d08bd553be 100644 --- a/tests/providers/aws/services/ec2/ec2_networkacl_allow_ingress_tcp_port_3389/ec2_networkacl_allow_ingress_tcp_port_3389_test.py +++ b/tests/providers/aws/services/ec2/ec2_networkacl_allow_ingress_tcp_port_3389/ec2_networkacl_allow_ingress_tcp_port_3389_test.py @@ -19,15 +19,12 @@ class Test_ec2_networkacl_allow_ingress_tcp_port_3389: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_networkacl_allow_ingress_tcp_port_3389 import ( @@ -48,15 +45,12 @@ class Test_ec2_networkacl_allow_ingress_tcp_port_3389: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_networkacl_allow_ingress_tcp_port_3389 import ( @@ -102,15 +96,12 @@ class Test_ec2_networkacl_allow_ingress_tcp_port_3389: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_networkacl_allow_ingress_tcp_port_3389 import ( @@ -161,15 +152,12 @@ class Test_ec2_networkacl_allow_ingress_tcp_port_3389: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_networkacl_allow_ingress_tcp_port_3389 import ( @@ -220,15 +208,12 @@ class Test_ec2_networkacl_allow_ingress_tcp_port_3389: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_networkacl_allow_ingress_tcp_port_3389 import ( @@ -265,15 +250,12 @@ class Test_ec2_networkacl_allow_ingress_tcp_port_3389: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_allow_ingress_tcp_port_3389.ec2_networkacl_allow_ingress_tcp_port_3389 import ( diff --git a/tests/providers/aws/services/ec2/ec2_networkacl_unused/ec2_networkacl_unused_test.py b/tests/providers/aws/services/ec2/ec2_networkacl_unused/ec2_networkacl_unused_test.py index de9fb191bf..b141f51d0a 100644 --- a/tests/providers/aws/services/ec2/ec2_networkacl_unused/ec2_networkacl_unused_test.py +++ b/tests/providers/aws/services/ec2/ec2_networkacl_unused/ec2_networkacl_unused_test.py @@ -19,15 +19,12 @@ class Test_ec2_networkacl_unused: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_unused.ec2_networkacl_unused.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_unused.ec2_networkacl_unused.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_unused.ec2_networkacl_unused import ( @@ -54,15 +51,12 @@ class Test_ec2_networkacl_unused: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_unused.ec2_networkacl_unused.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_unused.ec2_networkacl_unused.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_unused.ec2_networkacl_unused import ( @@ -105,15 +99,12 @@ class Test_ec2_networkacl_unused: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_networkacl_unused.ec2_networkacl_unused.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_networkacl_unused.ec2_networkacl_unused.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_networkacl_unused.ec2_networkacl_unused import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_all_ports/ec2_securitygroup_allow_ingress_from_internet_to_all_ports_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_all_ports/ec2_securitygroup_allow_ingress_from_internet_to_all_ports_test.py index 85c1fd3c3d..a1dd6497d4 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_all_ports/ec2_securitygroup_allow_ingress_from_internet_to_all_ports_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_all_ports/ec2_securitygroup_allow_ingress_from_internet_to_all_ports_test.py @@ -27,19 +27,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_all_ports: ], ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( @@ -86,19 +82,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_all_ports: ], ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( @@ -156,19 +148,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_all_ports: ], ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( @@ -231,19 +219,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_all_ports: ], ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( @@ -288,19 +272,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_all_ports: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( @@ -336,19 +316,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_all_ports: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( @@ -392,23 +368,18 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_all_ports: from prowler.providers.aws.services.ec2.ec2_service import EC2 from prowler.providers.aws.services.vpc.vpc_service import VPC - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_service.VPC", - new=VPC(aws_provider), - ), - mock.patch( - "prowler.providers.aws.lib.service.service.AWSService.set_failed_check" - ) as mock_set_failed_check, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_service.VPC", + new=VPC(aws_provider), + ), mock.patch( + "prowler.providers.aws.lib.service.service.AWSService.set_failed_check" + ) as mock_set_failed_check: from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( ec2_securitygroup_allow_ingress_from_internet_to_all_ports, diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_any_port/ec2_securitygroup_allow_ingress_from_internet_to_any_port_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_any_port/ec2_securitygroup_allow_ingress_from_internet_to_any_port_test.py index c31b6ad02c..2c83564bc9 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_any_port/ec2_securitygroup_allow_ingress_from_internet_to_any_port_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_any_port/ec2_securitygroup_allow_ingress_from_internet_to_any_port_test.py @@ -25,19 +25,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_any_port: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1], ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port import ( @@ -109,19 +105,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_any_port: aws_provider = set_mocked_aws_provider( [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1], ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port import ( @@ -162,15 +154,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_any_port: from prowler.providers.aws.services.ec2.ec2_service import EC2 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_client", + new=EC2(aws_provider), ): from unittest.mock import Mock @@ -286,19 +275,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_any_port: audit_config={"ec2_allowed_interface_types": [eni_type]}, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port import ( @@ -383,19 +368,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_any_port: audit_config={"ec2_allowed_instance_owners": [eni_owner]}, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port import ( @@ -450,19 +431,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_any_port: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1], ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port import ( @@ -554,19 +531,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_any_port: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port import ( @@ -599,19 +572,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_any_port: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port import ( @@ -649,19 +618,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_any_port: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port import ( @@ -709,20 +674,16 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_any_port: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ) as ec2_client_instance, - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ) as vpc_client_instance, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ) as ec2_client_instance, mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), + ) as vpc_client_instance: # Run check for all ports from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( ec2_securitygroup_allow_ingress_from_internet_to_all_ports, @@ -740,15 +701,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_any_port: ) # use the same mock objects for the specific port check - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_client", - new=ec2_client_instance, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.vpc_client", - new=vpc_client_instance, - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_client", + new=ec2_client_instance, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port.vpc_client", + new=vpc_client_instance, ): # Now run the specific port check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_any_port.ec2_securitygroup_allow_ingress_from_internet_to_any_port import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports/ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_fixer_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports/ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_fixer_test.py index a67818c90c..f206a28940 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports/ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_fixer_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports/ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_fixer_test.py @@ -66,15 +66,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_ aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_fixer import ( @@ -133,15 +130,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_ aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_fixer import ( @@ -190,15 +184,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_ aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_fixer import ( @@ -247,15 +238,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_ aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_fixer import ( @@ -304,15 +292,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_ aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_fixer.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_fixer.ec2_client", + new=EC2(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_fixer import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports/ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports/ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_test.py index 13ea331779..2811c00f0b 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports/ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports/ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports_test.py @@ -39,19 +39,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports import ( @@ -114,19 +110,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports import ( @@ -201,19 +193,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports import ( @@ -273,19 +261,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports import ( @@ -330,19 +314,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports import ( @@ -409,20 +389,16 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ) as ec2_client_instance, - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ) as vpc_client_instance, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ) as ec2_client_instance, mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), + ) as vpc_client_instance: # Run check for all ports from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( ec2_securitygroup_allow_ingress_from_internet_to_all_ports, @@ -440,15 +416,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports: ) # use the same mock objects for the specific port check - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_client", - new=ec2_client_instance, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.vpc_client", - new=vpc_client_instance, - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_client", + new=ec2_client_instance, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.vpc_client", + new=vpc_client_instance, ): # Now run the specific port check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports.ec2_securitygroup_allow_ingress_from_internet_to_high_risk_tcp_ports import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018/ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018/ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018_test.py index 3cd1b87612..b3e5c005c8 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018/ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018/ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018_test.py @@ -24,19 +24,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_2 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018 import ( @@ -84,19 +80,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_2 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018 import ( @@ -155,19 +147,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_2 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018 import ( @@ -211,19 +199,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_2 scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018 import ( @@ -258,19 +242,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_2 scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018 import ( @@ -321,20 +301,16 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_2 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ) as ec2_client_instance, - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ) as vpc_client_instance, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ) as ec2_client_instance, mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), + ) as vpc_client_instance: # Run check for all ports from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( ec2_securitygroup_allow_ingress_from_internet_to_all_ports, @@ -352,15 +328,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_2 ) # use the same mock objects for the specific port check - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_client", - new=ec2_client_instance, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.vpc_client", - new=vpc_client_instance, - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_client", + new=ec2_client_instance, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.vpc_client", + new=vpc_client_instance, ): # Now run the specific port check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018.ec2_securitygroup_allow_ingress_from_internet_to_port_mongodb_27017_27018 import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21/ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21/ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21_test.py index 5b753e4981..4000111b9b 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21/ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21/ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21_test.py @@ -24,19 +24,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21 import ( @@ -84,19 +80,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21 import ( @@ -155,19 +147,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21 import ( @@ -211,19 +199,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21 import ( @@ -258,19 +242,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21 import ( @@ -321,20 +301,16 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ) as ec2_client_instance, - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ) as vpc_client_instance, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ) as ec2_client_instance, mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), + ) as vpc_client_instance: # Run check for all ports from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( ec2_securitygroup_allow_ingress_from_internet_to_all_ports, @@ -352,15 +328,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21: ) # use the same mock objects for the specific port check - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_client", - new=ec2_client_instance, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.vpc_client", - new=vpc_client_instance, - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_client", + new=ec2_client_instance, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.vpc_client", + new=vpc_client_instance, ): # Now run the specific port check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21.ec2_securitygroup_allow_ingress_from_internet_to_tcp_ftp_port_20_21 import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22_test.py index 9dc839e4aa..0223840d03 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22_test.py @@ -25,19 +25,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22 import ( @@ -83,19 +79,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22 import ( @@ -156,19 +148,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22 import ( @@ -210,19 +198,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22 import ( @@ -255,19 +239,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22 import ( @@ -316,20 +296,16 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ) as ec2_client_instance, - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ) as vpc_client_instance, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ) as ec2_client_instance, mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), + ) as vpc_client_instance: # Run check for all ports from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( ec2_securitygroup_allow_ingress_from_internet_to_all_ports, @@ -347,15 +323,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22: ) # use the same mock objects for the specific port check - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_client", - new=ec2_client_instance, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.vpc_client", - new=vpc_client_instance, - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_client", + new=ec2_client_instance, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.vpc_client", + new=vpc_client_instance, ): # Now run the specific port check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_22 import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389_test.py index 1902d7fffc..b598910d09 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389_test.py @@ -24,19 +24,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389 import ( @@ -67,19 +63,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389 import ( @@ -112,19 +104,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389 import ( @@ -167,19 +155,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389 import ( @@ -236,19 +220,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389 import ( @@ -311,20 +291,16 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ) as ec2_client_instance, - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ) as vpc_client_instance, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ) as ec2_client_instance, mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), + ) as vpc_client_instance: # Run check for all ports from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( ec2_securitygroup_allow_ingress_from_internet_to_all_ports, @@ -342,15 +318,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389: ) # use the same mock objects for the specific port check - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_client", - new=ec2_client_instance, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.vpc_client", - new=vpc_client_instance, - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_client", + new=ec2_client_instance, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.vpc_client", + new=vpc_client_instance, ): # Now run the specific port check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_3389 import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888_test.py index 756cf9a4fa..31a9f5b3c1 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888_test.py @@ -24,19 +24,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888 import ( @@ -84,19 +80,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888 import ( @@ -155,19 +147,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888 import ( @@ -211,19 +199,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7 scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888 import ( @@ -258,19 +242,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7 scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888 import ( @@ -321,20 +301,16 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ) as ec2_client_instance, - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ) as vpc_client_instance, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ) as ec2_client_instance, mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), + ) as vpc_client_instance: # Run check for all ports from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( ec2_securitygroup_allow_ingress_from_internet_to_all_ports, @@ -352,15 +328,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7 ) # use the same mock objects for the specific port check - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_client", - new=ec2_client_instance, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.vpc_client", - new=vpc_client_instance, - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_client", + new=ec2_client_instance, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.vpc_client", + new=vpc_client_instance, ): # Now run the specific port check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_cassandra_7199_9160_8888 import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601_test.py index 876193f087..40529aecd1 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601_test.py @@ -24,19 +24,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsear [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601 import ( @@ -84,19 +80,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsear [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601 import ( @@ -155,19 +147,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsear [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601 import ( @@ -211,19 +199,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsear scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601 import ( @@ -258,19 +242,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsear scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601 import ( @@ -321,20 +301,16 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsear [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ) as ec2_client_instance, - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ) as vpc_client_instance, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ) as ec2_client_instance, mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), + ) as vpc_client_instance: # Run check for all ports from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( ec2_securitygroup_allow_ingress_from_internet_to_all_ports, @@ -352,15 +328,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsear ) # use the same mock objects for the specific port check - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_client", - new=ec2_client_instance, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.vpc_client", - new=vpc_client_instance, - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_client", + new=ec2_client_instance, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.vpc_client", + new=vpc_client_instance, ): # Now run the specific port check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_elasticsearch_kibana_9200_9300_5601 import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092_test.py index 949a7f1d58..6c4a2f97c5 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092_test.py @@ -24,19 +24,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092 import ( @@ -84,19 +80,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092 import ( @@ -155,19 +147,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092 import ( @@ -211,19 +199,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092 import ( @@ -258,19 +242,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092 import ( @@ -321,20 +301,16 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ) as ec2_client_instance, - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ) as vpc_client_instance, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ) as ec2_client_instance, mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), + ) as vpc_client_instance: # Run check for all ports from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( ec2_securitygroup_allow_ingress_from_internet_to_all_ports, @@ -352,15 +328,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092: ) # use the same mock objects for the specific port check - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_client", - new=ec2_client_instance, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.vpc_client", - new=vpc_client_instance, - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_client", + new=ec2_client_instance, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.vpc_client", + new=vpc_client_instance, ): # Now run the specific port check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_kafka_9092 import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211_test.py index 1b5db361af..03c352f140 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211_test.py @@ -24,19 +24,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_1 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211 import ( @@ -84,19 +80,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_1 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211 import ( @@ -155,19 +147,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_1 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211 import ( @@ -211,19 +199,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_1 scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211 import ( @@ -258,19 +242,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_1 scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211 import ( @@ -321,20 +301,16 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_1 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ) as ec2_client_instance, - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ) as vpc_client_instance, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ) as ec2_client_instance, mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), + ) as vpc_client_instance: # Run check for all ports from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( ec2_securitygroup_allow_ingress_from_internet_to_all_ports, @@ -352,15 +328,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_1 ) # use the same mock objects for the specific port check - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_client", - new=ec2_client_instance, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.vpc_client", - new=vpc_client_instance, - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_client", + new=ec2_client_instance, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.vpc_client", + new=vpc_client_instance, ): # Now run the specific port check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_memcached_11211 import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306_test.py index 2be40347af..ab98c91201 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306_test.py @@ -24,19 +24,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306 import ( @@ -84,19 +80,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306 import ( @@ -155,19 +147,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306 import ( @@ -211,19 +199,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306 import ( @@ -258,19 +242,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306 import ( @@ -321,20 +301,16 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ) as ec2_client_instance, - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ) as vpc_client_instance, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ) as ec2_client_instance, mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), + ) as vpc_client_instance: # Run check for all ports from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( ec2_securitygroup_allow_ingress_from_internet_to_all_ports, @@ -352,15 +328,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306: ) # use the same mock objects for the specific port check - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_client", - new=ec2_client_instance, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.vpc_client", - new=vpc_client_instance, - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_client", + new=ec2_client_instance, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.vpc_client", + new=vpc_client_instance, ): # Now run the specific port check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_mysql_3306 import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483_test.py index c7f89c4ae8..74df6597a3 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483_test.py @@ -24,19 +24,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483 import ( @@ -84,19 +80,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483 import ( @@ -155,19 +147,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483 import ( @@ -211,19 +199,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521 scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483 import ( @@ -258,19 +242,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521 scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483 import ( @@ -321,20 +301,16 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ) as ec2_client_instance, - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ) as vpc_client_instance, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ) as ec2_client_instance, mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), + ) as vpc_client_instance: # Run check for all ports from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( ec2_securitygroup_allow_ingress_from_internet_to_all_ports, @@ -352,15 +328,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521 ) # use the same mock objects for the specific port check - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_client", - new=ec2_client_instance, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.vpc_client", - new=vpc_client_instance, - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_client", + new=ec2_client_instance, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.vpc_client", + new=vpc_client_instance, ): # Now run the specific port check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_oracle_1521_2483 import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432_test.py index 1140431594..7a434faf31 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432_test.py @@ -24,19 +24,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_54 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432 import ( @@ -84,19 +80,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_54 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432 import ( @@ -155,19 +147,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_54 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432 import ( @@ -231,19 +219,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_54 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432 import ( @@ -287,19 +271,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_54 scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432 import ( @@ -334,19 +314,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_54 scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432 import ( @@ -397,20 +373,16 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_54 [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ) as ec2_client_instance, - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ) as vpc_client_instance, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ) as ec2_client_instance, mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), + ) as vpc_client_instance: # Run check for all ports from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( ec2_securitygroup_allow_ingress_from_internet_to_all_ports, @@ -428,15 +400,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_54 ) # use the same mock objects for the specific port check - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_client", - new=ec2_client_instance, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.vpc_client", - new=vpc_client_instance, - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_client", + new=ec2_client_instance, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.vpc_client", + new=vpc_client_instance, ): # Now run the specific port check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_postgres_5432 import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379_test.py index e4fa340d41..4e11f7360c 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379_test.py @@ -34,19 +34,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379: ] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379 import ( @@ -128,19 +124,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379: ] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379 import ( @@ -201,19 +193,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379: AWS_REGION_EU_WEST_1, ] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379 import ( @@ -260,19 +248,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379: ) aws_provider._scan_unused_services = False - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379 import ( @@ -303,19 +287,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379: aws_provider = set_mocked_aws_provider() aws_provider._scan_unused_services = False - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379 import ( @@ -379,20 +359,16 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ) as ec2_client_instance, - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ) as vpc_client_instance, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ) as ec2_client_instance, mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), + ) as vpc_client_instance: # Run check for all ports from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( ec2_securitygroup_allow_ingress_from_internet_to_all_ports, @@ -410,15 +386,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379: ) # use the same mock objects for the specific port check - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_client", - new=ec2_client_instance, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.vpc_client", - new=vpc_client_instance, - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_client", + new=ec2_client_instance, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.vpc_client", + new=vpc_client_instance, ): # Now run the specific port check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_redis_6379 import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434_test.py index 6600af4e48..95d955659b 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434_test.py @@ -24,19 +24,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_ [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434 import ( @@ -84,19 +80,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_ [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434 import ( @@ -155,19 +147,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_ [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434 import ( @@ -211,19 +199,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_ scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434 import ( @@ -258,19 +242,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_ scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434 import ( @@ -321,20 +301,16 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_ [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ) as ec2_client_instance, - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ) as vpc_client_instance, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ) as ec2_client_instance, mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), + ) as vpc_client_instance: # Run check for all ports from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( ec2_securitygroup_allow_ingress_from_internet_to_all_ports, @@ -352,15 +328,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_ ) # use the same mock objects for the specific port check - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_client", - new=ec2_client_instance, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.vpc_client", - new=vpc_client_instance, - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_client", + new=ec2_client_instance, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.vpc_client", + new=vpc_client_instance, ): # Now run the specific port check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_sql_server_1433_1434 import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23_test.py index c8f8241a33..1bb95311de 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23/ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23_test.py @@ -24,19 +24,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23 import ( @@ -84,19 +80,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23 import ( @@ -155,19 +147,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23 import ( @@ -211,19 +199,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23 import ( @@ -258,19 +242,15 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23: scan_unused_services=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23 import ( @@ -321,20 +301,16 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", - new=EC2(aws_provider), - ) as ec2_client_instance, - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", - new=VPC(aws_provider), - ) as vpc_client_instance, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_client", + new=EC2(aws_provider), + ) as ec2_client_instance, mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.vpc_client", + new=VPC(aws_provider), + ) as vpc_client_instance: # Run check for all ports from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_all_ports.ec2_securitygroup_allow_ingress_from_internet_to_all_ports import ( ec2_securitygroup_allow_ingress_from_internet_to_all_ports, @@ -352,15 +328,12 @@ class Test_ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23: ) # use the same mock objects for the specific port check - with ( - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_client", - new=ec2_client_instance, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.vpc_client", - new=vpc_client_instance, - ), + with mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_client", + new=ec2_client_instance, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.vpc_client", + new=vpc_client_instance, ): # Now run the specific port check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23.ec2_securitygroup_allow_ingress_from_internet_to_tcp_port_telnet_23 import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_wide_open_public_ipv4/ec2_securitygroup_allow_wide_open_public_ipv4_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_wide_open_public_ipv4/ec2_securitygroup_allow_wide_open_public_ipv4_test.py index d42e9f128a..cf7ad847d7 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_allow_wide_open_public_ipv4/ec2_securitygroup_allow_wide_open_public_ipv4_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_allow_wide_open_public_ipv4/ec2_securitygroup_allow_wide_open_public_ipv4_test.py @@ -24,19 +24,15 @@ class Test_ec2_securitygroup_allow_wide_open_public_ipv4: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_securitygroup_allow_wide_open_public_ipv4.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_securitygroup_allow_wide_open_public_ipv4.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_securitygroup_allow_wide_open_public_ipv4 import ( @@ -80,19 +76,15 @@ class Test_ec2_securitygroup_allow_wide_open_public_ipv4: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_securitygroup_allow_wide_open_public_ipv4.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_securitygroup_allow_wide_open_public_ipv4.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_securitygroup_allow_wide_open_public_ipv4 import ( @@ -147,19 +139,15 @@ class Test_ec2_securitygroup_allow_wide_open_public_ipv4: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_securitygroup_allow_wide_open_public_ipv4.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_securitygroup_allow_wide_open_public_ipv4.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_allow_wide_open_public_ipv4.ec2_securitygroup_allow_wide_open_public_ipv4 import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_default_restrict_traffic/ec2_securitygroup_default_restrict_traffic_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_default_restrict_traffic/ec2_securitygroup_default_restrict_traffic_test.py index d3223f7629..3c3a4b084d 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_default_restrict_traffic/ec2_securitygroup_default_restrict_traffic_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_default_restrict_traffic/ec2_securitygroup_default_restrict_traffic_test.py @@ -34,19 +34,15 @@ class Test_ec2_securitygroup_default_restrict_traffic: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_default_restrict_traffic.ec2_securitygroup_default_restrict_traffic.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_default_restrict_traffic.ec2_securitygroup_default_restrict_traffic.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_default_restrict_traffic.ec2_securitygroup_default_restrict_traffic.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_default_restrict_traffic.ec2_securitygroup_default_restrict_traffic.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_default_restrict_traffic.ec2_securitygroup_default_restrict_traffic import ( @@ -105,19 +101,15 @@ class Test_ec2_securitygroup_default_restrict_traffic: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_default_restrict_traffic.ec2_securitygroup_default_restrict_traffic.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_default_restrict_traffic.ec2_securitygroup_default_restrict_traffic.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_default_restrict_traffic.ec2_securitygroup_default_restrict_traffic.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_default_restrict_traffic.ec2_securitygroup_default_restrict_traffic.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_default_restrict_traffic.ec2_securitygroup_default_restrict_traffic import ( @@ -178,19 +170,15 @@ class Test_ec2_securitygroup_default_restrict_traffic: audited_regions=[AWS_REGION_US_EAST_1], scan_unused_services=False ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_default_restrict_traffic.ec2_securitygroup_default_restrict_traffic.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_default_restrict_traffic.ec2_securitygroup_default_restrict_traffic.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_default_restrict_traffic.ec2_securitygroup_default_restrict_traffic.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_default_restrict_traffic.ec2_securitygroup_default_restrict_traffic.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_default_restrict_traffic.ec2_securitygroup_default_restrict_traffic import ( @@ -218,19 +206,15 @@ class Test_ec2_securitygroup_default_restrict_traffic: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_default_restrict_traffic.ec2_securitygroup_default_restrict_traffic.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_default_restrict_traffic.ec2_securitygroup_default_restrict_traffic.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_default_restrict_traffic.ec2_securitygroup_default_restrict_traffic.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_default_restrict_traffic.ec2_securitygroup_default_restrict_traffic.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_default_restrict_traffic.ec2_securitygroup_default_restrict_traffic import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_from_launch_wizard/ec2_securitygroup_from_launch_wizard_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_from_launch_wizard/ec2_securitygroup_from_launch_wizard_test.py index 048ec4eb02..ad07cedd3e 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_from_launch_wizard/ec2_securitygroup_from_launch_wizard_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_from_launch_wizard/ec2_securitygroup_from_launch_wizard_test.py @@ -25,15 +25,12 @@ class Test_ec2_securitygroup_from_launch_wizard: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_from_launch_wizard.ec2_securitygroup_from_launch_wizard.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_from_launch_wizard.ec2_securitygroup_from_launch_wizard.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_from_launch_wizard.ec2_securitygroup_from_launch_wizard import ( @@ -67,15 +64,12 @@ class Test_ec2_securitygroup_from_launch_wizard: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_from_launch_wizard.ec2_securitygroup_from_launch_wizard.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_from_launch_wizard.ec2_securitygroup_from_launch_wizard.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_from_launch_wizard.ec2_securitygroup_from_launch_wizard import ( @@ -129,15 +123,12 @@ class Test_ec2_securitygroup_from_launch_wizard: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_from_launch_wizard.ec2_securitygroup_from_launch_wizard.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_from_launch_wizard.ec2_securitygroup_from_launch_wizard.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_from_launch_wizard.ec2_securitygroup_from_launch_wizard import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_not_used/ec2_securitygroup_not_used_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_not_used/ec2_securitygroup_not_used_test.py index d1b77b9f8b..cddb4b2aac 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_not_used/ec2_securitygroup_not_used_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_not_used/ec2_securitygroup_not_used_test.py @@ -27,19 +27,15 @@ class Test_ec2_securitygroup_not_used: audited_regions=["us-east-1", "eu-west-1"] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used.awslambda_client", - new=Lambda(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used.awslambda_client", + new=Lambda(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used import ( @@ -70,19 +66,15 @@ class Test_ec2_securitygroup_not_used: audited_regions=["us-east-1", "eu-west-1"] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used.awslambda_client", - new=Lambda(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used.awslambda_client", + new=Lambda(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used import ( @@ -128,19 +120,15 @@ class Test_ec2_securitygroup_not_used: audited_regions=["us-east-1", "eu-west-1"] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used.awslambda_client", - new=Lambda(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used.awslambda_client", + new=Lambda(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used import ( @@ -214,19 +202,15 @@ class Test_ec2_securitygroup_not_used: audited_regions=["us-east-1", "eu-west-1"] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used.awslambda_client", - new=Lambda(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used.awslambda_client", + new=Lambda(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used import ( @@ -293,19 +277,15 @@ class Test_ec2_securitygroup_not_used: audited_regions=["us-east-1", "eu-west-1"] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used.ec2_client", - new=EC2(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used.awslambda_client", - new=Lambda(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used.ec2_client", + new=EC2(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used.awslambda_client", + new=Lambda(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_not_used.ec2_securitygroup_not_used import ( diff --git a/tests/providers/aws/services/ec2/ec2_securitygroup_with_many_ingress_egress_rules/ec2_securitygroup_with_many_ingress_egress_rules_test.py b/tests/providers/aws/services/ec2/ec2_securitygroup_with_many_ingress_egress_rules/ec2_securitygroup_with_many_ingress_egress_rules_test.py index 68dd1237dc..417578e0a1 100644 --- a/tests/providers/aws/services/ec2/ec2_securitygroup_with_many_ingress_egress_rules/ec2_securitygroup_with_many_ingress_egress_rules_test.py +++ b/tests/providers/aws/services/ec2/ec2_securitygroup_with_many_ingress_egress_rules/ec2_securitygroup_with_many_ingress_egress_rules_test.py @@ -24,15 +24,12 @@ class Test_ec2_securitygroup_with_many_ingress_egress_rules: ) aws_provider._audit_config = {"max_security_group_rules": 50} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_with_many_ingress_egress_rules.ec2_securitygroup_with_many_ingress_egress_rules.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_with_many_ingress_egress_rules.ec2_securitygroup_with_many_ingress_egress_rules.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_with_many_ingress_egress_rules.ec2_securitygroup_with_many_ingress_egress_rules import ( @@ -79,15 +76,12 @@ class Test_ec2_securitygroup_with_many_ingress_egress_rules: ) aws_provider._audit_config = {"max_security_group_rules": 50} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_with_many_ingress_egress_rules.ec2_securitygroup_with_many_ingress_egress_rules.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_with_many_ingress_egress_rules.ec2_securitygroup_with_many_ingress_egress_rules.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_with_many_ingress_egress_rules.ec2_securitygroup_with_many_ingress_egress_rules import ( @@ -144,15 +138,12 @@ class Test_ec2_securitygroup_with_many_ingress_egress_rules: ) aws_provider._audit_config = {"max_security_group_rules": 50} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_securitygroup_with_many_ingress_egress_rules.ec2_securitygroup_with_many_ingress_egress_rules.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_securitygroup_with_many_ingress_egress_rules.ec2_securitygroup_with_many_ingress_egress_rules.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_securitygroup_with_many_ingress_egress_rules.ec2_securitygroup_with_many_ingress_egress_rules import ( diff --git a/tests/providers/aws/services/ec2/ec2_service_test.py b/tests/providers/aws/services/ec2/ec2_service_test.py index 327d85312a..629e319f38 100644 --- a/tests/providers/aws/services/ec2/ec2_service_test.py +++ b/tests/providers/aws/services/ec2/ec2_service_test.py @@ -398,15 +398,12 @@ class Test_EC2_Service: ec2_client.audited_account = AWS_ACCOUNT_NUMBER ec2_client.region = AWS_REGION_US_EAST_1 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_client.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_client.ec2_client", + new=ec2_client, ): assert ( ec2_client.ebs_block_public_access_snapshots_states[0].status @@ -477,15 +474,12 @@ class Test_EC2_Service: ec2_client.audited_account = AWS_ACCOUNT_NUMBER ec2_client.region = AWS_REGION_US_EAST_1 - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_client.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_client.ec2_client", + new=ec2_client, ): assert ec2_client.instance_metadata_defaults[0].http_tokens == "required" diff --git a/tests/providers/aws/services/ec2/ec2_transitgateway_auto_accept_vpc_attachments/ec2_transitgateway_auto_accept_vpc_attachments_test.py b/tests/providers/aws/services/ec2/ec2_transitgateway_auto_accept_vpc_attachments/ec2_transitgateway_auto_accept_vpc_attachments_test.py index c159b6253c..fa31325833 100644 --- a/tests/providers/aws/services/ec2/ec2_transitgateway_auto_accept_vpc_attachments/ec2_transitgateway_auto_accept_vpc_attachments_test.py +++ b/tests/providers/aws/services/ec2/ec2_transitgateway_auto_accept_vpc_attachments/ec2_transitgateway_auto_accept_vpc_attachments_test.py @@ -16,15 +16,12 @@ class Test_ec2_transitgateway_auto_accept_vpc_attachments: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_transitgateway_auto_accept_vpc_attachments.ec2_transitgateway_auto_accept_vpc_attachments.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_transitgateway_auto_accept_vpc_attachments.ec2_transitgateway_auto_accept_vpc_attachments.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_transitgateway_auto_accept_vpc_attachments.ec2_transitgateway_auto_accept_vpc_attachments import ( @@ -48,15 +45,12 @@ class Test_ec2_transitgateway_auto_accept_vpc_attachments: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_transitgateway_auto_accept_vpc_attachments.ec2_transitgateway_auto_accept_vpc_attachments.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_transitgateway_auto_accept_vpc_attachments.ec2_transitgateway_auto_accept_vpc_attachments.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_transitgateway_auto_accept_vpc_attachments.ec2_transitgateway_auto_accept_vpc_attachments import ( @@ -90,15 +84,12 @@ class Test_ec2_transitgateway_auto_accept_vpc_attachments: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_transitgateway_auto_accept_vpc_attachments.ec2_transitgateway_auto_accept_vpc_attachments.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_transitgateway_auto_accept_vpc_attachments.ec2_transitgateway_auto_accept_vpc_attachments.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_transitgateway_auto_accept_vpc_attachments.ec2_transitgateway_auto_accept_vpc_attachments import ( @@ -132,15 +123,12 @@ class Test_ec2_transitgateway_auto_accept_vpc_attachments: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_transitgateway_auto_accept_vpc_attachments.ec2_transitgateway_auto_accept_vpc_attachments.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_transitgateway_auto_accept_vpc_attachments.ec2_transitgateway_auto_accept_vpc_attachments.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_transitgateway_auto_accept_vpc_attachments.ec2_transitgateway_auto_accept_vpc_attachments import ( @@ -189,15 +177,12 @@ class Test_ec2_transitgateway_auto_accept_vpc_attachments: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ec2.ec2_transitgateway_auto_accept_vpc_attachments.ec2_transitgateway_auto_accept_vpc_attachments.ec2_client", - new=EC2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ec2.ec2_transitgateway_auto_accept_vpc_attachments.ec2_transitgateway_auto_accept_vpc_attachments.ec2_client", + new=EC2(aws_provider), ): # Test Check from prowler.providers.aws.services.ec2.ec2_transitgateway_auto_accept_vpc_attachments.ec2_transitgateway_auto_accept_vpc_attachments import ( diff --git a/tests/providers/aws/services/ecr/ecr_registry_scan_images_on_push_enabled/ecr_registry_scan_images_on_push_enabled_test.py b/tests/providers/aws/services/ecr/ecr_registry_scan_images_on_push_enabled/ecr_registry_scan_images_on_push_enabled_test.py index f736866742..3db3dd4d6f 100644 --- a/tests/providers/aws/services/ecr/ecr_registry_scan_images_on_push_enabled/ecr_registry_scan_images_on_push_enabled_test.py +++ b/tests/providers/aws/services/ecr/ecr_registry_scan_images_on_push_enabled/ecr_registry_scan_images_on_push_enabled_test.py @@ -23,15 +23,12 @@ class Test_ecr_registry_scan_images_on_push_enabled: ecr_client = mock.MagicMock ecr_client.registries = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_registry_scan_images_on_push_enabled.ecr_registry_scan_images_on_push_enabled.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_registry_scan_images_on_push_enabled.ecr_registry_scan_images_on_push_enabled.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_registry_scan_images_on_push_enabled.ecr_registry_scan_images_on_push_enabled import ( ecr_registry_scan_images_on_push_enabled, @@ -53,15 +50,12 @@ class Test_ecr_registry_scan_images_on_push_enabled: rules=[], ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_registry_scan_images_on_push_enabled.ecr_registry_scan_images_on_push_enabled.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_registry_scan_images_on_push_enabled.ecr_registry_scan_images_on_push_enabled.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_registry_scan_images_on_push_enabled.ecr_registry_scan_images_on_push_enabled import ( ecr_registry_scan_images_on_push_enabled, @@ -99,15 +93,12 @@ class Test_ecr_registry_scan_images_on_push_enabled: ], ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_registry_scan_images_on_push_enabled.ecr_registry_scan_images_on_push_enabled.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_registry_scan_images_on_push_enabled.ecr_registry_scan_images_on_push_enabled.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_registry_scan_images_on_push_enabled.ecr_registry_scan_images_on_push_enabled import ( ecr_registry_scan_images_on_push_enabled, @@ -156,15 +147,12 @@ class Test_ecr_registry_scan_images_on_push_enabled: ], ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_registry_scan_images_on_push_enabled.ecr_registry_scan_images_on_push_enabled.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_registry_scan_images_on_push_enabled.ecr_registry_scan_images_on_push_enabled.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_registry_scan_images_on_push_enabled.ecr_registry_scan_images_on_push_enabled import ( ecr_registry_scan_images_on_push_enabled, @@ -208,15 +196,12 @@ class Test_ecr_registry_scan_images_on_push_enabled: rules=[], ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_registry_scan_images_on_push_enabled.ecr_registry_scan_images_on_push_enabled.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_registry_scan_images_on_push_enabled.ecr_registry_scan_images_on_push_enabled.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_registry_scan_images_on_push_enabled.ecr_registry_scan_images_on_push_enabled import ( ecr_registry_scan_images_on_push_enabled, diff --git a/tests/providers/aws/services/ecr/ecr_repositories_lifecycle_policy_enabled/ecr_repositories_lifecycle_policy_enabled_test.py b/tests/providers/aws/services/ecr/ecr_repositories_lifecycle_policy_enabled/ecr_repositories_lifecycle_policy_enabled_test.py index e40d3ffc5f..3e430e8095 100644 --- a/tests/providers/aws/services/ecr/ecr_repositories_lifecycle_policy_enabled/ecr_repositories_lifecycle_policy_enabled_test.py +++ b/tests/providers/aws/services/ecr/ecr_repositories_lifecycle_policy_enabled/ecr_repositories_lifecycle_policy_enabled_test.py @@ -31,15 +31,12 @@ class Test_ecr_repositories_lifecycle_policy_enabled: ecr_client = mock.MagicMock ecr_client.registries = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_lifecycle_policy_enabled.ecr_repositories_lifecycle_policy_enabled.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_lifecycle_policy_enabled.ecr_repositories_lifecycle_policy_enabled.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_lifecycle_policy_enabled.ecr_repositories_lifecycle_policy_enabled import ( ecr_repositories_lifecycle_policy_enabled, @@ -61,15 +58,12 @@ class Test_ecr_repositories_lifecycle_policy_enabled: rules=[], ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_lifecycle_policy_enabled.ecr_repositories_lifecycle_policy_enabled.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_lifecycle_policy_enabled.ecr_repositories_lifecycle_policy_enabled.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_lifecycle_policy_enabled.ecr_repositories_lifecycle_policy_enabled import ( ecr_repositories_lifecycle_policy_enabled, @@ -101,15 +95,12 @@ class Test_ecr_repositories_lifecycle_policy_enabled: ], ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_lifecycle_policy_enabled.ecr_repositories_lifecycle_policy_enabled.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_lifecycle_policy_enabled.ecr_repositories_lifecycle_policy_enabled.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_lifecycle_policy_enabled.ecr_repositories_lifecycle_policy_enabled import ( ecr_repositories_lifecycle_policy_enabled, @@ -149,15 +140,12 @@ class Test_ecr_repositories_lifecycle_policy_enabled: ], ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_lifecycle_policy_enabled.ecr_repositories_lifecycle_policy_enabled.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_lifecycle_policy_enabled.ecr_repositories_lifecycle_policy_enabled.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_lifecycle_policy_enabled.ecr_repositories_lifecycle_policy_enabled import ( ecr_repositories_lifecycle_policy_enabled, diff --git a/tests/providers/aws/services/ecr/ecr_repositories_not_publicly_accessible/ecr_repositories_not_publicly_accessible_fixer_test.py b/tests/providers/aws/services/ecr/ecr_repositories_not_publicly_accessible/ecr_repositories_not_publicly_accessible_fixer_test.py index e0e68c571f..d2b4953cdc 100644 --- a/tests/providers/aws/services/ecr/ecr_repositories_not_publicly_accessible/ecr_repositories_not_publicly_accessible_fixer_test.py +++ b/tests/providers/aws/services/ecr/ecr_repositories_not_publicly_accessible/ecr_repositories_not_publicly_accessible_fixer_test.py @@ -36,15 +36,12 @@ class Test_ecr_repositories_not_publicly_accessible_fixer: from prowler.providers.aws.services.ecr.ecr_service import ECR - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_not_publicly_accessible.ecr_repositories_not_publicly_accessible_fixer.ecr_client", - new=ECR(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_not_publicly_accessible.ecr_repositories_not_publicly_accessible_fixer.ecr_client", + new=ECR(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ecr.ecr_repositories_not_publicly_accessible.ecr_repositories_not_publicly_accessible_fixer import ( @@ -81,15 +78,12 @@ class Test_ecr_repositories_not_publicly_accessible_fixer: from prowler.providers.aws.services.ecr.ecr_service import ECR - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_not_publicly_accessible.ecr_repositories_not_publicly_accessible_fixer.ecr_client", - new=ECR(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_not_publicly_accessible.ecr_repositories_not_publicly_accessible_fixer.ecr_client", + new=ECR(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ecr.ecr_repositories_not_publicly_accessible.ecr_repositories_not_publicly_accessible_fixer import ( @@ -126,15 +120,12 @@ class Test_ecr_repositories_not_publicly_accessible_fixer: from prowler.providers.aws.services.ecr.ecr_service import ECR - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_not_publicly_accessible.ecr_repositories_not_publicly_accessible_fixer.ecr_client", - new=ECR(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_not_publicly_accessible.ecr_repositories_not_publicly_accessible_fixer.ecr_client", + new=ECR(aws_provider), ): # Test Fixer from prowler.providers.aws.services.ecr.ecr_repositories_not_publicly_accessible.ecr_repositories_not_publicly_accessible_fixer import ( diff --git a/tests/providers/aws/services/ecr/ecr_repositories_scan_images_on_push_enabled/ecr_repositories_scan_images_on_push_enabled_test.py b/tests/providers/aws/services/ecr/ecr_repositories_scan_images_on_push_enabled/ecr_repositories_scan_images_on_push_enabled_test.py index f633228dab..3e86ba0829 100644 --- a/tests/providers/aws/services/ecr/ecr_repositories_scan_images_on_push_enabled/ecr_repositories_scan_images_on_push_enabled_test.py +++ b/tests/providers/aws/services/ecr/ecr_repositories_scan_images_on_push_enabled/ecr_repositories_scan_images_on_push_enabled_test.py @@ -31,15 +31,12 @@ class Test_ecr_repositories_scan_images_on_push_enabled: ecr_client = mock.MagicMock ecr_client.registries = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_scan_images_on_push_enabled.ecr_repositories_scan_images_on_push_enabled.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_scan_images_on_push_enabled.ecr_repositories_scan_images_on_push_enabled.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_scan_images_on_push_enabled.ecr_repositories_scan_images_on_push_enabled import ( ecr_repositories_scan_images_on_push_enabled, @@ -61,15 +58,12 @@ class Test_ecr_repositories_scan_images_on_push_enabled: rules=[], ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_scan_images_on_push_enabled.ecr_repositories_scan_images_on_push_enabled.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_scan_images_on_push_enabled.ecr_repositories_scan_images_on_push_enabled.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_scan_images_on_push_enabled.ecr_repositories_scan_images_on_push_enabled import ( ecr_repositories_scan_images_on_push_enabled, @@ -101,15 +95,12 @@ class Test_ecr_repositories_scan_images_on_push_enabled: rules=[], ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_scan_images_on_push_enabled.ecr_repositories_scan_images_on_push_enabled.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_scan_images_on_push_enabled.ecr_repositories_scan_images_on_push_enabled.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_scan_images_on_push_enabled.ecr_repositories_scan_images_on_push_enabled import ( ecr_repositories_scan_images_on_push_enabled, @@ -148,15 +139,12 @@ class Test_ecr_repositories_scan_images_on_push_enabled: rules=[], ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_scan_images_on_push_enabled.ecr_repositories_scan_images_on_push_enabled.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_scan_images_on_push_enabled.ecr_repositories_scan_images_on_push_enabled.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_scan_images_on_push_enabled.ecr_repositories_scan_images_on_push_enabled import ( ecr_repositories_scan_images_on_push_enabled, diff --git a/tests/providers/aws/services/ecr/ecr_repositories_scan_vulnerabilities_in_latest_image/ecr_repositories_scan_vulnerabilities_in_latest_image_test.py b/tests/providers/aws/services/ecr/ecr_repositories_scan_vulnerabilities_in_latest_image/ecr_repositories_scan_vulnerabilities_in_latest_image_test.py index d4a1daeb03..8bc08f005b 100644 --- a/tests/providers/aws/services/ecr/ecr_repositories_scan_vulnerabilities_in_latest_image/ecr_repositories_scan_vulnerabilities_in_latest_image_test.py +++ b/tests/providers/aws/services/ecr/ecr_repositories_scan_vulnerabilities_in_latest_image/ecr_repositories_scan_vulnerabilities_in_latest_image_test.py @@ -42,15 +42,12 @@ class Test_ecr_repositories_scan_vulnerabilities_in_latest_image: ecr_client.registries = {} ecr_client.audit_config = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image import ( ecr_repositories_scan_vulnerabilities_in_latest_image, @@ -73,15 +70,12 @@ class Test_ecr_repositories_scan_vulnerabilities_in_latest_image: ) ecr_client.audit_config = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image import ( ecr_repositories_scan_vulnerabilities_in_latest_image, @@ -114,15 +108,12 @@ class Test_ecr_repositories_scan_vulnerabilities_in_latest_image: ) ecr_client.audit_config = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image import ( ecr_repositories_scan_vulnerabilities_in_latest_image, @@ -167,15 +158,12 @@ class Test_ecr_repositories_scan_vulnerabilities_in_latest_image: ) ecr_client.audit_config = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image import ( ecr_repositories_scan_vulnerabilities_in_latest_image, @@ -229,15 +217,12 @@ class Test_ecr_repositories_scan_vulnerabilities_in_latest_image: ) ecr_client.audit_config = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image import ( ecr_repositories_scan_vulnerabilities_in_latest_image, @@ -295,15 +280,12 @@ class Test_ecr_repositories_scan_vulnerabilities_in_latest_image: "ecr_repository_vulnerability_minimum_severity": "MEDIUM" } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image import ( ecr_repositories_scan_vulnerabilities_in_latest_image, @@ -361,15 +343,12 @@ class Test_ecr_repositories_scan_vulnerabilities_in_latest_image: "ecr_repository_vulnerability_minimum_severity": "HIGH" } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image import ( ecr_repositories_scan_vulnerabilities_in_latest_image, @@ -427,15 +406,12 @@ class Test_ecr_repositories_scan_vulnerabilities_in_latest_image: "ecr_repository_vulnerability_minimum_severity": "CRITICAL" } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image import ( ecr_repositories_scan_vulnerabilities_in_latest_image, @@ -493,15 +469,12 @@ class Test_ecr_repositories_scan_vulnerabilities_in_latest_image: "ecr_repository_vulnerability_minimum_severity": "CRITICAL" } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image import ( ecr_repositories_scan_vulnerabilities_in_latest_image, @@ -559,15 +532,12 @@ class Test_ecr_repositories_scan_vulnerabilities_in_latest_image: "ecr_repository_vulnerability_minimum_severity": "HIGH" } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image import ( ecr_repositories_scan_vulnerabilities_in_latest_image, @@ -621,15 +591,12 @@ class Test_ecr_repositories_scan_vulnerabilities_in_latest_image: ) ecr_client.audit_config = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image import ( ecr_repositories_scan_vulnerabilities_in_latest_image, @@ -683,15 +650,12 @@ class Test_ecr_repositories_scan_vulnerabilities_in_latest_image: ) ecr_client.audit_config = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_scan_vulnerabilities_in_latest_image.ecr_repositories_scan_vulnerabilities_in_latest_image import ( ecr_repositories_scan_vulnerabilities_in_latest_image, diff --git a/tests/providers/aws/services/ecr/ecr_repositories_tag_immutability/ecr_repositories_tag_immutability_test.py b/tests/providers/aws/services/ecr/ecr_repositories_tag_immutability/ecr_repositories_tag_immutability_test.py index f320b52be2..128291b129 100644 --- a/tests/providers/aws/services/ecr/ecr_repositories_tag_immutability/ecr_repositories_tag_immutability_test.py +++ b/tests/providers/aws/services/ecr/ecr_repositories_tag_immutability/ecr_repositories_tag_immutability_test.py @@ -31,15 +31,12 @@ class Test_ecr_repositories_tag_immutability: ecr_client = mock.MagicMock ecr_client.registries = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_tag_immutability.ecr_repositories_tag_immutability.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_tag_immutability.ecr_repositories_tag_immutability.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_tag_immutability.ecr_repositories_tag_immutability import ( ecr_repositories_tag_immutability, @@ -61,15 +58,12 @@ class Test_ecr_repositories_tag_immutability: rules=[], ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_tag_immutability.ecr_repositories_tag_immutability.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_tag_immutability.ecr_repositories_tag_immutability.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_tag_immutability.ecr_repositories_tag_immutability import ( ecr_repositories_tag_immutability, @@ -102,15 +96,12 @@ class Test_ecr_repositories_tag_immutability: ], ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_tag_immutability.ecr_repositories_tag_immutability.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_tag_immutability.ecr_repositories_tag_immutability.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_tag_immutability.ecr_repositories_tag_immutability import ( ecr_repositories_tag_immutability, @@ -151,15 +142,12 @@ class Test_ecr_repositories_tag_immutability: ], ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider(), - ), - mock.patch( - "prowler.providers.aws.services.ecr.ecr_repositories_tag_immutability.ecr_repositories_tag_immutability.ecr_client", - ecr_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider(), + ), mock.patch( + "prowler.providers.aws.services.ecr.ecr_repositories_tag_immutability.ecr_repositories_tag_immutability.ecr_client", + ecr_client, ): from prowler.providers.aws.services.ecr.ecr_repositories_tag_immutability.ecr_repositories_tag_immutability import ( ecr_repositories_tag_immutability, diff --git a/tests/providers/aws/services/ecs/ecs_cluster_container_insights_enabled/ecs_cluster_container_insights_enabled_test.py b/tests/providers/aws/services/ecs/ecs_cluster_container_insights_enabled/ecs_cluster_container_insights_enabled_test.py index 45483453c7..780ccdca8c 100644 --- a/tests/providers/aws/services/ecs/ecs_cluster_container_insights_enabled/ecs_cluster_container_insights_enabled_test.py +++ b/tests/providers/aws/services/ecs/ecs_cluster_container_insights_enabled/ecs_cluster_container_insights_enabled_test.py @@ -15,15 +15,12 @@ class Test_ecs_clusters_container_insights_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ecs.ecs_cluster_container_insights_enabled.ecs_cluster_container_insights_enabled.ecs_client", - new=ECS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ecs.ecs_cluster_container_insights_enabled.ecs_cluster_container_insights_enabled.ecs_client", + new=ECS(aws_provider), ): from prowler.providers.aws.services.ecs.ecs_cluster_container_insights_enabled.ecs_cluster_container_insights_enabled import ( ecs_cluster_container_insights_enabled, @@ -46,15 +43,12 @@ class Test_ecs_clusters_container_insights_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ecs.ecs_cluster_container_insights_enabled.ecs_cluster_container_insights_enabled.ecs_client", - new=ECS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ecs.ecs_cluster_container_insights_enabled.ecs_cluster_container_insights_enabled.ecs_client", + new=ECS(aws_provider), ): from prowler.providers.aws.services.ecs.ecs_cluster_container_insights_enabled.ecs_cluster_container_insights_enabled import ( ecs_cluster_container_insights_enabled, @@ -85,15 +79,12 @@ class Test_ecs_clusters_container_insights_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ecs.ecs_cluster_container_insights_enabled.ecs_cluster_container_insights_enabled.ecs_client", - new=ECS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ecs.ecs_cluster_container_insights_enabled.ecs_cluster_container_insights_enabled.ecs_client", + new=ECS(aws_provider), ): from prowler.providers.aws.services.ecs.ecs_cluster_container_insights_enabled.ecs_cluster_container_insights_enabled import ( ecs_cluster_container_insights_enabled, @@ -124,15 +115,12 @@ class Test_ecs_clusters_container_insights_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ecs.ecs_cluster_container_insights_enabled.ecs_cluster_container_insights_enabled.ecs_client", - new=ECS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ecs.ecs_cluster_container_insights_enabled.ecs_cluster_container_insights_enabled.ecs_client", + new=ECS(aws_provider), ): from prowler.providers.aws.services.ecs.ecs_cluster_container_insights_enabled.ecs_cluster_container_insights_enabled import ( ecs_cluster_container_insights_enabled, diff --git a/tests/providers/aws/services/ecs/ecs_task_definitions_containers_readonly_access/ecs_task_definitions_containers_readonly_access_test.py b/tests/providers/aws/services/ecs/ecs_task_definitions_containers_readonly_access/ecs_task_definitions_containers_readonly_access_test.py index a9aa9901a8..a162ef87d8 100644 --- a/tests/providers/aws/services/ecs/ecs_task_definitions_containers_readonly_access/ecs_task_definitions_containers_readonly_access_test.py +++ b/tests/providers/aws/services/ecs/ecs_task_definitions_containers_readonly_access/ecs_task_definitions_containers_readonly_access_test.py @@ -21,15 +21,12 @@ class Test_ecs_task_definitions_containers_readonly_access: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_containers_readonly_access.ecs_task_definitions_containers_readonly_access.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_containers_readonly_access.ecs_task_definitions_containers_readonly_access.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_containers_readonly_access.ecs_task_definitions_containers_readonly_access import ( ecs_task_definitions_containers_readonly_access, @@ -62,15 +59,12 @@ class Test_ecs_task_definitions_containers_readonly_access: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_containers_readonly_access.ecs_task_definitions_containers_readonly_access.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_containers_readonly_access.ecs_task_definitions_containers_readonly_access.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_containers_readonly_access.ecs_task_definitions_containers_readonly_access import ( ecs_task_definitions_containers_readonly_access, @@ -112,15 +106,12 @@ class Test_ecs_task_definitions_containers_readonly_access: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_containers_readonly_access.ecs_task_definitions_containers_readonly_access.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_containers_readonly_access.ecs_task_definitions_containers_readonly_access.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_containers_readonly_access.ecs_task_definitions_containers_readonly_access import ( ecs_task_definitions_containers_readonly_access, @@ -171,15 +162,12 @@ class Test_ecs_task_definitions_containers_readonly_access: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_containers_readonly_access.ecs_task_definitions_containers_readonly_access.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_containers_readonly_access.ecs_task_definitions_containers_readonly_access.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_containers_readonly_access.ecs_task_definitions_containers_readonly_access import ( ecs_task_definitions_containers_readonly_access, diff --git a/tests/providers/aws/services/ecs/ecs_task_definitions_host_namespace_not_shared/ecs_task_definitions_host_namespace_not_shared_test.py b/tests/providers/aws/services/ecs/ecs_task_definitions_host_namespace_not_shared/ecs_task_definitions_host_namespace_not_shared_test.py index 0ad2650266..92346124e0 100644 --- a/tests/providers/aws/services/ecs/ecs_task_definitions_host_namespace_not_shared/ecs_task_definitions_host_namespace_not_shared_test.py +++ b/tests/providers/aws/services/ecs/ecs_task_definitions_host_namespace_not_shared/ecs_task_definitions_host_namespace_not_shared_test.py @@ -17,15 +17,12 @@ class Test_ecs_task_definitions_host_namespace_not_shared: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_host_namespace_not_shared.ecs_task_definitions_host_namespace_not_shared.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_host_namespace_not_shared.ecs_task_definitions_host_namespace_not_shared.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_host_namespace_not_shared.ecs_task_definitions_host_namespace_not_shared import ( ecs_task_definitions_host_namespace_not_shared, @@ -59,15 +56,12 @@ class Test_ecs_task_definitions_host_namespace_not_shared: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_host_namespace_not_shared.ecs_task_definitions_host_namespace_not_shared.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_host_namespace_not_shared.ecs_task_definitions_host_namespace_not_shared.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_host_namespace_not_shared.ecs_task_definitions_host_namespace_not_shared import ( ecs_task_definitions_host_namespace_not_shared, @@ -110,15 +104,12 @@ class Test_ecs_task_definitions_host_namespace_not_shared: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_host_namespace_not_shared.ecs_task_definitions_host_namespace_not_shared.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_host_namespace_not_shared.ecs_task_definitions_host_namespace_not_shared.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_host_namespace_not_shared.ecs_task_definitions_host_namespace_not_shared import ( ecs_task_definitions_host_namespace_not_shared, @@ -160,15 +151,12 @@ class Test_ecs_task_definitions_host_namespace_not_shared: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_host_namespace_not_shared.ecs_task_definitions_host_namespace_not_shared.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_host_namespace_not_shared.ecs_task_definitions_host_namespace_not_shared.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_host_namespace_not_shared.ecs_task_definitions_host_namespace_not_shared import ( ecs_task_definitions_host_namespace_not_shared, diff --git a/tests/providers/aws/services/ecs/ecs_task_definitions_host_networking_mode_users/ecs_task_definitions_host_networking_mode_users_test.py b/tests/providers/aws/services/ecs/ecs_task_definitions_host_networking_mode_users/ecs_task_definitions_host_networking_mode_users_test.py index 7047028a76..6458a4e527 100644 --- a/tests/providers/aws/services/ecs/ecs_task_definitions_host_networking_mode_users/ecs_task_definitions_host_networking_mode_users_test.py +++ b/tests/providers/aws/services/ecs/ecs_task_definitions_host_networking_mode_users/ecs_task_definitions_host_networking_mode_users_test.py @@ -16,15 +16,12 @@ class Test_ecs_task_definitions_host_networking_mode_users: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_host_networking_mode_users.ecs_task_definitions_host_networking_mode_users.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_host_networking_mode_users.ecs_task_definitions_host_networking_mode_users.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_host_networking_mode_users.ecs_task_definitions_host_networking_mode_users import ( ecs_task_definitions_host_networking_mode_users, @@ -58,15 +55,12 @@ class Test_ecs_task_definitions_host_networking_mode_users: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_host_networking_mode_users.ecs_task_definitions_host_networking_mode_users.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_host_networking_mode_users.ecs_task_definitions_host_networking_mode_users.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_host_networking_mode_users.ecs_task_definitions_host_networking_mode_users import ( ecs_task_definitions_host_networking_mode_users, @@ -109,15 +103,12 @@ class Test_ecs_task_definitions_host_networking_mode_users: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_host_networking_mode_users.ecs_task_definitions_host_networking_mode_users.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_host_networking_mode_users.ecs_task_definitions_host_networking_mode_users.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_host_networking_mode_users.ecs_task_definitions_host_networking_mode_users import ( ecs_task_definitions_host_networking_mode_users, @@ -160,15 +151,12 @@ class Test_ecs_task_definitions_host_networking_mode_users: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_host_networking_mode_users.ecs_task_definitions_host_networking_mode_users.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_host_networking_mode_users.ecs_task_definitions_host_networking_mode_users.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_host_networking_mode_users.ecs_task_definitions_host_networking_mode_users import ( ecs_task_definitions_host_networking_mode_users, @@ -211,15 +199,12 @@ class Test_ecs_task_definitions_host_networking_mode_users: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_host_networking_mode_users.ecs_task_definitions_host_networking_mode_users.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_host_networking_mode_users.ecs_task_definitions_host_networking_mode_users.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_host_networking_mode_users.ecs_task_definitions_host_networking_mode_users import ( ecs_task_definitions_host_networking_mode_users, diff --git a/tests/providers/aws/services/ecs/ecs_task_definitions_logging_block_mode/ecs_task_definitions_logging_block_mode_test.py b/tests/providers/aws/services/ecs/ecs_task_definitions_logging_block_mode/ecs_task_definitions_logging_block_mode_test.py index 586a1b57d5..9ab3c16f96 100644 --- a/tests/providers/aws/services/ecs/ecs_task_definitions_logging_block_mode/ecs_task_definitions_logging_block_mode_test.py +++ b/tests/providers/aws/services/ecs/ecs_task_definitions_logging_block_mode/ecs_task_definitions_logging_block_mode_test.py @@ -16,15 +16,12 @@ class Test_ecs_task_definitions_logging_block_mode: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_logging_block_mode.ecs_task_definitions_logging_block_mode.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_logging_block_mode.ecs_task_definitions_logging_block_mode.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_logging_block_mode.ecs_task_definitions_logging_block_mode import ( ecs_task_definitions_logging_block_mode, @@ -57,15 +54,12 @@ class Test_ecs_task_definitions_logging_block_mode: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_logging_block_mode.ecs_task_definitions_logging_block_mode.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_logging_block_mode.ecs_task_definitions_logging_block_mode.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_logging_block_mode.ecs_task_definitions_logging_block_mode import ( ecs_task_definitions_logging_block_mode, @@ -99,15 +93,12 @@ class Test_ecs_task_definitions_logging_block_mode: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_logging_block_mode.ecs_task_definitions_logging_block_mode.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_logging_block_mode.ecs_task_definitions_logging_block_mode.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_logging_block_mode.ecs_task_definitions_logging_block_mode import ( ecs_task_definitions_logging_block_mode, @@ -156,15 +147,12 @@ class Test_ecs_task_definitions_logging_block_mode: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_logging_block_mode.ecs_task_definitions_logging_block_mode.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_logging_block_mode.ecs_task_definitions_logging_block_mode.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_logging_block_mode.ecs_task_definitions_logging_block_mode import ( ecs_task_definitions_logging_block_mode, diff --git a/tests/providers/aws/services/ecs/ecs_task_definitions_logging_enabled/ecs_task_definitions_logging_enabled_test.py b/tests/providers/aws/services/ecs/ecs_task_definitions_logging_enabled/ecs_task_definitions_logging_enabled_test.py index 0423e4467d..d49da4eaa9 100644 --- a/tests/providers/aws/services/ecs/ecs_task_definitions_logging_enabled/ecs_task_definitions_logging_enabled_test.py +++ b/tests/providers/aws/services/ecs/ecs_task_definitions_logging_enabled/ecs_task_definitions_logging_enabled_test.py @@ -16,15 +16,12 @@ class Test_ecs_task_definitions_logging_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_logging_enabled.ecs_task_definitions_logging_enabled.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_logging_enabled.ecs_task_definitions_logging_enabled.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_logging_enabled.ecs_task_definitions_logging_enabled import ( ecs_task_definitions_logging_enabled, @@ -57,15 +54,12 @@ class Test_ecs_task_definitions_logging_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_logging_enabled.ecs_task_definitions_logging_enabled.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_logging_enabled.ecs_task_definitions_logging_enabled.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_logging_enabled.ecs_task_definitions_logging_enabled import ( ecs_task_definitions_logging_enabled, @@ -108,15 +102,12 @@ class Test_ecs_task_definitions_logging_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_logging_enabled.ecs_task_definitions_logging_enabled.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_logging_enabled.ecs_task_definitions_logging_enabled.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_logging_enabled.ecs_task_definitions_logging_enabled import ( ecs_task_definitions_logging_enabled, diff --git a/tests/providers/aws/services/ecs/ecs_task_definitions_no_environment_secrets/ecs_task_definitions_no_environment_secrets_test.py b/tests/providers/aws/services/ecs/ecs_task_definitions_no_environment_secrets/ecs_task_definitions_no_environment_secrets_test.py index 24c27ccf0b..f6b5c411c2 100644 --- a/tests/providers/aws/services/ecs/ecs_task_definitions_no_environment_secrets/ecs_task_definitions_no_environment_secrets_test.py +++ b/tests/providers/aws/services/ecs/ecs_task_definitions_no_environment_secrets/ecs_task_definitions_no_environment_secrets_test.py @@ -2,7 +2,6 @@ from unittest.mock import patch from boto3 import client from moto import mock_aws - from tests.providers.aws.utils import AWS_REGION_US_EAST_1, set_mocked_aws_provider TASK_NAME = "test-task" @@ -22,15 +21,12 @@ class Test_ecs_task_definitions_no_environment_secrets: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets import ( ecs_task_definitions_no_environment_secrets, @@ -68,15 +64,12 @@ class Test_ecs_task_definitions_no_environment_secrets: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets import ( ecs_task_definitions_no_environment_secrets, @@ -123,15 +116,12 @@ class Test_ecs_task_definitions_no_environment_secrets: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets import ( ecs_task_definitions_no_environment_secrets, @@ -178,15 +168,12 @@ class Test_ecs_task_definitions_no_environment_secrets: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets import ( ecs_task_definitions_no_environment_secrets, @@ -233,15 +220,12 @@ class Test_ecs_task_definitions_no_environment_secrets: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets import ( ecs_task_definitions_no_environment_secrets, @@ -292,15 +276,12 @@ class Test_ecs_task_definitions_no_environment_secrets: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets import ( ecs_task_definitions_no_environment_secrets, @@ -351,15 +332,12 @@ class Test_ecs_task_definitions_no_environment_secrets: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_no_environment_secrets.ecs_task_definitions_no_environment_secrets import ( ecs_task_definitions_no_environment_secrets, diff --git a/tests/providers/aws/services/ecs/ecs_task_definitions_no_privileged_containers/ecs_task_definitions_no_privileged_containers_test.py b/tests/providers/aws/services/ecs/ecs_task_definitions_no_privileged_containers/ecs_task_definitions_no_privileged_containers_test.py index ae670e7f4b..a18f0c1205 100644 --- a/tests/providers/aws/services/ecs/ecs_task_definitions_no_privileged_containers/ecs_task_definitions_no_privileged_containers_test.py +++ b/tests/providers/aws/services/ecs/ecs_task_definitions_no_privileged_containers/ecs_task_definitions_no_privileged_containers_test.py @@ -16,15 +16,12 @@ class Test_ecs_task_definitions_no_privileged_containers: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_no_privileged_containers.ecs_task_definitions_no_privileged_containers.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_no_privileged_containers.ecs_task_definitions_no_privileged_containers.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_no_privileged_containers.ecs_task_definitions_no_privileged_containers import ( ecs_task_definitions_no_privileged_containers, @@ -57,15 +54,12 @@ class Test_ecs_task_definitions_no_privileged_containers: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_no_privileged_containers.ecs_task_definitions_no_privileged_containers.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_no_privileged_containers.ecs_task_definitions_no_privileged_containers.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_no_privileged_containers.ecs_task_definitions_no_privileged_containers import ( ecs_task_definitions_no_privileged_containers, @@ -107,15 +101,12 @@ class Test_ecs_task_definitions_no_privileged_containers: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.ecs.ecs_task_definitions_no_privileged_containers.ecs_task_definitions_no_privileged_containers.ecs_client", - new=ECS(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.ecs.ecs_task_definitions_no_privileged_containers.ecs_task_definitions_no_privileged_containers.ecs_client", + new=ECS(mocked_aws_provider), ): from prowler.providers.aws.services.ecs.ecs_task_definitions_no_privileged_containers.ecs_task_definitions_no_privileged_containers import ( ecs_task_definitions_no_privileged_containers, diff --git a/tests/providers/aws/services/efs/efs_access_point_enforce_root_directory/efs_access_point_enforce_root_directory_test.py b/tests/providers/aws/services/efs/efs_access_point_enforce_root_directory/efs_access_point_enforce_root_directory_test.py index 9e7b4aaf2c..f7d1fe2cf9 100644 --- a/tests/providers/aws/services/efs/efs_access_point_enforce_root_directory/efs_access_point_enforce_root_directory_test.py +++ b/tests/providers/aws/services/efs/efs_access_point_enforce_root_directory/efs_access_point_enforce_root_directory_test.py @@ -19,15 +19,12 @@ class Test_efs_access_point_enforce_root_directory: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_access_point_enforce_root_directory.efs_access_point_enforce_root_directory.efs_client", - new=EFS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_access_point_enforce_root_directory.efs_access_point_enforce_root_directory.efs_client", + new=EFS(aws_provider), ): from prowler.providers.aws.services.efs.efs_access_point_enforce_root_directory.efs_access_point_enforce_root_directory import ( efs_access_point_enforce_root_directory, @@ -45,15 +42,12 @@ class Test_efs_access_point_enforce_root_directory: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_access_point_enforce_root_directory.efs_access_point_enforce_root_directory.efs_client", - new=EFS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_access_point_enforce_root_directory.efs_access_point_enforce_root_directory.efs_client", + new=EFS(aws_provider), ): from prowler.providers.aws.services.efs.efs_access_point_enforce_root_directory.efs_access_point_enforce_root_directory import ( efs_access_point_enforce_root_directory, @@ -78,15 +72,12 @@ class Test_efs_access_point_enforce_root_directory: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_access_point_enforce_root_directory.efs_access_point_enforce_root_directory.efs_client", - new=EFS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_access_point_enforce_root_directory.efs_access_point_enforce_root_directory.efs_client", + new=EFS(aws_provider), ): from prowler.providers.aws.services.efs.efs_access_point_enforce_root_directory.efs_access_point_enforce_root_directory import ( efs_access_point_enforce_root_directory, @@ -121,15 +112,12 @@ class Test_efs_access_point_enforce_root_directory: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_access_point_enforce_root_directory.efs_access_point_enforce_root_directory.efs_client", - new=EFS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_access_point_enforce_root_directory.efs_access_point_enforce_root_directory.efs_client", + new=EFS(aws_provider), ): from prowler.providers.aws.services.efs.efs_access_point_enforce_root_directory.efs_access_point_enforce_root_directory import ( efs_access_point_enforce_root_directory, diff --git a/tests/providers/aws/services/efs/efs_access_point_enforce_user_identity/efs_access_point_enforce_user_identity_test.py b/tests/providers/aws/services/efs/efs_access_point_enforce_user_identity/efs_access_point_enforce_user_identity_test.py index d45f176d1c..51d20e1668 100644 --- a/tests/providers/aws/services/efs/efs_access_point_enforce_user_identity/efs_access_point_enforce_user_identity_test.py +++ b/tests/providers/aws/services/efs/efs_access_point_enforce_user_identity/efs_access_point_enforce_user_identity_test.py @@ -19,15 +19,12 @@ class Test_efs_access_point_enforce_user_identity: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_access_point_enforce_user_identity.efs_access_point_enforce_user_identity.efs_client", - new=EFS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_access_point_enforce_user_identity.efs_access_point_enforce_user_identity.efs_client", + new=EFS(aws_provider), ): from prowler.providers.aws.services.efs.efs_access_point_enforce_user_identity.efs_access_point_enforce_user_identity import ( efs_access_point_enforce_user_identity, @@ -45,15 +42,12 @@ class Test_efs_access_point_enforce_user_identity: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_access_point_enforce_user_identity.efs_access_point_enforce_user_identity.efs_client", - new=EFS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_access_point_enforce_user_identity.efs_access_point_enforce_user_identity.efs_client", + new=EFS(aws_provider), ): from prowler.providers.aws.services.efs.efs_access_point_enforce_user_identity.efs_access_point_enforce_user_identity import ( efs_access_point_enforce_user_identity, @@ -77,15 +71,12 @@ class Test_efs_access_point_enforce_user_identity: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_access_point_enforce_user_identity.efs_access_point_enforce_user_identity.efs_client", - new=EFS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_access_point_enforce_user_identity.efs_access_point_enforce_user_identity.efs_client", + new=EFS(aws_provider), ): from prowler.providers.aws.services.efs.efs_access_point_enforce_user_identity.efs_access_point_enforce_user_identity import ( efs_access_point_enforce_user_identity, @@ -120,15 +111,12 @@ class Test_efs_access_point_enforce_user_identity: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_access_point_enforce_user_identity.efs_access_point_enforce_user_identity.efs_client", - new=EFS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_access_point_enforce_user_identity.efs_access_point_enforce_user_identity.efs_client", + new=EFS(aws_provider), ): from prowler.providers.aws.services.efs.efs_access_point_enforce_user_identity.efs_access_point_enforce_user_identity import ( efs_access_point_enforce_user_identity, diff --git a/tests/providers/aws/services/efs/efs_encryption_at_rest_enabled/efs_encryption_at_rest_enabled_test.py b/tests/providers/aws/services/efs/efs_encryption_at_rest_enabled/efs_encryption_at_rest_enabled_test.py index 0ace020fd1..9670fceb7c 100644 --- a/tests/providers/aws/services/efs/efs_encryption_at_rest_enabled/efs_encryption_at_rest_enabled_test.py +++ b/tests/providers/aws/services/efs/efs_encryption_at_rest_enabled/efs_encryption_at_rest_enabled_test.py @@ -26,15 +26,12 @@ class Test_efs_encryption_at_rest_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_encryption_at_rest_enabled.efs_encryption_at_rest_enabled.efs_client", - new=EFS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_encryption_at_rest_enabled.efs_encryption_at_rest_enabled.efs_client", + new=EFS(aws_provider), ): from prowler.providers.aws.services.efs.efs_encryption_at_rest_enabled.efs_encryption_at_rest_enabled import ( efs_encryption_at_rest_enabled, @@ -65,15 +62,12 @@ class Test_efs_encryption_at_rest_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_encryption_at_rest_enabled.efs_encryption_at_rest_enabled.efs_client", - new=EFS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_encryption_at_rest_enabled.efs_encryption_at_rest_enabled.efs_client", + new=EFS(aws_provider), ): from prowler.providers.aws.services.efs.efs_encryption_at_rest_enabled.efs_encryption_at_rest_enabled import ( efs_encryption_at_rest_enabled, diff --git a/tests/providers/aws/services/efs/efs_have_backup_enabled/efs_have_backup_enabled_test.py b/tests/providers/aws/services/efs/efs_have_backup_enabled/efs_have_backup_enabled_test.py index 4e3152c0c3..9052639d0c 100644 --- a/tests/providers/aws/services/efs/efs_have_backup_enabled/efs_have_backup_enabled_test.py +++ b/tests/providers/aws/services/efs/efs_have_backup_enabled/efs_have_backup_enabled_test.py @@ -24,15 +24,12 @@ class Test_efs_have_backup_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_have_backup_enabled.efs_have_backup_enabled.efs_client", - new=EFS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_have_backup_enabled.efs_have_backup_enabled.efs_client", + new=EFS(aws_provider), ): from prowler.providers.aws.services.efs.efs_have_backup_enabled.efs_have_backup_enabled import ( efs_have_backup_enabled, @@ -63,15 +60,12 @@ class Test_efs_have_backup_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_have_backup_enabled.efs_have_backup_enabled.efs_client", - new=EFS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_have_backup_enabled.efs_have_backup_enabled.efs_client", + new=EFS(aws_provider), ): from prowler.providers.aws.services.efs.efs_have_backup_enabled.efs_have_backup_enabled import ( efs_have_backup_enabled, @@ -102,15 +96,12 @@ class Test_efs_have_backup_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_have_backup_enabled.efs_have_backup_enabled.efs_client", - new=EFS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_have_backup_enabled.efs_have_backup_enabled.efs_client", + new=EFS(aws_provider), ): from prowler.providers.aws.services.efs.efs_have_backup_enabled.efs_have_backup_enabled import ( efs_have_backup_enabled, diff --git a/tests/providers/aws/services/efs/efs_mount_target_not_publicly_accessible/efs_mount_target_not_publicly_accessible_test.py b/tests/providers/aws/services/efs/efs_mount_target_not_publicly_accessible/efs_mount_target_not_publicly_accessible_test.py index bb0cd77c22..68391fd66f 100644 --- a/tests/providers/aws/services/efs/efs_mount_target_not_publicly_accessible/efs_mount_target_not_publicly_accessible_test.py +++ b/tests/providers/aws/services/efs/efs_mount_target_not_publicly_accessible/efs_mount_target_not_publicly_accessible_test.py @@ -19,15 +19,12 @@ class Test_efs_mount_target_not_publicly_accessible: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_mount_target_not_publicly_accessible.efs_mount_target_not_publicly_accessible.efs_client", - new=EFS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_mount_target_not_publicly_accessible.efs_mount_target_not_publicly_accessible.efs_client", + new=EFS(aws_provider), ): from prowler.providers.aws.services.efs.efs_mount_target_not_publicly_accessible.efs_mount_target_not_publicly_accessible import ( efs_mount_target_not_publicly_accessible, @@ -46,15 +43,12 @@ class Test_efs_mount_target_not_publicly_accessible: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_mount_target_not_publicly_accessible.efs_mount_target_not_publicly_accessible.efs_client", - new=EFS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_mount_target_not_publicly_accessible.efs_mount_target_not_publicly_accessible.efs_client", + new=EFS(aws_provider), ): from prowler.providers.aws.services.efs.efs_mount_target_not_publicly_accessible.efs_mount_target_not_publicly_accessible import ( efs_mount_target_not_publicly_accessible, @@ -113,19 +107,15 @@ class Test_efs_mount_target_not_publicly_accessible: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_mount_target_not_publicly_accessible.efs_mount_target_not_publicly_accessible.efs_client", - new=EFS(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_mount_target_not_publicly_accessible.efs_mount_target_not_publicly_accessible.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_mount_target_not_publicly_accessible.efs_mount_target_not_publicly_accessible.efs_client", + new=EFS(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.efs.efs_mount_target_not_publicly_accessible.efs_mount_target_not_publicly_accessible.vpc_client", + new=VPC(aws_provider), ): from prowler.providers.aws.services.efs.efs_mount_target_not_publicly_accessible.efs_mount_target_not_publicly_accessible import ( efs_mount_target_not_publicly_accessible, @@ -169,19 +159,15 @@ class Test_efs_mount_target_not_publicly_accessible: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_mount_target_not_publicly_accessible.efs_mount_target_not_publicly_accessible.efs_client", - new=EFS(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_mount_target_not_publicly_accessible.efs_mount_target_not_publicly_accessible.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_mount_target_not_publicly_accessible.efs_mount_target_not_publicly_accessible.efs_client", + new=EFS(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.efs.efs_mount_target_not_publicly_accessible.efs_mount_target_not_publicly_accessible.vpc_client", + new=VPC(aws_provider), ): from prowler.providers.aws.services.efs.efs_mount_target_not_publicly_accessible.efs_mount_target_not_publicly_accessible import ( efs_mount_target_not_publicly_accessible, diff --git a/tests/providers/aws/services/efs/efs_not_publicly_accessible/efs_not_publicly_accessible_test.py b/tests/providers/aws/services/efs/efs_not_publicly_accessible/efs_not_publicly_accessible_test.py index ef2a5e9084..14adf7ac22 100644 --- a/tests/providers/aws/services/efs/efs_not_publicly_accessible/efs_not_publicly_accessible_test.py +++ b/tests/providers/aws/services/efs/efs_not_publicly_accessible/efs_not_publicly_accessible_test.py @@ -126,15 +126,12 @@ class Test_efs_not_publicly_accessible: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_not_publicly_accessible.efs_not_publicly_accessible.efs_client", - new=EFS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_not_publicly_accessible.efs_not_publicly_accessible.efs_client", + new=EFS(aws_provider), ): from prowler.providers.aws.services.efs.efs_not_publicly_accessible.efs_not_publicly_accessible import ( efs_not_publicly_accessible, @@ -166,15 +163,12 @@ class Test_efs_not_publicly_accessible: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_not_publicly_accessible.efs_not_publicly_accessible.efs_client", - new=EFS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_not_publicly_accessible.efs_not_publicly_accessible.efs_client", + new=EFS(aws_provider), ): from prowler.providers.aws.services.efs.efs_not_publicly_accessible.efs_not_publicly_accessible import ( efs_not_publicly_accessible, @@ -206,15 +200,12 @@ class Test_efs_not_publicly_accessible: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_not_publicly_accessible.efs_not_publicly_accessible.efs_client", - new=EFS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_not_publicly_accessible.efs_not_publicly_accessible.efs_client", + new=EFS(aws_provider), ): from prowler.providers.aws.services.efs.efs_not_publicly_accessible.efs_not_publicly_accessible import ( efs_not_publicly_accessible, @@ -246,15 +237,12 @@ class Test_efs_not_publicly_accessible: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_not_publicly_accessible.efs_not_publicly_accessible.efs_client", - new=EFS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_not_publicly_accessible.efs_not_publicly_accessible.efs_client", + new=EFS(aws_provider), ): from prowler.providers.aws.services.efs.efs_not_publicly_accessible.efs_not_publicly_accessible import ( efs_not_publicly_accessible, @@ -285,15 +273,12 @@ class Test_efs_not_publicly_accessible: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.efs.efs_not_publicly_accessible.efs_not_publicly_accessible.efs_client", - new=EFS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.efs.efs_not_publicly_accessible.efs_not_publicly_accessible.efs_client", + new=EFS(aws_provider), ): from prowler.providers.aws.services.efs.efs_not_publicly_accessible.efs_not_publicly_accessible import ( efs_not_publicly_accessible, diff --git a/tests/providers/aws/services/elasticache/elasticache_cluster_uses_public_subnet/elasticache_cluster_uses_public_subnet_test.py b/tests/providers/aws/services/elasticache/elasticache_cluster_uses_public_subnet/elasticache_cluster_uses_public_subnet_test.py index 4c92f3e754..8529f0ad00 100644 --- a/tests/providers/aws/services/elasticache/elasticache_cluster_uses_public_subnet/elasticache_cluster_uses_public_subnet_test.py +++ b/tests/providers/aws/services/elasticache/elasticache_cluster_uses_public_subnet/elasticache_cluster_uses_public_subnet_test.py @@ -39,23 +39,18 @@ class Test_elasticache_cluster_uses_public_subnet: elasticache_service = MagicMock elasticache_service.clusters = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_service.VPC", - new=vpc_client, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_client.vpc_client", - new=vpc_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_service.VPC", + new=vpc_client, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_client.vpc_client", + new=vpc_client, ): from prowler.providers.aws.services.elasticache.elasticache_cluster_uses_public_subnet.elasticache_cluster_uses_public_subnet import ( elasticache_cluster_uses_public_subnet, @@ -84,23 +79,18 @@ class Test_elasticache_cluster_uses_public_subnet: vpc_client = MagicMock vpc_client.vpc_subnets = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_service.VPC", - new=vpc_client, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_client.vpc_client", - new=vpc_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_service.VPC", + new=vpc_client, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_client.vpc_client", + new=vpc_client, ): from prowler.providers.aws.services.elasticache.elasticache_cluster_uses_public_subnet.elasticache_cluster_uses_public_subnet import ( elasticache_cluster_uses_public_subnet, @@ -167,23 +157,18 @@ class Test_elasticache_cluster_uses_public_subnet: mapPublicIpOnLaunch=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_service.VPC", - new=vpc_client, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_client.vpc_client", - new=vpc_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_service.VPC", + new=vpc_client, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_client.vpc_client", + new=vpc_client, ): from prowler.providers.aws.services.elasticache.elasticache_cluster_uses_public_subnet.elasticache_cluster_uses_public_subnet import ( elasticache_cluster_uses_public_subnet, @@ -250,23 +235,18 @@ class Test_elasticache_cluster_uses_public_subnet: mapPublicIpOnLaunch=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_service.VPC", - new=vpc_client, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_client.vpc_client", - new=vpc_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_service.VPC", + new=vpc_client, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_client.vpc_client", + new=vpc_client, ): from prowler.providers.aws.services.elasticache.elasticache_cluster_uses_public_subnet.elasticache_cluster_uses_public_subnet import ( elasticache_cluster_uses_public_subnet, @@ -333,23 +313,18 @@ class Test_elasticache_cluster_uses_public_subnet: mapPublicIpOnLaunch=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_service.VPC", - new=vpc_client, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_client.vpc_client", - new=vpc_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_service.VPC", + new=vpc_client, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_client.vpc_client", + new=vpc_client, ): from prowler.providers.aws.services.elasticache.elasticache_cluster_uses_public_subnet.elasticache_cluster_uses_public_subnet import ( elasticache_cluster_uses_public_subnet, @@ -416,23 +391,18 @@ class Test_elasticache_cluster_uses_public_subnet: mapPublicIpOnLaunch=False, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_service.VPC", - new=vpc_client, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_client.vpc_client", - new=vpc_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_service.VPC", + new=vpc_client, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_client.vpc_client", + new=vpc_client, ): from prowler.providers.aws.services.elasticache.elasticache_cluster_uses_public_subnet.elasticache_cluster_uses_public_subnet import ( elasticache_cluster_uses_public_subnet, diff --git a/tests/providers/aws/services/elasticache/elasticache_redis_cluster_auto_minor_version_upgrades/elasticache_redis_cluster_auto_minor_version_upgrades_test.py b/tests/providers/aws/services/elasticache/elasticache_redis_cluster_auto_minor_version_upgrades/elasticache_redis_cluster_auto_minor_version_upgrades_test.py index 9ed07a61b5..553b378820 100644 --- a/tests/providers/aws/services/elasticache/elasticache_redis_cluster_auto_minor_version_upgrades/elasticache_redis_cluster_auto_minor_version_upgrades_test.py +++ b/tests/providers/aws/services/elasticache/elasticache_redis_cluster_auto_minor_version_upgrades/elasticache_redis_cluster_auto_minor_version_upgrades_test.py @@ -31,23 +31,18 @@ class Test_elasticache_redis_cluster_auto_minor_version_upgrades: elasticache_service = MagicMock elasticache_service.replication_groups = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_service.VPC", - new=vpc_client, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_client.vpc_client", - new=vpc_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_service.VPC", + new=vpc_client, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_client.vpc_client", + new=vpc_client, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_auto_minor_version_upgrades.elasticache_redis_cluster_auto_minor_version_upgrades import ( elasticache_redis_cluster_auto_minor_version_upgrades, @@ -80,15 +75,12 @@ class Test_elasticache_redis_cluster_auto_minor_version_upgrades: ) ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_auto_minor_version_upgrades.elasticache_redis_cluster_auto_minor_version_upgrades import ( elasticache_redis_cluster_auto_minor_version_upgrades, @@ -130,15 +122,12 @@ class Test_elasticache_redis_cluster_auto_minor_version_upgrades: ) ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_auto_minor_version_upgrades.elasticache_redis_cluster_auto_minor_version_upgrades import ( elasticache_redis_cluster_auto_minor_version_upgrades, diff --git a/tests/providers/aws/services/elasticache/elasticache_redis_cluster_automatic_failover_enabled/elasticache_redis_cluster_automatic_failover_enabled_test.py b/tests/providers/aws/services/elasticache/elasticache_redis_cluster_automatic_failover_enabled/elasticache_redis_cluster_automatic_failover_enabled_test.py index c9477ed515..a9863be907 100644 --- a/tests/providers/aws/services/elasticache/elasticache_redis_cluster_automatic_failover_enabled/elasticache_redis_cluster_automatic_failover_enabled_test.py +++ b/tests/providers/aws/services/elasticache/elasticache_redis_cluster_automatic_failover_enabled/elasticache_redis_cluster_automatic_failover_enabled_test.py @@ -31,23 +31,18 @@ class Test_elasticache_redis_cluster_automatic_failover_enabled: elasticache_service = MagicMock elasticache_service.replication_groups = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_service.VPC", - new=vpc_client, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_client.vpc_client", - new=vpc_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_service.VPC", + new=vpc_client, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_client.vpc_client", + new=vpc_client, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_automatic_failover_enabled.elasticache_redis_cluster_automatic_failover_enabled import ( elasticache_redis_cluster_automatic_failover_enabled, @@ -80,15 +75,12 @@ class Test_elasticache_redis_cluster_automatic_failover_enabled: ) ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_automatic_failover_enabled.elasticache_redis_cluster_automatic_failover_enabled import ( elasticache_redis_cluster_automatic_failover_enabled, @@ -130,15 +122,12 @@ class Test_elasticache_redis_cluster_automatic_failover_enabled: ) ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_automatic_failover_enabled.elasticache_redis_cluster_automatic_failover_enabled import ( elasticache_redis_cluster_automatic_failover_enabled, diff --git a/tests/providers/aws/services/elasticache/elasticache_redis_cluster_backup_enabled/elasticache_redis_cluster_backup_enabled_test.py b/tests/providers/aws/services/elasticache/elasticache_redis_cluster_backup_enabled/elasticache_redis_cluster_backup_enabled_test.py index 958e2fd346..881db612fb 100644 --- a/tests/providers/aws/services/elasticache/elasticache_redis_cluster_backup_enabled/elasticache_redis_cluster_backup_enabled_test.py +++ b/tests/providers/aws/services/elasticache/elasticache_redis_cluster_backup_enabled/elasticache_redis_cluster_backup_enabled_test.py @@ -39,15 +39,12 @@ class Test_elasticache_redis_cluster_backup_enabled: elasticache_client = MagicMock elasticache_client.replication_groups = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_client, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_backup_enabled.elasticache_redis_cluster_backup_enabled import ( elasticache_redis_cluster_backup_enabled, @@ -79,15 +76,12 @@ class Test_elasticache_redis_cluster_backup_enabled: ) elasticache_client.audit_config = {"minimum_snapshot_retention_period": 7} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_client, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_backup_enabled.elasticache_redis_cluster_backup_enabled import ( elasticache_redis_cluster_backup_enabled, @@ -130,15 +124,12 @@ class Test_elasticache_redis_cluster_backup_enabled: elasticache_client.audit_config = {"minimum_snapshot_retention_period": 7} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_client, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_backup_enabled.elasticache_redis_cluster_backup_enabled import ( elasticache_redis_cluster_backup_enabled, @@ -181,15 +172,12 @@ class Test_elasticache_redis_cluster_backup_enabled: elasticache_client.audit_config = {"minimum_snapshot_retention_period": 7} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_client, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_backup_enabled.elasticache_redis_cluster_backup_enabled import ( elasticache_redis_cluster_backup_enabled, @@ -231,15 +219,12 @@ class Test_elasticache_redis_cluster_backup_enabled: ) elasticache_client.audit_config = {"minimum_snapshot_retention_period": 1} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_client, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_backup_enabled.elasticache_redis_cluster_backup_enabled import ( elasticache_redis_cluster_backup_enabled, @@ -281,15 +266,12 @@ class Test_elasticache_redis_cluster_backup_enabled: ) elasticache_client.audit_config = {"minimum_snapshot_retention_period": 3} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_client, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_backup_enabled.elasticache_redis_cluster_backup_enabled import ( elasticache_redis_cluster_backup_enabled, diff --git a/tests/providers/aws/services/elasticache/elasticache_redis_cluster_in_transit_encryption_enabled/elasticache_redis_cluster_in_transit_encryption_enabled_test.py b/tests/providers/aws/services/elasticache/elasticache_redis_cluster_in_transit_encryption_enabled/elasticache_redis_cluster_in_transit_encryption_enabled_test.py index 363751f3fb..e6db0f4997 100644 --- a/tests/providers/aws/services/elasticache/elasticache_redis_cluster_in_transit_encryption_enabled/elasticache_redis_cluster_in_transit_encryption_enabled_test.py +++ b/tests/providers/aws/services/elasticache/elasticache_redis_cluster_in_transit_encryption_enabled/elasticache_redis_cluster_in_transit_encryption_enabled_test.py @@ -41,15 +41,12 @@ class Test_elasticache_replication_group_in_transit_encryption_enabled: elasticache_service = MagicMock elasticache_service.replication_groups = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_in_transit_encryption_enabled.elasticache_redis_cluster_in_transit_encryption_enabled import ( elasticache_redis_cluster_in_transit_encryption_enabled, @@ -82,15 +79,12 @@ class Test_elasticache_replication_group_in_transit_encryption_enabled: ) ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_in_transit_encryption_enabled.elasticache_redis_cluster_in_transit_encryption_enabled import ( elasticache_redis_cluster_in_transit_encryption_enabled, @@ -133,15 +127,12 @@ class Test_elasticache_replication_group_in_transit_encryption_enabled: ) ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_in_transit_encryption_enabled.elasticache_redis_cluster_in_transit_encryption_enabled import ( elasticache_redis_cluster_in_transit_encryption_enabled, diff --git a/tests/providers/aws/services/elasticache/elasticache_redis_cluster_multi_az_enabled/elasticache_redis_cluster_multi_az_enabled_test.py b/tests/providers/aws/services/elasticache/elasticache_redis_cluster_multi_az_enabled/elasticache_redis_cluster_multi_az_enabled_test.py index 9c493c7c07..c82679ccd0 100644 --- a/tests/providers/aws/services/elasticache/elasticache_redis_cluster_multi_az_enabled/elasticache_redis_cluster_multi_az_enabled_test.py +++ b/tests/providers/aws/services/elasticache/elasticache_redis_cluster_multi_az_enabled/elasticache_redis_cluster_multi_az_enabled_test.py @@ -41,15 +41,12 @@ class Test_elasticache_replication_group_multi_az_enabled: elasticache_service = MagicMock elasticache_service.replication_groups = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_multi_az_enabled.elasticache_redis_cluster_multi_az_enabled import ( elasticache_redis_cluster_multi_az_enabled, @@ -82,15 +79,12 @@ class Test_elasticache_replication_group_multi_az_enabled: ) ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_multi_az_enabled.elasticache_redis_cluster_multi_az_enabled import ( elasticache_redis_cluster_multi_az_enabled, @@ -133,15 +127,12 @@ class Test_elasticache_replication_group_multi_az_enabled: ) ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_multi_az_enabled.elasticache_redis_cluster_multi_az_enabled import ( elasticache_redis_cluster_multi_az_enabled, diff --git a/tests/providers/aws/services/elasticache/elasticache_redis_cluster_rest_encryption_enabled/elasticache_redis_cluster_rest_encryption_enabled_test.py b/tests/providers/aws/services/elasticache/elasticache_redis_cluster_rest_encryption_enabled/elasticache_redis_cluster_rest_encryption_enabled_test.py index 42c0bef20b..913f5f8948 100644 --- a/tests/providers/aws/services/elasticache/elasticache_redis_cluster_rest_encryption_enabled/elasticache_redis_cluster_rest_encryption_enabled_test.py +++ b/tests/providers/aws/services/elasticache/elasticache_redis_cluster_rest_encryption_enabled/elasticache_redis_cluster_rest_encryption_enabled_test.py @@ -40,15 +40,12 @@ class Test_elasticache_replication_group_at_rest_encryption_enabled: elasticache_service = MagicMock elasticache_service.replication_groups = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_rest_encryption_enabled.elasticache_redis_cluster_rest_encryption_enabled import ( elasticache_redis_cluster_rest_encryption_enabled, @@ -81,15 +78,12 @@ class Test_elasticache_replication_group_at_rest_encryption_enabled: ) ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_rest_encryption_enabled.elasticache_redis_cluster_rest_encryption_enabled import ( elasticache_redis_cluster_rest_encryption_enabled, @@ -132,15 +126,12 @@ class Test_elasticache_replication_group_at_rest_encryption_enabled: ) ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, ): from prowler.providers.aws.services.elasticache.elasticache_redis_cluster_rest_encryption_enabled.elasticache_redis_cluster_rest_encryption_enabled import ( elasticache_redis_cluster_rest_encryption_enabled, diff --git a/tests/providers/aws/services/elasticache/elasticache_redis_replication_group_auth_enabled/elasticache_redis_replication_group_auth_enabled_test.py b/tests/providers/aws/services/elasticache/elasticache_redis_replication_group_auth_enabled/elasticache_redis_replication_group_auth_enabled_test.py index 1978373ce3..f7425d5398 100644 --- a/tests/providers/aws/services/elasticache/elasticache_redis_replication_group_auth_enabled/elasticache_redis_replication_group_auth_enabled_test.py +++ b/tests/providers/aws/services/elasticache/elasticache_redis_replication_group_auth_enabled/elasticache_redis_replication_group_auth_enabled_test.py @@ -35,23 +35,18 @@ class Test_elasticache_redis_replication_group_auth_enabled: elasticache_service = MagicMock elasticache_service.replication_groups = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_service.VPC", - new=vpc_client, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_client.vpc_client", - new=vpc_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_service.VPC", + new=vpc_client, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_client.vpc_client", + new=vpc_client, ): from prowler.providers.aws.services.elasticache.elasticache_redis_replication_group_auth_enabled.elasticache_redis_replication_group_auth_enabled import ( elasticache_redis_replication_group_auth_enabled, @@ -87,23 +82,18 @@ class Test_elasticache_redis_replication_group_auth_enabled: vpc_client = MagicMock vpc_client.vpc_subnets = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_service.VPC", - new=vpc_client, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_client.vpc_client", - new=vpc_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_service.VPC", + new=vpc_client, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_client.vpc_client", + new=vpc_client, ): from prowler.providers.aws.services.elasticache.elasticache_redis_replication_group_auth_enabled.elasticache_redis_replication_group_auth_enabled import ( elasticache_redis_replication_group_auth_enabled, @@ -148,23 +138,18 @@ class Test_elasticache_redis_replication_group_auth_enabled: vpc_client = MagicMock vpc_client.vpc_subnets = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_service.VPC", - new=vpc_client, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_client.vpc_client", - new=vpc_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_service.VPC", + new=vpc_client, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_client.vpc_client", + new=vpc_client, ): from prowler.providers.aws.services.elasticache.elasticache_redis_replication_group_auth_enabled.elasticache_redis_replication_group_auth_enabled import ( elasticache_redis_replication_group_auth_enabled, @@ -210,23 +195,18 @@ class Test_elasticache_redis_replication_group_auth_enabled: vpc_client = MagicMock vpc_client.vpc_subnets = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", - new=elasticache_service, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_service.VPC", - new=vpc_client, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_client.vpc_client", - new=vpc_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.elasticache.elasticache_service.ElastiCache", + new=elasticache_service, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_service.VPC", + new=vpc_client, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_client.vpc_client", + new=vpc_client, ): from prowler.providers.aws.services.elasticache.elasticache_redis_replication_group_auth_enabled.elasticache_redis_replication_group_auth_enabled import ( elasticache_redis_replication_group_auth_enabled, diff --git a/tests/providers/aws/services/elasticbeanstalk/elasticbeanstalk_environment_cloudwatch_logging_enabled/elasticbeanstalk_environment_cloudwatch_logging_enabled_test.py b/tests/providers/aws/services/elasticbeanstalk/elasticbeanstalk_environment_cloudwatch_logging_enabled/elasticbeanstalk_environment_cloudwatch_logging_enabled_test.py index c9323bc0af..58c6484acc 100644 --- a/tests/providers/aws/services/elasticbeanstalk/elasticbeanstalk_environment_cloudwatch_logging_enabled/elasticbeanstalk_environment_cloudwatch_logging_enabled_test.py +++ b/tests/providers/aws/services/elasticbeanstalk/elasticbeanstalk_environment_cloudwatch_logging_enabled/elasticbeanstalk_environment_cloudwatch_logging_enabled_test.py @@ -56,15 +56,12 @@ class Test_elasticbeanstalk_environment_cloudwatch_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_cloudwatch_logging_enabled.elasticbeanstalk_environment_cloudwatch_logging_enabled.elasticbeanstalk_client", - new=ElasticBeanstalk(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_cloudwatch_logging_enabled.elasticbeanstalk_environment_cloudwatch_logging_enabled.elasticbeanstalk_client", + new=ElasticBeanstalk(aws_provider), ): from prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_cloudwatch_logging_enabled.elasticbeanstalk_environment_cloudwatch_logging_enabled import ( elasticbeanstalk_environment_cloudwatch_logging_enabled, @@ -88,15 +85,12 @@ class Test_elasticbeanstalk_environment_cloudwatch_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_cloudwatch_logging_enabled.elasticbeanstalk_environment_cloudwatch_logging_enabled.elasticbeanstalk_client", - new=ElasticBeanstalk(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_cloudwatch_logging_enabled.elasticbeanstalk_environment_cloudwatch_logging_enabled.elasticbeanstalk_client", + new=ElasticBeanstalk(aws_provider), ): from prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_cloudwatch_logging_enabled.elasticbeanstalk_environment_cloudwatch_logging_enabled import ( elasticbeanstalk_environment_cloudwatch_logging_enabled, @@ -128,15 +122,12 @@ class Test_elasticbeanstalk_environment_cloudwatch_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_cloudwatch_logging_enabled.elasticbeanstalk_environment_cloudwatch_logging_enabled.elasticbeanstalk_client", - new=ElasticBeanstalk(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_cloudwatch_logging_enabled.elasticbeanstalk_environment_cloudwatch_logging_enabled.elasticbeanstalk_client", + new=ElasticBeanstalk(aws_provider), ): from prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_cloudwatch_logging_enabled.elasticbeanstalk_environment_cloudwatch_logging_enabled import ( elasticbeanstalk_environment_cloudwatch_logging_enabled, diff --git a/tests/providers/aws/services/elasticbeanstalk/elasticbeanstalk_environment_enhanced_health_reporting/elasticbeanstalk_environment_enhanced_health_reporting_test.py b/tests/providers/aws/services/elasticbeanstalk/elasticbeanstalk_environment_enhanced_health_reporting/elasticbeanstalk_environment_enhanced_health_reporting_test.py index 9cb7f92ba1..2716bb9c03 100644 --- a/tests/providers/aws/services/elasticbeanstalk/elasticbeanstalk_environment_enhanced_health_reporting/elasticbeanstalk_environment_enhanced_health_reporting_test.py +++ b/tests/providers/aws/services/elasticbeanstalk/elasticbeanstalk_environment_enhanced_health_reporting/elasticbeanstalk_environment_enhanced_health_reporting_test.py @@ -56,15 +56,12 @@ class Test_elasticbeanstalk_environment_enhanced_health_reporting: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_enhanced_health_reporting.elasticbeanstalk_environment_enhanced_health_reporting.elasticbeanstalk_client", - new=ElasticBeanstalk(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_enhanced_health_reporting.elasticbeanstalk_environment_enhanced_health_reporting.elasticbeanstalk_client", + new=ElasticBeanstalk(aws_provider), ): from prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_enhanced_health_reporting.elasticbeanstalk_environment_enhanced_health_reporting import ( elasticbeanstalk_environment_enhanced_health_reporting, @@ -88,15 +85,12 @@ class Test_elasticbeanstalk_environment_enhanced_health_reporting: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_enhanced_health_reporting.elasticbeanstalk_environment_enhanced_health_reporting.elasticbeanstalk_client", - new=ElasticBeanstalk(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_enhanced_health_reporting.elasticbeanstalk_environment_enhanced_health_reporting.elasticbeanstalk_client", + new=ElasticBeanstalk(aws_provider), ): from prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_enhanced_health_reporting.elasticbeanstalk_environment_enhanced_health_reporting import ( elasticbeanstalk_environment_enhanced_health_reporting, @@ -128,15 +122,12 @@ class Test_elasticbeanstalk_environment_enhanced_health_reporting: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_enhanced_health_reporting.elasticbeanstalk_environment_enhanced_health_reporting.elasticbeanstalk_client", - new=ElasticBeanstalk(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_enhanced_health_reporting.elasticbeanstalk_environment_enhanced_health_reporting.elasticbeanstalk_client", + new=ElasticBeanstalk(aws_provider), ): from prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_enhanced_health_reporting.elasticbeanstalk_environment_enhanced_health_reporting import ( elasticbeanstalk_environment_enhanced_health_reporting, diff --git a/tests/providers/aws/services/elasticbeanstalk/elasticbeanstalk_environment_managed_updates_enabled/elasticbeanstalk_environment_managed_updates_enabled_test.py b/tests/providers/aws/services/elasticbeanstalk/elasticbeanstalk_environment_managed_updates_enabled/elasticbeanstalk_environment_managed_updates_enabled_test.py index 6a80bfe039..01b19a9a8d 100644 --- a/tests/providers/aws/services/elasticbeanstalk/elasticbeanstalk_environment_managed_updates_enabled/elasticbeanstalk_environment_managed_updates_enabled_test.py +++ b/tests/providers/aws/services/elasticbeanstalk/elasticbeanstalk_environment_managed_updates_enabled/elasticbeanstalk_environment_managed_updates_enabled_test.py @@ -60,15 +60,12 @@ class Test_elasticbeanstalk_environment_managed_updates_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_managed_updates_enabled.elasticbeanstalk_environment_managed_updates_enabled.elasticbeanstalk_client", - new=ElasticBeanstalk(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_managed_updates_enabled.elasticbeanstalk_environment_managed_updates_enabled.elasticbeanstalk_client", + new=ElasticBeanstalk(aws_provider), ): from prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_managed_updates_enabled.elasticbeanstalk_environment_managed_updates_enabled import ( elasticbeanstalk_environment_managed_updates_enabled, @@ -92,15 +89,12 @@ class Test_elasticbeanstalk_environment_managed_updates_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_managed_updates_enabled.elasticbeanstalk_environment_managed_updates_enabled.elasticbeanstalk_client", - new=ElasticBeanstalk(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_managed_updates_enabled.elasticbeanstalk_environment_managed_updates_enabled.elasticbeanstalk_client", + new=ElasticBeanstalk(aws_provider), ): from prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_managed_updates_enabled.elasticbeanstalk_environment_managed_updates_enabled import ( elasticbeanstalk_environment_managed_updates_enabled, @@ -132,15 +126,12 @@ class Test_elasticbeanstalk_environment_managed_updates_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_managed_updates_enabled.elasticbeanstalk_environment_managed_updates_enabled.elasticbeanstalk_client", - new=ElasticBeanstalk(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_managed_updates_enabled.elasticbeanstalk_environment_managed_updates_enabled.elasticbeanstalk_client", + new=ElasticBeanstalk(aws_provider), ): from prowler.providers.aws.services.elasticbeanstalk.elasticbeanstalk_environment_managed_updates_enabled.elasticbeanstalk_environment_managed_updates_enabled import ( elasticbeanstalk_environment_managed_updates_enabled, diff --git a/tests/providers/aws/services/elb/elb_connection_draining_enabled/elb_connection_draining_enabled_test.py b/tests/providers/aws/services/elb/elb_connection_draining_enabled/elb_connection_draining_enabled_test.py index b729e27af2..f77c434622 100644 --- a/tests/providers/aws/services/elb/elb_connection_draining_enabled/elb_connection_draining_enabled_test.py +++ b/tests/providers/aws/services/elb/elb_connection_draining_enabled/elb_connection_draining_enabled_test.py @@ -15,15 +15,12 @@ class Test_elb_connection_draining_enabled: def test_elb_no_balancers(self): from prowler.providers.aws.services.elb.elb_service import ELB - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_EU_WEST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elb.elb_connection_draining_enabled.elb_connection_draining_enabled.elb_client", - new=ELB(set_mocked_aws_provider([AWS_REGION_EU_WEST_1])), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_EU_WEST_1]), + ), mock.patch( + "prowler.providers.aws.services.elb.elb_connection_draining_enabled.elb_connection_draining_enabled.elb_client", + new=ELB(set_mocked_aws_provider([AWS_REGION_EU_WEST_1])), ): # Test Check from prowler.providers.aws.services.elb.elb_connection_draining_enabled.elb_connection_draining_enabled import ( @@ -63,15 +60,12 @@ class Test_elb_connection_draining_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.elb.elb_connection_draining_enabled.elb_connection_draining_enabled.elb_client", - new=ELB(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.elb.elb_connection_draining_enabled.elb_connection_draining_enabled.elb_client", + new=ELB(aws_provider), ): from prowler.providers.aws.services.elb.elb_connection_draining_enabled.elb_connection_draining_enabled import ( elb_connection_draining_enabled, @@ -122,15 +116,12 @@ class Test_elb_connection_draining_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.elb.elb_connection_draining_enabled.elb_connection_draining_enabled.elb_client", - new=ELB(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.elb.elb_connection_draining_enabled.elb_connection_draining_enabled.elb_client", + new=ELB(aws_provider), ): from prowler.providers.aws.services.elb.elb_connection_draining_enabled.elb_connection_draining_enabled import ( elb_connection_draining_enabled, diff --git a/tests/providers/aws/services/elb/elb_cross_zone_load_balancing_enabled/elb_cross_zone_load_balancing_enabled_test.py b/tests/providers/aws/services/elb/elb_cross_zone_load_balancing_enabled/elb_cross_zone_load_balancing_enabled_test.py index 35b03183fe..42d68f6f35 100644 --- a/tests/providers/aws/services/elb/elb_cross_zone_load_balancing_enabled/elb_cross_zone_load_balancing_enabled_test.py +++ b/tests/providers/aws/services/elb/elb_cross_zone_load_balancing_enabled/elb_cross_zone_load_balancing_enabled_test.py @@ -14,15 +14,12 @@ class Test_elb_cross_zone_load_balancing_enabled: def test_elb_no_balancers(self): from prowler.providers.aws.services.elb.elb_service import ELB - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_EU_WEST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elb.elb_ssl_listeners.elb_ssl_listeners.elb_client", - new=ELB(set_mocked_aws_provider([AWS_REGION_EU_WEST_1])), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_EU_WEST_1]), + ), mock.patch( + "prowler.providers.aws.services.elb.elb_ssl_listeners.elb_ssl_listeners.elb_client", + new=ELB(set_mocked_aws_provider([AWS_REGION_EU_WEST_1])), ): # Test Check from prowler.providers.aws.services.elb.elb_ssl_listeners.elb_ssl_listeners import ( @@ -51,15 +48,12 @@ class Test_elb_cross_zone_load_balancing_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.elb.elb_cross_zone_load_balancing_enabled.elb_cross_zone_load_balancing_enabled.elb_client", - new=ELB(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.elb.elb_cross_zone_load_balancing_enabled.elb_cross_zone_load_balancing_enabled.elb_client", + new=ELB(aws_provider), ): from prowler.providers.aws.services.elb.elb_cross_zone_load_balancing_enabled.elb_cross_zone_load_balancing_enabled import ( elb_cross_zone_load_balancing_enabled, @@ -108,15 +102,12 @@ class Test_elb_cross_zone_load_balancing_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.elb.elb_cross_zone_load_balancing_enabled.elb_cross_zone_load_balancing_enabled.elb_client", - new=ELB(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.elb.elb_cross_zone_load_balancing_enabled.elb_cross_zone_load_balancing_enabled.elb_client", + new=ELB(aws_provider), ): from prowler.providers.aws.services.elb.elb_cross_zone_load_balancing_enabled.elb_cross_zone_load_balancing_enabled import ( elb_cross_zone_load_balancing_enabled, @@ -163,15 +154,12 @@ class Test_elb_cross_zone_load_balancing_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.elb.elb_cross_zone_load_balancing_enabled.elb_cross_zone_load_balancing_enabled.elb_client", - new=ELB(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.elb.elb_cross_zone_load_balancing_enabled.elb_cross_zone_load_balancing_enabled.elb_client", + new=ELB(aws_provider), ): from prowler.providers.aws.services.elb.elb_cross_zone_load_balancing_enabled.elb_cross_zone_load_balancing_enabled import ( elb_cross_zone_load_balancing_enabled, diff --git a/tests/providers/aws/services/elb/elb_is_in_multiple_az/elb_is_in_multiple_az_test.py b/tests/providers/aws/services/elb/elb_is_in_multiple_az/elb_is_in_multiple_az_test.py index d6f0a3c6d3..a47f2c2a1f 100644 --- a/tests/providers/aws/services/elb/elb_is_in_multiple_az/elb_is_in_multiple_az_test.py +++ b/tests/providers/aws/services/elb/elb_is_in_multiple_az/elb_is_in_multiple_az_test.py @@ -16,15 +16,12 @@ class Test_elb_is_in_multiple_az: def test_elb_no_balancers(self): from prowler.providers.aws.services.elb.elb_service import ELB - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_EU_WEST_1]), - ), - mock.patch( - "prowler.providers.aws.services.elb.elb_is_in_multiple_az.elb_is_in_multiple_az.elb_client", - new=ELB(set_mocked_aws_provider([AWS_REGION_EU_WEST_1])), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_EU_WEST_1]), + ), mock.patch( + "prowler.providers.aws.services.elb.elb_is_in_multiple_az.elb_is_in_multiple_az.elb_client", + new=ELB(set_mocked_aws_provider([AWS_REGION_EU_WEST_1])), ): # Test Check from prowler.providers.aws.services.elb.elb_is_in_multiple_az.elb_is_in_multiple_az import ( @@ -55,15 +52,12 @@ class Test_elb_is_in_multiple_az: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.elb.elb_is_in_multiple_az.elb_is_in_multiple_az.elb_client", - new=ELB(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.elb.elb_is_in_multiple_az.elb_is_in_multiple_az.elb_client", + new=ELB(aws_provider), ): from prowler.providers.aws.services.elb.elb_is_in_multiple_az.elb_is_in_multiple_az import ( elb_is_in_multiple_az, @@ -105,15 +99,12 @@ class Test_elb_is_in_multiple_az: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.elb.elb_is_in_multiple_az.elb_is_in_multiple_az.elb_client", - new=ELB(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.elb.elb_is_in_multiple_az.elb_is_in_multiple_az.elb_client", + new=ELB(aws_provider), ): from prowler.providers.aws.services.elb.elb_is_in_multiple_az.elb_is_in_multiple_az import ( elb_is_in_multiple_az, diff --git a/tests/providers/aws/services/elb/elb_ssl_listeners_use_acm_certificate/elb_ssl_listeners_use_acm_certificate_test.py b/tests/providers/aws/services/elb/elb_ssl_listeners_use_acm_certificate/elb_ssl_listeners_use_acm_certificate_test.py index 1fef4bd613..87487e7721 100644 --- a/tests/providers/aws/services/elb/elb_ssl_listeners_use_acm_certificate/elb_ssl_listeners_use_acm_certificate_test.py +++ b/tests/providers/aws/services/elb/elb_ssl_listeners_use_acm_certificate/elb_ssl_listeners_use_acm_certificate_test.py @@ -83,19 +83,15 @@ class Test_elb_ssl_listeners_use_acm_certificate: aws_mocked_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_mocked_provider, - ), - mock.patch( - "prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate.elb_client", - new=ELB(aws_mocked_provider), - ), - mock.patch( - "prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate.acm_client", - new=ACM(aws_mocked_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_mocked_provider, + ), mock.patch( + "prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate.elb_client", + new=ELB(aws_mocked_provider), + ), mock.patch( + "prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate.acm_client", + new=ACM(aws_mocked_provider), ): # Test Check from prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate import ( @@ -141,19 +137,15 @@ class Test_elb_ssl_listeners_use_acm_certificate: aws_mocked_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_mocked_provider, - ), - mock.patch( - "prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate.elb_client", - new=ELB(aws_mocked_provider), - ), - mock.patch( - "prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate.acm_client", - new=ACM(aws_mocked_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_mocked_provider, + ), mock.patch( + "prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate.elb_client", + new=ELB(aws_mocked_provider), + ), mock.patch( + "prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate.acm_client", + new=ACM(aws_mocked_provider), ): from prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate import ( elb_ssl_listeners_use_acm_certificate, @@ -207,19 +199,15 @@ class Test_elb_ssl_listeners_use_acm_certificate: aws_mocked_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_mocked_provider, - ), - mock.patch( - "prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate.elb_client", - new=ELB(aws_mocked_provider), - ), - mock.patch( - "prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate.acm_client", - new=ACM(aws_mocked_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_mocked_provider, + ), mock.patch( + "prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate.elb_client", + new=ELB(aws_mocked_provider), + ), mock.patch( + "prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate.acm_client", + new=ACM(aws_mocked_provider), ): from prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate import ( elb_ssl_listeners_use_acm_certificate, @@ -270,19 +258,15 @@ class Test_elb_ssl_listeners_use_acm_certificate: aws_mocked_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_mocked_provider, - ), - mock.patch( - "prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate.elb_client", - new=ELB(aws_mocked_provider), - ), - mock.patch( - "prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate.acm_client", - new=ACM(aws_mocked_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_mocked_provider, + ), mock.patch( + "prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate.elb_client", + new=ELB(aws_mocked_provider), + ), mock.patch( + "prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate.acm_client", + new=ACM(aws_mocked_provider), ): from prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate import ( elb_ssl_listeners_use_acm_certificate, @@ -333,19 +317,15 @@ class Test_elb_ssl_listeners_use_acm_certificate: aws_mocked_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_mocked_provider, - ), - mock.patch( - "prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate.elb_client", - new=ELB(aws_mocked_provider), - ), - mock.patch( - "prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate.acm_client", - new=ACM(aws_mocked_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_mocked_provider, + ), mock.patch( + "prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate.elb_client", + new=ELB(aws_mocked_provider), + ), mock.patch( + "prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate.acm_client", + new=ACM(aws_mocked_provider), ): from prowler.providers.aws.services.elb.elb_ssl_listeners_use_acm_certificate.elb_ssl_listeners_use_acm_certificate import ( elb_ssl_listeners_use_acm_certificate, diff --git a/tests/providers/aws/services/elbv2/elbv2_is_in_multiple_az/elbv2_is_in_multiple_az_test.py b/tests/providers/aws/services/elbv2/elbv2_is_in_multiple_az/elbv2_is_in_multiple_az_test.py index 68f7b43b71..ab4f505898 100644 --- a/tests/providers/aws/services/elbv2/elbv2_is_in_multiple_az/elbv2_is_in_multiple_az_test.py +++ b/tests/providers/aws/services/elbv2/elbv2_is_in_multiple_az/elbv2_is_in_multiple_az_test.py @@ -18,15 +18,12 @@ class Test_elbv2_is_in_multiple_az: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.elbv2.elbv2_is_in_multiple_az.elbv2_is_in_multiple_az.elbv2_client", - new=ELBv2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.elbv2.elbv2_is_in_multiple_az.elbv2_is_in_multiple_az.elbv2_client", + new=ELBv2(aws_provider), ): # Test Check from prowler.providers.aws.services.elbv2.elbv2_is_in_multiple_az.elbv2_is_in_multiple_az import ( @@ -67,15 +64,12 @@ class Test_elbv2_is_in_multiple_az: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.elbv2.elbv2_is_in_multiple_az.elbv2_is_in_multiple_az.elbv2_client", - new=ELBv2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.elbv2.elbv2_is_in_multiple_az.elbv2_is_in_multiple_az.elbv2_client", + new=ELBv2(aws_provider), ): from prowler.providers.aws.services.elbv2.elbv2_is_in_multiple_az.elbv2_is_in_multiple_az import ( elbv2_is_in_multiple_az, @@ -130,15 +124,12 @@ class Test_elbv2_is_in_multiple_az: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.elbv2.elbv2_is_in_multiple_az.elbv2_is_in_multiple_az.elbv2_client", - new=ELBv2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.elbv2.elbv2_is_in_multiple_az.elbv2_is_in_multiple_az.elbv2_client", + new=ELBv2(aws_provider), ): from prowler.providers.aws.services.elbv2.elbv2_is_in_multiple_az.elbv2_is_in_multiple_az import ( elbv2_is_in_multiple_az, diff --git a/tests/providers/aws/services/eventbridge/eventbridge_bus_cross_account_access/eventbridge_bus_cross_account_access_test.py b/tests/providers/aws/services/eventbridge/eventbridge_bus_cross_account_access/eventbridge_bus_cross_account_access_test.py index c4adf8ae51..280d3d268d 100644 --- a/tests/providers/aws/services/eventbridge/eventbridge_bus_cross_account_access/eventbridge_bus_cross_account_access_test.py +++ b/tests/providers/aws/services/eventbridge/eventbridge_bus_cross_account_access/eventbridge_bus_cross_account_access_test.py @@ -23,15 +23,12 @@ class Test_eventbridge_bus_cross_account_access: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.eventbridge.eventbridge_bus_cross_account_access.eventbridge_bus_cross_account_access.eventbridge_client", - new=EventBridge(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.eventbridge.eventbridge_bus_cross_account_access.eventbridge_bus_cross_account_access.eventbridge_client", + new=EventBridge(aws_provider), ): from prowler.providers.aws.services.eventbridge.eventbridge_bus_cross_account_access.eventbridge_bus_cross_account_access import ( eventbridge_bus_cross_account_access, @@ -66,15 +63,12 @@ class Test_eventbridge_bus_cross_account_access: Principal=AWS_ACCOUNT_NUMBER, StatementId="test-statement", ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.eventbridge.eventbridge_bus_cross_account_access.eventbridge_bus_cross_account_access.eventbridge_client", - new=EventBridge(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.eventbridge.eventbridge_bus_cross_account_access.eventbridge_bus_cross_account_access.eventbridge_client", + new=EventBridge(aws_provider), ): from prowler.providers.aws.services.eventbridge.eventbridge_bus_cross_account_access.eventbridge_bus_cross_account_access import ( eventbridge_bus_cross_account_access, @@ -109,15 +103,12 @@ class Test_eventbridge_bus_cross_account_access: Principal="111122223333", StatementId="test-statement", ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.eventbridge.eventbridge_bus_cross_account_access.eventbridge_bus_cross_account_access.eventbridge_client", - new=EventBridge(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.eventbridge.eventbridge_bus_cross_account_access.eventbridge_bus_cross_account_access.eventbridge_client", + new=EventBridge(aws_provider), ): from prowler.providers.aws.services.eventbridge.eventbridge_bus_cross_account_access.eventbridge_bus_cross_account_access import ( eventbridge_bus_cross_account_access, @@ -152,15 +143,12 @@ class Test_eventbridge_bus_cross_account_access: Principal="*", StatementId="test-statement", ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.eventbridge.eventbridge_bus_cross_account_access.eventbridge_bus_cross_account_access.eventbridge_client", - new=EventBridge(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.eventbridge.eventbridge_bus_cross_account_access.eventbridge_bus_cross_account_access.eventbridge_client", + new=EventBridge(aws_provider), ): from prowler.providers.aws.services.eventbridge.eventbridge_bus_cross_account_access.eventbridge_bus_cross_account_access import ( eventbridge_bus_cross_account_access, diff --git a/tests/providers/aws/services/eventbridge/eventbridge_bus_exposed/eventbridge_bus_exposed_test.py b/tests/providers/aws/services/eventbridge/eventbridge_bus_exposed/eventbridge_bus_exposed_test.py index 374b7b1103..44df4ff2d3 100644 --- a/tests/providers/aws/services/eventbridge/eventbridge_bus_exposed/eventbridge_bus_exposed_test.py +++ b/tests/providers/aws/services/eventbridge/eventbridge_bus_exposed/eventbridge_bus_exposed_test.py @@ -23,15 +23,12 @@ class Test_eventbridge_bus_exposed: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.eventbridge.eventbridge_bus_exposed.eventbridge_bus_exposed.eventbridge_client", - new=EventBridge(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.eventbridge.eventbridge_bus_exposed.eventbridge_bus_exposed.eventbridge_client", + new=EventBridge(aws_provider), ): from prowler.providers.aws.services.eventbridge.eventbridge_bus_exposed.eventbridge_bus_exposed import ( eventbridge_bus_exposed, @@ -66,15 +63,12 @@ class Test_eventbridge_bus_exposed: Principal=AWS_ACCOUNT_NUMBER, StatementId="test-statement", ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.eventbridge.eventbridge_bus_exposed.eventbridge_bus_exposed.eventbridge_client", - new=EventBridge(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.eventbridge.eventbridge_bus_exposed.eventbridge_bus_exposed.eventbridge_client", + new=EventBridge(aws_provider), ): from prowler.providers.aws.services.eventbridge.eventbridge_bus_exposed.eventbridge_bus_exposed import ( eventbridge_bus_exposed, @@ -109,15 +103,12 @@ class Test_eventbridge_bus_exposed: Principal="111122223333", StatementId="test-statement", ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.eventbridge.eventbridge_bus_exposed.eventbridge_bus_exposed.eventbridge_client", - new=EventBridge(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.eventbridge.eventbridge_bus_exposed.eventbridge_bus_exposed.eventbridge_client", + new=EventBridge(aws_provider), ): from prowler.providers.aws.services.eventbridge.eventbridge_bus_exposed.eventbridge_bus_exposed import ( eventbridge_bus_exposed, @@ -152,15 +143,12 @@ class Test_eventbridge_bus_exposed: Principal="*", StatementId="test-statement", ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.eventbridge.eventbridge_bus_exposed.eventbridge_bus_exposed.eventbridge_client", - new=EventBridge(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.eventbridge.eventbridge_bus_exposed.eventbridge_bus_exposed.eventbridge_client", + new=EventBridge(aws_provider), ): from prowler.providers.aws.services.eventbridge.eventbridge_bus_exposed.eventbridge_bus_exposed import ( eventbridge_bus_exposed, diff --git a/tests/providers/aws/services/eventbridge/eventbridge_global_endpoint_event_replication_enabled/eventbridge_global_endpoint_event_replication_enabled_test.py b/tests/providers/aws/services/eventbridge/eventbridge_global_endpoint_event_replication_enabled/eventbridge_global_endpoint_event_replication_enabled_test.py index 1690d2f920..bc0c14694b 100644 --- a/tests/providers/aws/services/eventbridge/eventbridge_global_endpoint_event_replication_enabled/eventbridge_global_endpoint_event_replication_enabled_test.py +++ b/tests/providers/aws/services/eventbridge/eventbridge_global_endpoint_event_replication_enabled/eventbridge_global_endpoint_event_replication_enabled_test.py @@ -51,15 +51,12 @@ class Test_eventbridge_global_endpoint_event_replication_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.eventbridge.eventbridge_global_endpoint_event_replication_enabled.eventbridge_global_endpoint_event_replication_enabled.eventbridge_client", - new=EventBridge(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.eventbridge.eventbridge_global_endpoint_event_replication_enabled.eventbridge_global_endpoint_event_replication_enabled.eventbridge_client", + new=EventBridge(aws_provider), ): from prowler.providers.aws.services.eventbridge.eventbridge_global_endpoint_event_replication_enabled.eventbridge_global_endpoint_event_replication_enabled import ( eventbridge_global_endpoint_event_replication_enabled, @@ -77,15 +74,12 @@ class Test_eventbridge_global_endpoint_event_replication_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.eventbridge.eventbridge_global_endpoint_event_replication_enabled.eventbridge_global_endpoint_event_replication_enabled.eventbridge_client", - new=EventBridge(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.eventbridge.eventbridge_global_endpoint_event_replication_enabled.eventbridge_global_endpoint_event_replication_enabled.eventbridge_client", + new=EventBridge(aws_provider), ): from prowler.providers.aws.services.eventbridge.eventbridge_global_endpoint_event_replication_enabled.eventbridge_global_endpoint_event_replication_enabled import ( eventbridge_global_endpoint_event_replication_enabled, @@ -114,15 +108,12 @@ class Test_eventbridge_global_endpoint_event_replication_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.eventbridge.eventbridge_global_endpoint_event_replication_enabled.eventbridge_global_endpoint_event_replication_enabled.eventbridge_client", - new=EventBridge(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.eventbridge.eventbridge_global_endpoint_event_replication_enabled.eventbridge_global_endpoint_event_replication_enabled.eventbridge_client", + new=EventBridge(aws_provider), ): from prowler.providers.aws.services.eventbridge.eventbridge_global_endpoint_event_replication_enabled.eventbridge_global_endpoint_event_replication_enabled import ( eventbridge_global_endpoint_event_replication_enabled, diff --git a/tests/providers/aws/services/eventbridge/eventbridge_schema_registry_cross_account_access/eventbridge_schema_registry_cross_account_access_test.py b/tests/providers/aws/services/eventbridge/eventbridge_schema_registry_cross_account_access/eventbridge_schema_registry_cross_account_access_test.py index f48700db7b..b12f28af4e 100644 --- a/tests/providers/aws/services/eventbridge/eventbridge_schema_registry_cross_account_access/eventbridge_schema_registry_cross_account_access_test.py +++ b/tests/providers/aws/services/eventbridge/eventbridge_schema_registry_cross_account_access/eventbridge_schema_registry_cross_account_access_test.py @@ -51,15 +51,12 @@ class Test_eventbridge_schema_registry_cross_account_access: schema_client = mock.MagicMock schema_client.registries = {} - with ( - mock.patch( - "prowler.providers.aws.services.eventbridge.eventbridge_service.Schema", - new=schema_client, - ), - mock.patch( - "prowler.providers.aws.services.eventbridge.schema_client.schema_client", - new=schema_client, - ), + with mock.patch( + "prowler.providers.aws.services.eventbridge.eventbridge_service.Schema", + new=schema_client, + ), mock.patch( + "prowler.providers.aws.services.eventbridge.schema_client.schema_client", + new=schema_client, ): from prowler.providers.aws.services.eventbridge.eventbridge_schema_registry_cross_account_access.eventbridge_schema_registry_cross_account_access import ( eventbridge_schema_registry_cross_account_access, @@ -86,15 +83,12 @@ class Test_eventbridge_schema_registry_cross_account_access: ) } - with ( - mock.patch( - "prowler.providers.aws.services.eventbridge.eventbridge_service.Schema", - new=schema_client, - ), - mock.patch( - "prowler.providers.aws.services.eventbridge.schema_client.schema_client", - new=schema_client, - ), + with mock.patch( + "prowler.providers.aws.services.eventbridge.eventbridge_service.Schema", + new=schema_client, + ), mock.patch( + "prowler.providers.aws.services.eventbridge.schema_client.schema_client", + new=schema_client, ): from prowler.providers.aws.services.eventbridge.eventbridge_schema_registry_cross_account_access.eventbridge_schema_registry_cross_account_access import ( eventbridge_schema_registry_cross_account_access, @@ -129,15 +123,12 @@ class Test_eventbridge_schema_registry_cross_account_access: ) } - with ( - mock.patch( - "prowler.providers.aws.services.eventbridge.eventbridge_service.Schema", - new=schema_client, - ), - mock.patch( - "prowler.providers.aws.services.eventbridge.schema_client.schema_client", - new=schema_client, - ), + with mock.patch( + "prowler.providers.aws.services.eventbridge.eventbridge_service.Schema", + new=schema_client, + ), mock.patch( + "prowler.providers.aws.services.eventbridge.schema_client.schema_client", + new=schema_client, ): from prowler.providers.aws.services.eventbridge.eventbridge_schema_registry_cross_account_access.eventbridge_schema_registry_cross_account_access import ( eventbridge_schema_registry_cross_account_access, @@ -172,15 +163,12 @@ class Test_eventbridge_schema_registry_cross_account_access: ) } - with ( - mock.patch( - "prowler.providers.aws.services.eventbridge.eventbridge_service.Schema", - new=schema_client, - ), - mock.patch( - "prowler.providers.aws.services.eventbridge.schema_client.schema_client", - new=schema_client, - ), + with mock.patch( + "prowler.providers.aws.services.eventbridge.eventbridge_service.Schema", + new=schema_client, + ), mock.patch( + "prowler.providers.aws.services.eventbridge.schema_client.schema_client", + new=schema_client, ): from prowler.providers.aws.services.eventbridge.eventbridge_schema_registry_cross_account_access.eventbridge_schema_registry_cross_account_access import ( eventbridge_schema_registry_cross_account_access, diff --git a/tests/providers/aws/services/fsx/fsx_file_system_copy_tags_to_backups_enabled/fsx_file_system_copy_tags_to_backups_enabled_test.py b/tests/providers/aws/services/fsx/fsx_file_system_copy_tags_to_backups_enabled/fsx_file_system_copy_tags_to_backups_enabled_test.py index 5b4a6b322b..6da5bb98ac 100644 --- a/tests/providers/aws/services/fsx/fsx_file_system_copy_tags_to_backups_enabled/fsx_file_system_copy_tags_to_backups_enabled_test.py +++ b/tests/providers/aws/services/fsx/fsx_file_system_copy_tags_to_backups_enabled/fsx_file_system_copy_tags_to_backups_enabled_test.py @@ -14,15 +14,12 @@ class Test_fsx_file_system_copy_tags_to_backups_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled.fsx_client", - new=FSx(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled.fsx_client", + new=FSx(aws_provider), ): from prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled import ( fsx_file_system_copy_tags_to_backups_enabled, @@ -44,15 +41,12 @@ class Test_fsx_file_system_copy_tags_to_backups_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled.fsx_client", - new=FSx(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled.fsx_client", + new=FSx(aws_provider), ): from prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled import ( fsx_file_system_copy_tags_to_backups_enabled, @@ -79,15 +73,12 @@ class Test_fsx_file_system_copy_tags_to_backups_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled.fsx_client", - new=FSx(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled.fsx_client", + new=FSx(aws_provider), ): from prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled import ( fsx_file_system_copy_tags_to_backups_enabled, @@ -125,15 +116,12 @@ class Test_fsx_file_system_copy_tags_to_backups_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled.fsx_client", - new=FSx(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled.fsx_client", + new=FSx(aws_provider), ): from prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled import ( fsx_file_system_copy_tags_to_backups_enabled, @@ -167,15 +155,12 @@ class Test_fsx_file_system_copy_tags_to_backups_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled.fsx_client", - new=FSx(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled.fsx_client", + new=FSx(aws_provider), ): from prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled import ( fsx_file_system_copy_tags_to_backups_enabled, @@ -209,15 +194,12 @@ class Test_fsx_file_system_copy_tags_to_backups_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled.fsx_client", - new=FSx(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled.fsx_client", + new=FSx(aws_provider), ): from prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled import ( fsx_file_system_copy_tags_to_backups_enabled, @@ -254,15 +236,12 @@ class Test_fsx_file_system_copy_tags_to_backups_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled.fsx_client", - new=FSx(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled.fsx_client", + new=FSx(aws_provider), ): from prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled import ( fsx_file_system_copy_tags_to_backups_enabled, @@ -299,15 +278,12 @@ class Test_fsx_file_system_copy_tags_to_backups_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled.fsx_client", - new=FSx(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled.fsx_client", + new=FSx(aws_provider), ): from prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_backups_enabled.fsx_file_system_copy_tags_to_backups_enabled import ( fsx_file_system_copy_tags_to_backups_enabled, diff --git a/tests/providers/aws/services/fsx/fsx_file_system_copy_tags_to_volumes_enabled/fsx_file_system_copy_tags_to_volumes_enabled_test.py b/tests/providers/aws/services/fsx/fsx_file_system_copy_tags_to_volumes_enabled/fsx_file_system_copy_tags_to_volumes_enabled_test.py index 0feb32ac38..30ad6fc0f7 100644 --- a/tests/providers/aws/services/fsx/fsx_file_system_copy_tags_to_volumes_enabled/fsx_file_system_copy_tags_to_volumes_enabled_test.py +++ b/tests/providers/aws/services/fsx/fsx_file_system_copy_tags_to_volumes_enabled/fsx_file_system_copy_tags_to_volumes_enabled_test.py @@ -14,15 +14,12 @@ class Test_fsx_file_system_copy_tags_to_volumes_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_volumes_enabled.fsx_file_system_copy_tags_to_volumes_enabled.fsx_client", - new=FSx(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_volumes_enabled.fsx_file_system_copy_tags_to_volumes_enabled.fsx_client", + new=FSx(aws_provider), ): from prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_volumes_enabled.fsx_file_system_copy_tags_to_volumes_enabled import ( fsx_file_system_copy_tags_to_volumes_enabled, @@ -45,15 +42,12 @@ class Test_fsx_file_system_copy_tags_to_volumes_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_volumes_enabled.fsx_file_system_copy_tags_to_volumes_enabled.fsx_client", - new=FSx(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_volumes_enabled.fsx_file_system_copy_tags_to_volumes_enabled.fsx_client", + new=FSx(aws_provider), ): from prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_volumes_enabled.fsx_file_system_copy_tags_to_volumes_enabled import ( fsx_file_system_copy_tags_to_volumes_enabled, @@ -80,15 +74,12 @@ class Test_fsx_file_system_copy_tags_to_volumes_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_volumes_enabled.fsx_file_system_copy_tags_to_volumes_enabled.fsx_client", - new=FSx(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_volumes_enabled.fsx_file_system_copy_tags_to_volumes_enabled.fsx_client", + new=FSx(aws_provider), ): from prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_volumes_enabled.fsx_file_system_copy_tags_to_volumes_enabled import ( fsx_file_system_copy_tags_to_volumes_enabled, @@ -126,15 +117,12 @@ class Test_fsx_file_system_copy_tags_to_volumes_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_volumes_enabled.fsx_file_system_copy_tags_to_volumes_enabled.fsx_client", - new=FSx(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_volumes_enabled.fsx_file_system_copy_tags_to_volumes_enabled.fsx_client", + new=FSx(aws_provider), ): from prowler.providers.aws.services.fsx.fsx_file_system_copy_tags_to_volumes_enabled.fsx_file_system_copy_tags_to_volumes_enabled import ( fsx_file_system_copy_tags_to_volumes_enabled, diff --git a/tests/providers/aws/services/fsx/fsx_file_system_multi_az_enabled/fsx_file_system_multi_az_enabled_test.py b/tests/providers/aws/services/fsx/fsx_file_system_multi_az_enabled/fsx_file_system_multi_az_enabled_test.py index de7f4a22e2..a4b6bdf000 100644 --- a/tests/providers/aws/services/fsx/fsx_file_system_multi_az_enabled/fsx_file_system_multi_az_enabled_test.py +++ b/tests/providers/aws/services/fsx/fsx_file_system_multi_az_enabled/fsx_file_system_multi_az_enabled_test.py @@ -14,15 +14,12 @@ class Test_fsx_windows_file_system_multi_az: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.fsx.fsx_windows_file_system_multi_az_enabled.fsx_windows_file_system_multi_az_enabled.fsx_client", - new=FSx(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.fsx.fsx_windows_file_system_multi_az_enabled.fsx_windows_file_system_multi_az_enabled.fsx_client", + new=FSx(aws_provider), ): from prowler.providers.aws.services.fsx.fsx_windows_file_system_multi_az_enabled.fsx_windows_file_system_multi_az_enabled import ( fsx_windows_file_system_multi_az_enabled, @@ -45,15 +42,12 @@ class Test_fsx_windows_file_system_multi_az: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.fsx.fsx_windows_file_system_multi_az_enabled.fsx_windows_file_system_multi_az_enabled.fsx_client", - new=FSx(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.fsx.fsx_windows_file_system_multi_az_enabled.fsx_windows_file_system_multi_az_enabled.fsx_client", + new=FSx(aws_provider), ): from prowler.providers.aws.services.fsx.fsx_windows_file_system_multi_az_enabled.fsx_windows_file_system_multi_az_enabled import ( fsx_windows_file_system_multi_az_enabled, @@ -80,15 +74,12 @@ class Test_fsx_windows_file_system_multi_az: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.fsx.fsx_windows_file_system_multi_az_enabled.fsx_windows_file_system_multi_az_enabled.fsx_client", - new=FSx(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.fsx.fsx_windows_file_system_multi_az_enabled.fsx_windows_file_system_multi_az_enabled.fsx_client", + new=FSx(aws_provider), ): from prowler.providers.aws.services.fsx.fsx_windows_file_system_multi_az_enabled.fsx_windows_file_system_multi_az_enabled import ( fsx_windows_file_system_multi_az_enabled, @@ -126,15 +117,12 @@ class Test_fsx_windows_file_system_multi_az: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.fsx.fsx_windows_file_system_multi_az_enabled.fsx_windows_file_system_multi_az_enabled.fsx_client", - new=FSx(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.fsx.fsx_windows_file_system_multi_az_enabled.fsx_windows_file_system_multi_az_enabled.fsx_client", + new=FSx(aws_provider), ): from prowler.providers.aws.services.fsx.fsx_windows_file_system_multi_az_enabled.fsx_windows_file_system_multi_az_enabled import ( fsx_windows_file_system_multi_az_enabled, diff --git a/tests/providers/aws/services/glacier/glacier_vaults_policy_public_access/glacier_vaults_policy_public_access_fixer_test.py b/tests/providers/aws/services/glacier/glacier_vaults_policy_public_access/glacier_vaults_policy_public_access_fixer_test.py index 3bd5b277da..87c808d722 100644 --- a/tests/providers/aws/services/glacier/glacier_vaults_policy_public_access/glacier_vaults_policy_public_access_fixer_test.py +++ b/tests/providers/aws/services/glacier/glacier_vaults_policy_public_access/glacier_vaults_policy_public_access_fixer_test.py @@ -44,15 +44,12 @@ class Test_glacier_vaults_policy_public_access_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.glacier.glacier_vaults_policy_public_access.glacier_vaults_policy_public_access_fixer.glacier_client", - new=Glacier(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.glacier.glacier_vaults_policy_public_access.glacier_vaults_policy_public_access_fixer.glacier_client", + new=Glacier(aws_provider), ): from prowler.providers.aws.services.glacier.glacier_vaults_policy_public_access.glacier_vaults_policy_public_access_fixer import ( fixer, @@ -70,15 +67,12 @@ class Test_glacier_vaults_policy_public_access_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.glacier.glacier_vaults_policy_public_access.glacier_vaults_policy_public_access_fixer.glacier_client", - new=Glacier(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.glacier.glacier_vaults_policy_public_access.glacier_vaults_policy_public_access_fixer.glacier_client", + new=Glacier(aws_provider), ): from prowler.providers.aws.services.glacier.glacier_vaults_policy_public_access.glacier_vaults_policy_public_access_fixer import ( fixer, diff --git a/tests/providers/aws/services/glacier/glacier_vaults_policy_public_access/glacier_vaults_policy_public_access_test.py b/tests/providers/aws/services/glacier/glacier_vaults_policy_public_access/glacier_vaults_policy_public_access_test.py index 1364d15bee..d267e1bc66 100644 --- a/tests/providers/aws/services/glacier/glacier_vaults_policy_public_access/glacier_vaults_policy_public_access_test.py +++ b/tests/providers/aws/services/glacier/glacier_vaults_policy_public_access/glacier_vaults_policy_public_access_test.py @@ -8,15 +8,12 @@ class Test_glacier_vaults_policy_public_access: def test_no_vaults(self): glacier_client = mock.MagicMock glacier_client.vaults = {} - with ( - mock.patch( - "prowler.providers.aws.services.glacier.glacier_service.Glacier", - new=glacier_client, - ), - mock.patch( - "prowler.providers.aws.services.glacier.glacier_client.glacier_client", - new=glacier_client, - ), + with mock.patch( + "prowler.providers.aws.services.glacier.glacier_service.Glacier", + new=glacier_client, + ), mock.patch( + "prowler.providers.aws.services.glacier.glacier_client.glacier_client", + new=glacier_client, ): # Test Check from prowler.providers.aws.services.glacier.glacier_vaults_policy_public_access.glacier_vaults_policy_public_access import ( @@ -40,15 +37,12 @@ class Test_glacier_vaults_policy_public_access: region=AWS_REGION_EU_WEST_1, ) } - with ( - mock.patch( - "prowler.providers.aws.services.glacier.glacier_service.Glacier", - new=glacier_client, - ), - mock.patch( - "prowler.providers.aws.services.glacier.glacier_client.glacier_client", - new=glacier_client, - ), + with mock.patch( + "prowler.providers.aws.services.glacier.glacier_service.Glacier", + new=glacier_client, + ), mock.patch( + "prowler.providers.aws.services.glacier.glacier_client.glacier_client", + new=glacier_client, ): # Test Check from prowler.providers.aws.services.glacier.glacier_vaults_policy_public_access.glacier_vaults_policy_public_access import ( @@ -98,15 +92,12 @@ class Test_glacier_vaults_policy_public_access: region=AWS_REGION_EU_WEST_1, ) } - with ( - mock.patch( - "prowler.providers.aws.services.glacier.glacier_service.Glacier", - new=glacier_client, - ), - mock.patch( - "prowler.providers.aws.services.glacier.glacier_client.glacier_client", - new=glacier_client, - ), + with mock.patch( + "prowler.providers.aws.services.glacier.glacier_service.Glacier", + new=glacier_client, + ), mock.patch( + "prowler.providers.aws.services.glacier.glacier_client.glacier_client", + new=glacier_client, ): # Test Check from prowler.providers.aws.services.glacier.glacier_vaults_policy_public_access.glacier_vaults_policy_public_access import ( @@ -156,15 +147,12 @@ class Test_glacier_vaults_policy_public_access: region=AWS_REGION_EU_WEST_1, ) } - with ( - mock.patch( - "prowler.providers.aws.services.glacier.glacier_service.Glacier", - new=glacier_client, - ), - mock.patch( - "prowler.providers.aws.services.glacier.glacier_client.glacier_client", - new=glacier_client, - ), + with mock.patch( + "prowler.providers.aws.services.glacier.glacier_service.Glacier", + new=glacier_client, + ), mock.patch( + "prowler.providers.aws.services.glacier.glacier_client.glacier_client", + new=glacier_client, ): # Test Check from prowler.providers.aws.services.glacier.glacier_vaults_policy_public_access.glacier_vaults_policy_public_access import ( @@ -214,15 +202,12 @@ class Test_glacier_vaults_policy_public_access: region=AWS_REGION_EU_WEST_1, ) } - with ( - mock.patch( - "prowler.providers.aws.services.glacier.glacier_service.Glacier", - new=glacier_client, - ), - mock.patch( - "prowler.providers.aws.services.glacier.glacier_client.glacier_client", - new=glacier_client, - ), + with mock.patch( + "prowler.providers.aws.services.glacier.glacier_service.Glacier", + new=glacier_client, + ), mock.patch( + "prowler.providers.aws.services.glacier.glacier_client.glacier_client", + new=glacier_client, ): # Test Check from prowler.providers.aws.services.glacier.glacier_vaults_policy_public_access.glacier_vaults_policy_public_access import ( @@ -276,15 +261,12 @@ class Test_glacier_vaults_policy_public_access: region=AWS_REGION_EU_WEST_1, ) } - with ( - mock.patch( - "prowler.providers.aws.services.glacier.glacier_service.Glacier", - new=glacier_client, - ), - mock.patch( - "prowler.providers.aws.services.glacier.glacier_client.glacier_client", - new=glacier_client, - ), + with mock.patch( + "prowler.providers.aws.services.glacier.glacier_service.Glacier", + new=glacier_client, + ), mock.patch( + "prowler.providers.aws.services.glacier.glacier_client.glacier_client", + new=glacier_client, ): # Test Check from prowler.providers.aws.services.glacier.glacier_vaults_policy_public_access.glacier_vaults_policy_public_access import ( diff --git a/tests/providers/aws/services/glue/glue_data_catalogs_connection_passwords_encryption_enabled/glue_data_catalogs_connection_passwords_encryption_enabled_test.py b/tests/providers/aws/services/glue/glue_data_catalogs_connection_passwords_encryption_enabled/glue_data_catalogs_connection_passwords_encryption_enabled_test.py index 88fa11fe26..f9bca0e2c6 100644 --- a/tests/providers/aws/services/glue/glue_data_catalogs_connection_passwords_encryption_enabled/glue_data_catalogs_connection_passwords_encryption_enabled_test.py +++ b/tests/providers/aws/services/glue/glue_data_catalogs_connection_passwords_encryption_enabled/glue_data_catalogs_connection_passwords_encryption_enabled_test.py @@ -81,15 +81,12 @@ class Test_glue_data_catalogs_connection_passwords_encryption_enabled: client("glue", region_name=AWS_REGION_EU_WEST_1) aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.glue.glue_data_catalogs_connection_passwords_encryption_enabled.glue_data_catalogs_connection_passwords_encryption_enabled.glue_client", - new=Glue(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.glue.glue_data_catalogs_connection_passwords_encryption_enabled.glue_data_catalogs_connection_passwords_encryption_enabled.glue_client", + new=Glue(aws_provider), ): # Test Check from prowler.providers.aws.services.glue.glue_data_catalogs_connection_passwords_encryption_enabled.glue_data_catalogs_connection_passwords_encryption_enabled import ( @@ -107,15 +104,12 @@ class Test_glue_data_catalogs_connection_passwords_encryption_enabled: client("glue", region_name=AWS_REGION_EU_WEST_1) aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.glue.glue_data_catalogs_connection_passwords_encryption_enabled.glue_data_catalogs_connection_passwords_encryption_enabled.glue_client", - new=Glue(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.glue.glue_data_catalogs_connection_passwords_encryption_enabled.glue_data_catalogs_connection_passwords_encryption_enabled.glue_client", + new=Glue(aws_provider), ): # Test Check from prowler.providers.aws.services.glue.glue_data_catalogs_connection_passwords_encryption_enabled.glue_data_catalogs_connection_passwords_encryption_enabled import ( @@ -145,15 +139,12 @@ class Test_glue_data_catalogs_connection_passwords_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) aws_provider._scan_unused_services = False - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.glue.glue_data_catalogs_connection_passwords_encryption_enabled.glue_data_catalogs_connection_passwords_encryption_enabled.glue_client", - new=Glue(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.glue.glue_data_catalogs_connection_passwords_encryption_enabled.glue_data_catalogs_connection_passwords_encryption_enabled.glue_client", + new=Glue(aws_provider), ): # Test Check from prowler.providers.aws.services.glue.glue_data_catalogs_connection_passwords_encryption_enabled.glue_data_catalogs_connection_passwords_encryption_enabled import ( @@ -172,15 +163,12 @@ class Test_glue_data_catalogs_connection_passwords_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) aws_provider._scan_unused_services = False - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.glue.glue_data_catalogs_connection_passwords_encryption_enabled.glue_data_catalogs_connection_passwords_encryption_enabled.glue_client", - new=Glue(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.glue.glue_data_catalogs_connection_passwords_encryption_enabled.glue_data_catalogs_connection_passwords_encryption_enabled.glue_client", + new=Glue(aws_provider), ): # Test Check from prowler.providers.aws.services.glue.glue_data_catalogs_connection_passwords_encryption_enabled.glue_data_catalogs_connection_passwords_encryption_enabled import ( @@ -209,15 +197,12 @@ class Test_glue_data_catalogs_connection_passwords_encryption_enabled: client("glue", region_name=AWS_REGION_EU_WEST_1) aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.glue.glue_data_catalogs_connection_passwords_encryption_enabled.glue_data_catalogs_connection_passwords_encryption_enabled.glue_client", - new=Glue(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.glue.glue_data_catalogs_connection_passwords_encryption_enabled.glue_data_catalogs_connection_passwords_encryption_enabled.glue_client", + new=Glue(aws_provider), ): # Test Check from prowler.providers.aws.services.glue.glue_data_catalogs_connection_passwords_encryption_enabled.glue_data_catalogs_connection_passwords_encryption_enabled import ( diff --git a/tests/providers/aws/services/glue/glue_data_catalogs_metadata_encryption_enabled/glue_data_catalogs_metadata_encryption_enabled_test.py b/tests/providers/aws/services/glue/glue_data_catalogs_metadata_encryption_enabled/glue_data_catalogs_metadata_encryption_enabled_test.py index 4c79c820f9..46e7db6c09 100644 --- a/tests/providers/aws/services/glue/glue_data_catalogs_metadata_encryption_enabled/glue_data_catalogs_metadata_encryption_enabled_test.py +++ b/tests/providers/aws/services/glue/glue_data_catalogs_metadata_encryption_enabled/glue_data_catalogs_metadata_encryption_enabled_test.py @@ -81,15 +81,12 @@ class Test_glue_data_catalogs_metadata_encryption_enabled: client("glue", region_name=AWS_REGION_EU_WEST_1) aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.glue.glue_data_catalogs_metadata_encryption_enabled.glue_data_catalogs_metadata_encryption_enabled.glue_client", - new=Glue(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.glue.glue_data_catalogs_metadata_encryption_enabled.glue_data_catalogs_metadata_encryption_enabled.glue_client", + new=Glue(aws_provider), ): # Test Check from prowler.providers.aws.services.glue.glue_data_catalogs_metadata_encryption_enabled.glue_data_catalogs_metadata_encryption_enabled import ( @@ -107,15 +104,12 @@ class Test_glue_data_catalogs_metadata_encryption_enabled: client("glue", region_name=AWS_REGION_EU_WEST_1) aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.glue.glue_data_catalogs_metadata_encryption_enabled.glue_data_catalogs_metadata_encryption_enabled.glue_client", - new=Glue(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.glue.glue_data_catalogs_metadata_encryption_enabled.glue_data_catalogs_metadata_encryption_enabled.glue_client", + new=Glue(aws_provider), ): # Test Check from prowler.providers.aws.services.glue.glue_data_catalogs_metadata_encryption_enabled.glue_data_catalogs_metadata_encryption_enabled import ( @@ -144,15 +138,12 @@ class Test_glue_data_catalogs_metadata_encryption_enabled: client("glue", region_name=AWS_REGION_EU_WEST_1) aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) aws_provider._scan_unused_services = False - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.glue.glue_data_catalogs_metadata_encryption_enabled.glue_data_catalogs_metadata_encryption_enabled.glue_client", - new=Glue(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.glue.glue_data_catalogs_metadata_encryption_enabled.glue_data_catalogs_metadata_encryption_enabled.glue_client", + new=Glue(aws_provider), ): # Test Check from prowler.providers.aws.services.glue.glue_data_catalogs_metadata_encryption_enabled.glue_data_catalogs_metadata_encryption_enabled import ( @@ -170,15 +161,12 @@ class Test_glue_data_catalogs_metadata_encryption_enabled: client("glue", region_name=AWS_REGION_EU_WEST_1) aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) aws_provider._scan_unused_services = False - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.glue.glue_data_catalogs_metadata_encryption_enabled.glue_data_catalogs_metadata_encryption_enabled.glue_client", - new=Glue(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.glue.glue_data_catalogs_metadata_encryption_enabled.glue_data_catalogs_metadata_encryption_enabled.glue_client", + new=Glue(aws_provider), ): # Test Check from prowler.providers.aws.services.glue.glue_data_catalogs_metadata_encryption_enabled.glue_data_catalogs_metadata_encryption_enabled import ( @@ -207,15 +195,12 @@ class Test_glue_data_catalogs_metadata_encryption_enabled: client("glue", region_name=AWS_REGION_EU_WEST_1) aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.glue.glue_data_catalogs_metadata_encryption_enabled.glue_data_catalogs_metadata_encryption_enabled.glue_client", - new=Glue(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.glue.glue_data_catalogs_metadata_encryption_enabled.glue_data_catalogs_metadata_encryption_enabled.glue_client", + new=Glue(aws_provider), ): # Test Check from prowler.providers.aws.services.glue.glue_data_catalogs_metadata_encryption_enabled.glue_data_catalogs_metadata_encryption_enabled import ( diff --git a/tests/providers/aws/services/glue/glue_data_catalogs_not_publicly_accessible/glue_data_catalogs_not_publicly_accessible_test.py b/tests/providers/aws/services/glue/glue_data_catalogs_not_publicly_accessible/glue_data_catalogs_not_publicly_accessible_test.py index 3448ccd1ca..1e802e38f6 100644 --- a/tests/providers/aws/services/glue/glue_data_catalogs_not_publicly_accessible/glue_data_catalogs_not_publicly_accessible_test.py +++ b/tests/providers/aws/services/glue/glue_data_catalogs_not_publicly_accessible/glue_data_catalogs_not_publicly_accessible_test.py @@ -62,15 +62,12 @@ class Test_glue_data_catalogs_not_publicly_accessible: client("glue", region_name=AWS_REGION_EU_WEST_1) aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.glue.glue_data_catalogs_not_publicly_accessible.glue_data_catalogs_not_publicly_accessible.glue_client", - new=Glue(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.glue.glue_data_catalogs_not_publicly_accessible.glue_data_catalogs_not_publicly_accessible.glue_client", + new=Glue(aws_provider), ): # Test Check from prowler.providers.aws.services.glue.glue_data_catalogs_not_publicly_accessible.glue_data_catalogs_not_publicly_accessible import ( @@ -88,15 +85,12 @@ class Test_glue_data_catalogs_not_publicly_accessible: client("glue", region_name=AWS_REGION_EU_WEST_1) aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.glue.glue_data_catalogs_not_publicly_accessible.glue_data_catalogs_not_publicly_accessible.glue_client", - new=Glue(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.glue.glue_data_catalogs_not_publicly_accessible.glue_data_catalogs_not_publicly_accessible.glue_client", + new=Glue(aws_provider), ): # Test Check from prowler.providers.aws.services.glue.glue_data_catalogs_not_publicly_accessible.glue_data_catalogs_not_publicly_accessible import ( @@ -124,15 +118,12 @@ class Test_glue_data_catalogs_not_publicly_accessible: client("glue", region_name=AWS_REGION_EU_WEST_1) aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.glue.glue_data_catalogs_not_publicly_accessible.glue_data_catalogs_not_publicly_accessible.glue_client", - new=Glue(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.glue.glue_data_catalogs_not_publicly_accessible.glue_data_catalogs_not_publicly_accessible.glue_client", + new=Glue(aws_provider), ): # Test Check from prowler.providers.aws.services.glue.glue_data_catalogs_not_publicly_accessible.glue_data_catalogs_not_publicly_accessible import ( diff --git a/tests/providers/aws/services/glue/glue_database_connections_ssl_enabled/glue_database_connections_ssl_enabled_test.py b/tests/providers/aws/services/glue/glue_database_connections_ssl_enabled/glue_database_connections_ssl_enabled_test.py index 8e0e76e3ba..9cf73ff830 100644 --- a/tests/providers/aws/services/glue/glue_database_connections_ssl_enabled/glue_database_connections_ssl_enabled_test.py +++ b/tests/providers/aws/services/glue/glue_database_connections_ssl_enabled/glue_database_connections_ssl_enabled_test.py @@ -9,15 +9,12 @@ class Test_glue_database_connections_ssl_enabled: glue_client = MagicMock glue_client.connections = [] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_database_connections_ssl_enabled.glue_database_connections_ssl_enabled import ( @@ -47,15 +44,12 @@ class Test_glue_database_connections_ssl_enabled: ) ] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_database_connections_ssl_enabled.glue_database_connections_ssl_enabled import ( @@ -94,15 +88,12 @@ class Test_glue_database_connections_ssl_enabled: ) ] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_database_connections_ssl_enabled.glue_database_connections_ssl_enabled import ( diff --git a/tests/providers/aws/services/glue/glue_development_endpoints_cloudwatch_logs_encryption_enabled/glue_development_endpoints_cloudwatch_logs_encryption_enabled_test.py b/tests/providers/aws/services/glue/glue_development_endpoints_cloudwatch_logs_encryption_enabled/glue_development_endpoints_cloudwatch_logs_encryption_enabled_test.py index 78ca8db1a7..37eb7531cf 100644 --- a/tests/providers/aws/services/glue/glue_development_endpoints_cloudwatch_logs_encryption_enabled/glue_development_endpoints_cloudwatch_logs_encryption_enabled_test.py +++ b/tests/providers/aws/services/glue/glue_development_endpoints_cloudwatch_logs_encryption_enabled/glue_development_endpoints_cloudwatch_logs_encryption_enabled_test.py @@ -9,15 +9,12 @@ class Test_glue_development_endpoints_cloudwatch_logs_encryption_enabled: glue_client = MagicMock glue_client.dev_endpoints = [] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_development_endpoints_cloudwatch_logs_encryption_enabled.glue_development_endpoints_cloudwatch_logs_encryption_enabled import ( @@ -51,15 +48,12 @@ class Test_glue_development_endpoints_cloudwatch_logs_encryption_enabled: ) ] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_development_endpoints_cloudwatch_logs_encryption_enabled.glue_development_endpoints_cloudwatch_logs_encryption_enabled import ( @@ -100,15 +94,12 @@ class Test_glue_development_endpoints_cloudwatch_logs_encryption_enabled: ) ] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_development_endpoints_cloudwatch_logs_encryption_enabled.glue_development_endpoints_cloudwatch_logs_encryption_enabled import ( @@ -141,15 +132,12 @@ class Test_glue_development_endpoints_cloudwatch_logs_encryption_enabled: ] glue_client.security_configs = [] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_development_endpoints_cloudwatch_logs_encryption_enabled.glue_development_endpoints_cloudwatch_logs_encryption_enabled import ( diff --git a/tests/providers/aws/services/glue/glue_development_endpoints_job_bookmark_encryption_enabled/glue_development_endpoints_job_bookmark_encryption_enabled_test.py b/tests/providers/aws/services/glue/glue_development_endpoints_job_bookmark_encryption_enabled/glue_development_endpoints_job_bookmark_encryption_enabled_test.py index 084b1f1bac..c7b73394e1 100644 --- a/tests/providers/aws/services/glue/glue_development_endpoints_job_bookmark_encryption_enabled/glue_development_endpoints_job_bookmark_encryption_enabled_test.py +++ b/tests/providers/aws/services/glue/glue_development_endpoints_job_bookmark_encryption_enabled/glue_development_endpoints_job_bookmark_encryption_enabled_test.py @@ -9,15 +9,12 @@ class Test_glue_development_endpoints_job_bookmark_encryption_enabled: glue_client = MagicMock glue_client.dev_endpoints = [] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_development_endpoints_job_bookmark_encryption_enabled.glue_development_endpoints_job_bookmark_encryption_enabled import ( @@ -51,15 +48,12 @@ class Test_glue_development_endpoints_job_bookmark_encryption_enabled: ) ] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_development_endpoints_job_bookmark_encryption_enabled.glue_development_endpoints_job_bookmark_encryption_enabled import ( @@ -100,15 +94,12 @@ class Test_glue_development_endpoints_job_bookmark_encryption_enabled: ) ] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_development_endpoints_job_bookmark_encryption_enabled.glue_development_endpoints_job_bookmark_encryption_enabled import ( @@ -141,15 +132,12 @@ class Test_glue_development_endpoints_job_bookmark_encryption_enabled: ] glue_client.security_configs = [] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_development_endpoints_job_bookmark_encryption_enabled.glue_development_endpoints_job_bookmark_encryption_enabled import ( diff --git a/tests/providers/aws/services/glue/glue_development_endpoints_s3_encryption_enabled/glue_development_endpoints_s3_encryption_enabled_test.py b/tests/providers/aws/services/glue/glue_development_endpoints_s3_encryption_enabled/glue_development_endpoints_s3_encryption_enabled_test.py index 8dbbce6d89..f43bab581d 100644 --- a/tests/providers/aws/services/glue/glue_development_endpoints_s3_encryption_enabled/glue_development_endpoints_s3_encryption_enabled_test.py +++ b/tests/providers/aws/services/glue/glue_development_endpoints_s3_encryption_enabled/glue_development_endpoints_s3_encryption_enabled_test.py @@ -9,15 +9,12 @@ class Test_glue_development_endpoints_s3_encryption_enabled: glue_client = MagicMock glue_client.dev_endpoints = [] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_development_endpoints_s3_encryption_enabled.glue_development_endpoints_s3_encryption_enabled import ( @@ -51,15 +48,12 @@ class Test_glue_development_endpoints_s3_encryption_enabled: ) ] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_development_endpoints_s3_encryption_enabled.glue_development_endpoints_s3_encryption_enabled import ( @@ -100,15 +94,12 @@ class Test_glue_development_endpoints_s3_encryption_enabled: ) ] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_development_endpoints_s3_encryption_enabled.glue_development_endpoints_s3_encryption_enabled import ( @@ -141,15 +132,12 @@ class Test_glue_development_endpoints_s3_encryption_enabled: ] glue_client.security_configs = [] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_development_endpoints_s3_encryption_enabled.glue_development_endpoints_s3_encryption_enabled import ( diff --git a/tests/providers/aws/services/glue/glue_etl_jobs_amazon_s3_encryption_enabled/glue_etl_jobs_amazon_s3_encryption_enabled_test.py b/tests/providers/aws/services/glue/glue_etl_jobs_amazon_s3_encryption_enabled/glue_etl_jobs_amazon_s3_encryption_enabled_test.py index 23f6d9343f..f2865f6aac 100644 --- a/tests/providers/aws/services/glue/glue_etl_jobs_amazon_s3_encryption_enabled/glue_etl_jobs_amazon_s3_encryption_enabled_test.py +++ b/tests/providers/aws/services/glue/glue_etl_jobs_amazon_s3_encryption_enabled/glue_etl_jobs_amazon_s3_encryption_enabled_test.py @@ -9,15 +9,12 @@ class Test_glue_etl_jobs_amazon_s3_encryption_enabled: glue_client = MagicMock glue_client.jobs = [] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_etl_jobs_amazon_s3_encryption_enabled.glue_etl_jobs_amazon_s3_encryption_enabled import ( @@ -52,15 +49,12 @@ class Test_glue_etl_jobs_amazon_s3_encryption_enabled: ) ] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_etl_jobs_amazon_s3_encryption_enabled.glue_etl_jobs_amazon_s3_encryption_enabled import ( @@ -102,15 +96,12 @@ class Test_glue_etl_jobs_amazon_s3_encryption_enabled: ) ] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_etl_jobs_amazon_s3_encryption_enabled.glue_etl_jobs_amazon_s3_encryption_enabled import ( @@ -143,15 +134,12 @@ class Test_glue_etl_jobs_amazon_s3_encryption_enabled: ] glue_client.security_configs = [] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_etl_jobs_amazon_s3_encryption_enabled.glue_etl_jobs_amazon_s3_encryption_enabled import ( @@ -188,15 +176,12 @@ class Test_glue_etl_jobs_amazon_s3_encryption_enabled: ] glue_client.security_configs = [] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_etl_jobs_amazon_s3_encryption_enabled.glue_etl_jobs_amazon_s3_encryption_enabled import ( diff --git a/tests/providers/aws/services/glue/glue_etl_jobs_cloudwatch_logs_encryption_enabled/glue_etl_jobs_cloudwatch_logs_encryption_enabled_test.py b/tests/providers/aws/services/glue/glue_etl_jobs_cloudwatch_logs_encryption_enabled/glue_etl_jobs_cloudwatch_logs_encryption_enabled_test.py index 29c5d08da7..1683ea5a18 100644 --- a/tests/providers/aws/services/glue/glue_etl_jobs_cloudwatch_logs_encryption_enabled/glue_etl_jobs_cloudwatch_logs_encryption_enabled_test.py +++ b/tests/providers/aws/services/glue/glue_etl_jobs_cloudwatch_logs_encryption_enabled/glue_etl_jobs_cloudwatch_logs_encryption_enabled_test.py @@ -9,15 +9,12 @@ class Test_glue_etl_jobs_cloudwatch_logs_encryption_enabled: glue_client = MagicMock glue_client.jobs = [] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_etl_jobs_cloudwatch_logs_encryption_enabled.glue_etl_jobs_cloudwatch_logs_encryption_enabled import ( @@ -52,15 +49,12 @@ class Test_glue_etl_jobs_cloudwatch_logs_encryption_enabled: ) ] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_etl_jobs_cloudwatch_logs_encryption_enabled.glue_etl_jobs_cloudwatch_logs_encryption_enabled import ( @@ -102,15 +96,12 @@ class Test_glue_etl_jobs_cloudwatch_logs_encryption_enabled: ) ] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_etl_jobs_cloudwatch_logs_encryption_enabled.glue_etl_jobs_cloudwatch_logs_encryption_enabled import ( @@ -143,15 +134,12 @@ class Test_glue_etl_jobs_cloudwatch_logs_encryption_enabled: ] glue_client.security_configs = [] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_etl_jobs_cloudwatch_logs_encryption_enabled.glue_etl_jobs_cloudwatch_logs_encryption_enabled import ( diff --git a/tests/providers/aws/services/glue/glue_etl_jobs_job_bookmark_encryption_enabled/glue_etl_jobs_job_bookmark_encryption_enabled_test.py b/tests/providers/aws/services/glue/glue_etl_jobs_job_bookmark_encryption_enabled/glue_etl_jobs_job_bookmark_encryption_enabled_test.py index 86e42e031c..4999e27e48 100644 --- a/tests/providers/aws/services/glue/glue_etl_jobs_job_bookmark_encryption_enabled/glue_etl_jobs_job_bookmark_encryption_enabled_test.py +++ b/tests/providers/aws/services/glue/glue_etl_jobs_job_bookmark_encryption_enabled/glue_etl_jobs_job_bookmark_encryption_enabled_test.py @@ -9,15 +9,12 @@ class Test_glue_etl_jobs_job_bookmark_encryption_enabled: glue_client = MagicMock glue_client.jobs = [] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_etl_jobs_job_bookmark_encryption_enabled.glue_etl_jobs_job_bookmark_encryption_enabled import ( @@ -52,15 +49,12 @@ class Test_glue_etl_jobs_job_bookmark_encryption_enabled: ) ] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_etl_jobs_job_bookmark_encryption_enabled.glue_etl_jobs_job_bookmark_encryption_enabled import ( @@ -102,15 +96,12 @@ class Test_glue_etl_jobs_job_bookmark_encryption_enabled: ) ] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_etl_jobs_job_bookmark_encryption_enabled.glue_etl_jobs_job_bookmark_encryption_enabled import ( @@ -143,15 +134,12 @@ class Test_glue_etl_jobs_job_bookmark_encryption_enabled: ] glue_client.security_configs = [] - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): # Test Check from prowler.providers.aws.services.glue.glue_etl_jobs_job_bookmark_encryption_enabled.glue_etl_jobs_job_bookmark_encryption_enabled import ( diff --git a/tests/providers/aws/services/glue/glue_ml_transform_encrypted_at_rest/glue_ml_transform_encrypted_at_rest_test.py b/tests/providers/aws/services/glue/glue_ml_transform_encrypted_at_rest/glue_ml_transform_encrypted_at_rest_test.py index db4f85bae6..19e9f88cbe 100644 --- a/tests/providers/aws/services/glue/glue_ml_transform_encrypted_at_rest/glue_ml_transform_encrypted_at_rest_test.py +++ b/tests/providers/aws/services/glue/glue_ml_transform_encrypted_at_rest/glue_ml_transform_encrypted_at_rest_test.py @@ -8,15 +8,12 @@ class Test_glue_ml_transform_encrypted_at_rest: glue_client = MagicMock glue_client.ml_transforms = {} - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): from prowler.providers.aws.services.glue.glue_ml_transform_encrypted_at_rest.glue_ml_transform_encrypted_at_rest import ( glue_ml_transform_encrypted_at_rest, @@ -45,15 +42,12 @@ class Test_glue_ml_transform_encrypted_at_rest: ) } - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): from prowler.providers.aws.services.glue.glue_ml_transform_encrypted_at_rest.glue_ml_transform_encrypted_at_rest import ( glue_ml_transform_encrypted_at_rest, @@ -90,15 +84,12 @@ class Test_glue_ml_transform_encrypted_at_rest: ) } - with ( - patch( - "prowler.providers.aws.services.glue.glue_service.Glue", - new=glue_client, - ), - patch( - "prowler.providers.aws.services.glue.glue_client.glue_client", - new=glue_client, - ), + with patch( + "prowler.providers.aws.services.glue.glue_service.Glue", + new=glue_client, + ), patch( + "prowler.providers.aws.services.glue.glue_client.glue_client", + new=glue_client, ): from prowler.providers.aws.services.glue.glue_ml_transform_encrypted_at_rest.glue_ml_transform_encrypted_at_rest import ( glue_ml_transform_encrypted_at_rest, diff --git a/tests/providers/aws/services/guardduty/guardduty_centrally_managed/guardduty_centrally_managed_test.py b/tests/providers/aws/services/guardduty/guardduty_centrally_managed/guardduty_centrally_managed_test.py index 5459a969b0..739d9728df 100644 --- a/tests/providers/aws/services/guardduty/guardduty_centrally_managed/guardduty_centrally_managed_test.py +++ b/tests/providers/aws/services/guardduty/guardduty_centrally_managed/guardduty_centrally_managed_test.py @@ -43,15 +43,12 @@ class Test_guardduty_centrally_managed: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_centrally_managed.guardduty_centrally_managed.guardduty_client", - new=GuardDuty(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_centrally_managed.guardduty_centrally_managed.guardduty_client", + new=GuardDuty(aws_provider), ): from prowler.providers.aws.services.guardduty.guardduty_centrally_managed.guardduty_centrally_managed import ( guardduty_centrally_managed, @@ -71,15 +68,12 @@ class Test_guardduty_centrally_managed: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_centrally_managed.guardduty_centrally_managed.guardduty_client", - new=GuardDuty(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_centrally_managed.guardduty_centrally_managed.guardduty_client", + new=GuardDuty(aws_provider), ): # Test Check from prowler.providers.aws.services.guardduty.guardduty_centrally_managed.guardduty_centrally_managed import ( @@ -116,15 +110,12 @@ class Test_guardduty_centrally_managed: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_centrally_managed.guardduty_centrally_managed.guardduty_client", - new=GuardDuty(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_centrally_managed.guardduty_centrally_managed.guardduty_client", + new=GuardDuty(aws_provider), ): # Test Check from prowler.providers.aws.services.guardduty.guardduty_centrally_managed.guardduty_centrally_managed import ( @@ -160,15 +151,12 @@ class Test_guardduty_centrally_managed: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_centrally_managed.guardduty_centrally_managed.guardduty_client", - new=GuardDuty(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_centrally_managed.guardduty_centrally_managed.guardduty_client", + new=GuardDuty(aws_provider), ): # Test Check from prowler.providers.aws.services.guardduty.guardduty_centrally_managed.guardduty_centrally_managed import ( diff --git a/tests/providers/aws/services/guardduty/guardduty_ec2_malware_protection_enabled/guardduty_ec2_malware_protection_enabled_test.py b/tests/providers/aws/services/guardduty/guardduty_ec2_malware_protection_enabled/guardduty_ec2_malware_protection_enabled_test.py index 2b5b127adf..afd1e90da8 100644 --- a/tests/providers/aws/services/guardduty/guardduty_ec2_malware_protection_enabled/guardduty_ec2_malware_protection_enabled_test.py +++ b/tests/providers/aws/services/guardduty/guardduty_ec2_malware_protection_enabled/guardduty_ec2_malware_protection_enabled_test.py @@ -48,15 +48,12 @@ class Test_guardduty_ec2_malware_protection_enabled: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_ec2_malware_protection_enabled.guardduty_ec2_malware_protection_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_ec2_malware_protection_enabled.guardduty_ec2_malware_protection_enabled.guardduty_client", + new=GuardDuty(aws_provider), ): # Test Check from prowler.providers.aws.services.guardduty.guardduty_ec2_malware_protection_enabled.guardduty_ec2_malware_protection_enabled import ( @@ -78,15 +75,12 @@ class Test_guardduty_ec2_malware_protection_enabled: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_ec2_malware_protection_enabled.guardduty_ec2_malware_protection_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_ec2_malware_protection_enabled.guardduty_ec2_malware_protection_enabled.guardduty_client", + new=GuardDuty(aws_provider), ): # Test Check from prowler.providers.aws.services.guardduty.guardduty_ec2_malware_protection_enabled.guardduty_ec2_malware_protection_enabled import ( @@ -116,15 +110,12 @@ class Test_guardduty_ec2_malware_protection_enabled: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_ec2_malware_protection_enabled.guardduty_ec2_malware_protection_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_ec2_malware_protection_enabled.guardduty_ec2_malware_protection_enabled.guardduty_client", + new=GuardDuty(aws_provider), ): # Test Check from prowler.providers.aws.services.guardduty.guardduty_ec2_malware_protection_enabled.guardduty_ec2_malware_protection_enabled import ( @@ -165,15 +156,12 @@ class Test_guardduty_ec2_malware_protection_enabled: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_ec2_malware_protection_enabled.guardduty_ec2_malware_protection_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_ec2_malware_protection_enabled.guardduty_ec2_malware_protection_enabled.guardduty_client", + new=GuardDuty(aws_provider), ): # Test Check from prowler.providers.aws.services.guardduty.guardduty_ec2_malware_protection_enabled.guardduty_ec2_malware_protection_enabled import ( diff --git a/tests/providers/aws/services/guardduty/guardduty_eks_audit_log_enabled/guardduty_eks_audit_log_enabled_test.py b/tests/providers/aws/services/guardduty/guardduty_eks_audit_log_enabled/guardduty_eks_audit_log_enabled_test.py index 563a309a01..7a373f15cc 100644 --- a/tests/providers/aws/services/guardduty/guardduty_eks_audit_log_enabled/guardduty_eks_audit_log_enabled_test.py +++ b/tests/providers/aws/services/guardduty/guardduty_eks_audit_log_enabled/guardduty_eks_audit_log_enabled_test.py @@ -16,15 +16,12 @@ class Test_guardduty_eks_audit_log_enabled: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_eks_audit_log_enabled.guardduty_eks_audit_log_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_eks_audit_log_enabled.guardduty_eks_audit_log_enabled.guardduty_client", + new=GuardDuty(aws_provider), ): # Test Check from prowler.providers.aws.services.guardduty.guardduty_eks_audit_log_enabled.guardduty_eks_audit_log_enabled import ( @@ -46,15 +43,12 @@ class Test_guardduty_eks_audit_log_enabled: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_eks_audit_log_enabled.guardduty_eks_audit_log_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_eks_audit_log_enabled.guardduty_eks_audit_log_enabled.guardduty_client", + new=GuardDuty(aws_provider), ): # Test Check from prowler.providers.aws.services.guardduty.guardduty_eks_audit_log_enabled.guardduty_eks_audit_log_enabled import ( @@ -78,15 +72,12 @@ class Test_guardduty_eks_audit_log_enabled: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_eks_audit_log_enabled.guardduty_eks_audit_log_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_eks_audit_log_enabled.guardduty_eks_audit_log_enabled.guardduty_client", + new=GuardDuty(aws_provider), ): # Test Check from prowler.providers.aws.services.guardduty.guardduty_eks_audit_log_enabled.guardduty_eks_audit_log_enabled import ( @@ -122,15 +113,12 @@ class Test_guardduty_eks_audit_log_enabled: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_eks_audit_log_enabled.guardduty_eks_audit_log_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_eks_audit_log_enabled.guardduty_eks_audit_log_enabled.guardduty_client", + new=GuardDuty(aws_provider), ): # Test Check from prowler.providers.aws.services.guardduty.guardduty_eks_audit_log_enabled.guardduty_eks_audit_log_enabled import ( diff --git a/tests/providers/aws/services/guardduty/guardduty_eks_runtime_monitoring_enabled/guardduty_eks_runtime_monitoring_enabled_test.py b/tests/providers/aws/services/guardduty/guardduty_eks_runtime_monitoring_enabled/guardduty_eks_runtime_monitoring_enabled_test.py index fd55b81018..845dba57ce 100644 --- a/tests/providers/aws/services/guardduty/guardduty_eks_runtime_monitoring_enabled/guardduty_eks_runtime_monitoring_enabled_test.py +++ b/tests/providers/aws/services/guardduty/guardduty_eks_runtime_monitoring_enabled/guardduty_eks_runtime_monitoring_enabled_test.py @@ -18,15 +18,12 @@ class Test_guardduty_eks_runtime_monitoring_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_eks_runtime_monitoring_enabled.guardduty_eks_runtime_monitoring_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_eks_runtime_monitoring_enabled.guardduty_eks_runtime_monitoring_enabled.guardduty_client", + new=GuardDuty(aws_provider), ): from prowler.providers.aws.services.guardduty.guardduty_eks_runtime_monitoring_enabled.guardduty_eks_runtime_monitoring_enabled import ( @@ -52,15 +49,12 @@ class Test_guardduty_eks_runtime_monitoring_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_eks_runtime_monitoring_enabled.guardduty_eks_runtime_monitoring_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_eks_runtime_monitoring_enabled.guardduty_eks_runtime_monitoring_enabled.guardduty_client", + new=GuardDuty(aws_provider), ): from prowler.providers.aws.services.guardduty.guardduty_eks_runtime_monitoring_enabled.guardduty_eks_runtime_monitoring_enabled import ( @@ -86,15 +80,12 @@ class Test_guardduty_eks_runtime_monitoring_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_eks_runtime_monitoring_enabled.guardduty_eks_runtime_monitoring_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_eks_runtime_monitoring_enabled.guardduty_eks_runtime_monitoring_enabled.guardduty_client", + new=GuardDuty(aws_provider), ): from prowler.providers.aws.services.guardduty.guardduty_eks_runtime_monitoring_enabled.guardduty_eks_runtime_monitoring_enabled import ( @@ -132,15 +123,12 @@ class Test_guardduty_eks_runtime_monitoring_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_eks_runtime_monitoring_enabled.guardduty_eks_runtime_monitoring_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_eks_runtime_monitoring_enabled.guardduty_eks_runtime_monitoring_enabled.guardduty_client", + new=GuardDuty(aws_provider), ): from prowler.providers.aws.services.guardduty.guardduty_eks_runtime_monitoring_enabled.guardduty_eks_runtime_monitoring_enabled import ( guardduty_eks_runtime_monitoring_enabled, diff --git a/tests/providers/aws/services/guardduty/guardduty_is_enabled/guardduty_is_enabled_fixer_test.py b/tests/providers/aws/services/guardduty/guardduty_is_enabled/guardduty_is_enabled_fixer_test.py index e08524d4ce..8a7e3f4eb6 100644 --- a/tests/providers/aws/services/guardduty/guardduty_is_enabled/guardduty_is_enabled_fixer_test.py +++ b/tests/providers/aws/services/guardduty/guardduty_is_enabled/guardduty_is_enabled_fixer_test.py @@ -53,15 +53,12 @@ class Test_guardduty_is_enabled_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled_fixer.guardduty_client", - new=GuardDuty(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled_fixer.guardduty_client", + new=GuardDuty(aws_provider), ): from prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled_fixer import ( fixer, @@ -82,15 +79,12 @@ class Test_guardduty_is_enabled_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled_fixer.guardduty_client", - new=GuardDuty(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled_fixer.guardduty_client", + new=GuardDuty(aws_provider), ): from prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled_fixer import ( fixer, diff --git a/tests/providers/aws/services/guardduty/guardduty_is_enabled/guardduty_is_enabled_test.py b/tests/providers/aws/services/guardduty/guardduty_is_enabled/guardduty_is_enabled_test.py index 47b122d5e9..25e6aaa18d 100644 --- a/tests/providers/aws/services/guardduty/guardduty_is_enabled/guardduty_is_enabled_test.py +++ b/tests/providers/aws/services/guardduty/guardduty_is_enabled/guardduty_is_enabled_test.py @@ -17,16 +17,13 @@ class Test_guardduty_is_enabled: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ) as guardduty_client, - ): + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled.guardduty_client", + new=GuardDuty(aws_provider), + ) as guardduty_client: from prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled import ( guardduty_is_enabled, ) @@ -47,15 +44,12 @@ class Test_guardduty_is_enabled: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled.guardduty_client", + new=GuardDuty(aws_provider), ): from prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled import ( guardduty_is_enabled, @@ -88,16 +82,13 @@ class Test_guardduty_is_enabled: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ) as mock_guardduty_client, - ): + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled.guardduty_client", + new=GuardDuty(aws_provider), + ) as mock_guardduty_client: from prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled import ( guardduty_is_enabled, ) @@ -133,16 +124,13 @@ class Test_guardduty_is_enabled: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ) as mock_guardduty_client, - ): + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled.guardduty_client", + new=GuardDuty(aws_provider), + ) as mock_guardduty_client: from prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled import ( guardduty_is_enabled, ) @@ -178,16 +166,13 @@ class Test_guardduty_is_enabled: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ) as mock_guardduty_client, - ): + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled.guardduty_client", + new=GuardDuty(aws_provider), + ) as mock_guardduty_client: from prowler.providers.aws.services.guardduty.guardduty_is_enabled.guardduty_is_enabled import ( guardduty_is_enabled, ) diff --git a/tests/providers/aws/services/guardduty/guardduty_lambda_protection_enabled/guardduty_lambda_protection_enabled_test.py b/tests/providers/aws/services/guardduty/guardduty_lambda_protection_enabled/guardduty_lambda_protection_enabled_test.py index e156f8db84..78eb2cc9ad 100644 --- a/tests/providers/aws/services/guardduty/guardduty_lambda_protection_enabled/guardduty_lambda_protection_enabled_test.py +++ b/tests/providers/aws/services/guardduty/guardduty_lambda_protection_enabled/guardduty_lambda_protection_enabled_test.py @@ -19,15 +19,12 @@ class Test_guardduty_lambda_protection_enabled: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_lambda_protection_enabled.guardduty_lambda_protection_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_lambda_protection_enabled.guardduty_lambda_protection_enabled.guardduty_client", + new=GuardDuty(aws_provider), ): # Test Check from prowler.providers.aws.services.guardduty.guardduty_lambda_protection_enabled.guardduty_lambda_protection_enabled import ( @@ -49,15 +46,12 @@ class Test_guardduty_lambda_protection_enabled: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_lambda_protection_enabled.guardduty_lambda_protection_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_lambda_protection_enabled.guardduty_lambda_protection_enabled.guardduty_client", + new=GuardDuty(aws_provider), ): # Test Check from prowler.providers.aws.services.guardduty.guardduty_lambda_protection_enabled.guardduty_lambda_protection_enabled import ( @@ -82,15 +76,12 @@ class Test_guardduty_lambda_protection_enabled: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_lambda_protection_enabled.guardduty_lambda_protection_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_lambda_protection_enabled.guardduty_lambda_protection_enabled.guardduty_client", + new=GuardDuty(aws_provider), ): # Test Check from prowler.providers.aws.services.guardduty.guardduty_lambda_protection_enabled.guardduty_lambda_protection_enabled import ( @@ -127,15 +118,12 @@ class Test_guardduty_lambda_protection_enabled: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_lambda_protection_enabled.guardduty_lambda_protection_enabled.guardduty_client", - new=GuardDuty(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_lambda_protection_enabled.guardduty_lambda_protection_enabled.guardduty_client", + new=GuardDuty(aws_provider), ): # Test Check from prowler.providers.aws.services.guardduty.guardduty_lambda_protection_enabled.guardduty_lambda_protection_enabled import ( diff --git a/tests/providers/aws/services/guardduty/guardduty_no_high_severity_findings/guardduty_no_high_severity_findings_test.py b/tests/providers/aws/services/guardduty/guardduty_no_high_severity_findings/guardduty_no_high_severity_findings_test.py index b3a7694c55..b068b89225 100644 --- a/tests/providers/aws/services/guardduty/guardduty_no_high_severity_findings/guardduty_no_high_severity_findings_test.py +++ b/tests/providers/aws/services/guardduty/guardduty_no_high_severity_findings/guardduty_no_high_severity_findings_test.py @@ -32,15 +32,12 @@ class Test_guardduty_no_high_severity_findings: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_no_high_severity_findings.guardduty_no_high_severity_findings.guardduty_client", - new=GuardDuty(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_no_high_severity_findings.guardduty_no_high_severity_findings.guardduty_client", + new=GuardDuty(aws_provider), ): from prowler.providers.aws.services.guardduty.guardduty_no_high_severity_findings.guardduty_no_high_severity_findings import ( guardduty_no_high_severity_findings, @@ -60,15 +57,12 @@ class Test_guardduty_no_high_severity_findings: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_no_high_severity_findings.guardduty_no_high_severity_findings.guardduty_client", - new=GuardDuty(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_no_high_severity_findings.guardduty_no_high_severity_findings.guardduty_client", + new=GuardDuty(aws_provider), ): from prowler.providers.aws.services.guardduty.guardduty_no_high_severity_findings.guardduty_no_high_severity_findings import ( guardduty_no_high_severity_findings, @@ -101,15 +95,12 @@ class Test_guardduty_no_high_severity_findings: from prowler.providers.aws.services.guardduty.guardduty_service import GuardDuty - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.guardduty.guardduty_no_high_severity_findings.guardduty_no_high_severity_findings.guardduty_client", - new=GuardDuty(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.guardduty.guardduty_no_high_severity_findings.guardduty_no_high_severity_findings.guardduty_client", + new=GuardDuty(aws_provider), ): from prowler.providers.aws.services.guardduty.guardduty_no_high_severity_findings.guardduty_no_high_severity_findings import ( guardduty_no_high_severity_findings, diff --git a/tests/providers/aws/services/guardduty/guardduty_rds_protection_enabled/guardduty_rds_protection_enabled_test.py b/tests/providers/aws/services/guardduty/guardduty_rds_protection_enabled/guardduty_rds_protection_enabled_test.py index 38ca346870..5587d9e0d1 100644 --- a/tests/providers/aws/services/guardduty/guardduty_rds_protection_enabled/guardduty_rds_protection_enabled_test.py +++ b/tests/providers/aws/services/guardduty/guardduty_rds_protection_enabled/guardduty_rds_protection_enabled_test.py @@ -9,15 +9,12 @@ class Test_guardduty_rds_protection_enabled: guardduty_client = mock.MagicMock() guardduty_client.detectors = [] - with ( - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_service.GuardDuty", - new=guardduty_client, - ), - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_rds_protection_enabled.guardduty_rds_protection_enabled.guardduty_client", - new=guardduty_client, - ), + with mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_service.GuardDuty", + new=guardduty_client, + ), mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_rds_protection_enabled.guardduty_rds_protection_enabled.guardduty_client", + new=guardduty_client, ): from prowler.providers.aws.services.guardduty.guardduty_rds_protection_enabled.guardduty_rds_protection_enabled import ( @@ -43,15 +40,12 @@ class Test_guardduty_rds_protection_enabled: ) ] - with ( - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_service.GuardDuty", - new=guardduty_client, - ), - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_rds_protection_enabled.guardduty_rds_protection_enabled.guardduty_client", - new=guardduty_client, - ), + with mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_service.GuardDuty", + new=guardduty_client, + ), mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_rds_protection_enabled.guardduty_rds_protection_enabled.guardduty_client", + new=guardduty_client, ): from prowler.providers.aws.services.guardduty.guardduty_rds_protection_enabled.guardduty_rds_protection_enabled import ( @@ -78,15 +72,12 @@ class Test_guardduty_rds_protection_enabled: ) ] - with ( - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_service.GuardDuty", - new=guardduty_client, - ), - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_rds_protection_enabled.guardduty_rds_protection_enabled.guardduty_client", - new=guardduty_client, - ), + with mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_service.GuardDuty", + new=guardduty_client, + ), mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_rds_protection_enabled.guardduty_rds_protection_enabled.guardduty_client", + new=guardduty_client, ): from prowler.providers.aws.services.guardduty.guardduty_rds_protection_enabled.guardduty_rds_protection_enabled import ( @@ -122,15 +113,12 @@ class Test_guardduty_rds_protection_enabled: ) ] - with ( - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_service.GuardDuty", - new=guardduty_client, - ), - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_rds_protection_enabled.guardduty_rds_protection_enabled.guardduty_client", - new=guardduty_client, - ), + with mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_service.GuardDuty", + new=guardduty_client, + ), mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_rds_protection_enabled.guardduty_rds_protection_enabled.guardduty_client", + new=guardduty_client, ): from prowler.providers.aws.services.guardduty.guardduty_rds_protection_enabled.guardduty_rds_protection_enabled import ( guardduty_rds_protection_enabled, diff --git a/tests/providers/aws/services/guardduty/guardduty_s3_protection_enabled/guardduty_s3_protection_enabled_test.py b/tests/providers/aws/services/guardduty/guardduty_s3_protection_enabled/guardduty_s3_protection_enabled_test.py index 8f1a8afe75..731d075a15 100644 --- a/tests/providers/aws/services/guardduty/guardduty_s3_protection_enabled/guardduty_s3_protection_enabled_test.py +++ b/tests/providers/aws/services/guardduty/guardduty_s3_protection_enabled/guardduty_s3_protection_enabled_test.py @@ -9,15 +9,12 @@ class Test_guardduty_s3_protection_enabled: guardduty_client = mock.MagicMock() guardduty_client.detectors = [] - with ( - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_service.GuardDuty", - new=guardduty_client, - ), - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_s3_protection_enabled.guardduty_s3_protection_enabled.guardduty_client", - new=guardduty_client, - ), + with mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_service.GuardDuty", + new=guardduty_client, + ), mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_s3_protection_enabled.guardduty_s3_protection_enabled.guardduty_client", + new=guardduty_client, ): from prowler.providers.aws.services.guardduty.guardduty_s3_protection_enabled.guardduty_s3_protection_enabled import ( @@ -43,15 +40,12 @@ class Test_guardduty_s3_protection_enabled: ) ] - with ( - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_service.GuardDuty", - new=guardduty_client, - ), - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_s3_protection_enabled.guardduty_s3_protection_enabled.guardduty_client", - new=guardduty_client, - ), + with mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_service.GuardDuty", + new=guardduty_client, + ), mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_s3_protection_enabled.guardduty_s3_protection_enabled.guardduty_client", + new=guardduty_client, ): from prowler.providers.aws.services.guardduty.guardduty_s3_protection_enabled.guardduty_s3_protection_enabled import ( @@ -78,15 +72,12 @@ class Test_guardduty_s3_protection_enabled: ) ] - with ( - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_service.GuardDuty", - new=guardduty_client, - ), - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_s3_protection_enabled.guardduty_s3_protection_enabled.guardduty_client", - new=guardduty_client, - ), + with mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_service.GuardDuty", + new=guardduty_client, + ), mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_s3_protection_enabled.guardduty_s3_protection_enabled.guardduty_client", + new=guardduty_client, ): from prowler.providers.aws.services.guardduty.guardduty_s3_protection_enabled.guardduty_s3_protection_enabled import ( @@ -122,15 +113,12 @@ class Test_guardduty_s3_protection_enabled: ) ] - with ( - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_service.GuardDuty", - new=guardduty_client, - ), - mock.patch( - "prowler.providers.aws.services.guardduty.guardduty_s3_protection_enabled.guardduty_s3_protection_enabled.guardduty_client", - new=guardduty_client, - ), + with mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_service.GuardDuty", + new=guardduty_client, + ), mock.patch( + "prowler.providers.aws.services.guardduty.guardduty_s3_protection_enabled.guardduty_s3_protection_enabled.guardduty_client", + new=guardduty_client, ): from prowler.providers.aws.services.guardduty.guardduty_s3_protection_enabled.guardduty_s3_protection_enabled import ( guardduty_s3_protection_enabled, diff --git a/tests/providers/aws/services/iam/iam_aws_attached_policy_no_administrative_privileges/iam_aws_attached_policy_no_administrative_privileges_test.py b/tests/providers/aws/services/iam/iam_aws_attached_policy_no_administrative_privileges/iam_aws_attached_policy_no_administrative_privileges_test.py index 48ceef1530..45efbeed53 100644 --- a/tests/providers/aws/services/iam/iam_aws_attached_policy_no_administrative_privileges/iam_aws_attached_policy_no_administrative_privileges_test.py +++ b/tests/providers/aws/services/iam/iam_aws_attached_policy_no_administrative_privileges/iam_aws_attached_policy_no_administrative_privileges_test.py @@ -21,15 +21,12 @@ class Test_iam_aws_attached_policy_no_administrative_privileges_test: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_aws_attached_policy_no_administrative_privileges.iam_aws_attached_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_aws_attached_policy_no_administrative_privileges.iam_aws_attached_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_aws_attached_policy_no_administrative_privileges.iam_aws_attached_policy_no_administrative_privileges import ( iam_aws_attached_policy_no_administrative_privileges, @@ -64,15 +61,12 @@ class Test_iam_aws_attached_policy_no_administrative_privileges_test: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_aws_attached_policy_no_administrative_privileges.iam_aws_attached_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_aws_attached_policy_no_administrative_privileges.iam_aws_attached_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_aws_attached_policy_no_administrative_privileges.iam_aws_attached_policy_no_administrative_privileges import ( iam_aws_attached_policy_no_administrative_privileges, @@ -110,15 +104,12 @@ class Test_iam_aws_attached_policy_no_administrative_privileges_test: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_aws_attached_policy_no_administrative_privileges.iam_aws_attached_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_aws_attached_policy_no_administrative_privileges.iam_aws_attached_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_aws_attached_policy_no_administrative_privileges.iam_aws_attached_policy_no_administrative_privileges import ( iam_aws_attached_policy_no_administrative_privileges, diff --git a/tests/providers/aws/services/iam/iam_customer_attached_policy_no_administrative_privileges/iam_customer_attached_policy_no_administrative_privileges_test.py b/tests/providers/aws/services/iam/iam_customer_attached_policy_no_administrative_privileges/iam_customer_attached_policy_no_administrative_privileges_test.py index 2716307123..d38060e93e 100644 --- a/tests/providers/aws/services/iam/iam_customer_attached_policy_no_administrative_privileges/iam_customer_attached_policy_no_administrative_privileges_test.py +++ b/tests/providers/aws/services/iam/iam_customer_attached_policy_no_administrative_privileges/iam_customer_attached_policy_no_administrative_privileges_test.py @@ -29,15 +29,12 @@ class Test_iam_customer_attached_policy_no_administrative_privileges_test: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_customer_attached_policy_no_administrative_privileges.iam_customer_attached_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_customer_attached_policy_no_administrative_privileges.iam_customer_attached_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_customer_attached_policy_no_administrative_privileges.iam_customer_attached_policy_no_administrative_privileges import ( iam_customer_attached_policy_no_administrative_privileges, @@ -75,15 +72,12 @@ class Test_iam_customer_attached_policy_no_administrative_privileges_test: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_customer_attached_policy_no_administrative_privileges.iam_customer_attached_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_customer_attached_policy_no_administrative_privileges.iam_customer_attached_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_customer_attached_policy_no_administrative_privileges.iam_customer_attached_policy_no_administrative_privileges import ( iam_customer_attached_policy_no_administrative_privileges, @@ -136,15 +130,12 @@ class Test_iam_customer_attached_policy_no_administrative_privileges_test: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_customer_attached_policy_no_administrative_privileges.iam_customer_attached_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_customer_attached_policy_no_administrative_privileges.iam_customer_attached_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_customer_attached_policy_no_administrative_privileges.iam_customer_attached_policy_no_administrative_privileges import ( iam_customer_attached_policy_no_administrative_privileges, diff --git a/tests/providers/aws/services/iam/iam_customer_unattached_policy_no_administrative_privileges/iam_customer_unattached_policy_no_administrative_privileges_test.py b/tests/providers/aws/services/iam/iam_customer_unattached_policy_no_administrative_privileges/iam_customer_unattached_policy_no_administrative_privileges_test.py index dce7da0a6b..3a4a9a0613 100644 --- a/tests/providers/aws/services/iam/iam_customer_unattached_policy_no_administrative_privileges/iam_customer_unattached_policy_no_administrative_privileges_test.py +++ b/tests/providers/aws/services/iam/iam_customer_unattached_policy_no_administrative_privileges/iam_customer_unattached_policy_no_administrative_privileges_test.py @@ -26,15 +26,12 @@ class Test_iam_customer_unattached_policy_no_administrative_privileges_test: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_customer_unattached_policy_no_administrative_privileges.iam_customer_unattached_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_customer_unattached_policy_no_administrative_privileges.iam_customer_unattached_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_customer_unattached_policy_no_administrative_privileges.iam_customer_unattached_policy_no_administrative_privileges import ( iam_customer_unattached_policy_no_administrative_privileges, @@ -69,15 +66,12 @@ class Test_iam_customer_unattached_policy_no_administrative_privileges_test: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_customer_unattached_policy_no_administrative_privileges.iam_customer_unattached_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_customer_unattached_policy_no_administrative_privileges.iam_customer_unattached_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_customer_unattached_policy_no_administrative_privileges.iam_customer_unattached_policy_no_administrative_privileges import ( iam_customer_unattached_policy_no_administrative_privileges, @@ -124,15 +118,12 @@ class Test_iam_customer_unattached_policy_no_administrative_privileges_test: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_customer_unattached_policy_no_administrative_privileges.iam_customer_unattached_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_customer_unattached_policy_no_administrative_privileges.iam_customer_unattached_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_customer_unattached_policy_no_administrative_privileges.iam_customer_unattached_policy_no_administrative_privileges import ( iam_customer_unattached_policy_no_administrative_privileges, diff --git a/tests/providers/aws/services/iam/iam_group_administrator_access_policy/iam_group_administrator_access_policy_test.py b/tests/providers/aws/services/iam/iam_group_administrator_access_policy/iam_group_administrator_access_policy_test.py index a70ca28acd..3c7c86d395 100644 --- a/tests/providers/aws/services/iam/iam_group_administrator_access_policy/iam_group_administrator_access_policy_test.py +++ b/tests/providers/aws/services/iam/iam_group_administrator_access_policy/iam_group_administrator_access_policy_test.py @@ -13,15 +13,12 @@ class Test_iam_group_administrator_access_policy: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_group_administrator_access_policy.iam_group_administrator_access_policy.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_group_administrator_access_policy.iam_group_administrator_access_policy.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_group_administrator_access_policy.iam_group_administrator_access_policy import ( @@ -42,15 +39,12 @@ class Test_iam_group_administrator_access_policy: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_group_administrator_access_policy.iam_group_administrator_access_policy.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_group_administrator_access_policy.iam_group_administrator_access_policy.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_group_administrator_access_policy.iam_group_administrator_access_policy import ( @@ -96,15 +90,12 @@ class Test_iam_group_administrator_access_policy: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_group_administrator_access_policy.iam_group_administrator_access_policy.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_group_administrator_access_policy.iam_group_administrator_access_policy.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_group_administrator_access_policy.iam_group_administrator_access_policy import ( diff --git a/tests/providers/aws/services/iam/iam_inline_policy_no_administrative_privileges/iam_inline_policy_no_administrative_privileges_test.py b/tests/providers/aws/services/iam/iam_inline_policy_no_administrative_privileges/iam_inline_policy_no_administrative_privileges_test.py index 4c5c2470a4..2a65ac0fe9 100644 --- a/tests/providers/aws/services/iam/iam_inline_policy_no_administrative_privileges/iam_inline_policy_no_administrative_privileges_test.py +++ b/tests/providers/aws/services/iam/iam_inline_policy_no_administrative_privileges/iam_inline_policy_no_administrative_privileges_test.py @@ -47,15 +47,12 @@ class Test_iam_inline_policy_no_administrative_privileges: from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges import ( iam_inline_policy_no_administrative_privileges, @@ -85,15 +82,12 @@ class Test_iam_inline_policy_no_administrative_privileges: from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges import ( iam_inline_policy_no_administrative_privileges, @@ -132,15 +126,12 @@ class Test_iam_inline_policy_no_administrative_privileges: from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges import ( iam_inline_policy_no_administrative_privileges, @@ -187,15 +178,12 @@ class Test_iam_inline_policy_no_administrative_privileges: from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges import ( iam_inline_policy_no_administrative_privileges, @@ -244,15 +232,12 @@ class Test_iam_inline_policy_no_administrative_privileges: from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges import ( iam_inline_policy_no_administrative_privileges, @@ -285,15 +270,12 @@ class Test_iam_inline_policy_no_administrative_privileges: from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges import ( iam_inline_policy_no_administrative_privileges, @@ -335,15 +317,12 @@ class Test_iam_inline_policy_no_administrative_privileges: from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges import ( iam_inline_policy_no_administrative_privileges, @@ -392,15 +371,12 @@ class Test_iam_inline_policy_no_administrative_privileges: from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges import ( iam_inline_policy_no_administrative_privileges, @@ -448,15 +424,12 @@ class Test_iam_inline_policy_no_administrative_privileges: from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges import ( iam_inline_policy_no_administrative_privileges, @@ -490,15 +463,12 @@ class Test_iam_inline_policy_no_administrative_privileges: from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges import ( iam_inline_policy_no_administrative_privileges, @@ -541,15 +511,12 @@ class Test_iam_inline_policy_no_administrative_privileges: from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges import ( iam_inline_policy_no_administrative_privileges, @@ -599,15 +566,12 @@ class Test_iam_inline_policy_no_administrative_privileges: from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_administrative_privileges.iam_inline_policy_no_administrative_privileges import ( iam_inline_policy_no_administrative_privileges, diff --git a/tests/providers/aws/services/iam/iam_inline_policy_no_full_access_to_cloudtrail/iam_inline_policy_no_full_access_to_cloudtrail_test.py b/tests/providers/aws/services/iam/iam_inline_policy_no_full_access_to_cloudtrail/iam_inline_policy_no_full_access_to_cloudtrail_test.py index 14d008711e..a5e9da2496 100644 --- a/tests/providers/aws/services/iam/iam_inline_policy_no_full_access_to_cloudtrail/iam_inline_policy_no_full_access_to_cloudtrail_test.py +++ b/tests/providers/aws/services/iam/iam_inline_policy_no_full_access_to_cloudtrail/iam_inline_policy_no_full_access_to_cloudtrail_test.py @@ -38,15 +38,12 @@ class Test_iam_inline_policy_no_full_access_to_cloudtrail: PolicyDocument=dumps(policy_document_full_access), ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_cloudtrail.iam_inline_policy_no_full_access_to_cloudtrail.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_cloudtrail.iam_inline_policy_no_full_access_to_cloudtrail.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_cloudtrail.iam_inline_policy_no_full_access_to_cloudtrail import ( iam_inline_policy_no_full_access_to_cloudtrail, @@ -88,15 +85,12 @@ class Test_iam_inline_policy_no_full_access_to_cloudtrail: PolicyDocument=dumps(policy_document_full_access), ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_cloudtrail.iam_inline_policy_no_full_access_to_cloudtrail.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_cloudtrail.iam_inline_policy_no_full_access_to_cloudtrail.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_cloudtrail.iam_inline_policy_no_full_access_to_cloudtrail import ( iam_inline_policy_no_full_access_to_cloudtrail, @@ -138,15 +132,12 @@ class Test_iam_inline_policy_no_full_access_to_cloudtrail: PolicyDocument=dumps(policy_document_full_access), ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_cloudtrail.iam_inline_policy_no_full_access_to_cloudtrail.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_cloudtrail.iam_inline_policy_no_full_access_to_cloudtrail.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_cloudtrail.iam_inline_policy_no_full_access_to_cloudtrail import ( iam_inline_policy_no_full_access_to_cloudtrail, @@ -192,15 +183,12 @@ class Test_iam_inline_policy_no_full_access_to_cloudtrail: PolicyDocument=dumps(policy_document_full_access), ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_cloudtrail.iam_inline_policy_no_full_access_to_cloudtrail.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_cloudtrail.iam_inline_policy_no_full_access_to_cloudtrail.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_cloudtrail.iam_inline_policy_no_full_access_to_cloudtrail import ( iam_inline_policy_no_full_access_to_cloudtrail, @@ -246,15 +234,12 @@ class Test_iam_inline_policy_no_full_access_to_cloudtrail: PolicyDocument=dumps(policy_document_full_access), ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_cloudtrail.iam_inline_policy_no_full_access_to_cloudtrail.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_cloudtrail.iam_inline_policy_no_full_access_to_cloudtrail.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_cloudtrail.iam_inline_policy_no_full_access_to_cloudtrail import ( iam_inline_policy_no_full_access_to_cloudtrail, diff --git a/tests/providers/aws/services/iam/iam_inline_policy_no_full_access_to_kms/iam_inline_policy_no_full_access_to_kms_test.py b/tests/providers/aws/services/iam/iam_inline_policy_no_full_access_to_kms/iam_inline_policy_no_full_access_to_kms_test.py index 17a001a589..67d7e27a94 100644 --- a/tests/providers/aws/services/iam/iam_inline_policy_no_full_access_to_kms/iam_inline_policy_no_full_access_to_kms_test.py +++ b/tests/providers/aws/services/iam/iam_inline_policy_no_full_access_to_kms/iam_inline_policy_no_full_access_to_kms_test.py @@ -38,15 +38,12 @@ class Test_iam_inline_policy_no_full_access_to_kms: PolicyDocument=dumps(policy_document_full_access), ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_kms.iam_inline_policy_no_full_access_to_kms.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_kms.iam_inline_policy_no_full_access_to_kms.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_kms.iam_inline_policy_no_full_access_to_kms import ( iam_inline_policy_no_full_access_to_kms, @@ -88,15 +85,12 @@ class Test_iam_inline_policy_no_full_access_to_kms: PolicyDocument=dumps(policy_document_full_access), ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_kms.iam_inline_policy_no_full_access_to_kms.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_kms.iam_inline_policy_no_full_access_to_kms.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_kms.iam_inline_policy_no_full_access_to_kms import ( iam_inline_policy_no_full_access_to_kms, @@ -138,15 +132,12 @@ class Test_iam_inline_policy_no_full_access_to_kms: PolicyDocument=dumps(policy_document_full_access), ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_kms.iam_inline_policy_no_full_access_to_kms.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_kms.iam_inline_policy_no_full_access_to_kms.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_kms.iam_inline_policy_no_full_access_to_kms import ( iam_inline_policy_no_full_access_to_kms, @@ -192,15 +183,12 @@ class Test_iam_inline_policy_no_full_access_to_kms: PolicyDocument=dumps(policy_document_full_access), ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_kms.iam_inline_policy_no_full_access_to_kms.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_kms.iam_inline_policy_no_full_access_to_kms.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_kms.iam_inline_policy_no_full_access_to_kms import ( iam_inline_policy_no_full_access_to_kms, @@ -246,15 +234,12 @@ class Test_iam_inline_policy_no_full_access_to_kms: PolicyDocument=dumps(policy_document_full_access), ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_kms.iam_inline_policy_no_full_access_to_kms.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_kms.iam_inline_policy_no_full_access_to_kms.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_inline_policy_no_full_access_to_kms.iam_inline_policy_no_full_access_to_kms import ( iam_inline_policy_no_full_access_to_kms, diff --git a/tests/providers/aws/services/iam/iam_password_policy_expires_passwords_within_90_days_or_less/iam_password_policy_expires_passwords_within_90_days_or_less_fixer_test.py b/tests/providers/aws/services/iam/iam_password_policy_expires_passwords_within_90_days_or_less/iam_password_policy_expires_passwords_within_90_days_or_less_fixer_test.py index 14f4cfeec7..2c8ed801de 100644 --- a/tests/providers/aws/services/iam/iam_password_policy_expires_passwords_within_90_days_or_less/iam_password_policy_expires_passwords_within_90_days_or_less_fixer_test.py +++ b/tests/providers/aws/services/iam/iam_password_policy_expires_passwords_within_90_days_or_less/iam_password_policy_expires_passwords_within_90_days_or_less_fixer_test.py @@ -17,16 +17,13 @@ class Test_iam_password_policy_expires_passwords_within_90_days_or_less_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_password_policy_expires_passwords_within_90_days_or_less.iam_password_policy_expires_passwords_within_90_days_or_less_fixer.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_password_policy_expires_passwords_within_90_days_or_less.iam_password_policy_expires_passwords_within_90_days_or_less_fixer.iam_client", + new=IAM(aws_provider), + ) as service_client: service_client.password_policy = PasswordPolicy( length=10, symbols=True, diff --git a/tests/providers/aws/services/iam/iam_password_policy_lowercase/iam_password_policy_lowercase_fixer_test.py b/tests/providers/aws/services/iam/iam_password_policy_lowercase/iam_password_policy_lowercase_fixer_test.py index 10103edffc..99bcb31bca 100644 --- a/tests/providers/aws/services/iam/iam_password_policy_lowercase/iam_password_policy_lowercase_fixer_test.py +++ b/tests/providers/aws/services/iam/iam_password_policy_lowercase/iam_password_policy_lowercase_fixer_test.py @@ -17,16 +17,13 @@ class Test_iam_password_policy_lowercase_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_password_policy_lowercase.iam_password_policy_lowercase_fixer.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_password_policy_lowercase.iam_password_policy_lowercase_fixer.iam_client", + new=IAM(aws_provider), + ) as service_client: service_client.password_policy = PasswordPolicy( length=10, symbols=True, diff --git a/tests/providers/aws/services/iam/iam_password_policy_lowercase/iam_password_policy_lowercase_test.py b/tests/providers/aws/services/iam/iam_password_policy_lowercase/iam_password_policy_lowercase_test.py index 57071cc504..66b76b10ca 100644 --- a/tests/providers/aws/services/iam/iam_password_policy_lowercase/iam_password_policy_lowercase_test.py +++ b/tests/providers/aws/services/iam/iam_password_policy_lowercase/iam_password_policy_lowercase_test.py @@ -22,15 +22,12 @@ class Test_iam_password_policy_lowercase: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_password_policy_lowercase.iam_password_policy_lowercase.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_password_policy_lowercase.iam_password_policy_lowercase.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_password_policy_lowercase.iam_password_policy_lowercase import ( @@ -62,15 +59,12 @@ class Test_iam_password_policy_lowercase: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_password_policy_lowercase.iam_password_policy_lowercase.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_password_policy_lowercase.iam_password_policy_lowercase.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_password_policy_lowercase.iam_password_policy_lowercase import ( diff --git a/tests/providers/aws/services/iam/iam_password_policy_minimum_length_14/iam_password_policy_minimum_length_14_fixer_test.py b/tests/providers/aws/services/iam/iam_password_policy_minimum_length_14/iam_password_policy_minimum_length_14_fixer_test.py index b837d02768..095b9b4cb3 100644 --- a/tests/providers/aws/services/iam/iam_password_policy_minimum_length_14/iam_password_policy_minimum_length_14_fixer_test.py +++ b/tests/providers/aws/services/iam/iam_password_policy_minimum_length_14/iam_password_policy_minimum_length_14_fixer_test.py @@ -17,16 +17,13 @@ class Test_iam_password_policy_minimum_length_14_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_password_policy_minimum_length_14.iam_password_policy_minimum_length_14_fixer.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_password_policy_minimum_length_14.iam_password_policy_minimum_length_14_fixer.iam_client", + new=IAM(aws_provider), + ) as service_client: service_client.password_policy = PasswordPolicy( length=10, symbols=True, diff --git a/tests/providers/aws/services/iam/iam_password_policy_minimum_length_14/iam_password_policy_minimum_length_14_test.py b/tests/providers/aws/services/iam/iam_password_policy_minimum_length_14/iam_password_policy_minimum_length_14_test.py index 3261648ba6..4c4eacac91 100644 --- a/tests/providers/aws/services/iam/iam_password_policy_minimum_length_14/iam_password_policy_minimum_length_14_test.py +++ b/tests/providers/aws/services/iam/iam_password_policy_minimum_length_14/iam_password_policy_minimum_length_14_test.py @@ -29,15 +29,12 @@ class Test_iam_password_policy_minimum_length_14: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_password_policy_minimum_length_14.iam_password_policy_minimum_length_14.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_password_policy_minimum_length_14.iam_password_policy_minimum_length_14.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_password_policy_minimum_length_14.iam_password_policy_minimum_length_14 import ( @@ -69,15 +66,12 @@ class Test_iam_password_policy_minimum_length_14: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_password_policy_minimum_length_14.iam_password_policy_minimum_length_14.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_password_policy_minimum_length_14.iam_password_policy_minimum_length_14.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_password_policy_minimum_length_14.iam_password_policy_minimum_length_14 import ( @@ -109,15 +103,12 @@ class Test_iam_password_policy_minimum_length_14: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_password_policy_minimum_length_14.iam_password_policy_minimum_length_14.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_password_policy_minimum_length_14.iam_password_policy_minimum_length_14.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_password_policy_minimum_length_14.iam_password_policy_minimum_length_14 import ( diff --git a/tests/providers/aws/services/iam/iam_password_policy_number/iam_password_policy_number_fixer_test.py b/tests/providers/aws/services/iam/iam_password_policy_number/iam_password_policy_number_fixer_test.py index ad7cf513e4..df485424ac 100644 --- a/tests/providers/aws/services/iam/iam_password_policy_number/iam_password_policy_number_fixer_test.py +++ b/tests/providers/aws/services/iam/iam_password_policy_number/iam_password_policy_number_fixer_test.py @@ -17,16 +17,13 @@ class Test_iam_password_policy_number_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_password_policy_number.iam_password_policy_number_fixer.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_password_policy_number.iam_password_policy_number_fixer.iam_client", + new=IAM(aws_provider), + ) as service_client: service_client.password_policy = PasswordPolicy( length=10, symbols=True, diff --git a/tests/providers/aws/services/iam/iam_password_policy_number/iam_password_policy_number_test.py b/tests/providers/aws/services/iam/iam_password_policy_number/iam_password_policy_number_test.py index 677eacc06c..099abf7296 100644 --- a/tests/providers/aws/services/iam/iam_password_policy_number/iam_password_policy_number_test.py +++ b/tests/providers/aws/services/iam/iam_password_policy_number/iam_password_policy_number_test.py @@ -29,15 +29,12 @@ class Test_iam_password_policy_number: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_password_policy_number.iam_password_policy_number.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_password_policy_number.iam_password_policy_number.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_password_policy_number.iam_password_policy_number import ( @@ -69,15 +66,12 @@ class Test_iam_password_policy_number: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_password_policy_number.iam_password_policy_number.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_password_policy_number.iam_password_policy_number.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_password_policy_number.iam_password_policy_number import ( diff --git a/tests/providers/aws/services/iam/iam_password_policy_reuse_24/iam_password_policy_reuse_24_fixer_test.py b/tests/providers/aws/services/iam/iam_password_policy_reuse_24/iam_password_policy_reuse_24_fixer_test.py index 0585211708..1a0952e1fe 100644 --- a/tests/providers/aws/services/iam/iam_password_policy_reuse_24/iam_password_policy_reuse_24_fixer_test.py +++ b/tests/providers/aws/services/iam/iam_password_policy_reuse_24/iam_password_policy_reuse_24_fixer_test.py @@ -17,16 +17,13 @@ class Test_iam_password_policy_reuse_24_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_password_policy_reuse_24.iam_password_policy_reuse_24_fixer.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_password_policy_reuse_24.iam_password_policy_reuse_24_fixer.iam_client", + new=IAM(aws_provider), + ) as service_client: service_client.password_policy = PasswordPolicy( length=10, symbols=True, diff --git a/tests/providers/aws/services/iam/iam_password_policy_reuse_24/iam_password_policy_reuse_24_test.py b/tests/providers/aws/services/iam/iam_password_policy_reuse_24/iam_password_policy_reuse_24_test.py index 1ba08e37dc..9402a631c0 100644 --- a/tests/providers/aws/services/iam/iam_password_policy_reuse_24/iam_password_policy_reuse_24_test.py +++ b/tests/providers/aws/services/iam/iam_password_policy_reuse_24/iam_password_policy_reuse_24_test.py @@ -27,15 +27,12 @@ class Test_iam_password_policy_reuse_24: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_password_policy_reuse_24.iam_password_policy_reuse_24.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_password_policy_reuse_24.iam_password_policy_reuse_24.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_password_policy_reuse_24.iam_password_policy_reuse_24 import ( @@ -66,15 +63,12 @@ class Test_iam_password_policy_reuse_24: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_password_policy_reuse_24.iam_password_policy_reuse_24.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_password_policy_reuse_24.iam_password_policy_reuse_24.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_password_policy_reuse_24.iam_password_policy_reuse_24 import ( diff --git a/tests/providers/aws/services/iam/iam_password_policy_symbol/iam_password_policy_symbol_fixer_test.py b/tests/providers/aws/services/iam/iam_password_policy_symbol/iam_password_policy_symbol_fixer_test.py index b3ad1bebda..fb6875d4ab 100644 --- a/tests/providers/aws/services/iam/iam_password_policy_symbol/iam_password_policy_symbol_fixer_test.py +++ b/tests/providers/aws/services/iam/iam_password_policy_symbol/iam_password_policy_symbol_fixer_test.py @@ -17,16 +17,13 @@ class Test_iam_password_policy_symbol_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_password_policy_symbol.iam_password_policy_symbol_fixer.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_password_policy_symbol.iam_password_policy_symbol_fixer.iam_client", + new=IAM(aws_provider), + ) as service_client: service_client.password_policy = PasswordPolicy( length=10, symbols=True, diff --git a/tests/providers/aws/services/iam/iam_password_policy_symbol/iam_password_policy_symbol_test.py b/tests/providers/aws/services/iam/iam_password_policy_symbol/iam_password_policy_symbol_test.py index 6b44af4054..13644917d2 100644 --- a/tests/providers/aws/services/iam/iam_password_policy_symbol/iam_password_policy_symbol_test.py +++ b/tests/providers/aws/services/iam/iam_password_policy_symbol/iam_password_policy_symbol_test.py @@ -29,15 +29,12 @@ class Test_iam_password_policy_symbol: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_password_policy_symbol.iam_password_policy_symbol.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_password_policy_symbol.iam_password_policy_symbol.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_password_policy_symbol.iam_password_policy_symbol import ( @@ -69,15 +66,12 @@ class Test_iam_password_policy_symbol: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_password_policy_symbol.iam_password_policy_symbol.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_password_policy_symbol.iam_password_policy_symbol.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_password_policy_symbol.iam_password_policy_symbol import ( diff --git a/tests/providers/aws/services/iam/iam_password_policy_uppercase/iam_password_policy_uppercase_fixer_test.py b/tests/providers/aws/services/iam/iam_password_policy_uppercase/iam_password_policy_uppercase_fixer_test.py index 8e37edc93c..5dee7e591e 100644 --- a/tests/providers/aws/services/iam/iam_password_policy_uppercase/iam_password_policy_uppercase_fixer_test.py +++ b/tests/providers/aws/services/iam/iam_password_policy_uppercase/iam_password_policy_uppercase_fixer_test.py @@ -17,16 +17,13 @@ class Test_iam_password_policy_uppercase_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_password_policy_uppercase.iam_password_policy_uppercase_fixer.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_password_policy_uppercase.iam_password_policy_uppercase_fixer.iam_client", + new=IAM(aws_provider), + ) as service_client: service_client.password_policy = PasswordPolicy( length=10, symbols=True, diff --git a/tests/providers/aws/services/iam/iam_password_policy_uppercase/iam_password_policy_uppercase_test.py b/tests/providers/aws/services/iam/iam_password_policy_uppercase/iam_password_policy_uppercase_test.py index 1c476b7496..ee3b7a10b6 100644 --- a/tests/providers/aws/services/iam/iam_password_policy_uppercase/iam_password_policy_uppercase_test.py +++ b/tests/providers/aws/services/iam/iam_password_policy_uppercase/iam_password_policy_uppercase_test.py @@ -27,15 +27,12 @@ class Test_iam_password_policy_uppercase: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_password_policy_uppercase.iam_password_policy_uppercase.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_password_policy_uppercase.iam_password_policy_uppercase.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_password_policy_uppercase.iam_password_policy_uppercase import ( @@ -66,15 +63,12 @@ class Test_iam_password_policy_uppercase: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_password_policy_uppercase.iam_password_policy_uppercase.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_password_policy_uppercase.iam_password_policy_uppercase.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_password_policy_uppercase.iam_password_policy_uppercase import ( diff --git a/tests/providers/aws/services/iam/iam_policy_allows_privilege_escalation/iam_policy_allows_privilege_escalation_test.py b/tests/providers/aws/services/iam/iam_policy_allows_privilege_escalation/iam_policy_allows_privilege_escalation_test.py index 6dbe167e47..5b0e747f41 100644 --- a/tests/providers/aws/services/iam/iam_policy_allows_privilege_escalation/iam_policy_allows_privilege_escalation_test.py +++ b/tests/providers/aws/services/iam/iam_policy_allows_privilege_escalation/iam_policy_allows_privilege_escalation_test.py @@ -79,15 +79,12 @@ class Test_iam_policy_allows_privilege_escalation: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation import ( @@ -130,15 +127,12 @@ class Test_iam_policy_allows_privilege_escalation: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation import ( @@ -192,15 +186,12 @@ class Test_iam_policy_allows_privilege_escalation: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation import ( @@ -250,15 +241,12 @@ class Test_iam_policy_allows_privilege_escalation: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation import ( @@ -304,15 +292,12 @@ class Test_iam_policy_allows_privilege_escalation: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation import ( @@ -376,15 +361,12 @@ class Test_iam_policy_allows_privilege_escalation: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation import ( @@ -437,15 +419,12 @@ class Test_iam_policy_allows_privilege_escalation: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation import ( @@ -492,15 +471,12 @@ class Test_iam_policy_allows_privilege_escalation: from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation import ( @@ -580,15 +556,12 @@ class Test_iam_policy_allows_privilege_escalation: from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation import ( @@ -683,15 +656,12 @@ class Test_iam_policy_allows_privilege_escalation: from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation import ( @@ -767,15 +737,12 @@ class Test_iam_policy_allows_privilege_escalation: from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation import ( @@ -826,15 +793,12 @@ class Test_iam_policy_allows_privilege_escalation: from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation import ( @@ -887,15 +851,12 @@ class Test_iam_policy_allows_privilege_escalation: from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation import ( @@ -943,15 +904,12 @@ class Test_iam_policy_allows_privilege_escalation: from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation import ( @@ -1005,15 +963,12 @@ class Test_iam_policy_allows_privilege_escalation: from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation import ( @@ -1056,15 +1011,12 @@ class Test_iam_policy_allows_privilege_escalation: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_policy_allows_privilege_escalation.iam_policy_allows_privilege_escalation import ( iam_policy_allows_privilege_escalation, diff --git a/tests/providers/aws/services/iam/iam_policy_attached_only_to_group_or_roles/iam_policy_attached_only_to_group_or_roles_test.py b/tests/providers/aws/services/iam/iam_policy_attached_only_to_group_or_roles/iam_policy_attached_only_to_group_or_roles_test.py index 876e013f3d..1ca291c135 100644 --- a/tests/providers/aws/services/iam/iam_policy_attached_only_to_group_or_roles/iam_policy_attached_only_to_group_or_roles_test.py +++ b/tests/providers/aws/services/iam/iam_policy_attached_only_to_group_or_roles/iam_policy_attached_only_to_group_or_roles_test.py @@ -33,15 +33,12 @@ class Test_iam_policy_attached_only_to_group_or_roles: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_attached_only_to_group_or_roles.iam_policy_attached_only_to_group_or_roles.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_attached_only_to_group_or_roles.iam_policy_attached_only_to_group_or_roles.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_policy_attached_only_to_group_or_roles.iam_policy_attached_only_to_group_or_roles import ( iam_policy_attached_only_to_group_or_roles, @@ -85,15 +82,12 @@ class Test_iam_policy_attached_only_to_group_or_roles: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_attached_only_to_group_or_roles.iam_policy_attached_only_to_group_or_roles.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_attached_only_to_group_or_roles.iam_policy_attached_only_to_group_or_roles.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_policy_attached_only_to_group_or_roles.iam_policy_attached_only_to_group_or_roles import ( iam_policy_attached_only_to_group_or_roles, @@ -145,15 +139,12 @@ class Test_iam_policy_attached_only_to_group_or_roles: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_attached_only_to_group_or_roles.iam_policy_attached_only_to_group_or_roles.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_attached_only_to_group_or_roles.iam_policy_attached_only_to_group_or_roles.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_policy_attached_only_to_group_or_roles.iam_policy_attached_only_to_group_or_roles import ( iam_policy_attached_only_to_group_or_roles, @@ -187,15 +178,12 @@ class Test_iam_policy_attached_only_to_group_or_roles: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_attached_only_to_group_or_roles.iam_policy_attached_only_to_group_or_roles.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_attached_only_to_group_or_roles.iam_policy_attached_only_to_group_or_roles.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_policy_attached_only_to_group_or_roles.iam_policy_attached_only_to_group_or_roles import ( iam_policy_attached_only_to_group_or_roles, diff --git a/tests/providers/aws/services/iam/iam_policy_cloudshell_admin_not_attached/iam_policy_cloudshell_admin_not_attached_test.py b/tests/providers/aws/services/iam/iam_policy_cloudshell_admin_not_attached/iam_policy_cloudshell_admin_not_attached_test.py index 8df2e710a4..123683c9f5 100644 --- a/tests/providers/aws/services/iam/iam_policy_cloudshell_admin_not_attached/iam_policy_cloudshell_admin_not_attached_test.py +++ b/tests/providers/aws/services/iam/iam_policy_cloudshell_admin_not_attached/iam_policy_cloudshell_admin_not_attached_test.py @@ -18,16 +18,13 @@ class Test_iam_policy_cloudshell_admin_not_attached: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_policy_cloudshell_admin_not_attached.iam_policy_cloudshell_admin_not_attached.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_policy_cloudshell_admin_not_attached.iam_policy_cloudshell_admin_not_attached.iam_client", + new=IAM(aws_provider), + ) as service_client: from prowler.providers.aws.services.iam.iam_policy_cloudshell_admin_not_attached.iam_policy_cloudshell_admin_not_attached import ( iam_policy_cloudshell_admin_not_attached, ) diff --git a/tests/providers/aws/services/iam/iam_role_administratoraccess_policy/iam_role_administratoraccess_policy_test.py b/tests/providers/aws/services/iam/iam_role_administratoraccess_policy/iam_role_administratoraccess_policy_test.py index b48d25a47c..bb1b35463f 100644 --- a/tests/providers/aws/services/iam/iam_role_administratoraccess_policy/iam_role_administratoraccess_policy_test.py +++ b/tests/providers/aws/services/iam/iam_role_administratoraccess_policy/iam_role_administratoraccess_policy_test.py @@ -17,15 +17,12 @@ class Test_iam_role_administratoraccess_policy: from prowler.providers.aws.services.iam.iam_service import IAM aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy import ( @@ -57,15 +54,12 @@ class Test_iam_role_administratoraccess_policy: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy import ( @@ -109,15 +103,12 @@ class Test_iam_role_administratoraccess_policy: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy import ( @@ -161,15 +152,12 @@ class Test_iam_role_administratoraccess_policy: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy import ( @@ -213,15 +201,12 @@ class Test_iam_role_administratoraccess_policy: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy import ( @@ -264,15 +249,12 @@ class Test_iam_role_administratoraccess_policy: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy.iam_client", + new=iam_client, ): # Test Check from prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy import ( @@ -290,15 +272,12 @@ class Test_iam_role_administratoraccess_policy: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy.iam_client", + new=iam_client, ): # Test Check from prowler.providers.aws.services.iam.iam_role_administratoraccess_policy.iam_role_administratoraccess_policy import ( diff --git a/tests/providers/aws/services/iam/iam_role_cross_account_readonlyaccess_policy/iam_role_cross_account_readonlyaccess_policy_test.py b/tests/providers/aws/services/iam/iam_role_cross_account_readonlyaccess_policy/iam_role_cross_account_readonlyaccess_policy_test.py index a288b55495..211d670d54 100644 --- a/tests/providers/aws/services/iam/iam_role_cross_account_readonlyaccess_policy/iam_role_cross_account_readonlyaccess_policy_test.py +++ b/tests/providers/aws/services/iam/iam_role_cross_account_readonlyaccess_policy/iam_role_cross_account_readonlyaccess_policy_test.py @@ -17,15 +17,12 @@ class Test_iam_role_cross_account_readonlyaccess_policy: from prowler.providers.aws.services.iam.iam_service import IAM aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_role_cross_account_readonlyaccess_policy.iam_role_cross_account_readonlyaccess_policy.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_role_cross_account_readonlyaccess_policy.iam_role_cross_account_readonlyaccess_policy.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_role_cross_account_readonlyaccess_policy.iam_role_cross_account_readonlyaccess_policy import ( @@ -57,15 +54,12 @@ class Test_iam_role_cross_account_readonlyaccess_policy: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_role_cross_account_readonlyaccess_policy.iam_role_cross_account_readonlyaccess_policy.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_role_cross_account_readonlyaccess_policy.iam_role_cross_account_readonlyaccess_policy.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_role_cross_account_readonlyaccess_policy.iam_role_cross_account_readonlyaccess_policy import ( @@ -109,15 +103,12 @@ class Test_iam_role_cross_account_readonlyaccess_policy: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_role_cross_account_readonlyaccess_policy.iam_role_cross_account_readonlyaccess_policy.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_role_cross_account_readonlyaccess_policy.iam_role_cross_account_readonlyaccess_policy.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_role_cross_account_readonlyaccess_policy.iam_role_cross_account_readonlyaccess_policy import ( @@ -161,15 +152,12 @@ class Test_iam_role_cross_account_readonlyaccess_policy: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_role_cross_account_readonlyaccess_policy.iam_role_cross_account_readonlyaccess_policy.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_role_cross_account_readonlyaccess_policy.iam_role_cross_account_readonlyaccess_policy.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_role_cross_account_readonlyaccess_policy.iam_role_cross_account_readonlyaccess_policy import ( @@ -213,15 +201,12 @@ class Test_iam_role_cross_account_readonlyaccess_policy: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_role_cross_account_readonlyaccess_policy.iam_role_cross_account_readonlyaccess_policy.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_role_cross_account_readonlyaccess_policy.iam_role_cross_account_readonlyaccess_policy.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_role_cross_account_readonlyaccess_policy.iam_role_cross_account_readonlyaccess_policy import ( @@ -264,15 +249,12 @@ class Test_iam_role_cross_account_readonlyaccess_policy: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_role_cross_account_readonlyaccess_policy.iam_role_cross_account_readonlyaccess_policy.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_role_cross_account_readonlyaccess_policy.iam_role_cross_account_readonlyaccess_policy.iam_client", + new=iam_client, ): # Test Check from prowler.providers.aws.services.iam.iam_role_cross_account_readonlyaccess_policy.iam_role_cross_account_readonlyaccess_policy import ( diff --git a/tests/providers/aws/services/iam/iam_role_cross_service_confused_deputy_prevention/iam_role_cross_service_confused_deputy_prevention_test.py b/tests/providers/aws/services/iam/iam_role_cross_service_confused_deputy_prevention/iam_role_cross_service_confused_deputy_prevention_test.py index b5554da266..353846eca9 100644 --- a/tests/providers/aws/services/iam/iam_role_cross_service_confused_deputy_prevention/iam_role_cross_service_confused_deputy_prevention_test.py +++ b/tests/providers/aws/services/iam/iam_role_cross_service_confused_deputy_prevention/iam_role_cross_service_confused_deputy_prevention_test.py @@ -18,15 +18,12 @@ class Test_iam_role_cross_service_confused_deputy_prevention: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) aws_provider.identity.account = AWS_ACCOUNT_ID - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention import ( @@ -62,15 +59,12 @@ class Test_iam_role_cross_service_confused_deputy_prevention: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) aws_provider.identity.account = AWS_ACCOUNT_ID - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention.iam_client", + new=iam_client, ): # Test Check from prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention import ( @@ -103,15 +97,12 @@ class Test_iam_role_cross_service_confused_deputy_prevention: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) aws_provider.identity.account = AWS_ACCOUNT_ID - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention import ( @@ -154,15 +145,12 @@ class Test_iam_role_cross_service_confused_deputy_prevention: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) aws_provider.identity.account = AWS_ACCOUNT_ID - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention import ( @@ -207,15 +195,12 @@ class Test_iam_role_cross_service_confused_deputy_prevention: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) aws_provider.identity.account = AWS_ACCOUNT_ID - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention import ( @@ -260,15 +245,12 @@ class Test_iam_role_cross_service_confused_deputy_prevention: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) aws_provider.identity.account = AWS_ACCOUNT_ID - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention import ( @@ -313,15 +295,12 @@ class Test_iam_role_cross_service_confused_deputy_prevention: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) aws_provider.identity.account = AWS_ACCOUNT_ID - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_role_cross_service_confused_deputy_prevention.iam_role_cross_service_confused_deputy_prevention import ( diff --git a/tests/providers/aws/services/iam/iam_root_hardware_mfa_enabled/iam_root_hardware_mfa_enabled_test.py b/tests/providers/aws/services/iam/iam_root_hardware_mfa_enabled/iam_root_hardware_mfa_enabled_test.py index 22ac46dd4a..9c2bca4ff3 100644 --- a/tests/providers/aws/services/iam/iam_root_hardware_mfa_enabled/iam_root_hardware_mfa_enabled_test.py +++ b/tests/providers/aws/services/iam/iam_root_hardware_mfa_enabled/iam_root_hardware_mfa_enabled_test.py @@ -33,15 +33,12 @@ class Test_iam_root_hardware_mfa_enabled_test: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_root_hardware_mfa_enabled.iam_root_hardware_mfa_enabled.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_root_hardware_mfa_enabled.iam_root_hardware_mfa_enabled.iam_client", + new=iam_client, ): from prowler.providers.aws.services.iam.iam_root_hardware_mfa_enabled.iam_root_hardware_mfa_enabled import ( iam_root_hardware_mfa_enabled, @@ -71,15 +68,12 @@ class Test_iam_root_hardware_mfa_enabled_test: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_root_hardware_mfa_enabled.iam_root_hardware_mfa_enabled.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_root_hardware_mfa_enabled.iam_root_hardware_mfa_enabled.iam_client", + new=iam_client, ): from prowler.providers.aws.services.iam.iam_root_hardware_mfa_enabled.iam_root_hardware_mfa_enabled import ( iam_root_hardware_mfa_enabled, @@ -107,15 +101,12 @@ class Test_iam_root_hardware_mfa_enabled_test: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_root_hardware_mfa_enabled.iam_root_hardware_mfa_enabled.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_root_hardware_mfa_enabled.iam_root_hardware_mfa_enabled.iam_client", + new=iam_client, ): from prowler.providers.aws.services.iam.iam_root_hardware_mfa_enabled.iam_root_hardware_mfa_enabled import ( iam_root_hardware_mfa_enabled, diff --git a/tests/providers/aws/services/iam/iam_root_mfa_enabled/iam_root_mfa_enabled_test.py b/tests/providers/aws/services/iam/iam_root_mfa_enabled/iam_root_mfa_enabled_test.py index 67381e1208..1932a5f2e6 100644 --- a/tests/providers/aws/services/iam/iam_root_mfa_enabled/iam_root_mfa_enabled_test.py +++ b/tests/providers/aws/services/iam/iam_root_mfa_enabled/iam_root_mfa_enabled_test.py @@ -26,16 +26,13 @@ class Test_iam_root_mfa_enabled_test: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_root_mfa_enabled.iam_root_mfa_enabled.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_root_mfa_enabled.iam_root_mfa_enabled.iam_client", + new=IAM(aws_provider), + ) as service_client: from prowler.providers.aws.services.iam.iam_root_mfa_enabled.iam_root_mfa_enabled import ( iam_root_mfa_enabled, ) @@ -64,16 +61,13 @@ class Test_iam_root_mfa_enabled_test: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_root_mfa_enabled.iam_root_mfa_enabled.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_root_mfa_enabled.iam_root_mfa_enabled.iam_client", + new=IAM(aws_provider), + ) as service_client: from prowler.providers.aws.services.iam.iam_root_mfa_enabled.iam_root_mfa_enabled import ( iam_root_mfa_enabled, ) diff --git a/tests/providers/aws/services/iam/iam_securityaudit_role_created/iam_securityaudit_role_created_test.py b/tests/providers/aws/services/iam/iam_securityaudit_role_created/iam_securityaudit_role_created_test.py index 405ec780e7..1e675ec209 100644 --- a/tests/providers/aws/services/iam/iam_securityaudit_role_created/iam_securityaudit_role_created_test.py +++ b/tests/providers/aws/services/iam/iam_securityaudit_role_created/iam_securityaudit_role_created_test.py @@ -91,16 +91,13 @@ class Test_iam_securityaudit_role_created: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_securityaudit_role_created.iam_securityaudit_role_created.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_securityaudit_role_created.iam_securityaudit_role_created.iam_client", + new=IAM(aws_provider), + ) as service_client: from prowler.providers.aws.services.iam.iam_securityaudit_role_created.iam_securityaudit_role_created import ( iam_securityaudit_role_created, ) diff --git a/tests/providers/aws/services/iam/iam_support_role_created/iam_support_role_created_test.py b/tests/providers/aws/services/iam/iam_support_role_created/iam_support_role_created_test.py index 4a9dfbf8da..39ef5a8c41 100644 --- a/tests/providers/aws/services/iam/iam_support_role_created/iam_support_role_created_test.py +++ b/tests/providers/aws/services/iam/iam_support_role_created/iam_support_role_created_test.py @@ -43,15 +43,12 @@ class Test_iam_support_role_created: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_support_role_created.iam_support_role_created.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_support_role_created.iam_support_role_created.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_support_role_created.iam_support_role_created import ( iam_support_role_created, @@ -72,15 +69,12 @@ class Test_iam_support_role_created: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_support_role_created.iam_support_role_created.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_support_role_created.iam_support_role_created.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_support_role_created.iam_support_role_created import ( iam_support_role_created, @@ -101,16 +95,13 @@ class Test_iam_support_role_created: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_support_role_created.iam_support_role_created.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_support_role_created.iam_support_role_created.iam_client", + new=IAM(aws_provider), + ) as service_client: from prowler.providers.aws.services.iam.iam_support_role_created.iam_support_role_created import ( iam_support_role_created, ) diff --git a/tests/providers/aws/services/iam/iam_user_administrator_access_policy/iam_user_administrator_access_policy_test.py b/tests/providers/aws/services/iam/iam_user_administrator_access_policy/iam_user_administrator_access_policy_test.py index 7a757fbf66..f47927ba35 100644 --- a/tests/providers/aws/services/iam/iam_user_administrator_access_policy/iam_user_administrator_access_policy_test.py +++ b/tests/providers/aws/services/iam/iam_user_administrator_access_policy/iam_user_administrator_access_policy_test.py @@ -14,15 +14,12 @@ class Test_iam_user_administrator_access_policy: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_user_administrator_access_policy.iam_user_administrator_access_policy.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_user_administrator_access_policy.iam_user_administrator_access_policy.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_user_administrator_access_policy.iam_user_administrator_access_policy import ( @@ -44,15 +41,12 @@ class Test_iam_user_administrator_access_policy: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_user_administrator_access_policy.iam_user_administrator_access_policy.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_user_administrator_access_policy.iam_user_administrator_access_policy.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_user_administrator_access_policy.iam_user_administrator_access_policy import ( iam_user_administrator_access_policy, @@ -102,15 +96,12 @@ class Test_iam_user_administrator_access_policy: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_user_administrator_access_policy.iam_user_administrator_access_policy.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_user_administrator_access_policy.iam_user_administrator_access_policy.iam_client", + new=IAM(aws_provider), ): from prowler.providers.aws.services.iam.iam_user_administrator_access_policy.iam_user_administrator_access_policy import ( iam_user_administrator_access_policy, diff --git a/tests/providers/aws/services/iam/iam_user_hardware_mfa_enabled/iam_user_hardware_mfa_enabled_test.py b/tests/providers/aws/services/iam/iam_user_hardware_mfa_enabled/iam_user_hardware_mfa_enabled_test.py index 8087fe53e3..18e2a6a4aa 100644 --- a/tests/providers/aws/services/iam/iam_user_hardware_mfa_enabled/iam_user_hardware_mfa_enabled_test.py +++ b/tests/providers/aws/services/iam/iam_user_hardware_mfa_enabled/iam_user_hardware_mfa_enabled_test.py @@ -27,16 +27,13 @@ class Test_iam_user_hardware_mfa_enabled_test: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_user_hardware_mfa_enabled.iam_user_hardware_mfa_enabled.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_user_hardware_mfa_enabled.iam_user_hardware_mfa_enabled.iam_client", + new=IAM(aws_provider), + ) as service_client: from prowler.providers.aws.services.iam.iam_user_hardware_mfa_enabled.iam_user_hardware_mfa_enabled import ( iam_user_hardware_mfa_enabled, ) @@ -63,16 +60,13 @@ class Test_iam_user_hardware_mfa_enabled_test: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_user_hardware_mfa_enabled.iam_user_hardware_mfa_enabled.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_user_hardware_mfa_enabled.iam_user_hardware_mfa_enabled.iam_client", + new=IAM(aws_provider), + ) as service_client: from prowler.providers.aws.services.iam.iam_user_hardware_mfa_enabled.iam_user_hardware_mfa_enabled import ( iam_user_hardware_mfa_enabled, ) @@ -105,16 +99,13 @@ class Test_iam_user_hardware_mfa_enabled_test: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_user_hardware_mfa_enabled.iam_user_hardware_mfa_enabled.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_user_hardware_mfa_enabled.iam_user_hardware_mfa_enabled.iam_client", + new=IAM(aws_provider), + ) as service_client: from prowler.providers.aws.services.iam.iam_user_hardware_mfa_enabled.iam_user_hardware_mfa_enabled import ( iam_user_hardware_mfa_enabled, ) diff --git a/tests/providers/aws/services/iam/iam_user_mfa_enabled_console_access/iam_user_mfa_enabled_console_access_test.py b/tests/providers/aws/services/iam/iam_user_mfa_enabled_console_access/iam_user_mfa_enabled_console_access_test.py index 55baaf712c..92588f9d38 100644 --- a/tests/providers/aws/services/iam/iam_user_mfa_enabled_console_access/iam_user_mfa_enabled_console_access_test.py +++ b/tests/providers/aws/services/iam/iam_user_mfa_enabled_console_access/iam_user_mfa_enabled_console_access_test.py @@ -21,16 +21,13 @@ class Test_iam_user_mfa_enabled_console_access_test: from prowler.providers.aws.services.iam.iam_service import IAM aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_user_mfa_enabled_console_access.iam_user_mfa_enabled_console_access.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_user_mfa_enabled_console_access.iam_user_mfa_enabled_console_access.iam_client", + new=IAM(aws_provider), + ) as service_client: from prowler.providers.aws.services.iam.iam_user_mfa_enabled_console_access.iam_user_mfa_enabled_console_access import ( iam_user_mfa_enabled_console_access, ) @@ -78,16 +75,13 @@ class Test_iam_user_mfa_enabled_console_access_test: from prowler.providers.aws.services.iam.iam_service import IAM aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_user_mfa_enabled_console_access.iam_user_mfa_enabled_console_access.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_user_mfa_enabled_console_access.iam_user_mfa_enabled_console_access.iam_client", + new=IAM(aws_provider), + ) as service_client: from prowler.providers.aws.services.iam.iam_user_mfa_enabled_console_access.iam_user_mfa_enabled_console_access import ( iam_user_mfa_enabled_console_access, ) @@ -117,16 +111,13 @@ class Test_iam_user_mfa_enabled_console_access_test: from prowler.providers.aws.services.iam.iam_service import IAM aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_user_mfa_enabled_console_access.iam_user_mfa_enabled_console_access.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_user_mfa_enabled_console_access.iam_user_mfa_enabled_console_access.iam_client", + new=IAM(aws_provider), + ) as service_client: from prowler.providers.aws.services.iam.iam_user_mfa_enabled_console_access.iam_user_mfa_enabled_console_access import ( iam_user_mfa_enabled_console_access, ) @@ -157,16 +148,13 @@ class Test_iam_user_mfa_enabled_console_access_test: from prowler.providers.aws.services.iam.iam_service import IAM aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_user_mfa_enabled_console_access.iam_user_mfa_enabled_console_access.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_user_mfa_enabled_console_access.iam_user_mfa_enabled_console_access.iam_client", + new=IAM(aws_provider), + ) as service_client: from prowler.providers.aws.services.iam.iam_user_mfa_enabled_console_access.iam_user_mfa_enabled_console_access import ( iam_user_mfa_enabled_console_access, ) diff --git a/tests/providers/aws/services/iam/iam_user_no_setup_initial_access_key/iam_user_no_setup_initial_access_key_test.py b/tests/providers/aws/services/iam/iam_user_no_setup_initial_access_key/iam_user_no_setup_initial_access_key_test.py index a4c1b23595..f22a3d6292 100644 --- a/tests/providers/aws/services/iam/iam_user_no_setup_initial_access_key/iam_user_no_setup_initial_access_key_test.py +++ b/tests/providers/aws/services/iam/iam_user_no_setup_initial_access_key/iam_user_no_setup_initial_access_key_test.py @@ -16,16 +16,13 @@ test_false_access_key_1,arn:aws:iam::123456789012:test_false_access_key_1,2022-0 aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_user_no_setup_initial_access_key.iam_user_no_setup_initial_access_key.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_user_no_setup_initial_access_key.iam_user_no_setup_initial_access_key.iam_client", + new=IAM(aws_provider), + ) as service_client: from prowler.providers.aws.services.iam.iam_user_no_setup_initial_access_key.iam_user_no_setup_initial_access_key import ( iam_user_no_setup_initial_access_key, ) @@ -63,16 +60,13 @@ test_false_access_key_2,arn:aws:iam::123456789012:test_false_access_key_2,2022-0 aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_user_no_setup_initial_access_key.iam_user_no_setup_initial_access_key.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_user_no_setup_initial_access_key.iam_user_no_setup_initial_access_key.iam_client", + new=IAM(aws_provider), + ) as service_client: from prowler.providers.aws.services.iam.iam_user_no_setup_initial_access_key.iam_user_no_setup_initial_access_key import ( iam_user_no_setup_initial_access_key, ) @@ -110,16 +104,13 @@ test_false_both_access_keys,arn:aws:iam::123456789012:test_false_both_access_key aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_user_no_setup_initial_access_key.iam_user_no_setup_initial_access_key.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_user_no_setup_initial_access_key.iam_user_no_setup_initial_access_key.iam_client", + new=IAM(aws_provider), + ) as service_client: from prowler.providers.aws.services.iam.iam_user_no_setup_initial_access_key.iam_user_no_setup_initial_access_key import ( iam_user_no_setup_initial_access_key, ) @@ -168,16 +159,13 @@ test_pass,arn:aws:iam::123456789012:test_pass,2022-02-17T14:59:38+00:00,not_supp aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) from prowler.providers.aws.services.iam.iam_service import IAM - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_user_no_setup_initial_access_key.iam_user_no_setup_initial_access_key.iam_client", - new=IAM(aws_provider), - ) as service_client, - ): + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_user_no_setup_initial_access_key.iam_user_no_setup_initial_access_key.iam_client", + new=IAM(aws_provider), + ) as service_client: from prowler.providers.aws.services.iam.iam_user_no_setup_initial_access_key.iam_user_no_setup_initial_access_key import ( iam_user_no_setup_initial_access_key, ) diff --git a/tests/providers/aws/services/iam/iam_user_two_active_access_key/iam_user_two_active_access_key_test.py b/tests/providers/aws/services/iam/iam_user_two_active_access_key/iam_user_two_active_access_key_test.py index 78cec9b400..6337334613 100644 --- a/tests/providers/aws/services/iam/iam_user_two_active_access_key/iam_user_two_active_access_key_test.py +++ b/tests/providers/aws/services/iam/iam_user_two_active_access_key/iam_user_two_active_access_key_test.py @@ -33,19 +33,15 @@ class Test_iam_user_two_active_access_key: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_user_two_active_access_key.iam_user_two_active_access_key.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_user_two_active_access_key.iam_user_two_active_access_key.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_user_two_active_access_key.iam_user_two_active_access_key import ( @@ -78,15 +74,12 @@ class Test_iam_user_two_active_access_key: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_user_two_active_access_key.iam_user_two_active_access_key.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_user_two_active_access_key.iam_user_two_active_access_key.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_user_two_active_access_key.iam_user_two_active_access_key import ( @@ -118,15 +111,12 @@ class Test_iam_user_two_active_access_key: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_user_two_active_access_key.iam_user_two_active_access_key.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_user_two_active_access_key.iam_user_two_active_access_key.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_user_two_active_access_key.iam_user_two_active_access_key import ( @@ -152,15 +142,12 @@ class Test_iam_user_two_active_access_key: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.iam.iam_user_two_active_access_key.iam_user_two_active_access_key.iam_client", - new=IAM(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.iam.iam_user_two_active_access_key.iam_user_two_active_access_key.iam_client", + new=IAM(aws_provider), ): # Test Check from prowler.providers.aws.services.iam.iam_user_two_active_access_key.iam_user_two_active_access_key import ( diff --git a/tests/providers/aws/services/iam/iam_user_with_temporary_credentials/iam_user_with_temporary_credentials_test.py b/tests/providers/aws/services/iam/iam_user_with_temporary_credentials/iam_user_with_temporary_credentials_test.py index d7086d54be..e1b9c31587 100644 --- a/tests/providers/aws/services/iam/iam_user_with_temporary_credentials/iam_user_with_temporary_credentials_test.py +++ b/tests/providers/aws/services/iam/iam_user_with_temporary_credentials/iam_user_with_temporary_credentials_test.py @@ -30,15 +30,12 @@ class Test_iam_user_with_temporary_credentials: ) iam_client._get_user_temporary_credentials_usage(iam_client) - with ( - mock.patch( - "prowler.providers.aws.services.iam.iam_service.IAM", - new=iam_client, - ) as iam_service, - mock.patch( - "prowler.providers.aws.services.iam.iam_client.iam_client", - new=iam_service, - ), + with mock.patch( + "prowler.providers.aws.services.iam.iam_service.IAM", + new=iam_client, + ) as iam_service, mock.patch( + "prowler.providers.aws.services.iam.iam_client.iam_client", + new=iam_service, ): from prowler.providers.aws.services.iam.iam_user_with_temporary_credentials.iam_user_with_temporary_credentials import ( iam_user_with_temporary_credentials, @@ -69,15 +66,12 @@ class Test_iam_user_with_temporary_credentials: ) iam_client._get_user_temporary_credentials_usage(iam_client) - with ( - mock.patch( - "prowler.providers.aws.services.iam.iam_service.IAM", - new=iam_client, - ) as iam_service, - mock.patch( - "prowler.providers.aws.services.iam.iam_client.iam_client", - new=iam_service, - ), + with mock.patch( + "prowler.providers.aws.services.iam.iam_service.IAM", + new=iam_client, + ) as iam_service, mock.patch( + "prowler.providers.aws.services.iam.iam_client.iam_client", + new=iam_service, ): from prowler.providers.aws.services.iam.iam_user_with_temporary_credentials.iam_user_with_temporary_credentials import ( iam_user_with_temporary_credentials, @@ -117,15 +111,12 @@ class Test_iam_user_with_temporary_credentials: ) iam_client._get_user_temporary_credentials_usage(iam_client) - with ( - mock.patch( - "prowler.providers.aws.services.iam.iam_service.IAM", - new=iam_client, - ) as iam_service, - mock.patch( - "prowler.providers.aws.services.iam.iam_client.iam_client", - new=iam_service, - ), + with mock.patch( + "prowler.providers.aws.services.iam.iam_service.IAM", + new=iam_client, + ) as iam_service, mock.patch( + "prowler.providers.aws.services.iam.iam_client.iam_client", + new=iam_service, ): from prowler.providers.aws.services.iam.iam_user_with_temporary_credentials.iam_user_with_temporary_credentials import ( iam_user_with_temporary_credentials, @@ -165,15 +156,12 @@ class Test_iam_user_with_temporary_credentials: ) iam_client._get_user_temporary_credentials_usage(iam_client) - with ( - mock.patch( - "prowler.providers.aws.services.iam.iam_service.IAM", - new=iam_client, - ) as iam_service, - mock.patch( - "prowler.providers.aws.services.iam.iam_client.iam_client", - new=iam_service, - ), + with mock.patch( + "prowler.providers.aws.services.iam.iam_service.IAM", + new=iam_client, + ) as iam_service, mock.patch( + "prowler.providers.aws.services.iam.iam_client.iam_client", + new=iam_service, ): from prowler.providers.aws.services.iam.iam_user_with_temporary_credentials.iam_user_with_temporary_credentials import ( iam_user_with_temporary_credentials, @@ -215,15 +203,12 @@ class Test_iam_user_with_temporary_credentials: ) iam_client._get_user_temporary_credentials_usage(iam_client) - with ( - mock.patch( - "prowler.providers.aws.services.iam.iam_service.IAM", - new=iam_client, - ) as iam_service, - mock.patch( - "prowler.providers.aws.services.iam.iam_client.iam_client", - new=iam_service, - ), + with mock.patch( + "prowler.providers.aws.services.iam.iam_service.IAM", + new=iam_client, + ) as iam_service, mock.patch( + "prowler.providers.aws.services.iam.iam_client.iam_client", + new=iam_service, ): from prowler.providers.aws.services.iam.iam_user_with_temporary_credentials.iam_user_with_temporary_credentials import ( iam_user_with_temporary_credentials, @@ -265,15 +250,12 @@ class Test_iam_user_with_temporary_credentials: ) iam_client._get_user_temporary_credentials_usage(iam_client) - with ( - mock.patch( - "prowler.providers.aws.services.iam.iam_service.IAM", - new=iam_client, - ) as iam_service, - mock.patch( - "prowler.providers.aws.services.iam.iam_client.iam_client", - new=iam_service, - ), + with mock.patch( + "prowler.providers.aws.services.iam.iam_service.IAM", + new=iam_client, + ) as iam_service, mock.patch( + "prowler.providers.aws.services.iam.iam_client.iam_client", + new=iam_service, ): from prowler.providers.aws.services.iam.iam_user_with_temporary_credentials.iam_user_with_temporary_credentials import ( iam_user_with_temporary_credentials, diff --git a/tests/providers/aws/services/kafka/kafka_cluster_encryption_at_rest_uses_cmk/kafka_cluster_encryption_at_rest_uses_cmk_test.py b/tests/providers/aws/services/kafka/kafka_cluster_encryption_at_rest_uses_cmk/kafka_cluster_encryption_at_rest_uses_cmk_test.py index 236da74917..91e8ca5357 100644 --- a/tests/providers/aws/services/kafka/kafka_cluster_encryption_at_rest_uses_cmk/kafka_cluster_encryption_at_rest_uses_cmk_test.py +++ b/tests/providers/aws/services/kafka/kafka_cluster_encryption_at_rest_uses_cmk/kafka_cluster_encryption_at_rest_uses_cmk_test.py @@ -12,15 +12,12 @@ class Test_kafka_cluster_encryption_at_rest_uses_cmk: kafka_client = MagicMock kafka_client.clusters = {} - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_encryption_at_rest_uses_cmk.kafka_cluster_encryption_at_rest_uses_cmk import ( kafka_cluster_encryption_at_rest_uses_cmk, @@ -62,19 +59,15 @@ class Test_kafka_cluster_encryption_at_rest_uses_cmk: ) ] - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), - patch( - "prowler.providers.aws.services.kafka.kafka_cluster_encryption_at_rest_uses_cmk.kafka_cluster_encryption_at_rest_uses_cmk.kms_client", - new=kms_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, + ), patch( + "prowler.providers.aws.services.kafka.kafka_cluster_encryption_at_rest_uses_cmk.kafka_cluster_encryption_at_rest_uses_cmk.kms_client", + new=kms_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_encryption_at_rest_uses_cmk.kafka_cluster_encryption_at_rest_uses_cmk import ( kafka_cluster_encryption_at_rest_uses_cmk, @@ -128,19 +121,15 @@ class Test_kafka_cluster_encryption_at_rest_uses_cmk: ) ] - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), - patch( - "prowler.providers.aws.services.kafka.kafka_cluster_encryption_at_rest_uses_cmk.kafka_cluster_encryption_at_rest_uses_cmk.kms_client", - new=kms_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, + ), patch( + "prowler.providers.aws.services.kafka.kafka_cluster_encryption_at_rest_uses_cmk.kafka_cluster_encryption_at_rest_uses_cmk.kms_client", + new=kms_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_encryption_at_rest_uses_cmk.kafka_cluster_encryption_at_rest_uses_cmk import ( kafka_cluster_encryption_at_rest_uses_cmk, diff --git a/tests/providers/aws/services/kafka/kafka_cluster_enhanced_monitoring_enabled/kafka_cluster_enhanced_monitoring_enabled_test.py b/tests/providers/aws/services/kafka/kafka_cluster_enhanced_monitoring_enabled/kafka_cluster_enhanced_monitoring_enabled_test.py index 784d5ebff4..e12757c7c2 100644 --- a/tests/providers/aws/services/kafka/kafka_cluster_enhanced_monitoring_enabled/kafka_cluster_enhanced_monitoring_enabled_test.py +++ b/tests/providers/aws/services/kafka/kafka_cluster_enhanced_monitoring_enabled/kafka_cluster_enhanced_monitoring_enabled_test.py @@ -12,15 +12,12 @@ class Test_kafka_cluster_enhanced_monitoring_enabled: kafka_client = MagicMock kafka_client.clusters = {} - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_enhanced_monitoring_enabled.kafka_cluster_enhanced_monitoring_enabled import ( kafka_cluster_enhanced_monitoring_enabled, @@ -54,15 +51,12 @@ class Test_kafka_cluster_enhanced_monitoring_enabled: ) } - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_enhanced_monitoring_enabled.kafka_cluster_enhanced_monitoring_enabled import ( kafka_cluster_enhanced_monitoring_enabled, @@ -108,15 +102,12 @@ class Test_kafka_cluster_enhanced_monitoring_enabled: ) } - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_enhanced_monitoring_enabled.kafka_cluster_enhanced_monitoring_enabled import ( kafka_cluster_enhanced_monitoring_enabled, diff --git a/tests/providers/aws/services/kafka/kafka_cluster_in_transit_encryption_enabled/kafka_cluster_in_transit_encryption_enabled_test.py b/tests/providers/aws/services/kafka/kafka_cluster_in_transit_encryption_enabled/kafka_cluster_in_transit_encryption_enabled_test.py index 78368530ea..0f186a8e74 100644 --- a/tests/providers/aws/services/kafka/kafka_cluster_in_transit_encryption_enabled/kafka_cluster_in_transit_encryption_enabled_test.py +++ b/tests/providers/aws/services/kafka/kafka_cluster_in_transit_encryption_enabled/kafka_cluster_in_transit_encryption_enabled_test.py @@ -12,15 +12,12 @@ class Test_kafka_cluster_in_transit_encryption_enabled: kafka_client = MagicMock kafka_client.clusters = {} - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_in_transit_encryption_enabled.kafka_cluster_in_transit_encryption_enabled import ( kafka_cluster_in_transit_encryption_enabled, @@ -54,15 +51,12 @@ class Test_kafka_cluster_in_transit_encryption_enabled: ) } - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_in_transit_encryption_enabled.kafka_cluster_in_transit_encryption_enabled import ( kafka_cluster_in_transit_encryption_enabled, @@ -108,15 +102,12 @@ class Test_kafka_cluster_in_transit_encryption_enabled: ) } - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_in_transit_encryption_enabled.kafka_cluster_in_transit_encryption_enabled import ( kafka_cluster_in_transit_encryption_enabled, @@ -162,15 +153,12 @@ class Test_kafka_cluster_in_transit_encryption_enabled: ) } - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_in_transit_encryption_enabled.kafka_cluster_in_transit_encryption_enabled import ( kafka_cluster_in_transit_encryption_enabled, diff --git a/tests/providers/aws/services/kafka/kafka_cluster_is_public/kafka_cluster_is_public_test.py b/tests/providers/aws/services/kafka/kafka_cluster_is_public/kafka_cluster_is_public_test.py index 5925a435cd..44eb233ec3 100644 --- a/tests/providers/aws/services/kafka/kafka_cluster_is_public/kafka_cluster_is_public_test.py +++ b/tests/providers/aws/services/kafka/kafka_cluster_is_public/kafka_cluster_is_public_test.py @@ -12,15 +12,12 @@ class Test_kafka_cluster_is_public: kafka_client = MagicMock kafka_client.clusters = {} - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_is_public.kafka_cluster_is_public import ( kafka_cluster_is_public, @@ -54,15 +51,12 @@ class Test_kafka_cluster_is_public: ) } - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_is_public.kafka_cluster_is_public import ( kafka_cluster_is_public, @@ -108,15 +102,12 @@ class Test_kafka_cluster_is_public: ) } - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_is_public.kafka_cluster_is_public import ( kafka_cluster_is_public, diff --git a/tests/providers/aws/services/kafka/kafka_cluster_mutual_tls_authentication_enabled/kafka_cluster_mutual_tls_authentication_enabled_test.py b/tests/providers/aws/services/kafka/kafka_cluster_mutual_tls_authentication_enabled/kafka_cluster_mutual_tls_authentication_enabled_test.py index c5d7f42205..8aac138544 100644 --- a/tests/providers/aws/services/kafka/kafka_cluster_mutual_tls_authentication_enabled/kafka_cluster_mutual_tls_authentication_enabled_test.py +++ b/tests/providers/aws/services/kafka/kafka_cluster_mutual_tls_authentication_enabled/kafka_cluster_mutual_tls_authentication_enabled_test.py @@ -12,15 +12,12 @@ class Test_kafka_cluster_mutual_tls_authentication_enabled: kafka_client = MagicMock kafka_client.clusters = {} - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_mutual_tls_authentication_enabled.kafka_cluster_mutual_tls_authentication_enabled import ( kafka_cluster_mutual_tls_authentication_enabled, @@ -54,15 +51,12 @@ class Test_kafka_cluster_mutual_tls_authentication_enabled: ) } - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_mutual_tls_authentication_enabled.kafka_cluster_mutual_tls_authentication_enabled import ( kafka_cluster_mutual_tls_authentication_enabled, @@ -108,15 +102,12 @@ class Test_kafka_cluster_mutual_tls_authentication_enabled: ) } - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_mutual_tls_authentication_enabled.kafka_cluster_mutual_tls_authentication_enabled import ( kafka_cluster_mutual_tls_authentication_enabled, diff --git a/tests/providers/aws/services/kafka/kafka_cluster_unrestricted_access_disabled/kafka_cluster_unrestricted_access_disabled_test.py b/tests/providers/aws/services/kafka/kafka_cluster_unrestricted_access_disabled/kafka_cluster_unrestricted_access_disabled_test.py index 22a2c0eb71..f3c69cfa22 100644 --- a/tests/providers/aws/services/kafka/kafka_cluster_unrestricted_access_disabled/kafka_cluster_unrestricted_access_disabled_test.py +++ b/tests/providers/aws/services/kafka/kafka_cluster_unrestricted_access_disabled/kafka_cluster_unrestricted_access_disabled_test.py @@ -12,15 +12,12 @@ class Test_kafka_cluster_unrestricted_access_disabled: kafka_client = MagicMock kafka_client.clusters = {} - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_unrestricted_access_disabled.kafka_cluster_unrestricted_access_disabled import ( kafka_cluster_unrestricted_access_disabled, @@ -54,15 +51,12 @@ class Test_kafka_cluster_unrestricted_access_disabled: ) } - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_unrestricted_access_disabled.kafka_cluster_unrestricted_access_disabled import ( kafka_cluster_unrestricted_access_disabled, @@ -108,15 +102,12 @@ class Test_kafka_cluster_unrestricted_access_disabled: ) } - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_unrestricted_access_disabled.kafka_cluster_unrestricted_access_disabled import ( kafka_cluster_unrestricted_access_disabled, diff --git a/tests/providers/aws/services/kafka/kafka_cluster_uses_latest_version/kafka_cluster_uses_latest_version_test.py b/tests/providers/aws/services/kafka/kafka_cluster_uses_latest_version/kafka_cluster_uses_latest_version_test.py index ac44f868e0..ef4a7cf2eb 100644 --- a/tests/providers/aws/services/kafka/kafka_cluster_uses_latest_version/kafka_cluster_uses_latest_version_test.py +++ b/tests/providers/aws/services/kafka/kafka_cluster_uses_latest_version/kafka_cluster_uses_latest_version_test.py @@ -13,15 +13,12 @@ class Test_kafka_cluster_latest_version: kafka_client = MagicMock kafka_client.clusters = {} - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_uses_latest_version.kafka_cluster_uses_latest_version import ( kafka_cluster_uses_latest_version, @@ -60,15 +57,12 @@ class Test_kafka_cluster_latest_version: KafkaVersion(version="2.8.0", status="ACTIVE"), ] - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_uses_latest_version.kafka_cluster_uses_latest_version import ( kafka_cluster_uses_latest_version, @@ -119,15 +113,12 @@ class Test_kafka_cluster_latest_version: KafkaVersion(version="2.8.0", status="ACTIVE"), ] - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.kafka.kafka_service.Kafka", - new=kafka_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.kafka.kafka_service.Kafka", + new=kafka_client, ): from prowler.providers.aws.services.kafka.kafka_cluster_uses_latest_version.kafka_cluster_uses_latest_version import ( kafka_cluster_uses_latest_version, diff --git a/tests/providers/aws/services/kafka/kafka_connector_in_transit_encryption_enabled/kafka_connector_in_transit_encryption_enabled_test.py b/tests/providers/aws/services/kafka/kafka_connector_in_transit_encryption_enabled/kafka_connector_in_transit_encryption_enabled_test.py index 419e9405e0..9efc27b9e3 100644 --- a/tests/providers/aws/services/kafka/kafka_connector_in_transit_encryption_enabled/kafka_connector_in_transit_encryption_enabled_test.py +++ b/tests/providers/aws/services/kafka/kafka_connector_in_transit_encryption_enabled/kafka_connector_in_transit_encryption_enabled_test.py @@ -46,15 +46,12 @@ class Test_kafka_connector_in_transit_encryption_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.kafka.kafka_connector_in_transit_encryption_enabled.kafka_connector_in_transit_encryption_enabled.kafkaconnect_client", - new=KafkaConnect(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.kafka.kafka_connector_in_transit_encryption_enabled.kafka_connector_in_transit_encryption_enabled.kafkaconnect_client", + new=KafkaConnect(mocked_aws_provider), ): from prowler.providers.aws.services.kafka.kafka_connector_in_transit_encryption_enabled.kafka_connector_in_transit_encryption_enabled import ( kafka_connector_in_transit_encryption_enabled, @@ -71,15 +68,12 @@ class Test_kafka_connector_in_transit_encryption_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.kafka.kafka_connector_in_transit_encryption_enabled.kafka_connector_in_transit_encryption_enabled.kafkaconnect_client", - new=KafkaConnect(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.kafka.kafka_connector_in_transit_encryption_enabled.kafka_connector_in_transit_encryption_enabled.kafkaconnect_client", + new=KafkaConnect(mocked_aws_provider), ): from prowler.providers.aws.services.kafka.kafka_connector_in_transit_encryption_enabled.kafka_connector_in_transit_encryption_enabled import ( kafka_connector_in_transit_encryption_enabled, @@ -107,15 +101,12 @@ class Test_kafka_connector_in_transit_encryption_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - patch( - "prowler.providers.aws.services.kafka.kafka_connector_in_transit_encryption_enabled.kafka_connector_in_transit_encryption_enabled.kafkaconnect_client", - new=KafkaConnect(mocked_aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), patch( + "prowler.providers.aws.services.kafka.kafka_connector_in_transit_encryption_enabled.kafka_connector_in_transit_encryption_enabled.kafkaconnect_client", + new=KafkaConnect(mocked_aws_provider), ): from prowler.providers.aws.services.kafka.kafka_connector_in_transit_encryption_enabled.kafka_connector_in_transit_encryption_enabled import ( kafka_connector_in_transit_encryption_enabled, diff --git a/tests/providers/aws/services/kinesis/kinesis_stream_data_retention_period/kinesis_stream_data_retention_period_test.py b/tests/providers/aws/services/kinesis/kinesis_stream_data_retention_period/kinesis_stream_data_retention_period_test.py index 26265d3563..55f68c3604 100644 --- a/tests/providers/aws/services/kinesis/kinesis_stream_data_retention_period/kinesis_stream_data_retention_period_test.py +++ b/tests/providers/aws/services/kinesis/kinesis_stream_data_retention_period/kinesis_stream_data_retention_period_test.py @@ -17,15 +17,12 @@ class Test_kinesis_encrypted_at_rest: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kinesis.kinesis_stream_data_retention_period.kinesis_stream_data_retention_period.kinesis_client", - new=Kinesis(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kinesis.kinesis_stream_data_retention_period.kinesis_stream_data_retention_period.kinesis_client", + new=Kinesis(aws_provider), ): # Test Check from prowler.providers.aws.services.kinesis.kinesis_stream_data_retention_period.kinesis_stream_data_retention_period import ( @@ -60,15 +57,12 @@ class Test_kinesis_encrypted_at_rest: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kinesis.kinesis_stream_data_retention_period.kinesis_stream_data_retention_period.kinesis_client", - new=Kinesis(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kinesis.kinesis_stream_data_retention_period.kinesis_stream_data_retention_period.kinesis_client", + new=Kinesis(aws_provider), ): with mock.patch( "prowler.providers.aws.services.kinesis.kinesis_stream_data_retention_period.kinesis_stream_data_retention_period.kinesis_client.audit_config", @@ -119,15 +113,12 @@ class Test_kinesis_encrypted_at_rest: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kinesis.kinesis_stream_data_retention_period.kinesis_stream_data_retention_period.kinesis_client", - new=Kinesis(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kinesis.kinesis_stream_data_retention_period.kinesis_stream_data_retention_period.kinesis_client", + new=Kinesis(aws_provider), ): with mock.patch( "prowler.providers.aws.services.kinesis.kinesis_stream_data_retention_period.kinesis_stream_data_retention_period.kinesis_client.audit_config", diff --git a/tests/providers/aws/services/kinesis/kinesis_stream_encrypted_at_rest/kinesis_stream_encrypted_at_rest_test.py b/tests/providers/aws/services/kinesis/kinesis_stream_encrypted_at_rest/kinesis_stream_encrypted_at_rest_test.py index 21423c76c6..f277b6ad83 100644 --- a/tests/providers/aws/services/kinesis/kinesis_stream_encrypted_at_rest/kinesis_stream_encrypted_at_rest_test.py +++ b/tests/providers/aws/services/kinesis/kinesis_stream_encrypted_at_rest/kinesis_stream_encrypted_at_rest_test.py @@ -13,15 +13,12 @@ class Test_kinesis_encrypted_at_rest: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kinesis.kinesis_stream_encrypted_at_rest.kinesis_stream_encrypted_at_rest.kinesis_client", - new=Kinesis(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kinesis.kinesis_stream_encrypted_at_rest.kinesis_stream_encrypted_at_rest.kinesis_client", + new=Kinesis(aws_provider), ): # Test Check from prowler.providers.aws.services.kinesis.kinesis_stream_encrypted_at_rest.kinesis_stream_encrypted_at_rest import ( @@ -53,15 +50,12 @@ class Test_kinesis_encrypted_at_rest: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kinesis.kinesis_stream_encrypted_at_rest.kinesis_stream_encrypted_at_rest.kinesis_client", - new=Kinesis(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kinesis.kinesis_stream_encrypted_at_rest.kinesis_stream_encrypted_at_rest.kinesis_client", + new=Kinesis(aws_provider), ): # Test Check from prowler.providers.aws.services.kinesis.kinesis_stream_encrypted_at_rest.kinesis_stream_encrypted_at_rest import ( @@ -99,15 +93,12 @@ class Test_kinesis_encrypted_at_rest: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kinesis.kinesis_stream_encrypted_at_rest.kinesis_stream_encrypted_at_rest.kinesis_client", - new=Kinesis(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kinesis.kinesis_stream_encrypted_at_rest.kinesis_stream_encrypted_at_rest.kinesis_client", + new=Kinesis(aws_provider), ): # Test Check from prowler.providers.aws.services.kinesis.kinesis_stream_encrypted_at_rest.kinesis_stream_encrypted_at_rest import ( diff --git a/tests/providers/aws/services/kms/kms_cmk_are_used/kms_cmk_are_used_test.py b/tests/providers/aws/services/kms/kms_cmk_are_used/kms_cmk_are_used_test.py index 51fad0bfa8..2c8e162609 100644 --- a/tests/providers/aws/services/kms/kms_cmk_are_used/kms_cmk_are_used_test.py +++ b/tests/providers/aws/services/kms/kms_cmk_are_used/kms_cmk_are_used_test.py @@ -1,9 +1,9 @@ from typing import Any, List from unittest import mock -import pytest from boto3 import client from moto import mock_aws +import pytest from tests.providers.aws.utils import AWS_REGION_US_EAST_1, set_mocked_aws_provider @@ -15,15 +15,12 @@ class Test_kms_cmk_are_used: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kms.kms_cmk_are_used.kms_cmk_are_used.kms_client", - new=KMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kms.kms_cmk_are_used.kms_cmk_are_used.kms_client", + new=KMS(aws_provider), ): # Test Check from prowler.providers.aws.services.kms.kms_cmk_are_used.kms_cmk_are_used import ( @@ -46,15 +43,12 @@ class Test_kms_cmk_are_used: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kms.kms_cmk_are_used.kms_cmk_are_used.kms_client", - new=KMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kms.kms_cmk_are_used.kms_cmk_are_used.kms_client", + new=KMS(aws_provider), ): # Test Check from prowler.providers.aws.services.kms.kms_cmk_are_used.kms_cmk_are_used import ( @@ -109,19 +103,15 @@ class Test_kms_cmk_are_used: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.aws_provider.AwsProvider.generate_regional_clients", - return_value={AWS_REGION_US_EAST_1: kms_client}, - ), - mock.patch( - "prowler.providers.aws.services.kms.kms_cmk_are_used.kms_cmk_are_used.kms_client", - new=KMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.aws_provider.AwsProvider.generate_regional_clients", + return_value={AWS_REGION_US_EAST_1: kms_client}, + ), mock.patch( + "prowler.providers.aws.services.kms.kms_cmk_are_used.kms_cmk_are_used.kms_client", + new=KMS(aws_provider), ): # Test Check from prowler.providers.aws.services.kms.kms_cmk_are_used.kms_cmk_are_used import ( @@ -145,15 +135,12 @@ class Test_kms_cmk_are_used: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kms.kms_cmk_are_used.kms_cmk_are_used.kms_client", - new=KMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kms.kms_cmk_are_used.kms_cmk_are_used.kms_client", + new=KMS(aws_provider), ): # Test Check from prowler.providers.aws.services.kms.kms_cmk_are_used.kms_cmk_are_used import ( @@ -184,15 +171,12 @@ class Test_kms_cmk_are_used: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kms.kms_cmk_are_used.kms_cmk_are_used.kms_client", - new=KMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kms.kms_cmk_are_used.kms_cmk_are_used.kms_client", + new=KMS(aws_provider), ): # Test Check from prowler.providers.aws.services.kms.kms_cmk_are_used.kms_cmk_are_used import ( diff --git a/tests/providers/aws/services/kms/kms_cmk_not_deleted_unintentionally/kms_cmk_not_deleted_unintentionally_fixer_test.py b/tests/providers/aws/services/kms/kms_cmk_not_deleted_unintentionally/kms_cmk_not_deleted_unintentionally_fixer_test.py index c9836f0c28..47ff98d8f4 100644 --- a/tests/providers/aws/services/kms/kms_cmk_not_deleted_unintentionally/kms_cmk_not_deleted_unintentionally_fixer_test.py +++ b/tests/providers/aws/services/kms/kms_cmk_not_deleted_unintentionally/kms_cmk_not_deleted_unintentionally_fixer_test.py @@ -17,15 +17,12 @@ class Test_kms_cmk_not_deleted_unintentionally_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally_fixer.kms_client", - new=KMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally_fixer.kms_client", + new=KMS(aws_provider), ): from prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally_fixer import ( fixer, @@ -43,15 +40,12 @@ class Test_kms_cmk_not_deleted_unintentionally_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally_fixer.kms_client", - new=KMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally_fixer.kms_client", + new=KMS(aws_provider), ): from prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally_fixer import ( fixer, @@ -69,15 +63,12 @@ class Test_kms_cmk_not_deleted_unintentionally_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally_fixer.kms_client", - new=KMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally_fixer.kms_client", + new=KMS(aws_provider), ): from prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally_fixer import ( fixer, diff --git a/tests/providers/aws/services/kms/kms_cmk_not_deleted_unintentionally/kms_cmk_not_deleted_unintentionally_test.py b/tests/providers/aws/services/kms/kms_cmk_not_deleted_unintentionally/kms_cmk_not_deleted_unintentionally_test.py index 3b47474b10..07823dac77 100644 --- a/tests/providers/aws/services/kms/kms_cmk_not_deleted_unintentionally/kms_cmk_not_deleted_unintentionally_test.py +++ b/tests/providers/aws/services/kms/kms_cmk_not_deleted_unintentionally/kms_cmk_not_deleted_unintentionally_test.py @@ -13,15 +13,12 @@ class Test_kms_cmk_not_deleted_unintentionally: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally.kms_client", - new=KMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally.kms_client", + new=KMS(aws_provider), ): from prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally import ( kms_cmk_not_deleted_unintentionally, @@ -44,15 +41,12 @@ class Test_kms_cmk_not_deleted_unintentionally: [AWS_REGION_US_EAST_1], scan_unused_services=False ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally.kms_client", - new=KMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally.kms_client", + new=KMS(aws_provider), ): from prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally import ( kms_cmk_not_deleted_unintentionally, @@ -73,15 +67,12 @@ class Test_kms_cmk_not_deleted_unintentionally: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally.kms_client", - new=KMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally.kms_client", + new=KMS(aws_provider), ): from prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally import ( kms_cmk_not_deleted_unintentionally, @@ -109,15 +100,12 @@ class Test_kms_cmk_not_deleted_unintentionally: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally.kms_client", - new=KMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally.kms_client", + new=KMS(aws_provider), ): from prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally import ( kms_cmk_not_deleted_unintentionally, @@ -147,15 +135,12 @@ class Test_kms_cmk_not_deleted_unintentionally: [AWS_REGION_US_EAST_1], scan_unused_services=True ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally.kms_client", - new=KMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally.kms_client", + new=KMS(aws_provider), ): from prowler.providers.aws.services.kms.kms_cmk_not_deleted_unintentionally.kms_cmk_not_deleted_unintentionally import ( kms_cmk_not_deleted_unintentionally, diff --git a/tests/providers/aws/services/kms/kms_cmk_rotation_enabled/kms_cmk_rotation_enabled_fixer_test.py b/tests/providers/aws/services/kms/kms_cmk_rotation_enabled/kms_cmk_rotation_enabled_fixer_test.py index 9a9fe3b10f..48133aabca 100644 --- a/tests/providers/aws/services/kms/kms_cmk_rotation_enabled/kms_cmk_rotation_enabled_fixer_test.py +++ b/tests/providers/aws/services/kms/kms_cmk_rotation_enabled/kms_cmk_rotation_enabled_fixer_test.py @@ -18,15 +18,12 @@ class Test_kms_cmk_rotation_enabled_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kms.kms_cmk_rotation_enabled.kms_cmk_rotation_enabled_fixer.kms_client", - new=KMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kms.kms_cmk_rotation_enabled.kms_cmk_rotation_enabled_fixer.kms_client", + new=KMS(aws_provider), ): # Test Check from prowler.providers.aws.services.kms.kms_cmk_rotation_enabled.kms_cmk_rotation_enabled_fixer import ( diff --git a/tests/providers/aws/services/kms/kms_cmk_rotation_enabled/kms_cmk_rotation_enabled_test.py b/tests/providers/aws/services/kms/kms_cmk_rotation_enabled/kms_cmk_rotation_enabled_test.py index d026ef5ef3..c5562cc6f0 100644 --- a/tests/providers/aws/services/kms/kms_cmk_rotation_enabled/kms_cmk_rotation_enabled_test.py +++ b/tests/providers/aws/services/kms/kms_cmk_rotation_enabled/kms_cmk_rotation_enabled_test.py @@ -1,9 +1,9 @@ from typing import Any, List from unittest import mock -import pytest from boto3 import client from moto import mock_aws +import pytest from tests.providers.aws.utils import AWS_REGION_US_EAST_1, set_mocked_aws_provider @@ -15,15 +15,12 @@ class Test_kms_cmk_rotation_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kms.kms_cmk_rotation_enabled.kms_cmk_rotation_enabled.kms_client", - new=KMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kms.kms_cmk_rotation_enabled.kms_cmk_rotation_enabled.kms_client", + new=KMS(aws_provider), ): # Test Check from prowler.providers.aws.services.kms.kms_cmk_rotation_enabled.kms_cmk_rotation_enabled import ( @@ -47,15 +44,12 @@ class Test_kms_cmk_rotation_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kms.kms_cmk_rotation_enabled.kms_cmk_rotation_enabled.kms_client", - new=KMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kms.kms_cmk_rotation_enabled.kms_cmk_rotation_enabled.kms_client", + new=KMS(aws_provider), ): # Test Check from prowler.providers.aws.services.kms.kms_cmk_rotation_enabled.kms_cmk_rotation_enabled import ( @@ -111,19 +105,15 @@ class Test_kms_cmk_rotation_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.aws_provider.AwsProvider.generate_regional_clients", - return_value={AWS_REGION_US_EAST_1: kms_client}, - ), - mock.patch( - "prowler.providers.aws.services.kms.kms_cmk_rotation_enabled.kms_cmk_rotation_enabled.kms_client", - new=KMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.aws_provider.AwsProvider.generate_regional_clients", + return_value={AWS_REGION_US_EAST_1: kms_client}, + ), mock.patch( + "prowler.providers.aws.services.kms.kms_cmk_rotation_enabled.kms_cmk_rotation_enabled.kms_client", + new=KMS(aws_provider), ): # Test Check from prowler.providers.aws.services.kms.kms_cmk_rotation_enabled.kms_cmk_rotation_enabled import ( @@ -148,15 +138,12 @@ class Test_kms_cmk_rotation_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.kms.kms_cmk_rotation_enabled.kms_cmk_rotation_enabled.kms_client", - new=KMS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.kms.kms_cmk_rotation_enabled.kms_cmk_rotation_enabled.kms_client", + new=KMS(aws_provider), ): # Test Check from prowler.providers.aws.services.kms.kms_cmk_rotation_enabled.kms_cmk_rotation_enabled import ( diff --git a/tests/providers/aws/services/lightsail/lightsail_database_public/lightsail_database_public_test.py b/tests/providers/aws/services/lightsail/lightsail_database_public/lightsail_database_public_test.py index b73e9bc1aa..35b6e01972 100644 --- a/tests/providers/aws/services/lightsail/lightsail_database_public/lightsail_database_public_test.py +++ b/tests/providers/aws/services/lightsail/lightsail_database_public/lightsail_database_public_test.py @@ -14,15 +14,12 @@ class Test_lightsail_database_public: lightsail_client = MagicMock lightsail_client.databases = {} - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", - new=lightsail_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", + new=lightsail_client, ): from prowler.providers.aws.services.lightsail.lightsail_database_public.lightsail_database_public import ( lightsail_database_public, @@ -52,15 +49,12 @@ class Test_lightsail_database_public: ) } - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", - new=lightsail_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", + new=lightsail_client, ): from prowler.providers.aws.services.lightsail.lightsail_database_public.lightsail_database_public import ( lightsail_database_public, @@ -98,15 +92,12 @@ class Test_lightsail_database_public: ) } - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", - new=lightsail_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", + new=lightsail_client, ): from prowler.providers.aws.services.lightsail.lightsail_database_public.lightsail_database_public import ( lightsail_database_public, diff --git a/tests/providers/aws/services/lightsail/lightsail_instance_automated_snapshots/lightsail_instance_automated_snapshots_test.py b/tests/providers/aws/services/lightsail/lightsail_instance_automated_snapshots/lightsail_instance_automated_snapshots_test.py index fda58a990e..f1d67b7996 100644 --- a/tests/providers/aws/services/lightsail/lightsail_instance_automated_snapshots/lightsail_instance_automated_snapshots_test.py +++ b/tests/providers/aws/services/lightsail/lightsail_instance_automated_snapshots/lightsail_instance_automated_snapshots_test.py @@ -17,15 +17,12 @@ class Test_lightsail_instance_automated_snapshots: lightsail_client = MagicMock lightsail_client.instances = {} - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", - new=lightsail_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", + new=lightsail_client, ): from prowler.providers.aws.services.lightsail.lightsail_instance_automated_snapshots.lightsail_instance_automated_snapshots import ( lightsail_instance_automated_snapshots, @@ -69,15 +66,12 @@ class Test_lightsail_instance_automated_snapshots: ) } - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", - new=lightsail_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", + new=lightsail_client, ): from prowler.providers.aws.services.lightsail.lightsail_instance_automated_snapshots.lightsail_instance_automated_snapshots import ( lightsail_instance_automated_snapshots, @@ -132,15 +126,12 @@ class Test_lightsail_instance_automated_snapshots: ) } - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", - new=lightsail_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", + new=lightsail_client, ): from prowler.providers.aws.services.lightsail.lightsail_instance_automated_snapshots.lightsail_instance_automated_snapshots import ( lightsail_instance_automated_snapshots, diff --git a/tests/providers/aws/services/lightsail/lightsail_instance_public/lightsail_instance_public_test.py b/tests/providers/aws/services/lightsail/lightsail_instance_public/lightsail_instance_public_test.py index e023372e64..596ddd34c9 100644 --- a/tests/providers/aws/services/lightsail/lightsail_instance_public/lightsail_instance_public_test.py +++ b/tests/providers/aws/services/lightsail/lightsail_instance_public/lightsail_instance_public_test.py @@ -17,15 +17,12 @@ class Test_lightsail_instance_public: lightsail_client = MagicMock lightsail_client.instances = {} - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", - new=lightsail_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", + new=lightsail_client, ): from prowler.providers.aws.services.lightsail.lightsail_instance_public.lightsail_instance_public import ( lightsail_instance_public, @@ -69,15 +66,12 @@ class Test_lightsail_instance_public: ) } - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", - new=lightsail_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", + new=lightsail_client, ): from prowler.providers.aws.services.lightsail.lightsail_instance_public.lightsail_instance_public import ( lightsail_instance_public, diff --git a/tests/providers/aws/services/lightsail/lightsail_static_ip_unused/lightsail_static_ip_unused_test.py b/tests/providers/aws/services/lightsail/lightsail_static_ip_unused/lightsail_static_ip_unused_test.py index 4b1f58e31f..ad07d6d2b8 100644 --- a/tests/providers/aws/services/lightsail/lightsail_static_ip_unused/lightsail_static_ip_unused_test.py +++ b/tests/providers/aws/services/lightsail/lightsail_static_ip_unused/lightsail_static_ip_unused_test.py @@ -14,15 +14,12 @@ class Test_lightsail_static_ip_unused: lightsail_client = MagicMock lightsail_client.static_ips = {} - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", - new=lightsail_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", + new=lightsail_client, ): from prowler.providers.aws.services.lightsail.lightsail_static_ip_unused.lightsail_static_ip_unused import ( lightsail_static_ip_unused, @@ -48,15 +45,12 @@ class Test_lightsail_static_ip_unused: ) } - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", - new=lightsail_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", + new=lightsail_client, ): from prowler.providers.aws.services.lightsail.lightsail_static_ip_unused.lightsail_static_ip_unused import ( lightsail_static_ip_unused, @@ -94,15 +88,12 @@ class Test_lightsail_static_ip_unused: ) } - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - patch( - "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", - new=lightsail_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), patch( + "prowler.providers.aws.services.lightsail.lightsail_service.Lightsail", + new=lightsail_client, ): from prowler.providers.aws.services.lightsail.lightsail_static_ip_unused.lightsail_static_ip_unused import ( lightsail_static_ip_unused, diff --git a/tests/providers/aws/services/memorydb/memorydb_cluster_auto_minor_version_upgrades/memorydb_cluster_auto_minor_version_upgrades_test.py b/tests/providers/aws/services/memorydb/memorydb_cluster_auto_minor_version_upgrades/memorydb_cluster_auto_minor_version_upgrades_test.py index 067cf68102..3abba98ac3 100644 --- a/tests/providers/aws/services/memorydb/memorydb_cluster_auto_minor_version_upgrades/memorydb_cluster_auto_minor_version_upgrades_test.py +++ b/tests/providers/aws/services/memorydb/memorydb_cluster_auto_minor_version_upgrades/memorydb_cluster_auto_minor_version_upgrades_test.py @@ -13,15 +13,12 @@ class Test_memorydb_cluster_auto_minor_version_upgrades: memorydb_client = mock.MagicMock memorydb_client.clusters = {} - with ( - mock.patch( - "prowler.providers.aws.services.memorydb.memorydb_service.MemoryDB", - new=memorydb_client, - ), - mock.patch( - "prowler.providers.aws.services.memorydb.memorydb_cluster_auto_minor_version_upgrades.memorydb_cluster_auto_minor_version_upgrades.memorydb_client", - new=memorydb_client, - ), + with mock.patch( + "prowler.providers.aws.services.memorydb.memorydb_service.MemoryDB", + new=memorydb_client, + ), mock.patch( + "prowler.providers.aws.services.memorydb.memorydb_cluster_auto_minor_version_upgrades.memorydb_cluster_auto_minor_version_upgrades.memorydb_client", + new=memorydb_client, ): from prowler.providers.aws.services.memorydb.memorydb_cluster_auto_minor_version_upgrades.memorydb_cluster_auto_minor_version_upgrades import ( memorydb_cluster_auto_minor_version_upgrades, @@ -55,15 +52,12 @@ class Test_memorydb_cluster_auto_minor_version_upgrades: ) } - with ( - mock.patch( - "prowler.providers.aws.services.memorydb.memorydb_service.MemoryDB", - new=memorydb_client, - ), - mock.patch( - "prowler.providers.aws.services.memorydb.memorydb_cluster_auto_minor_version_upgrades.memorydb_cluster_auto_minor_version_upgrades.memorydb_client", - new=memorydb_client, - ), + with mock.patch( + "prowler.providers.aws.services.memorydb.memorydb_service.MemoryDB", + new=memorydb_client, + ), mock.patch( + "prowler.providers.aws.services.memorydb.memorydb_cluster_auto_minor_version_upgrades.memorydb_cluster_auto_minor_version_upgrades.memorydb_client", + new=memorydb_client, ): from prowler.providers.aws.services.memorydb.memorydb_cluster_auto_minor_version_upgrades.memorydb_cluster_auto_minor_version_upgrades import ( memorydb_cluster_auto_minor_version_upgrades, @@ -109,15 +103,12 @@ class Test_memorydb_cluster_auto_minor_version_upgrades: ) } - with ( - mock.patch( - "prowler.providers.aws.services.memorydb.memorydb_service.MemoryDB", - new=memorydb_client, - ), - mock.patch( - "prowler.providers.aws.services.memorydb.memorydb_cluster_auto_minor_version_upgrades.memorydb_cluster_auto_minor_version_upgrades.memorydb_client", - new=memorydb_client, - ), + with mock.patch( + "prowler.providers.aws.services.memorydb.memorydb_service.MemoryDB", + new=memorydb_client, + ), mock.patch( + "prowler.providers.aws.services.memorydb.memorydb_cluster_auto_minor_version_upgrades.memorydb_cluster_auto_minor_version_upgrades.memorydb_client", + new=memorydb_client, ): from prowler.providers.aws.services.memorydb.memorydb_cluster_auto_minor_version_upgrades.memorydb_cluster_auto_minor_version_upgrades import ( memorydb_cluster_auto_minor_version_upgrades, diff --git a/tests/providers/aws/services/mq/mq_broker_auto_minor_version_upgrades/mq_broker_auto_minor_version_upgrades_test.py b/tests/providers/aws/services/mq/mq_broker_auto_minor_version_upgrades/mq_broker_auto_minor_version_upgrades_test.py index e8ddaa082a..352a2ec5b5 100644 --- a/tests/providers/aws/services/mq/mq_broker_auto_minor_version_upgrades/mq_broker_auto_minor_version_upgrades_test.py +++ b/tests/providers/aws/services/mq/mq_broker_auto_minor_version_upgrades/mq_broker_auto_minor_version_upgrades_test.py @@ -17,15 +17,12 @@ class Test_mq_broker_auto_minor_version_upgrades: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.mq.mq_broker_auto_minor_version_upgrades.mq_broker_auto_minor_version_upgrades.mq_client", - new=MQ(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.mq.mq_broker_auto_minor_version_upgrades.mq_broker_auto_minor_version_upgrades.mq_client", + new=MQ(aws_provider), ): # Test Check from prowler.providers.aws.services.mq.mq_broker_auto_minor_version_upgrades.mq_broker_auto_minor_version_upgrades import ( @@ -61,15 +58,12 @@ class Test_mq_broker_auto_minor_version_upgrades: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.mq.mq_broker_auto_minor_version_upgrades.mq_broker_auto_minor_version_upgrades.mq_client", - new=MQ(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.mq.mq_broker_auto_minor_version_upgrades.mq_broker_auto_minor_version_upgrades.mq_client", + new=MQ(aws_provider), ): # Test Check from prowler.providers.aws.services.mq.mq_broker_auto_minor_version_upgrades.mq_broker_auto_minor_version_upgrades import ( @@ -116,15 +110,12 @@ class Test_mq_broker_auto_minor_version_upgrades: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.mq.mq_broker_auto_minor_version_upgrades.mq_broker_auto_minor_version_upgrades.mq_client", - new=MQ(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.mq.mq_broker_auto_minor_version_upgrades.mq_broker_auto_minor_version_upgrades.mq_client", + new=MQ(aws_provider), ): # Test Check from prowler.providers.aws.services.mq.mq_broker_auto_minor_version_upgrades.mq_broker_auto_minor_version_upgrades import ( diff --git a/tests/providers/aws/services/mq/mq_broker_logging_enabled/mq_broker_logging_enabled_test.py b/tests/providers/aws/services/mq/mq_broker_logging_enabled/mq_broker_logging_enabled_test.py index 148fdec872..831035ca22 100644 --- a/tests/providers/aws/services/mq/mq_broker_logging_enabled/mq_broker_logging_enabled_test.py +++ b/tests/providers/aws/services/mq/mq_broker_logging_enabled/mq_broker_logging_enabled_test.py @@ -17,15 +17,12 @@ class Test_mq_broker_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.mq.mq_broker_logging_enabled.mq_broker_logging_enabled.mq_client", - new=MQ(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.mq.mq_broker_logging_enabled.mq_broker_logging_enabled.mq_client", + new=MQ(aws_provider), ): # Test Check from prowler.providers.aws.services.mq.mq_broker_logging_enabled.mq_broker_logging_enabled import ( @@ -65,15 +62,12 @@ class Test_mq_broker_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.mq.mq_broker_logging_enabled.mq_broker_logging_enabled.mq_client", - new=MQ(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.mq.mq_broker_logging_enabled.mq_broker_logging_enabled.mq_client", + new=MQ(aws_provider), ): # Test Check from prowler.providers.aws.services.mq.mq_broker_logging_enabled.mq_broker_logging_enabled import ( @@ -124,15 +118,12 @@ class Test_mq_broker_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.mq.mq_broker_logging_enabled.mq_broker_logging_enabled.mq_client", - new=MQ(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.mq.mq_broker_logging_enabled.mq_broker_logging_enabled.mq_client", + new=MQ(aws_provider), ): # Test Check from prowler.providers.aws.services.mq.mq_broker_logging_enabled.mq_broker_logging_enabled import ( @@ -184,15 +175,12 @@ class Test_mq_broker_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.mq.mq_broker_logging_enabled.mq_broker_logging_enabled.mq_client", - new=MQ(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.mq.mq_broker_logging_enabled.mq_broker_logging_enabled.mq_client", + new=MQ(aws_provider), ): # Test Check from prowler.providers.aws.services.mq.mq_broker_logging_enabled.mq_broker_logging_enabled import ( @@ -244,15 +232,12 @@ class Test_mq_broker_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.mq.mq_broker_logging_enabled.mq_broker_logging_enabled.mq_client", - new=MQ(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.mq.mq_broker_logging_enabled.mq_broker_logging_enabled.mq_client", + new=MQ(aws_provider), ): # Test Check from prowler.providers.aws.services.mq.mq_broker_logging_enabled.mq_broker_logging_enabled import ( diff --git a/tests/providers/aws/services/mq/mq_broker_not_publicly_accessible/mq_broker_not_publicly_accessible_test.py b/tests/providers/aws/services/mq/mq_broker_not_publicly_accessible/mq_broker_not_publicly_accessible_test.py index fdff3a4f7a..5231d4fbcd 100644 --- a/tests/providers/aws/services/mq/mq_broker_not_publicly_accessible/mq_broker_not_publicly_accessible_test.py +++ b/tests/providers/aws/services/mq/mq_broker_not_publicly_accessible/mq_broker_not_publicly_accessible_test.py @@ -17,15 +17,12 @@ class Test_mq_broker_not_publicly_accessible: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.mq.mq_broker_not_publicly_accessible.mq_broker_not_publicly_accessible.mq_client", - new=MQ(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.mq.mq_broker_not_publicly_accessible.mq_broker_not_publicly_accessible.mq_client", + new=MQ(aws_provider), ): # Test Check from prowler.providers.aws.services.mq.mq_broker_not_publicly_accessible.mq_broker_not_publicly_accessible import ( @@ -61,15 +58,12 @@ class Test_mq_broker_not_publicly_accessible: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.mq.mq_broker_not_publicly_accessible.mq_broker_not_publicly_accessible.mq_client", - new=MQ(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.mq.mq_broker_not_publicly_accessible.mq_broker_not_publicly_accessible.mq_client", + new=MQ(aws_provider), ): # Test Check from prowler.providers.aws.services.mq.mq_broker_not_publicly_accessible.mq_broker_not_publicly_accessible import ( @@ -116,15 +110,12 @@ class Test_mq_broker_not_publicly_accessible: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.mq.mq_broker_not_publicly_accessible.mq_broker_not_publicly_accessible.mq_client", - new=MQ(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.mq.mq_broker_not_publicly_accessible.mq_broker_not_publicly_accessible.mq_client", + new=MQ(aws_provider), ): # Test Check from prowler.providers.aws.services.mq.mq_broker_not_publicly_accessible.mq_broker_not_publicly_accessible import ( diff --git a/tests/providers/aws/services/neptune/neptune_cluster_copy_tags_to_snapshots/neptune_cluster_copy_tags_to_snapshots_test.py b/tests/providers/aws/services/neptune/neptune_cluster_copy_tags_to_snapshots/neptune_cluster_copy_tags_to_snapshots_test.py index 1e4aa805d7..98ce7bc49b 100644 --- a/tests/providers/aws/services/neptune/neptune_cluster_copy_tags_to_snapshots/neptune_cluster_copy_tags_to_snapshots_test.py +++ b/tests/providers/aws/services/neptune/neptune_cluster_copy_tags_to_snapshots/neptune_cluster_copy_tags_to_snapshots_test.py @@ -9,15 +9,12 @@ class Test_neptune_cluster_copy_tags_to_snapshots: neptune_client = mock.MagicMock neptune_client.clusters = {} - with ( - mock.patch( - "prowler.providers.aws.services.neptune.neptune_service.Neptune", - new=neptune_client, - ), - mock.patch( - "prowler.providers.aws.services.neptune.neptune_cluster_copy_tags_to_snapshots.neptune_cluster_copy_tags_to_snapshots.neptune_client", - new=neptune_client, - ), + with mock.patch( + "prowler.providers.aws.services.neptune.neptune_service.Neptune", + new=neptune_client, + ), mock.patch( + "prowler.providers.aws.services.neptune.neptune_cluster_copy_tags_to_snapshots.neptune_cluster_copy_tags_to_snapshots.neptune_client", + new=neptune_client, ): # Test Check from prowler.providers.aws.services.neptune.neptune_cluster_copy_tags_to_snapshots.neptune_cluster_copy_tags_to_snapshots import ( @@ -50,15 +47,12 @@ class Test_neptune_cluster_copy_tags_to_snapshots: ) } - with ( - mock.patch( - "prowler.providers.aws.services.neptune.neptune_service.Neptune", - new=neptune_client, - ), - mock.patch( - "prowler.providers.aws.services.neptune.neptune_cluster_copy_tags_to_snapshots.neptune_cluster_copy_tags_to_snapshots.neptune_client", - new=neptune_client, - ), + with mock.patch( + "prowler.providers.aws.services.neptune.neptune_service.Neptune", + new=neptune_client, + ), mock.patch( + "prowler.providers.aws.services.neptune.neptune_cluster_copy_tags_to_snapshots.neptune_cluster_copy_tags_to_snapshots.neptune_client", + new=neptune_client, ): # Test Check from prowler.providers.aws.services.neptune.neptune_cluster_copy_tags_to_snapshots.neptune_cluster_copy_tags_to_snapshots import ( @@ -100,15 +94,12 @@ class Test_neptune_cluster_copy_tags_to_snapshots: ) } - with ( - mock.patch( - "prowler.providers.aws.services.neptune.neptune_service.Neptune", - new=neptune_client, - ), - mock.patch( - "prowler.providers.aws.services.neptune.neptune_cluster_copy_tags_to_snapshots.neptune_cluster_copy_tags_to_snapshots.neptune_client", - new=neptune_client, - ), + with mock.patch( + "prowler.providers.aws.services.neptune.neptune_service.Neptune", + new=neptune_client, + ), mock.patch( + "prowler.providers.aws.services.neptune.neptune_cluster_copy_tags_to_snapshots.neptune_cluster_copy_tags_to_snapshots.neptune_client", + new=neptune_client, ): # Test Check from prowler.providers.aws.services.neptune.neptune_cluster_copy_tags_to_snapshots.neptune_cluster_copy_tags_to_snapshots import ( diff --git a/tests/providers/aws/services/neptune/neptune_cluster_integration_cloudwatch_logs/neptune_cluster_integration_cloudwatch_logs_test.py b/tests/providers/aws/services/neptune/neptune_cluster_integration_cloudwatch_logs/neptune_cluster_integration_cloudwatch_logs_test.py index 5b6c149c49..1d3dc746f5 100644 --- a/tests/providers/aws/services/neptune/neptune_cluster_integration_cloudwatch_logs/neptune_cluster_integration_cloudwatch_logs_test.py +++ b/tests/providers/aws/services/neptune/neptune_cluster_integration_cloudwatch_logs/neptune_cluster_integration_cloudwatch_logs_test.py @@ -175,15 +175,12 @@ class Test_neptune_cluster_integration_cloudwatch_logs: ) } - with ( - mock.patch( - "prowler.providers.aws.services.neptune.neptune_service.Neptune", - new=neptune_client, - ), - mock.patch( - "prowler.providers.aws.services.neptune.neptune_cluster_integration_cloudwatch_logs.neptune_cluster_integration_cloudwatch_logs.neptune_client", - new=neptune_client, - ), + with mock.patch( + "prowler.providers.aws.services.neptune.neptune_service.Neptune", + new=neptune_client, + ), mock.patch( + "prowler.providers.aws.services.neptune.neptune_cluster_integration_cloudwatch_logs.neptune_cluster_integration_cloudwatch_logs.neptune_client", + new=neptune_client, ): # Test Check from prowler.providers.aws.services.neptune.neptune_cluster_integration_cloudwatch_logs.neptune_cluster_integration_cloudwatch_logs import ( diff --git a/tests/providers/aws/services/neptune/neptune_cluster_public_snapshot/neptune_cluster_public_snapshot_fixer_test.py b/tests/providers/aws/services/neptune/neptune_cluster_public_snapshot/neptune_cluster_public_snapshot_fixer_test.py index 52d98e2322..137b55c791 100644 --- a/tests/providers/aws/services/neptune/neptune_cluster_public_snapshot/neptune_cluster_public_snapshot_fixer_test.py +++ b/tests/providers/aws/services/neptune/neptune_cluster_public_snapshot/neptune_cluster_public_snapshot_fixer_test.py @@ -50,15 +50,12 @@ class Test_neptune_cluster_public_snapshot_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.neptune.neptune_cluster_public_snapshot.neptune_cluster_public_snapshot_fixer.neptune_client", - new=Neptune(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.neptune.neptune_cluster_public_snapshot.neptune_cluster_public_snapshot_fixer.neptune_client", + new=Neptune(aws_provider), ): from prowler.providers.aws.services.neptune.neptune_cluster_public_snapshot.neptune_cluster_public_snapshot_fixer import ( fixer, @@ -76,15 +73,12 @@ class Test_neptune_cluster_public_snapshot_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.neptune.neptune_cluster_public_snapshot.neptune_cluster_public_snapshot_fixer.neptune_client", - new=Neptune(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.neptune.neptune_cluster_public_snapshot.neptune_cluster_public_snapshot_fixer.neptune_client", + new=Neptune(aws_provider), ): from prowler.providers.aws.services.neptune.neptune_cluster_public_snapshot.neptune_cluster_public_snapshot_fixer import ( fixer, diff --git a/tests/providers/aws/services/neptune/neptune_cluster_public_snapshot/neptune_cluster_public_snapshot_test.py b/tests/providers/aws/services/neptune/neptune_cluster_public_snapshot/neptune_cluster_public_snapshot_test.py index 30eca9e78a..caecd222ab 100644 --- a/tests/providers/aws/services/neptune/neptune_cluster_public_snapshot/neptune_cluster_public_snapshot_test.py +++ b/tests/providers/aws/services/neptune/neptune_cluster_public_snapshot/neptune_cluster_public_snapshot_test.py @@ -18,15 +18,12 @@ class Test_neptune_cluster_public_snapshot: neptune_client.clusters = {} neptune_client.db_cluster_snapshots = [] - with ( - mock.patch( - "prowler.providers.aws.services.neptune.neptune_service.Neptune", - new=neptune_client, - ), - mock.patch( - "prowler.providers.aws.services.neptune.neptune_cluster_public_snapshot.neptune_cluster_public_snapshot.neptune_client", - new=neptune_client, - ), + with mock.patch( + "prowler.providers.aws.services.neptune.neptune_service.Neptune", + new=neptune_client, + ), mock.patch( + "prowler.providers.aws.services.neptune.neptune_cluster_public_snapshot.neptune_cluster_public_snapshot.neptune_client", + new=neptune_client, ): from prowler.providers.aws.services.neptune.neptune_cluster_public_snapshot.neptune_cluster_public_snapshot import ( neptune_cluster_public_snapshot, @@ -64,15 +61,12 @@ class Test_neptune_cluster_public_snapshot: ) ] - with ( - mock.patch( - "prowler.providers.aws.services.neptune.neptune_service.Neptune", - new=neptune_client, - ), - mock.patch( - "prowler.providers.aws.services.neptune.neptune_cluster_public_snapshot.neptune_cluster_public_snapshot.neptune_client", - new=neptune_client, - ), + with mock.patch( + "prowler.providers.aws.services.neptune.neptune_service.Neptune", + new=neptune_client, + ), mock.patch( + "prowler.providers.aws.services.neptune.neptune_cluster_public_snapshot.neptune_cluster_public_snapshot.neptune_client", + new=neptune_client, ): from prowler.providers.aws.services.neptune.neptune_cluster_public_snapshot.neptune_cluster_public_snapshot import ( neptune_cluster_public_snapshot, @@ -123,15 +117,12 @@ class Test_neptune_cluster_public_snapshot: ) ] - with ( - mock.patch( - "prowler.providers.aws.services.neptune.neptune_service.Neptune", - new=neptune_client, - ), - mock.patch( - "prowler.providers.aws.services.neptune.neptune_cluster_public_snapshot.neptune_cluster_public_snapshot.neptune_client", - new=neptune_client, - ), + with mock.patch( + "prowler.providers.aws.services.neptune.neptune_service.Neptune", + new=neptune_client, + ), mock.patch( + "prowler.providers.aws.services.neptune.neptune_cluster_public_snapshot.neptune_cluster_public_snapshot.neptune_client", + new=neptune_client, ): from prowler.providers.aws.services.neptune.neptune_cluster_public_snapshot.neptune_cluster_public_snapshot import ( neptune_cluster_public_snapshot, diff --git a/tests/providers/aws/services/neptune/neptune_cluster_snapshot_encrypted/neptune_cluster_snapshot_encrypted_test.py b/tests/providers/aws/services/neptune/neptune_cluster_snapshot_encrypted/neptune_cluster_snapshot_encrypted_test.py index c5bc7dd305..0bf8747fd0 100644 --- a/tests/providers/aws/services/neptune/neptune_cluster_snapshot_encrypted/neptune_cluster_snapshot_encrypted_test.py +++ b/tests/providers/aws/services/neptune/neptune_cluster_snapshot_encrypted/neptune_cluster_snapshot_encrypted_test.py @@ -9,15 +9,12 @@ class Test_neptune_cluster_snapshot_encrypted: neptune_client = mock.MagicMock() neptune_client.db_cluster_snapshots = [] - with ( - mock.patch( - "prowler.providers.aws.services.neptune.neptune_service.Neptune", - new=neptune_client, - ), - mock.patch( - "prowler.providers.aws.services.neptune.neptune_cluster_snapshot_encrypted.neptune_cluster_snapshot_encrypted.neptune_client", - new=neptune_client, - ), + with mock.patch( + "prowler.providers.aws.services.neptune.neptune_service.Neptune", + new=neptune_client, + ), mock.patch( + "prowler.providers.aws.services.neptune.neptune_cluster_snapshot_encrypted.neptune_cluster_snapshot_encrypted.neptune_client", + new=neptune_client, ): # Test Check from prowler.providers.aws.services.neptune.neptune_cluster_snapshot_encrypted.neptune_cluster_snapshot_encrypted import ( @@ -43,15 +40,12 @@ class Test_neptune_cluster_snapshot_encrypted: ) ] - with ( - mock.patch( - "prowler.providers.aws.services.neptune.neptune_service.Neptune", - new=neptune_client, - ), - mock.patch( - "prowler.providers.aws.services.neptune.neptune_cluster_snapshot_encrypted.neptune_cluster_snapshot_encrypted.neptune_client", - new=neptune_client, - ), + with mock.patch( + "prowler.providers.aws.services.neptune.neptune_service.Neptune", + new=neptune_client, + ), mock.patch( + "prowler.providers.aws.services.neptune.neptune_cluster_snapshot_encrypted.neptune_cluster_snapshot_encrypted.neptune_client", + new=neptune_client, ): # Test Check from prowler.providers.aws.services.neptune.neptune_cluster_snapshot_encrypted.neptune_cluster_snapshot_encrypted import ( @@ -86,15 +80,12 @@ class Test_neptune_cluster_snapshot_encrypted: ) ] - with ( - mock.patch( - "prowler.providers.aws.services.neptune.neptune_service.Neptune", - new=neptune_client, - ), - mock.patch( - "prowler.providers.aws.services.neptune.neptune_cluster_snapshot_encrypted.neptune_cluster_snapshot_encrypted.neptune_client", - new=neptune_client, - ), + with mock.patch( + "prowler.providers.aws.services.neptune.neptune_service.Neptune", + new=neptune_client, + ), mock.patch( + "prowler.providers.aws.services.neptune.neptune_cluster_snapshot_encrypted.neptune_cluster_snapshot_encrypted.neptune_client", + new=neptune_client, ): # Test Check from prowler.providers.aws.services.neptune.neptune_cluster_snapshot_encrypted.neptune_cluster_snapshot_encrypted import ( diff --git a/tests/providers/aws/services/neptune/neptune_cluster_uses_public_subnet/neptune_cluster_uses_public_subnet_test.py b/tests/providers/aws/services/neptune/neptune_cluster_uses_public_subnet/neptune_cluster_uses_public_subnet_test.py index c7d2ec68c9..007f4eea96 100644 --- a/tests/providers/aws/services/neptune/neptune_cluster_uses_public_subnet/neptune_cluster_uses_public_subnet_test.py +++ b/tests/providers/aws/services/neptune/neptune_cluster_uses_public_subnet/neptune_cluster_uses_public_subnet_test.py @@ -35,19 +35,15 @@ class Test_neptune_cluster_uses_public_subnet: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.neptune.neptune_cluster_uses_public_subnet.neptune_cluster_uses_public_subnet.neptune_client", - new=Neptune(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.neptune.neptune_cluster_uses_public_subnet.neptune_cluster_uses_public_subnet.vpc_client", - new=vpc_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.neptune.neptune_cluster_uses_public_subnet.neptune_cluster_uses_public_subnet.neptune_client", + new=Neptune(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.neptune.neptune_cluster_uses_public_subnet.neptune_cluster_uses_public_subnet.vpc_client", + new=vpc_client, ): from prowler.providers.aws.services.neptune.neptune_cluster_uses_public_subnet.neptune_cluster_uses_public_subnet import ( neptune_cluster_uses_public_subnet, @@ -111,19 +107,15 @@ class Test_neptune_cluster_uses_public_subnet: cluster_id = cluster["DbClusterResourceId"] aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.neptune.neptune_cluster_uses_public_subnet.neptune_cluster_uses_public_subnet.neptune_client", - new=Neptune(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.neptune.neptune_cluster_uses_public_subnet.neptune_cluster_uses_public_subnet.vpc_client", - new=vpc_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.neptune.neptune_cluster_uses_public_subnet.neptune_cluster_uses_public_subnet.neptune_client", + new=Neptune(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.neptune.neptune_cluster_uses_public_subnet.neptune_cluster_uses_public_subnet.vpc_client", + new=vpc_client, ): from prowler.providers.aws.services.neptune.neptune_cluster_uses_public_subnet.neptune_cluster_uses_public_subnet import ( neptune_cluster_uses_public_subnet, @@ -196,19 +188,15 @@ class Test_neptune_cluster_uses_public_subnet: cluster_id = cluster["DbClusterResourceId"] aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.neptune.neptune_cluster_uses_public_subnet.neptune_cluster_uses_public_subnet.neptune_client", - new=Neptune(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.neptune.neptune_cluster_uses_public_subnet.neptune_cluster_uses_public_subnet.vpc_client", - new=vpc_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.neptune.neptune_cluster_uses_public_subnet.neptune_cluster_uses_public_subnet.neptune_client", + new=Neptune(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.neptune.neptune_cluster_uses_public_subnet.neptune_cluster_uses_public_subnet.vpc_client", + new=vpc_client, ): from prowler.providers.aws.services.neptune.neptune_cluster_uses_public_subnet.neptune_cluster_uses_public_subnet import ( neptune_cluster_uses_public_subnet, diff --git a/tests/providers/aws/services/opensearch/opensearch_service_domains_access_control_enabled/opensearch_service_domains_access_control_enabled_test.py b/tests/providers/aws/services/opensearch/opensearch_service_domains_access_control_enabled/opensearch_service_domains_access_control_enabled_test.py index e9e8022df2..95f9af0208 100644 --- a/tests/providers/aws/services/opensearch/opensearch_service_domains_access_control_enabled/opensearch_service_domains_access_control_enabled_test.py +++ b/tests/providers/aws/services/opensearch/opensearch_service_domains_access_control_enabled/opensearch_service_domains_access_control_enabled_test.py @@ -21,15 +21,12 @@ class Test_opensearch_service_domains_access_control_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_access_control_enabled.opensearch_service_domains_access_control_enabled.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_access_control_enabled.opensearch_service_domains_access_control_enabled.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_access_control_enabled.opensearch_service_domains_access_control_enabled import ( opensearch_service_domains_access_control_enabled, @@ -53,15 +50,12 @@ class Test_opensearch_service_domains_access_control_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_access_control_enabled.opensearch_service_domains_access_control_enabled.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_access_control_enabled.opensearch_service_domains_access_control_enabled.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_access_control_enabled.opensearch_service_domains_access_control_enabled import ( opensearch_service_domains_access_control_enabled, @@ -95,15 +89,12 @@ class Test_opensearch_service_domains_access_control_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_access_control_enabled.opensearch_service_domains_access_control_enabled.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_access_control_enabled.opensearch_service_domains_access_control_enabled.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_access_control_enabled.opensearch_service_domains_access_control_enabled import ( opensearch_service_domains_access_control_enabled, diff --git a/tests/providers/aws/services/opensearch/opensearch_service_domains_audit_logging_enabled/opensearch_service_domains_audit_logging_enabled_test.py b/tests/providers/aws/services/opensearch/opensearch_service_domains_audit_logging_enabled/opensearch_service_domains_audit_logging_enabled_test.py index 42ab1384da..1c40fe5ef3 100644 --- a/tests/providers/aws/services/opensearch/opensearch_service_domains_audit_logging_enabled/opensearch_service_domains_audit_logging_enabled_test.py +++ b/tests/providers/aws/services/opensearch/opensearch_service_domains_audit_logging_enabled/opensearch_service_domains_audit_logging_enabled_test.py @@ -21,15 +21,12 @@ class Test_opensearch_service_domains_audit_logging_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_audit_logging_enabled.opensearch_service_domains_audit_logging_enabled.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_audit_logging_enabled.opensearch_service_domains_audit_logging_enabled.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_audit_logging_enabled.opensearch_service_domains_audit_logging_enabled import ( opensearch_service_domains_audit_logging_enabled, @@ -56,15 +53,12 @@ class Test_opensearch_service_domains_audit_logging_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_audit_logging_enabled.opensearch_service_domains_audit_logging_enabled.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_audit_logging_enabled.opensearch_service_domains_audit_logging_enabled.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_audit_logging_enabled.opensearch_service_domains_audit_logging_enabled import ( opensearch_service_domains_audit_logging_enabled, @@ -102,15 +96,12 @@ class Test_opensearch_service_domains_audit_logging_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_audit_logging_enabled.opensearch_service_domains_audit_logging_enabled.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_audit_logging_enabled.opensearch_service_domains_audit_logging_enabled.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_audit_logging_enabled.opensearch_service_domains_audit_logging_enabled import ( opensearch_service_domains_audit_logging_enabled, diff --git a/tests/providers/aws/services/opensearch/opensearch_service_domains_cloudwatch_logging_enabled/opensearch_service_domains_cloudwatch_logging_enabled_test.py b/tests/providers/aws/services/opensearch/opensearch_service_domains_cloudwatch_logging_enabled/opensearch_service_domains_cloudwatch_logging_enabled_test.py index 470da96032..a04d4eb714 100644 --- a/tests/providers/aws/services/opensearch/opensearch_service_domains_cloudwatch_logging_enabled/opensearch_service_domains_cloudwatch_logging_enabled_test.py +++ b/tests/providers/aws/services/opensearch/opensearch_service_domains_cloudwatch_logging_enabled/opensearch_service_domains_cloudwatch_logging_enabled_test.py @@ -21,15 +21,12 @@ class Test_opensearch_service_domains_cloudwatch_logging_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_service_domains_cloudwatch_logging_enabled import ( opensearch_service_domains_cloudwatch_logging_enabled, @@ -57,15 +54,12 @@ class Test_opensearch_service_domains_cloudwatch_logging_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_service_domains_cloudwatch_logging_enabled import ( opensearch_service_domains_cloudwatch_logging_enabled, @@ -103,15 +97,12 @@ class Test_opensearch_service_domains_cloudwatch_logging_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_service_domains_cloudwatch_logging_enabled import ( opensearch_service_domains_cloudwatch_logging_enabled, @@ -149,15 +140,12 @@ class Test_opensearch_service_domains_cloudwatch_logging_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_service_domains_cloudwatch_logging_enabled import ( opensearch_service_domains_cloudwatch_logging_enabled, @@ -199,15 +187,12 @@ class Test_opensearch_service_domains_cloudwatch_logging_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_cloudwatch_logging_enabled.opensearch_service_domains_cloudwatch_logging_enabled import ( opensearch_service_domains_cloudwatch_logging_enabled, diff --git a/tests/providers/aws/services/opensearch/opensearch_service_domains_encryption_at_rest_enabled/opensearch_service_domains_encryption_at_rest_enabled_test.py b/tests/providers/aws/services/opensearch/opensearch_service_domains_encryption_at_rest_enabled/opensearch_service_domains_encryption_at_rest_enabled_test.py index ef38c8363c..d1857bff1e 100644 --- a/tests/providers/aws/services/opensearch/opensearch_service_domains_encryption_at_rest_enabled/opensearch_service_domains_encryption_at_rest_enabled_test.py +++ b/tests/providers/aws/services/opensearch/opensearch_service_domains_encryption_at_rest_enabled/opensearch_service_domains_encryption_at_rest_enabled_test.py @@ -21,15 +21,12 @@ class Test_opensearch_service_domains_encryption_at_rest_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_encryption_at_rest_enabled.opensearch_service_domains_encryption_at_rest_enabled.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_encryption_at_rest_enabled.opensearch_service_domains_encryption_at_rest_enabled.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_encryption_at_rest_enabled.opensearch_service_domains_encryption_at_rest_enabled import ( opensearch_service_domains_encryption_at_rest_enabled, @@ -52,15 +49,12 @@ class Test_opensearch_service_domains_encryption_at_rest_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_encryption_at_rest_enabled.opensearch_service_domains_encryption_at_rest_enabled.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_encryption_at_rest_enabled.opensearch_service_domains_encryption_at_rest_enabled.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_encryption_at_rest_enabled.opensearch_service_domains_encryption_at_rest_enabled import ( opensearch_service_domains_encryption_at_rest_enabled, @@ -93,15 +87,12 @@ class Test_opensearch_service_domains_encryption_at_rest_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_encryption_at_rest_enabled.opensearch_service_domains_encryption_at_rest_enabled.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_encryption_at_rest_enabled.opensearch_service_domains_encryption_at_rest_enabled.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_encryption_at_rest_enabled.opensearch_service_domains_encryption_at_rest_enabled import ( opensearch_service_domains_encryption_at_rest_enabled, diff --git a/tests/providers/aws/services/opensearch/opensearch_service_domains_fault_tolerant_data_nodes/opensearch_service_domains_fault_tolerant_data_nodes_data_nodes_test.py b/tests/providers/aws/services/opensearch/opensearch_service_domains_fault_tolerant_data_nodes/opensearch_service_domains_fault_tolerant_data_nodes_data_nodes_test.py index 16da6f67ff..4a1ac5355c 100644 --- a/tests/providers/aws/services/opensearch/opensearch_service_domains_fault_tolerant_data_nodes/opensearch_service_domains_fault_tolerant_data_nodes_data_nodes_test.py +++ b/tests/providers/aws/services/opensearch/opensearch_service_domains_fault_tolerant_data_nodes/opensearch_service_domains_fault_tolerant_data_nodes_data_nodes_test.py @@ -21,15 +21,12 @@ class Test_opensearch_service_domains_fault_tolerant_data_nodes: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_service_domains_fault_tolerant_data_nodes import ( opensearch_service_domains_fault_tolerant_data_nodes, @@ -53,15 +50,12 @@ class Test_opensearch_service_domains_fault_tolerant_data_nodes: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_service_domains_fault_tolerant_data_nodes import ( opensearch_service_domains_fault_tolerant_data_nodes, @@ -95,15 +89,12 @@ class Test_opensearch_service_domains_fault_tolerant_data_nodes: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_service_domains_fault_tolerant_data_nodes import ( opensearch_service_domains_fault_tolerant_data_nodes, @@ -137,15 +128,12 @@ class Test_opensearch_service_domains_fault_tolerant_data_nodes: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_service_domains_fault_tolerant_data_nodes import ( opensearch_service_domains_fault_tolerant_data_nodes, @@ -179,15 +167,12 @@ class Test_opensearch_service_domains_fault_tolerant_data_nodes: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_data_nodes.opensearch_service_domains_fault_tolerant_data_nodes import ( opensearch_service_domains_fault_tolerant_data_nodes, diff --git a/tests/providers/aws/services/opensearch/opensearch_service_domains_fault_tolerant_master_nodes/opensearch_service_domains_fault_tolerant_master_nodes_test.py b/tests/providers/aws/services/opensearch/opensearch_service_domains_fault_tolerant_master_nodes/opensearch_service_domains_fault_tolerant_master_nodes_test.py index d8f7f9ca3f..56ab8ab596 100644 --- a/tests/providers/aws/services/opensearch/opensearch_service_domains_fault_tolerant_master_nodes/opensearch_service_domains_fault_tolerant_master_nodes_test.py +++ b/tests/providers/aws/services/opensearch/opensearch_service_domains_fault_tolerant_master_nodes/opensearch_service_domains_fault_tolerant_master_nodes_test.py @@ -17,15 +17,12 @@ class Test_opensearch_service_domains_fault_tolerant_master_nodes: OpenSearchService, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_master_nodes.opensearch_service_domains_fault_tolerant_master_nodes.opensearch_client", - new=OpenSearchService(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_master_nodes.opensearch_service_domains_fault_tolerant_master_nodes.opensearch_client", + new=OpenSearchService(aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_master_nodes.opensearch_service_domains_fault_tolerant_master_nodes import ( opensearch_service_domains_fault_tolerant_master_nodes, @@ -51,15 +48,12 @@ class Test_opensearch_service_domains_fault_tolerant_master_nodes: OpenSearchService, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_master_nodes.opensearch_service_domains_fault_tolerant_master_nodes.opensearch_client", - new=OpenSearchService(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_master_nodes.opensearch_service_domains_fault_tolerant_master_nodes.opensearch_client", + new=OpenSearchService(aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_master_nodes.opensearch_service_domains_fault_tolerant_master_nodes import ( opensearch_service_domains_fault_tolerant_master_nodes, @@ -98,15 +92,12 @@ class Test_opensearch_service_domains_fault_tolerant_master_nodes: OpenSearchService, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_master_nodes.opensearch_service_domains_fault_tolerant_master_nodes.opensearch_client", - new=OpenSearchService(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_master_nodes.opensearch_service_domains_fault_tolerant_master_nodes.opensearch_client", + new=OpenSearchService(aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_master_nodes.opensearch_service_domains_fault_tolerant_master_nodes import ( opensearch_service_domains_fault_tolerant_master_nodes, @@ -144,15 +135,12 @@ class Test_opensearch_service_domains_fault_tolerant_master_nodes: OpenSearchService, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_master_nodes.opensearch_service_domains_fault_tolerant_master_nodes.opensearch_client", - new=OpenSearchService(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_master_nodes.opensearch_service_domains_fault_tolerant_master_nodes.opensearch_client", + new=OpenSearchService(aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_fault_tolerant_master_nodes.opensearch_service_domains_fault_tolerant_master_nodes import ( opensearch_service_domains_fault_tolerant_master_nodes, diff --git a/tests/providers/aws/services/opensearch/opensearch_service_domains_https_communications_enforced/opensearch_service_domains_https_communications_enforced_test.py b/tests/providers/aws/services/opensearch/opensearch_service_domains_https_communications_enforced/opensearch_service_domains_https_communications_enforced_test.py index f84df30dab..6f499347c5 100644 --- a/tests/providers/aws/services/opensearch/opensearch_service_domains_https_communications_enforced/opensearch_service_domains_https_communications_enforced_test.py +++ b/tests/providers/aws/services/opensearch/opensearch_service_domains_https_communications_enforced/opensearch_service_domains_https_communications_enforced_test.py @@ -21,15 +21,12 @@ class Test_opensearch_service_domains_https_communications_enforced: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_https_communications_enforced.opensearch_service_domains_https_communications_enforced.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_https_communications_enforced.opensearch_service_domains_https_communications_enforced.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_https_communications_enforced.opensearch_service_domains_https_communications_enforced import ( opensearch_service_domains_https_communications_enforced, @@ -54,15 +51,12 @@ class Test_opensearch_service_domains_https_communications_enforced: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_https_communications_enforced.opensearch_service_domains_https_communications_enforced.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_https_communications_enforced.opensearch_service_domains_https_communications_enforced.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_https_communications_enforced.opensearch_service_domains_https_communications_enforced import ( opensearch_service_domains_https_communications_enforced, @@ -97,15 +91,12 @@ class Test_opensearch_service_domains_https_communications_enforced: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_https_communications_enforced.opensearch_service_domains_https_communications_enforced.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_https_communications_enforced.opensearch_service_domains_https_communications_enforced.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_https_communications_enforced.opensearch_service_domains_https_communications_enforced import ( opensearch_service_domains_https_communications_enforced, diff --git a/tests/providers/aws/services/opensearch/opensearch_service_domains_internal_user_database_enabled/opensearch_service_domains_internal_user_database_enabled_test.py b/tests/providers/aws/services/opensearch/opensearch_service_domains_internal_user_database_enabled/opensearch_service_domains_internal_user_database_enabled_test.py index 9e4a95112b..6c1f527fb1 100644 --- a/tests/providers/aws/services/opensearch/opensearch_service_domains_internal_user_database_enabled/opensearch_service_domains_internal_user_database_enabled_test.py +++ b/tests/providers/aws/services/opensearch/opensearch_service_domains_internal_user_database_enabled/opensearch_service_domains_internal_user_database_enabled_test.py @@ -21,15 +21,12 @@ class Test_opensearch_service_domains_internal_user_database_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_internal_user_database_enabled.opensearch_service_domains_internal_user_database_enabled.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_internal_user_database_enabled.opensearch_service_domains_internal_user_database_enabled.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_internal_user_database_enabled.opensearch_service_domains_internal_user_database_enabled import ( opensearch_service_domains_internal_user_database_enabled, @@ -52,15 +49,12 @@ class Test_opensearch_service_domains_internal_user_database_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_internal_user_database_enabled.opensearch_service_domains_internal_user_database_enabled.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_internal_user_database_enabled.opensearch_service_domains_internal_user_database_enabled.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_internal_user_database_enabled.opensearch_service_domains_internal_user_database_enabled import ( opensearch_service_domains_internal_user_database_enabled, @@ -93,15 +87,12 @@ class Test_opensearch_service_domains_internal_user_database_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_internal_user_database_enabled.opensearch_service_domains_internal_user_database_enabled.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_internal_user_database_enabled.opensearch_service_domains_internal_user_database_enabled.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_internal_user_database_enabled.opensearch_service_domains_internal_user_database_enabled import ( opensearch_service_domains_internal_user_database_enabled, diff --git a/tests/providers/aws/services/opensearch/opensearch_service_domains_node_to_node_encryption_enabled/opensearch_service_domains_node_to_node_encryption_enabled_test.py b/tests/providers/aws/services/opensearch/opensearch_service_domains_node_to_node_encryption_enabled/opensearch_service_domains_node_to_node_encryption_enabled_test.py index 96b3e0a867..83199814a2 100644 --- a/tests/providers/aws/services/opensearch/opensearch_service_domains_node_to_node_encryption_enabled/opensearch_service_domains_node_to_node_encryption_enabled_test.py +++ b/tests/providers/aws/services/opensearch/opensearch_service_domains_node_to_node_encryption_enabled/opensearch_service_domains_node_to_node_encryption_enabled_test.py @@ -21,15 +21,12 @@ class Test_opensearch_service_domains_node_to_node_encryption_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_node_to_node_encryption_enabled.opensearch_service_domains_node_to_node_encryption_enabled.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_node_to_node_encryption_enabled.opensearch_service_domains_node_to_node_encryption_enabled.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_node_to_node_encryption_enabled.opensearch_service_domains_node_to_node_encryption_enabled import ( opensearch_service_domains_node_to_node_encryption_enabled, @@ -52,15 +49,12 @@ class Test_opensearch_service_domains_node_to_node_encryption_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_node_to_node_encryption_enabled.opensearch_service_domains_node_to_node_encryption_enabled.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_node_to_node_encryption_enabled.opensearch_service_domains_node_to_node_encryption_enabled.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_node_to_node_encryption_enabled.opensearch_service_domains_node_to_node_encryption_enabled import ( opensearch_service_domains_node_to_node_encryption_enabled, @@ -93,15 +87,12 @@ class Test_opensearch_service_domains_node_to_node_encryption_enabled: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_node_to_node_encryption_enabled.opensearch_service_domains_node_to_node_encryption_enabled.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_node_to_node_encryption_enabled.opensearch_service_domains_node_to_node_encryption_enabled.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_node_to_node_encryption_enabled.opensearch_service_domains_node_to_node_encryption_enabled import ( opensearch_service_domains_node_to_node_encryption_enabled, diff --git a/tests/providers/aws/services/opensearch/opensearch_service_domains_not_publicly_accessible/opensearch_service_domains_not_publicly_accessible_fixer_test.py b/tests/providers/aws/services/opensearch/opensearch_service_domains_not_publicly_accessible/opensearch_service_domains_not_publicly_accessible_fixer_test.py index b68c4729d1..01bb37f03d 100644 --- a/tests/providers/aws/services/opensearch/opensearch_service_domains_not_publicly_accessible/opensearch_service_domains_not_publicly_accessible_fixer_test.py +++ b/tests/providers/aws/services/opensearch/opensearch_service_domains_not_publicly_accessible/opensearch_service_domains_not_publicly_accessible_fixer_test.py @@ -65,15 +65,12 @@ class Test_opensearch_service_domains_not_publicly_accessible_fixer: OpenSearchService, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_not_publicly_accessible.opensearch_service_domains_not_publicly_accessible_fixer.opensearch_client", - new=OpenSearchService(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_not_publicly_accessible.opensearch_service_domains_not_publicly_accessible_fixer.opensearch_client", + new=OpenSearchService(aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_not_publicly_accessible.opensearch_service_domains_not_publicly_accessible_fixer import ( fixer, @@ -96,15 +93,12 @@ class Test_opensearch_service_domains_not_publicly_accessible_fixer: OpenSearchService, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_not_publicly_accessible.opensearch_service_domains_not_publicly_accessible_fixer.opensearch_client", - new=OpenSearchService(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_not_publicly_accessible.opensearch_service_domains_not_publicly_accessible_fixer.opensearch_client", + new=OpenSearchService(aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_not_publicly_accessible.opensearch_service_domains_not_publicly_accessible_fixer import ( fixer, @@ -127,15 +121,12 @@ class Test_opensearch_service_domains_not_publicly_accessible_fixer: OpenSearchService, ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_not_publicly_accessible.opensearch_service_domains_not_publicly_accessible_fixer.opensearch_client", - new=OpenSearchService(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_not_publicly_accessible.opensearch_service_domains_not_publicly_accessible_fixer.opensearch_client", + new=OpenSearchService(aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_not_publicly_accessible.opensearch_service_domains_not_publicly_accessible_fixer import ( fixer, diff --git a/tests/providers/aws/services/opensearch/opensearch_service_domains_updated_to_the_latest_service_software_version/opensearch_service_domains_updated_to_the_latest_service_software_version_test.py b/tests/providers/aws/services/opensearch/opensearch_service_domains_updated_to_the_latest_service_software_version/opensearch_service_domains_updated_to_the_latest_service_software_version_test.py index 041d49994c..d548e8a9a4 100644 --- a/tests/providers/aws/services/opensearch/opensearch_service_domains_updated_to_the_latest_service_software_version/opensearch_service_domains_updated_to_the_latest_service_software_version_test.py +++ b/tests/providers/aws/services/opensearch/opensearch_service_domains_updated_to_the_latest_service_software_version/opensearch_service_domains_updated_to_the_latest_service_software_version_test.py @@ -65,15 +65,12 @@ class Test_opensearch_service_domains_updated_to_the_latest_service_software_ver mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_updated_to_the_latest_service_software_version.opensearch_service_domains_updated_to_the_latest_service_software_version.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_updated_to_the_latest_service_software_version.opensearch_service_domains_updated_to_the_latest_service_software_version.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_updated_to_the_latest_service_software_version.opensearch_service_domains_updated_to_the_latest_service_software_version import ( opensearch_service_domains_updated_to_the_latest_service_software_version, @@ -95,15 +92,12 @@ class Test_opensearch_service_domains_updated_to_the_latest_service_software_ver mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_updated_to_the_latest_service_software_version.opensearch_service_domains_updated_to_the_latest_service_software_version.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_updated_to_the_latest_service_software_version.opensearch_service_domains_updated_to_the_latest_service_software_version.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_updated_to_the_latest_service_software_version.opensearch_service_domains_updated_to_the_latest_service_software_version import ( opensearch_service_domains_updated_to_the_latest_service_software_version, @@ -138,15 +132,12 @@ class Test_opensearch_service_domains_updated_to_the_latest_service_software_ver mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_updated_to_the_latest_service_software_version.opensearch_service_domains_updated_to_the_latest_service_software_version.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_updated_to_the_latest_service_software_version.opensearch_service_domains_updated_to_the_latest_service_software_version.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_updated_to_the_latest_service_software_version.opensearch_service_domains_updated_to_the_latest_service_software_version import ( opensearch_service_domains_updated_to_the_latest_service_software_version, diff --git a/tests/providers/aws/services/opensearch/opensearch_service_domains_use_cognito_authentication_for_kibana/opensearch_service_domains_use_cognito_authentication_for_kibana_test.py b/tests/providers/aws/services/opensearch/opensearch_service_domains_use_cognito_authentication_for_kibana/opensearch_service_domains_use_cognito_authentication_for_kibana_test.py index fbfeb7f3c2..61740ffb95 100644 --- a/tests/providers/aws/services/opensearch/opensearch_service_domains_use_cognito_authentication_for_kibana/opensearch_service_domains_use_cognito_authentication_for_kibana_test.py +++ b/tests/providers/aws/services/opensearch/opensearch_service_domains_use_cognito_authentication_for_kibana/opensearch_service_domains_use_cognito_authentication_for_kibana_test.py @@ -21,15 +21,12 @@ class Test_opensearch_service_domains_use_cognito_authentication_for_kibana: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_use_cognito_authentication_for_kibana.opensearch_service_domains_use_cognito_authentication_for_kibana.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_use_cognito_authentication_for_kibana.opensearch_service_domains_use_cognito_authentication_for_kibana.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_use_cognito_authentication_for_kibana.opensearch_service_domains_use_cognito_authentication_for_kibana import ( opensearch_service_domains_use_cognito_authentication_for_kibana, @@ -51,15 +48,12 @@ class Test_opensearch_service_domains_use_cognito_authentication_for_kibana: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_use_cognito_authentication_for_kibana.opensearch_service_domains_use_cognito_authentication_for_kibana.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_use_cognito_authentication_for_kibana.opensearch_service_domains_use_cognito_authentication_for_kibana.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_use_cognito_authentication_for_kibana.opensearch_service_domains_use_cognito_authentication_for_kibana import ( opensearch_service_domains_use_cognito_authentication_for_kibana, @@ -92,15 +86,12 @@ class Test_opensearch_service_domains_use_cognito_authentication_for_kibana: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_use_cognito_authentication_for_kibana.opensearch_service_domains_use_cognito_authentication_for_kibana.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_use_cognito_authentication_for_kibana.opensearch_service_domains_use_cognito_authentication_for_kibana.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_use_cognito_authentication_for_kibana.opensearch_service_domains_use_cognito_authentication_for_kibana import ( opensearch_service_domains_use_cognito_authentication_for_kibana, @@ -133,15 +124,12 @@ class Test_opensearch_service_domains_use_cognito_authentication_for_kibana: mocked_aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=mocked_aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.opensearch.opensearch_service_domains_use_cognito_authentication_for_kibana.opensearch_service_domains_use_cognito_authentication_for_kibana.opensearch_client", - new=OpenSearchService(mocked_aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=mocked_aws_provider, + ), mock.patch( + "prowler.providers.aws.services.opensearch.opensearch_service_domains_use_cognito_authentication_for_kibana.opensearch_service_domains_use_cognito_authentication_for_kibana.opensearch_client", + new=OpenSearchService(mocked_aws_provider), ): from prowler.providers.aws.services.opensearch.opensearch_service_domains_use_cognito_authentication_for_kibana.opensearch_service_domains_use_cognito_authentication_for_kibana import ( opensearch_service_domains_use_cognito_authentication_for_kibana, diff --git a/tests/providers/aws/services/organizations/organizations_tags_policies_enabled_and_attached/organizations_tags_policies_enabled_and_attached_test.py b/tests/providers/aws/services/organizations/organizations_tags_policies_enabled_and_attached/organizations_tags_policies_enabled_and_attached_test.py index 6a09d10cb0..9e045c9116 100644 --- a/tests/providers/aws/services/organizations/organizations_tags_policies_enabled_and_attached/organizations_tags_policies_enabled_and_attached_test.py +++ b/tests/providers/aws/services/organizations/organizations_tags_policies_enabled_and_attached/organizations_tags_policies_enabled_and_attached_test.py @@ -139,15 +139,12 @@ class Test_organizations_tags_policies_enabled_and_attached: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.organizations.organizations_tags_policies_enabled_and_attached.organizations_tags_policies_enabled_and_attached.organizations_client", - new=organizations_client, - ), - mock.patch( - "prowler.providers.aws.services.organizations.organizations_tags_policies_enabled_and_attached.organizations_tags_policies_enabled_and_attached.organizations_client.get_unknown_arn", - return_value="arn:aws:organizations:eu-west-1:0123456789012:unknown", - ), + with mock.patch( + "prowler.providers.aws.services.organizations.organizations_tags_policies_enabled_and_attached.organizations_tags_policies_enabled_and_attached.organizations_client", + new=organizations_client, + ), mock.patch( + "prowler.providers.aws.services.organizations.organizations_tags_policies_enabled_and_attached.organizations_tags_policies_enabled_and_attached.organizations_client.get_unknown_arn", + return_value="arn:aws:organizations:eu-west-1:0123456789012:unknown", ): # Test Check from prowler.providers.aws.services.organizations.organizations_tags_policies_enabled_and_attached.organizations_tags_policies_enabled_and_attached import ( diff --git a/tests/providers/aws/services/rds/rds_cluster_minor_version_upgrade_enabled/rds_cluster_minor_version_upgrade_enabled_test.py b/tests/providers/aws/services/rds/rds_cluster_minor_version_upgrade_enabled/rds_cluster_minor_version_upgrade_enabled_test.py index 68f2fd5569..3d7a257fbc 100644 --- a/tests/providers/aws/services/rds/rds_cluster_minor_version_upgrade_enabled/rds_cluster_minor_version_upgrade_enabled_test.py +++ b/tests/providers/aws/services/rds/rds_cluster_minor_version_upgrade_enabled/rds_cluster_minor_version_upgrade_enabled_test.py @@ -9,15 +9,12 @@ class Test_rds_cluster_minor_version_upgrade_enabled: rds_client = mock.MagicMock rds_client.db_clusters = {} - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_service.RDS", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_cluster_minor_version_upgrade_enabled.rds_cluster_minor_version_upgrade_enabled.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_service.RDS", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_cluster_minor_version_upgrade_enabled.rds_cluster_minor_version_upgrade_enabled.rds_client", + new=rds_client, ): from prowler.providers.aws.services.rds.rds_cluster_minor_version_upgrade_enabled.rds_cluster_minor_version_upgrade_enabled import ( rds_cluster_minor_version_upgrade_enabled, @@ -53,15 +50,12 @@ class Test_rds_cluster_minor_version_upgrade_enabled: ) } - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_service.RDS", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_cluster_minor_version_upgrade_enabled.rds_cluster_minor_version_upgrade_enabled.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_service.RDS", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_cluster_minor_version_upgrade_enabled.rds_cluster_minor_version_upgrade_enabled.rds_client", + new=rds_client, ): from prowler.providers.aws.services.rds.rds_cluster_minor_version_upgrade_enabled.rds_cluster_minor_version_upgrade_enabled import ( rds_cluster_minor_version_upgrade_enabled, @@ -97,15 +91,12 @@ class Test_rds_cluster_minor_version_upgrade_enabled: ) } - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_service.RDS", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_cluster_minor_version_upgrade_enabled.rds_cluster_minor_version_upgrade_enabled.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_service.RDS", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_cluster_minor_version_upgrade_enabled.rds_cluster_minor_version_upgrade_enabled.rds_client", + new=rds_client, ): from prowler.providers.aws.services.rds.rds_cluster_minor_version_upgrade_enabled.rds_cluster_minor_version_upgrade_enabled import ( rds_cluster_minor_version_upgrade_enabled, @@ -153,15 +144,12 @@ class Test_rds_cluster_minor_version_upgrade_enabled: ) } - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_service.RDS", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_cluster_minor_version_upgrade_enabled.rds_cluster_minor_version_upgrade_enabled.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_service.RDS", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_cluster_minor_version_upgrade_enabled.rds_cluster_minor_version_upgrade_enabled.rds_client", + new=rds_client, ): from prowler.providers.aws.services.rds.rds_cluster_minor_version_upgrade_enabled.rds_cluster_minor_version_upgrade_enabled import ( rds_cluster_minor_version_upgrade_enabled, diff --git a/tests/providers/aws/services/rds/rds_cluster_multi_az/rds_cluster_multi_az_test.py b/tests/providers/aws/services/rds/rds_cluster_multi_az/rds_cluster_multi_az_test.py index ee1874f89c..650af65629 100644 --- a/tests/providers/aws/services/rds/rds_cluster_multi_az/rds_cluster_multi_az_test.py +++ b/tests/providers/aws/services/rds/rds_cluster_multi_az/rds_cluster_multi_az_test.py @@ -9,15 +9,12 @@ class Test_rds_cluster_multi_az: rds_client = mock.MagicMock rds_client.db_clusters = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_cluster_multi_az.rds_cluster_multi_az.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_cluster_multi_az.rds_cluster_multi_az.rds_client", + new=rds_client, ): from prowler.providers.aws.services.rds.rds_cluster_multi_az.rds_cluster_multi_az import ( rds_cluster_multi_az, @@ -56,15 +53,12 @@ class Test_rds_cluster_multi_az: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_cluster_multi_az.rds_cluster_multi_az.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_cluster_multi_az.rds_cluster_multi_az.rds_client", + new=rds_client, ): from prowler.providers.aws.services.rds.rds_cluster_multi_az.rds_cluster_multi_az import ( rds_cluster_multi_az, @@ -115,15 +109,12 @@ class Test_rds_cluster_multi_az: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_cluster_multi_az.rds_cluster_multi_az.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_cluster_multi_az.rds_cluster_multi_az.rds_client", + new=rds_client, ): from prowler.providers.aws.services.rds.rds_cluster_multi_az.rds_cluster_multi_az import ( rds_cluster_multi_az, diff --git a/tests/providers/aws/services/rds/rds_cluster_protected_by_backup_plan/rds_cluster_protected_by_backup_plan_test.py b/tests/providers/aws/services/rds/rds_cluster_protected_by_backup_plan/rds_cluster_protected_by_backup_plan_test.py index 0a4d5859ef..8f293c6e91 100644 --- a/tests/providers/aws/services/rds/rds_cluster_protected_by_backup_plan/rds_cluster_protected_by_backup_plan_test.py +++ b/tests/providers/aws/services/rds/rds_cluster_protected_by_backup_plan/rds_cluster_protected_by_backup_plan_test.py @@ -21,15 +21,12 @@ class Test_rds_cluster_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.rds_client", - new=RDS(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.backup_client", - new=Backup(aws_provider), - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.rds_client", + new=RDS(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.backup_client", + new=Backup(aws_provider), ): # Test Check from prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan import ( @@ -80,23 +77,18 @@ class Test_rds_cluster_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.rds_client", - new=cluster, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_client.rds_client", - new=cluster, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.backup_client", - new=backup, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_client.backup_client", - new=backup, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.rds_client", + new=cluster, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_client.rds_client", + new=cluster, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.backup_client", + new=backup, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_client.backup_client", + new=backup, ): # Test Check from prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan import ( @@ -162,23 +154,18 @@ class Test_rds_cluster_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.rds_client", - new=cluster, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_client.rds_client", - new=cluster, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.backup_client", - new=backup, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_client.backup_client", - new=backup, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.rds_client", + new=cluster, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_client.rds_client", + new=cluster, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.backup_client", + new=backup, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_client.backup_client", + new=backup, ): # Test Check from prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan import ( @@ -242,23 +229,18 @@ class Test_rds_cluster_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.rds_client", - new=cluster, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_client.rds_client", - new=cluster, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.backup_client", - new=backup, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_client.backup_client", - new=backup, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.rds_client", + new=cluster, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_client.rds_client", + new=cluster, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.backup_client", + new=backup, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_client.backup_client", + new=backup, ): # Test Check from prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan import ( @@ -323,23 +305,18 @@ class Test_rds_cluster_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.rds_client", - new=cluster, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_client.rds_client", - new=cluster, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.backup_client", - new=backup, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_client.backup_client", - new=backup, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.rds_client", + new=cluster, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_client.rds_client", + new=cluster, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.backup_client", + new=backup, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_client.backup_client", + new=backup, ): # Test Check from prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan import ( @@ -403,23 +380,18 @@ class Test_rds_cluster_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.rds_client", - new=cluster, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_client.rds_client", - new=cluster, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.backup_client", - new=backup, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_client.backup_client", - new=backup, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.rds_client", + new=cluster, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_client.rds_client", + new=cluster, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan.backup_client", + new=backup, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_client.backup_client", + new=backup, ): # Test Check from prowler.providers.aws.services.rds.rds_cluster_protected_by_backup_plan.rds_cluster_protected_by_backup_plan import ( diff --git a/tests/providers/aws/services/rds/rds_instance_certificate_expiration/rds_instance_certificate_expiration_test.py b/tests/providers/aws/services/rds/rds_instance_certificate_expiration/rds_instance_certificate_expiration_test.py index 0aa4716911..a492ff7d2b 100644 --- a/tests/providers/aws/services/rds/rds_instance_certificate_expiration/rds_instance_certificate_expiration_test.py +++ b/tests/providers/aws/services/rds/rds_instance_certificate_expiration/rds_instance_certificate_expiration_test.py @@ -34,15 +34,12 @@ class Test_rds_instance_certificate_expiration: rds_client = mock.MagicMock rds_client.db_instances = {} - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_service.RDS", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_client.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_service.RDS", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_client.rds_client", + new=rds_client, ): # Test Check from prowler.providers.aws.services.rds.rds_instance_certificate_expiration.rds_instance_certificate_expiration import ( @@ -97,15 +94,12 @@ class Test_rds_instance_certificate_expiration: ) } - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_service.RDS", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_client.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_service.RDS", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_client.rds_client", + new=rds_client, ): # Test Check from prowler.providers.aws.services.rds.rds_instance_certificate_expiration.rds_instance_certificate_expiration import ( @@ -173,15 +167,12 @@ class Test_rds_instance_certificate_expiration: ) } - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_service.RDS", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_client.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_service.RDS", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_client.rds_client", + new=rds_client, ): # Test Check from prowler.providers.aws.services.rds.rds_instance_certificate_expiration.rds_instance_certificate_expiration import ( @@ -248,15 +239,12 @@ class Test_rds_instance_certificate_expiration: ) } - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_service.RDS", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_client.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_service.RDS", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_client.rds_client", + new=rds_client, ): # Test Check from prowler.providers.aws.services.rds.rds_instance_certificate_expiration.rds_instance_certificate_expiration import ( @@ -323,15 +311,12 @@ class Test_rds_instance_certificate_expiration: ) } - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_service.RDS", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_client.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_service.RDS", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_client.rds_client", + new=rds_client, ): # Test Check from prowler.providers.aws.services.rds.rds_instance_certificate_expiration.rds_instance_certificate_expiration import ( @@ -398,15 +383,12 @@ class Test_rds_instance_certificate_expiration: ) } - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_service.RDS", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_client.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_service.RDS", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_client.rds_client", + new=rds_client, ): # Test Check from prowler.providers.aws.services.rds.rds_instance_certificate_expiration.rds_instance_certificate_expiration import ( @@ -473,15 +455,12 @@ class Test_rds_instance_certificate_expiration: ) } - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_service.RDS", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_client.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_service.RDS", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_client.rds_client", + new=rds_client, ): # Test Check from prowler.providers.aws.services.rds.rds_instance_certificate_expiration.rds_instance_certificate_expiration import ( @@ -548,15 +527,12 @@ class Test_rds_instance_certificate_expiration: ) } - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_service.RDS", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_client.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_service.RDS", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_client.rds_client", + new=rds_client, ): # Test Check from prowler.providers.aws.services.rds.rds_instance_certificate_expiration.rds_instance_certificate_expiration import ( @@ -623,15 +599,12 @@ class Test_rds_instance_certificate_expiration: ) } - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_service.RDS", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_client.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_service.RDS", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_client.rds_client", + new=rds_client, ): # Test Check from prowler.providers.aws.services.rds.rds_instance_certificate_expiration.rds_instance_certificate_expiration import ( @@ -698,15 +671,12 @@ class Test_rds_instance_certificate_expiration: ) } - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_service.RDS", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_client.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_service.RDS", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_client.rds_client", + new=rds_client, ): # Test Check from prowler.providers.aws.services.rds.rds_instance_certificate_expiration.rds_instance_certificate_expiration import ( @@ -773,15 +743,12 @@ class Test_rds_instance_certificate_expiration: ) } - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_service.RDS", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_client.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_service.RDS", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_client.rds_client", + new=rds_client, ): # Test Check from prowler.providers.aws.services.rds.rds_instance_certificate_expiration.rds_instance_certificate_expiration import ( diff --git a/tests/providers/aws/services/rds/rds_instance_inside_vpc/rds_instance_inside_vpc_test.py b/tests/providers/aws/services/rds/rds_instance_inside_vpc/rds_instance_inside_vpc_test.py index 1efd948a13..d9751f4cac 100644 --- a/tests/providers/aws/services/rds/rds_instance_inside_vpc/rds_instance_inside_vpc_test.py +++ b/tests/providers/aws/services/rds/rds_instance_inside_vpc/rds_instance_inside_vpc_test.py @@ -139,15 +139,12 @@ class Test_rds_instance_inside_vpc: ) } - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_service.RDS", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_instance_inside_vpc.rds_instance_inside_vpc.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_service.RDS", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_instance_inside_vpc.rds_instance_inside_vpc.rds_client", + new=rds_client, ): # Test Check from prowler.providers.aws.services.rds.rds_instance_inside_vpc.rds_instance_inside_vpc import ( diff --git a/tests/providers/aws/services/rds/rds_instance_multi_az/rds_instance_multi_az_test.py b/tests/providers/aws/services/rds/rds_instance_multi_az/rds_instance_multi_az_test.py index d29f1be601..ce8cc42811 100644 --- a/tests/providers/aws/services/rds/rds_instance_multi_az/rds_instance_multi_az_test.py +++ b/tests/providers/aws/services/rds/rds_instance_multi_az/rds_instance_multi_az_test.py @@ -179,15 +179,12 @@ class Test_rds_instance_multi_az: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_instance_multi_az.rds_instance_multi_az.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_instance_multi_az.rds_instance_multi_az.rds_client", + new=rds_client, ): # Test Check from prowler.providers.aws.services.rds.rds_instance_multi_az.rds_instance_multi_az import ( @@ -264,15 +261,12 @@ class Test_rds_instance_multi_az: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - new=rds_client, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_instance_multi_az.rds_instance_multi_az.rds_client", - new=rds_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + new=rds_client, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_instance_multi_az.rds_instance_multi_az.rds_client", + new=rds_client, ): # Test Check from prowler.providers.aws.services.rds.rds_instance_multi_az.rds_instance_multi_az import ( diff --git a/tests/providers/aws/services/rds/rds_instance_protected_by_backup_plan/rds_instance_protected_by_backup_plan_test.py b/tests/providers/aws/services/rds/rds_instance_protected_by_backup_plan/rds_instance_protected_by_backup_plan_test.py index 6079330d9a..6b2c052c56 100644 --- a/tests/providers/aws/services/rds/rds_instance_protected_by_backup_plan/rds_instance_protected_by_backup_plan_test.py +++ b/tests/providers/aws/services/rds/rds_instance_protected_by_backup_plan/rds_instance_protected_by_backup_plan_test.py @@ -22,15 +22,12 @@ class Test_rds_instance_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.rds_client", - new=RDS(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.backup_client", - new=Backup(aws_provider), - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.rds_client", + new=RDS(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.backup_client", + new=Backup(aws_provider), ): # Test Check from prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan import ( @@ -62,15 +59,12 @@ class Test_rds_instance_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.rds_client", - new=RDS(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.backup_client", - new=Backup(aws_provider), - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.rds_client", + new=RDS(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.backup_client", + new=Backup(aws_provider), ): # Test Check from prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan import ( @@ -138,23 +132,18 @@ class Test_rds_instance_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.rds_client", - new=instance, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_client.rds_client", - new=instance, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.backup_client", - new=backup, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_client.backup_client", - new=backup, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.rds_client", + new=instance, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_client.rds_client", + new=instance, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.backup_client", + new=backup, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_client.backup_client", + new=backup, ): # Test Check from prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan import ( @@ -220,23 +209,18 @@ class Test_rds_instance_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.rds_client", - new=instance, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_client.rds_client", - new=instance, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.backup_client", - new=backup, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_client.backup_client", - new=backup, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.rds_client", + new=instance, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_client.rds_client", + new=instance, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.backup_client", + new=backup, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_client.backup_client", + new=backup, ): # Test Check from prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan import ( @@ -303,23 +287,18 @@ class Test_rds_instance_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.rds_client", - new=instance, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_client.rds_client", - new=instance, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.backup_client", - new=backup, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_client.backup_client", - new=backup, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.rds_client", + new=instance, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_client.rds_client", + new=instance, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.backup_client", + new=backup, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_client.backup_client", + new=backup, ): # Test Check from prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan import ( @@ -385,23 +364,18 @@ class Test_rds_instance_protected_by_backup_plan: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.rds_client", - new=instance, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_client.rds_client", - new=instance, - ), - mock.patch( - "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.backup_client", - new=backup, - ), - mock.patch( - "prowler.providers.aws.services.backup.backup_client.backup_client", - new=backup, - ), + with mock.patch( + "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.rds_client", + new=instance, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_client.rds_client", + new=instance, + ), mock.patch( + "prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan.backup_client", + new=backup, + ), mock.patch( + "prowler.providers.aws.services.backup.backup_client.backup_client", + new=backup, ): # Test Check from prowler.providers.aws.services.rds.rds_instance_protected_by_backup_plan.rds_instance_protected_by_backup_plan import ( diff --git a/tests/providers/aws/services/redshift/redshift_cluster_audit_logging/redshift_cluster_audit_logging_test.py b/tests/providers/aws/services/redshift/redshift_cluster_audit_logging/redshift_cluster_audit_logging_test.py index dbdd5ef09e..3c6b9f98f2 100644 --- a/tests/providers/aws/services/redshift/redshift_cluster_audit_logging/redshift_cluster_audit_logging_test.py +++ b/tests/providers/aws/services/redshift/redshift_cluster_audit_logging/redshift_cluster_audit_logging_test.py @@ -14,15 +14,12 @@ class Test_redshift_cluster_audit_logging: def test_no_clusters(self): redshift_client = mock.MagicMock redshift_client.clusters = [] - with ( - mock.patch( - "prowler.providers.aws.services.redshift.redshift_service.Redshift", - redshift_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_client.redshift_client", - redshift_client, - ), + with mock.patch( + "prowler.providers.aws.services.redshift.redshift_service.Redshift", + redshift_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_client.redshift_client", + redshift_client, ): from prowler.providers.aws.services.redshift.redshift_cluster_audit_logging.redshift_cluster_audit_logging import ( redshift_cluster_audit_logging, @@ -43,15 +40,12 @@ class Test_redshift_cluster_audit_logging: logging_enabled=False, ) ) - with ( - mock.patch( - "prowler.providers.aws.services.redshift.redshift_service.Redshift", - redshift_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_client.redshift_client", - redshift_client, - ), + with mock.patch( + "prowler.providers.aws.services.redshift.redshift_service.Redshift", + redshift_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_client.redshift_client", + redshift_client, ): from prowler.providers.aws.services.redshift.redshift_cluster_audit_logging.redshift_cluster_audit_logging import ( redshift_cluster_audit_logging, @@ -79,15 +73,12 @@ class Test_redshift_cluster_audit_logging: endpoint_address="192.192.192.192", ) ) - with ( - mock.patch( - "prowler.providers.aws.services.redshift.redshift_service.Redshift", - redshift_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_client.redshift_client", - redshift_client, - ), + with mock.patch( + "prowler.providers.aws.services.redshift.redshift_service.Redshift", + redshift_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_client.redshift_client", + redshift_client, ): from prowler.providers.aws.services.redshift.redshift_cluster_audit_logging.redshift_cluster_audit_logging import ( redshift_cluster_audit_logging, diff --git a/tests/providers/aws/services/redshift/redshift_cluster_automated_snapshot/redshift_cluster_automated_snapshot_test.py b/tests/providers/aws/services/redshift/redshift_cluster_automated_snapshot/redshift_cluster_automated_snapshot_test.py index 20209f52a3..e13981a6d2 100644 --- a/tests/providers/aws/services/redshift/redshift_cluster_automated_snapshot/redshift_cluster_automated_snapshot_test.py +++ b/tests/providers/aws/services/redshift/redshift_cluster_automated_snapshot/redshift_cluster_automated_snapshot_test.py @@ -14,15 +14,12 @@ class Test_redshift_cluster_automated_snapshot: def test_no_clusters(self): redshift_client = mock.MagicMock redshift_client.clusters = [] - with ( - mock.patch( - "prowler.providers.aws.services.redshift.redshift_service.Redshift", - redshift_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_client.redshift_client", - redshift_client, - ), + with mock.patch( + "prowler.providers.aws.services.redshift.redshift_service.Redshift", + redshift_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_client.redshift_client", + redshift_client, ): from prowler.providers.aws.services.redshift.redshift_cluster_automated_snapshot.redshift_cluster_automated_snapshot import ( redshift_cluster_automated_snapshot, @@ -43,15 +40,12 @@ class Test_redshift_cluster_automated_snapshot: cluster_snapshots=False, ) ) - with ( - mock.patch( - "prowler.providers.aws.services.redshift.redshift_service.Redshift", - redshift_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_client.redshift_client", - redshift_client, - ), + with mock.patch( + "prowler.providers.aws.services.redshift.redshift_service.Redshift", + redshift_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_client.redshift_client", + redshift_client, ): from prowler.providers.aws.services.redshift.redshift_cluster_automated_snapshot.redshift_cluster_automated_snapshot import ( redshift_cluster_automated_snapshot, @@ -78,15 +72,12 @@ class Test_redshift_cluster_automated_snapshot: cluster_snapshots=True, ) ) - with ( - mock.patch( - "prowler.providers.aws.services.redshift.redshift_service.Redshift", - redshift_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_client.redshift_client", - redshift_client, - ), + with mock.patch( + "prowler.providers.aws.services.redshift.redshift_service.Redshift", + redshift_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_client.redshift_client", + redshift_client, ): from prowler.providers.aws.services.redshift.redshift_cluster_automated_snapshot.redshift_cluster_automated_snapshot import ( redshift_cluster_automated_snapshot, diff --git a/tests/providers/aws/services/redshift/redshift_cluster_automatic_upgrades/redshift_cluster_automatic_upgrades_test.py b/tests/providers/aws/services/redshift/redshift_cluster_automatic_upgrades/redshift_cluster_automatic_upgrades_test.py index 3c58247309..2f35eae7f4 100644 --- a/tests/providers/aws/services/redshift/redshift_cluster_automatic_upgrades/redshift_cluster_automatic_upgrades_test.py +++ b/tests/providers/aws/services/redshift/redshift_cluster_automatic_upgrades/redshift_cluster_automatic_upgrades_test.py @@ -14,15 +14,12 @@ class Test_redshift_cluster_automatic_upgrades: def test_no_clusters(self): redshift_client = mock.MagicMock redshift_client.clusters = [] - with ( - mock.patch( - "prowler.providers.aws.services.redshift.redshift_service.Redshift", - redshift_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_client.redshift_client", - redshift_client, - ), + with mock.patch( + "prowler.providers.aws.services.redshift.redshift_service.Redshift", + redshift_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_client.redshift_client", + redshift_client, ): from prowler.providers.aws.services.redshift.redshift_cluster_automatic_upgrades.redshift_cluster_automatic_upgrades import ( redshift_cluster_automatic_upgrades, @@ -43,15 +40,12 @@ class Test_redshift_cluster_automatic_upgrades: allow_version_upgrade=False, ) ) - with ( - mock.patch( - "prowler.providers.aws.services.redshift.redshift_service.Redshift", - redshift_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_client.redshift_client", - redshift_client, - ), + with mock.patch( + "prowler.providers.aws.services.redshift.redshift_service.Redshift", + redshift_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_client.redshift_client", + redshift_client, ): from prowler.providers.aws.services.redshift.redshift_cluster_automatic_upgrades.redshift_cluster_automatic_upgrades import ( redshift_cluster_automatic_upgrades, @@ -78,15 +72,12 @@ class Test_redshift_cluster_automatic_upgrades: allow_version_upgrade=True, ) ) - with ( - mock.patch( - "prowler.providers.aws.services.redshift.redshift_service.Redshift", - redshift_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_client.redshift_client", - redshift_client, - ), + with mock.patch( + "prowler.providers.aws.services.redshift.redshift_service.Redshift", + redshift_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_client.redshift_client", + redshift_client, ): from prowler.providers.aws.services.redshift.redshift_cluster_automatic_upgrades.redshift_cluster_automatic_upgrades import ( redshift_cluster_automatic_upgrades, diff --git a/tests/providers/aws/services/redshift/redshift_cluster_public_access/redshift_cluster_public_access_test.py b/tests/providers/aws/services/redshift/redshift_cluster_public_access/redshift_cluster_public_access_test.py index 816f52489a..534d706076 100644 --- a/tests/providers/aws/services/redshift/redshift_cluster_public_access/redshift_cluster_public_access_test.py +++ b/tests/providers/aws/services/redshift/redshift_cluster_public_access/redshift_cluster_public_access_test.py @@ -26,19 +26,15 @@ class Test_redshift_cluster_public_access: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.redshift_client", - new=redshift_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.vpc_client", - new=vpc_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.redshift_client", + new=redshift_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.vpc_client", + new=vpc_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.ec2_client", + new=ec2_client, ): from prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access import ( redshift_cluster_public_access, @@ -68,19 +64,15 @@ class Test_redshift_cluster_public_access: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.redshift_client", - new=redshift_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.vpc_client", - new=vpc_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.redshift_client", + new=redshift_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.vpc_client", + new=vpc_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.ec2_client", + new=ec2_client, ): from prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access import ( redshift_cluster_public_access, @@ -116,19 +108,15 @@ class Test_redshift_cluster_public_access: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.redshift_client", - new=redshift_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.vpc_client", - new=vpc_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.redshift_client", + new=redshift_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.vpc_client", + new=vpc_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.ec2_client", + new=ec2_client, ): from prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access import ( redshift_cluster_public_access, @@ -163,19 +151,15 @@ class Test_redshift_cluster_public_access: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.redshift_client", - new=redshift_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.vpc_client", - new=vpc_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.redshift_client", + new=redshift_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.vpc_client", + new=vpc_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.ec2_client", + new=ec2_client, ): from prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access import ( redshift_cluster_public_access, @@ -214,19 +198,15 @@ class Test_redshift_cluster_public_access: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.redshift_client", - new=redshift_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.vpc_client", - new=vpc_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.redshift_client", + new=redshift_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.vpc_client", + new=vpc_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.ec2_client", + new=ec2_client, ): from prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access import ( redshift_cluster_public_access, @@ -282,19 +262,15 @@ class Test_redshift_cluster_public_access: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.redshift_client", - new=redshift_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.vpc_client", - new=vpc_client, - ), - mock.patch( - "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.redshift_client", + new=redshift_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.vpc_client", + new=vpc_client, + ), mock.patch( + "prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access.ec2_client", + new=ec2_client, ): from prowler.providers.aws.services.redshift.redshift_cluster_public_access.redshift_cluster_public_access import ( redshift_cluster_public_access, diff --git a/tests/providers/aws/services/route53/route53_public_hosted_zones_cloudwatch_logging_enabled/route53_public_hosted_zones_cloudwatch_logging_enabled_test.py b/tests/providers/aws/services/route53/route53_public_hosted_zones_cloudwatch_logging_enabled/route53_public_hosted_zones_cloudwatch_logging_enabled_test.py index a5fd5f17c9..315fa73385 100644 --- a/tests/providers/aws/services/route53/route53_public_hosted_zones_cloudwatch_logging_enabled/route53_public_hosted_zones_cloudwatch_logging_enabled_test.py +++ b/tests/providers/aws/services/route53/route53_public_hosted_zones_cloudwatch_logging_enabled/route53_public_hosted_zones_cloudwatch_logging_enabled_test.py @@ -12,15 +12,12 @@ class Test_route53_public_hosted_zones_cloudwatch_logging_enabled: route53 = mock.MagicMock route53.hosted_zones = {} - with ( - mock.patch( - "prowler.providers.aws.services.route53.route53_service.Route53", - new=route53, - ), - mock.patch( - "prowler.providers.aws.services.route53.route53_public_hosted_zones_cloudwatch_logging_enabled.route53_public_hosted_zones_cloudwatch_logging_enabled.route53_client", - new=route53, - ), + with mock.patch( + "prowler.providers.aws.services.route53.route53_service.Route53", + new=route53, + ), mock.patch( + "prowler.providers.aws.services.route53.route53_public_hosted_zones_cloudwatch_logging_enabled.route53_public_hosted_zones_cloudwatch_logging_enabled.route53_client", + new=route53, ): # Test Check from prowler.providers.aws.services.route53.route53_public_hosted_zones_cloudwatch_logging_enabled.route53_public_hosted_zones_cloudwatch_logging_enabled import ( @@ -49,15 +46,12 @@ class Test_route53_public_hosted_zones_cloudwatch_logging_enabled: ) } - with ( - mock.patch( - "prowler.providers.aws.services.route53.route53_service.Route53", - new=route53, - ), - mock.patch( - "prowler.providers.aws.services.route53.route53_public_hosted_zones_cloudwatch_logging_enabled.route53_public_hosted_zones_cloudwatch_logging_enabled.route53_client", - new=route53, - ), + with mock.patch( + "prowler.providers.aws.services.route53.route53_service.Route53", + new=route53, + ), mock.patch( + "prowler.providers.aws.services.route53.route53_public_hosted_zones_cloudwatch_logging_enabled.route53_public_hosted_zones_cloudwatch_logging_enabled.route53_client", + new=route53, ): # Test Check from prowler.providers.aws.services.route53.route53_public_hosted_zones_cloudwatch_logging_enabled.route53_public_hosted_zones_cloudwatch_logging_enabled import ( @@ -90,15 +84,12 @@ class Test_route53_public_hosted_zones_cloudwatch_logging_enabled: ) } - with ( - mock.patch( - "prowler.providers.aws.services.route53.route53_service.Route53", - new=route53, - ), - mock.patch( - "prowler.providers.aws.services.route53.route53_public_hosted_zones_cloudwatch_logging_enabled.route53_public_hosted_zones_cloudwatch_logging_enabled.route53_client", - new=route53, - ), + with mock.patch( + "prowler.providers.aws.services.route53.route53_service.Route53", + new=route53, + ), mock.patch( + "prowler.providers.aws.services.route53.route53_public_hosted_zones_cloudwatch_logging_enabled.route53_public_hosted_zones_cloudwatch_logging_enabled.route53_client", + new=route53, ): # Test Check from prowler.providers.aws.services.route53.route53_public_hosted_zones_cloudwatch_logging_enabled.route53_public_hosted_zones_cloudwatch_logging_enabled import ( @@ -131,15 +122,12 @@ class Test_route53_public_hosted_zones_cloudwatch_logging_enabled: ) } - with ( - mock.patch( - "prowler.providers.aws.services.route53.route53_service.Route53", - new=route53, - ), - mock.patch( - "prowler.providers.aws.services.route53.route53_public_hosted_zones_cloudwatch_logging_enabled.route53_public_hosted_zones_cloudwatch_logging_enabled.route53_client", - new=route53, - ), + with mock.patch( + "prowler.providers.aws.services.route53.route53_service.Route53", + new=route53, + ), mock.patch( + "prowler.providers.aws.services.route53.route53_public_hosted_zones_cloudwatch_logging_enabled.route53_public_hosted_zones_cloudwatch_logging_enabled.route53_client", + new=route53, ): # Test Check from prowler.providers.aws.services.route53.route53_public_hosted_zones_cloudwatch_logging_enabled.route53_public_hosted_zones_cloudwatch_logging_enabled import ( diff --git a/tests/providers/aws/services/s3/s3_access_point_public_access_block/s3_access_point_public_access_block_test.py b/tests/providers/aws/services/s3/s3_access_point_public_access_block/s3_access_point_public_access_block_test.py index 60a7ef2582..322bc3405a 100644 --- a/tests/providers/aws/services/s3/s3_access_point_public_access_block/s3_access_point_public_access_block_test.py +++ b/tests/providers/aws/services/s3/s3_access_point_public_access_block/s3_access_point_public_access_block_test.py @@ -20,15 +20,12 @@ class Test_s3_access_point_public_access_block: [AWS_REGION_US_EAST_1, AWS_REGION_EU_WEST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_access_point_public_access_block.s3_access_point_public_access_block.s3control_client", - new=S3Control(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_access_point_public_access_block.s3_access_point_public_access_block.s3control_client", + new=S3Control(aws_provider), ): # Test Check from prowler.providers.aws.services.s3.s3_access_point_public_access_block.s3_access_point_public_access_block import ( @@ -62,15 +59,12 @@ class Test_s3_access_point_public_access_block: bucket_name_eu = "test-bucket-eu-west-1" arn_eu = f"arn:aws:s3:eu-west-1:{AWS_ACCOUNT_NUMBER}:accesspoint/{ap_name_eu}" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_access_point_public_access_block.s3_access_point_public_access_block.s3control_client", - new=S3Control(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_access_point_public_access_block.s3_access_point_public_access_block.s3control_client", + new=S3Control(aws_provider), ): # Test Check from prowler.providers.aws.services.s3.s3_access_point_public_access_block.s3_access_point_public_access_block import ( @@ -158,15 +152,12 @@ class Test_s3_access_point_public_access_block: bucket_name_eu = "test-bucket-eu-west-1" arn_eu = f"arn:aws:s3:eu-west-1:{AWS_ACCOUNT_NUMBER}:accesspoint/{ap_name_eu}" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_access_point_public_access_block.s3_access_point_public_access_block.s3control_client", - new=S3Control(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_access_point_public_access_block.s3_access_point_public_access_block.s3control_client", + new=S3Control(aws_provider), ): # Test Check from prowler.providers.aws.services.s3.s3_access_point_public_access_block.s3_access_point_public_access_block import ( @@ -254,15 +245,12 @@ class Test_s3_access_point_public_access_block: bucket_name_eu = "test-bucket-eu-west-1" arn_eu = f"arn:aws:s3:eu-west-1:{AWS_ACCOUNT_NUMBER}:accesspoint/{ap_name_eu}" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_access_point_public_access_block.s3_access_point_public_access_block.s3control_client", - new=S3Control(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_access_point_public_access_block.s3_access_point_public_access_block.s3control_client", + new=S3Control(aws_provider), ): # Test Check from prowler.providers.aws.services.s3.s3_access_point_public_access_block.s3_access_point_public_access_block import ( @@ -350,15 +338,12 @@ class Test_s3_access_point_public_access_block: f"arn:aws:s3:ap-southeast-2:{AWS_ACCOUNT_NUMBER}:accesspoint/{ap_name_ap}" ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_access_point_public_access_block.s3_access_point_public_access_block.s3control_client", - new=S3Control(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_access_point_public_access_block.s3_access_point_public_access_block.s3control_client", + new=S3Control(aws_provider), ): from prowler.providers.aws.services.s3.s3_access_point_public_access_block.s3_access_point_public_access_block import ( s3_access_point_public_access_block, diff --git a/tests/providers/aws/services/s3/s3_account_level_public_access_blocks/s3_account_level_public_access_blocks_fixer_test.py b/tests/providers/aws/services/s3/s3_account_level_public_access_blocks/s3_account_level_public_access_blocks_fixer_test.py index 3640a8b842..5fb2429a18 100644 --- a/tests/providers/aws/services/s3/s3_account_level_public_access_blocks/s3_account_level_public_access_blocks_fixer_test.py +++ b/tests/providers/aws/services/s3/s3_account_level_public_access_blocks/s3_account_level_public_access_blocks_fixer_test.py @@ -16,15 +16,12 @@ class Test_s3_account_level_public_access_block_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_account_level_public_access_blocks.s3_account_level_public_access_blocks_fixer.s3control_client", - new=S3Control(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_account_level_public_access_blocks.s3_account_level_public_access_blocks_fixer.s3control_client", + new=S3Control(aws_provider), ): from prowler.providers.aws.services.s3.s3_account_level_public_access_blocks.s3_account_level_public_access_blocks_fixer import ( fixer, diff --git a/tests/providers/aws/services/s3/s3_account_level_public_access_blocks/s3_account_level_public_access_blocks_test.py b/tests/providers/aws/services/s3/s3_account_level_public_access_blocks/s3_account_level_public_access_blocks_test.py index 994a6bcbe6..fa34cb1e56 100644 --- a/tests/providers/aws/services/s3/s3_account_level_public_access_blocks/s3_account_level_public_access_blocks_test.py +++ b/tests/providers/aws/services/s3/s3_account_level_public_access_blocks/s3_account_level_public_access_blocks_test.py @@ -28,19 +28,15 @@ class Test_s3_account_level_public_access_blocks: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_account_level_public_access_blocks.s3_account_level_public_access_blocks.s3_client", - new=S3(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_account_level_public_access_blocks.s3_account_level_public_access_blocks.s3control_client", - new=S3Control(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_account_level_public_access_blocks.s3_account_level_public_access_blocks.s3_client", + new=S3(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.s3.s3_account_level_public_access_blocks.s3_account_level_public_access_blocks.s3control_client", + new=S3Control(aws_provider), ): # Test Check from prowler.providers.aws.services.s3.s3_account_level_public_access_blocks.s3_account_level_public_access_blocks import ( @@ -80,19 +76,15 @@ class Test_s3_account_level_public_access_blocks: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_account_level_public_access_blocks.s3_account_level_public_access_blocks.s3_client", - new=S3(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_account_level_public_access_blocks.s3_account_level_public_access_blocks.s3control_client", - new=S3Control(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_account_level_public_access_blocks.s3_account_level_public_access_blocks.s3_client", + new=S3(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.s3.s3_account_level_public_access_blocks.s3_account_level_public_access_blocks.s3control_client", + new=S3Control(aws_provider), ): # Test Check from prowler.providers.aws.services.s3.s3_account_level_public_access_blocks.s3_account_level_public_access_blocks import ( @@ -133,19 +125,15 @@ class Test_s3_account_level_public_access_blocks: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) aws_provider._scan_unused_services = False - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_account_level_public_access_blocks.s3_account_level_public_access_blocks.s3_client", - new=S3(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_account_level_public_access_blocks.s3_account_level_public_access_blocks.s3control_client", - new=S3Control(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_account_level_public_access_blocks.s3_account_level_public_access_blocks.s3_client", + new=S3(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.s3.s3_account_level_public_access_blocks.s3_account_level_public_access_blocks.s3control_client", + new=S3Control(aws_provider), ): # Test Check from prowler.providers.aws.services.s3.s3_account_level_public_access_blocks.s3_account_level_public_access_blocks import ( diff --git a/tests/providers/aws/services/s3/s3_bucket_kms_encryption/s3_bucket_kms_encryption_test.py b/tests/providers/aws/services/s3/s3_bucket_kms_encryption/s3_bucket_kms_encryption_test.py index e9849df3ba..92e0112087 100644 --- a/tests/providers/aws/services/s3/s3_bucket_kms_encryption/s3_bucket_kms_encryption_test.py +++ b/tests/providers/aws/services/s3/s3_bucket_kms_encryption/s3_bucket_kms_encryption_test.py @@ -13,15 +13,12 @@ class Test_s3_bucket_kms_encryption: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_kms_encryption.s3_bucket_kms_encryption.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_kms_encryption.s3_bucket_kms_encryption.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.s3.s3_bucket_kms_encryption.s3_bucket_kms_encryption import ( @@ -43,15 +40,12 @@ class Test_s3_bucket_kms_encryption: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_kms_encryption.s3_bucket_kms_encryption.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_kms_encryption.s3_bucket_kms_encryption.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.s3.s3_bucket_kms_encryption.s3_bucket_kms_encryption import ( @@ -100,15 +94,12 @@ class Test_s3_bucket_kms_encryption: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_kms_encryption.s3_bucket_kms_encryption.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_kms_encryption.s3_bucket_kms_encryption.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.s3.s3_bucket_kms_encryption.s3_bucket_kms_encryption import ( diff --git a/tests/providers/aws/services/s3/s3_bucket_lifecycle_enabled/s3_bucket_lifecycle_enabled_test.py b/tests/providers/aws/services/s3/s3_bucket_lifecycle_enabled/s3_bucket_lifecycle_enabled_test.py index 2f18711fb5..f8ce8923d3 100644 --- a/tests/providers/aws/services/s3/s3_bucket_lifecycle_enabled/s3_bucket_lifecycle_enabled_test.py +++ b/tests/providers/aws/services/s3/s3_bucket_lifecycle_enabled/s3_bucket_lifecycle_enabled_test.py @@ -39,15 +39,12 @@ class Test_s3_bucket_lifecycle_enabled: def test_no_lifecycle_configuration(self): aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_lifecycle_enabled.s3_bucket_lifecycle_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_lifecycle_enabled.s3_bucket_lifecycle_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.s3.s3_bucket_lifecycle_enabled.s3_bucket_lifecycle_enabled import ( @@ -88,15 +85,12 @@ class Test_s3_bucket_lifecycle_enabled: def test_one_valid_lifecycle_configuration(self): aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_lifecycle_enabled.s3_bucket_lifecycle_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_lifecycle_enabled.s3_bucket_lifecycle_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.s3.s3_bucket_lifecycle_enabled.s3_bucket_lifecycle_enabled import ( @@ -140,15 +134,12 @@ class Test_s3_bucket_lifecycle_enabled: def test_several_lifecycle_configurations(self): aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_lifecycle_enabled.s3_bucket_lifecycle_enabled.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_lifecycle_enabled.s3_bucket_lifecycle_enabled.s3_client", + new=S3(aws_provider), ): # Test Check from prowler.providers.aws.services.s3.s3_bucket_lifecycle_enabled.s3_bucket_lifecycle_enabled import ( diff --git a/tests/providers/aws/services/s3/s3_bucket_policy_public_write_access/s3_bucket_policy_public_write_access_fixer_test.py b/tests/providers/aws/services/s3/s3_bucket_policy_public_write_access/s3_bucket_policy_public_write_access_fixer_test.py index 6d92c2261c..3f4687cab9 100644 --- a/tests/providers/aws/services/s3/s3_bucket_policy_public_write_access/s3_bucket_policy_public_write_access_fixer_test.py +++ b/tests/providers/aws/services/s3/s3_bucket_policy_public_write_access/s3_bucket_policy_public_write_access_fixer_test.py @@ -24,15 +24,12 @@ class Test_s3_bucket_policy_public_write_access_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access_fixer.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access_fixer.s3_client", + new=S3(aws_provider), ): # Test Fixer from prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access_fixer import ( @@ -58,15 +55,12 @@ class Test_s3_bucket_policy_public_write_access_fixer: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access_fixer.s3_client", - new=S3(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access_fixer.s3_client", + new=S3(aws_provider), ): # Test Fixer from prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access_fixer import ( diff --git a/tests/providers/aws/services/s3/s3_bucket_policy_public_write_access/s3_bucket_policy_public_write_access_test.py b/tests/providers/aws/services/s3/s3_bucket_policy_public_write_access/s3_bucket_policy_public_write_access_test.py index 2a5cdd9900..7e7f23970f 100644 --- a/tests/providers/aws/services/s3/s3_bucket_policy_public_write_access/s3_bucket_policy_public_write_access_test.py +++ b/tests/providers/aws/services/s3/s3_bucket_policy_public_write_access/s3_bucket_policy_public_write_access_test.py @@ -83,15 +83,12 @@ class Test_s3_bucket_policy_public_write_access: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access.s3_client", - new=S3(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access.s3control_client", - new=S3Control(aws_provider), - ), + with mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access.s3_client", + new=S3(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access.s3control_client", + new=S3Control(aws_provider), ): # Test Check from prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access import ( @@ -157,15 +154,12 @@ class Test_s3_bucket_policy_public_write_access: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access.s3_client", - new=S3(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access.s3control_client", - new=S3Control(aws_provider), - ), + with mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access.s3_client", + new=S3(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access.s3control_client", + new=S3Control(aws_provider), ): # Test Check from prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access import ( @@ -222,15 +216,12 @@ class Test_s3_bucket_policy_public_write_access: "prowler.providers.common.provider.Provider.get_global_provider", return_value=aws_provider, ): - with ( - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access.s3_client", - new=S3(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access.s3control_client", - new=S3Control(aws_provider), - ), + with mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access.s3_client", + new=S3(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access.s3control_client", + new=S3Control(aws_provider), ): # Test Check from prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access import ( @@ -282,15 +273,12 @@ class Test_s3_bucket_policy_public_write_access: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access.s3control_client", - new=S3Control(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access.s3control_client", + new=S3Control(aws_provider), ): with mock.patch( "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access.s3_client", @@ -346,15 +334,12 @@ class Test_s3_bucket_policy_public_write_access: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access.s3control_client", - new=S3Control(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access.s3control_client", + new=S3Control(aws_provider), ): with mock.patch( "prowler.providers.aws.services.s3.s3_bucket_policy_public_write_access.s3_bucket_policy_public_write_access.s3_client", diff --git a/tests/providers/aws/services/s3/s3_bucket_public_access/s3_bucket_public_access_test.py b/tests/providers/aws/services/s3/s3_bucket_public_access/s3_bucket_public_access_test.py index a5506dbe33..d7ce430e61 100644 --- a/tests/providers/aws/services/s3/s3_bucket_public_access/s3_bucket_public_access_test.py +++ b/tests/providers/aws/services/s3/s3_bucket_public_access/s3_bucket_public_access_test.py @@ -450,19 +450,15 @@ class Test_s3_bucket_public_access: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_public_access.s3_bucket_public_access.s3_client", - new=S3(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_public_access.s3_bucket_public_access.s3control_client", - new=S3Control(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_public_access.s3_bucket_public_access.s3_client", + new=S3(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_public_access.s3_bucket_public_access.s3control_client", + new=S3Control(aws_provider), ): # Test Check from prowler.providers.aws.services.s3.s3_bucket_public_access.s3_bucket_public_access import ( @@ -519,19 +515,15 @@ class Test_s3_bucket_public_access: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_public_access.s3_bucket_public_access.s3_client", - new=S3(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_public_access.s3_bucket_public_access.s3control_client", - new=S3Control(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_public_access.s3_bucket_public_access.s3_client", + new=S3(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_public_access.s3_bucket_public_access.s3control_client", + new=S3Control(aws_provider), ): # Test Check from prowler.providers.aws.services.s3.s3_bucket_public_access.s3_bucket_public_access import ( @@ -588,19 +580,15 @@ class Test_s3_bucket_public_access: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_public_access.s3_bucket_public_access.s3_client", - new=S3(aws_provider), - ), - mock.patch( - "prowler.providers.aws.services.s3.s3_bucket_public_access.s3_bucket_public_access.s3control_client", - new=S3Control(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_public_access.s3_bucket_public_access.s3_client", + new=S3(aws_provider), + ), mock.patch( + "prowler.providers.aws.services.s3.s3_bucket_public_access.s3_bucket_public_access.s3control_client", + new=S3Control(aws_provider), ): from prowler.providers.aws.services.s3.s3_bucket_public_access.s3_bucket_public_access import ( s3_bucket_public_access, diff --git a/tests/providers/aws/services/sagemaker/sagemaker_endpoint_config_prod_variant_instances/sagemaker_endpoint_config_prod_variant_instances_test.py b/tests/providers/aws/services/sagemaker/sagemaker_endpoint_config_prod_variant_instances/sagemaker_endpoint_config_prod_variant_instances_test.py index a4d39338cb..714726b50e 100644 --- a/tests/providers/aws/services/sagemaker/sagemaker_endpoint_config_prod_variant_instances/sagemaker_endpoint_config_prod_variant_instances_test.py +++ b/tests/providers/aws/services/sagemaker/sagemaker_endpoint_config_prod_variant_instances/sagemaker_endpoint_config_prod_variant_instances_test.py @@ -20,15 +20,12 @@ class Test_sagemaker_endpoint_config_prod_variant_instances: [AWS_REGION_EU_WEST_1, AWS_REGION_US_EAST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_endpoint_config_prod_variant_instances.sagemaker_endpoint_config_prod_variant_instances.sagemaker_client", - new=SageMaker(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_endpoint_config_prod_variant_instances.sagemaker_endpoint_config_prod_variant_instances.sagemaker_client", + new=SageMaker(aws_provider), ): from prowler.providers.aws.services.sagemaker.sagemaker_endpoint_config_prod_variant_instances.sagemaker_endpoint_config_prod_variant_instances import ( sagemaker_endpoint_config_prod_variant_instances, @@ -72,15 +69,12 @@ class Test_sagemaker_endpoint_config_prod_variant_instances: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_endpoint_config_prod_variant_instances.sagemaker_endpoint_config_prod_variant_instances.sagemaker_client", - new=SageMaker(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_endpoint_config_prod_variant_instances.sagemaker_endpoint_config_prod_variant_instances.sagemaker_client", + new=SageMaker(aws_provider), ): from prowler.providers.aws.services.sagemaker.sagemaker_endpoint_config_prod_variant_instances.sagemaker_endpoint_config_prod_variant_instances import ( sagemaker_endpoint_config_prod_variant_instances, @@ -131,15 +125,12 @@ class Test_sagemaker_endpoint_config_prod_variant_instances: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_endpoint_config_prod_variant_instances.sagemaker_endpoint_config_prod_variant_instances.sagemaker_client", - new=SageMaker(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_endpoint_config_prod_variant_instances.sagemaker_endpoint_config_prod_variant_instances.sagemaker_client", + new=SageMaker(aws_provider), ): from prowler.providers.aws.services.sagemaker.sagemaker_endpoint_config_prod_variant_instances.sagemaker_endpoint_config_prod_variant_instances import ( sagemaker_endpoint_config_prod_variant_instances, diff --git a/tests/providers/aws/services/sagemaker/sagemaker_models_network_isolation_enabled/sagemaker_models_network_isolation_enabled_test.py b/tests/providers/aws/services/sagemaker/sagemaker_models_network_isolation_enabled/sagemaker_models_network_isolation_enabled_test.py index 213fbcba24..b0a6db43be 100644 --- a/tests/providers/aws/services/sagemaker/sagemaker_models_network_isolation_enabled/sagemaker_models_network_isolation_enabled_test.py +++ b/tests/providers/aws/services/sagemaker/sagemaker_models_network_isolation_enabled/sagemaker_models_network_isolation_enabled_test.py @@ -20,15 +20,12 @@ class Test_sagemaker_models_network_isolation_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_models_network_isolation_enabled.sagemaker_models_network_isolation_enabled.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_models_network_isolation_enabled.sagemaker_models_network_isolation_enabled.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_models_network_isolation_enabled.sagemaker_models_network_isolation_enabled import ( sagemaker_models_network_isolation_enabled, @@ -52,15 +49,12 @@ class Test_sagemaker_models_network_isolation_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_models_network_isolation_enabled.sagemaker_models_network_isolation_enabled.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_models_network_isolation_enabled.sagemaker_models_network_isolation_enabled.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_models_network_isolation_enabled.sagemaker_models_network_isolation_enabled import ( sagemaker_models_network_isolation_enabled, @@ -91,15 +85,12 @@ class Test_sagemaker_models_network_isolation_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_models_network_isolation_enabled.sagemaker_models_network_isolation_enabled.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_models_network_isolation_enabled.sagemaker_models_network_isolation_enabled.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_models_network_isolation_enabled.sagemaker_models_network_isolation_enabled import ( sagemaker_models_network_isolation_enabled, diff --git a/tests/providers/aws/services/sagemaker/sagemaker_models_vpc_settings_configured/sagemaker_models_vpc_settings_configured_test.py b/tests/providers/aws/services/sagemaker/sagemaker_models_vpc_settings_configured/sagemaker_models_vpc_settings_configured_test.py index 35438e1f27..42df229c0a 100644 --- a/tests/providers/aws/services/sagemaker/sagemaker_models_vpc_settings_configured/sagemaker_models_vpc_settings_configured_test.py +++ b/tests/providers/aws/services/sagemaker/sagemaker_models_vpc_settings_configured/sagemaker_models_vpc_settings_configured_test.py @@ -20,15 +20,12 @@ class Test_sagemaker_models_vpc_settings_configured: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_models_vpc_settings_configured.sagemaker_models_vpc_settings_configured.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_models_vpc_settings_configured.sagemaker_models_vpc_settings_configured.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_models_vpc_settings_configured.sagemaker_models_vpc_settings_configured import ( sagemaker_models_vpc_settings_configured, @@ -52,15 +49,12 @@ class Test_sagemaker_models_vpc_settings_configured: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_models_vpc_settings_configured.sagemaker_models_vpc_settings_configured.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_models_vpc_settings_configured.sagemaker_models_vpc_settings_configured.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_models_vpc_settings_configured.sagemaker_models_vpc_settings_configured import ( sagemaker_models_vpc_settings_configured, @@ -90,15 +84,12 @@ class Test_sagemaker_models_vpc_settings_configured: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_models_vpc_settings_configured.sagemaker_models_vpc_settings_configured.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_models_vpc_settings_configured.sagemaker_models_vpc_settings_configured.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_models_vpc_settings_configured.sagemaker_models_vpc_settings_configured import ( sagemaker_models_vpc_settings_configured, diff --git a/tests/providers/aws/services/sagemaker/sagemaker_notebook_instance_encryption_enabled/sagemaker_notebook_instance_encryption_enabled_test.py b/tests/providers/aws/services/sagemaker/sagemaker_notebook_instance_encryption_enabled/sagemaker_notebook_instance_encryption_enabled_test.py index 0dd6c24078..5227206736 100644 --- a/tests/providers/aws/services/sagemaker/sagemaker_notebook_instance_encryption_enabled/sagemaker_notebook_instance_encryption_enabled_test.py +++ b/tests/providers/aws/services/sagemaker/sagemaker_notebook_instance_encryption_enabled/sagemaker_notebook_instance_encryption_enabled_test.py @@ -20,15 +20,12 @@ class Test_sagemaker_notebook_instance_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_encryption_enabled.sagemaker_notebook_instance_encryption_enabled.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_encryption_enabled.sagemaker_notebook_instance_encryption_enabled.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_encryption_enabled.sagemaker_notebook_instance_encryption_enabled import ( sagemaker_notebook_instance_encryption_enabled, @@ -52,15 +49,12 @@ class Test_sagemaker_notebook_instance_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_encryption_enabled.sagemaker_notebook_instance_encryption_enabled.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_encryption_enabled.sagemaker_notebook_instance_encryption_enabled.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_encryption_enabled.sagemaker_notebook_instance_encryption_enabled import ( sagemaker_notebook_instance_encryption_enabled, @@ -90,15 +84,12 @@ class Test_sagemaker_notebook_instance_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_encryption_enabled.sagemaker_notebook_instance_encryption_enabled.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_encryption_enabled.sagemaker_notebook_instance_encryption_enabled.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_encryption_enabled.sagemaker_notebook_instance_encryption_enabled import ( sagemaker_notebook_instance_encryption_enabled, diff --git a/tests/providers/aws/services/sagemaker/sagemaker_notebook_instance_root_access_disabled/sagemaker_notebook_instance_root_access_disabled_test.py b/tests/providers/aws/services/sagemaker/sagemaker_notebook_instance_root_access_disabled/sagemaker_notebook_instance_root_access_disabled_test.py index 18d9c1b856..3b0a38382a 100644 --- a/tests/providers/aws/services/sagemaker/sagemaker_notebook_instance_root_access_disabled/sagemaker_notebook_instance_root_access_disabled_test.py +++ b/tests/providers/aws/services/sagemaker/sagemaker_notebook_instance_root_access_disabled/sagemaker_notebook_instance_root_access_disabled_test.py @@ -18,15 +18,12 @@ class Test_sagemaker_notebook_instance_root_access_disabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_root_access_disabled.sagemaker_notebook_instance_root_access_disabled.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_root_access_disabled.sagemaker_notebook_instance_root_access_disabled.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_root_access_disabled.sagemaker_notebook_instance_root_access_disabled import ( sagemaker_notebook_instance_root_access_disabled, @@ -50,15 +47,12 @@ class Test_sagemaker_notebook_instance_root_access_disabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_root_access_disabled.sagemaker_notebook_instance_root_access_disabled.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_root_access_disabled.sagemaker_notebook_instance_root_access_disabled.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_root_access_disabled.sagemaker_notebook_instance_root_access_disabled import ( sagemaker_notebook_instance_root_access_disabled, @@ -89,15 +83,12 @@ class Test_sagemaker_notebook_instance_root_access_disabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_root_access_disabled.sagemaker_notebook_instance_root_access_disabled.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_root_access_disabled.sagemaker_notebook_instance_root_access_disabled.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_root_access_disabled.sagemaker_notebook_instance_root_access_disabled import ( sagemaker_notebook_instance_root_access_disabled, diff --git a/tests/providers/aws/services/sagemaker/sagemaker_notebook_instance_vpc_settings_configured/sagemaker_notebook_instance_vpc_settings_configured_test.py b/tests/providers/aws/services/sagemaker/sagemaker_notebook_instance_vpc_settings_configured/sagemaker_notebook_instance_vpc_settings_configured_test.py index 3171413b41..b613cd4c74 100644 --- a/tests/providers/aws/services/sagemaker/sagemaker_notebook_instance_vpc_settings_configured/sagemaker_notebook_instance_vpc_settings_configured_test.py +++ b/tests/providers/aws/services/sagemaker/sagemaker_notebook_instance_vpc_settings_configured/sagemaker_notebook_instance_vpc_settings_configured_test.py @@ -20,15 +20,12 @@ class Test_sagemaker_notebook_instance_vpc_settings_configured: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_vpc_settings_configured.sagemaker_notebook_instance_vpc_settings_configured.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_vpc_settings_configured.sagemaker_notebook_instance_vpc_settings_configured.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_vpc_settings_configured.sagemaker_notebook_instance_vpc_settings_configured import ( sagemaker_notebook_instance_vpc_settings_configured, @@ -52,15 +49,12 @@ class Test_sagemaker_notebook_instance_vpc_settings_configured: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_vpc_settings_configured.sagemaker_notebook_instance_vpc_settings_configured.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_vpc_settings_configured.sagemaker_notebook_instance_vpc_settings_configured.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_vpc_settings_configured.sagemaker_notebook_instance_vpc_settings_configured import ( sagemaker_notebook_instance_vpc_settings_configured, @@ -91,15 +85,12 @@ class Test_sagemaker_notebook_instance_vpc_settings_configured: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_vpc_settings_configured.sagemaker_notebook_instance_vpc_settings_configured.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_vpc_settings_configured.sagemaker_notebook_instance_vpc_settings_configured.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_vpc_settings_configured.sagemaker_notebook_instance_vpc_settings_configured import ( sagemaker_notebook_instance_vpc_settings_configured, diff --git a/tests/providers/aws/services/sagemaker/sagemaker_notebook_instance_without_direct_internet_access_configured/sagemaker_notebook_instance_without_direct_internet_access_configured_test.py b/tests/providers/aws/services/sagemaker/sagemaker_notebook_instance_without_direct_internet_access_configured/sagemaker_notebook_instance_without_direct_internet_access_configured_test.py index da0f9548bd..8c80572443 100644 --- a/tests/providers/aws/services/sagemaker/sagemaker_notebook_instance_without_direct_internet_access_configured/sagemaker_notebook_instance_without_direct_internet_access_configured_test.py +++ b/tests/providers/aws/services/sagemaker/sagemaker_notebook_instance_without_direct_internet_access_configured/sagemaker_notebook_instance_without_direct_internet_access_configured_test.py @@ -18,15 +18,12 @@ class Test_sagemaker_notebook_instance_without_direct_internet_access_configured aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_without_direct_internet_access_configured.sagemaker_notebook_instance_without_direct_internet_access_configured.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_without_direct_internet_access_configured.sagemaker_notebook_instance_without_direct_internet_access_configured.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_without_direct_internet_access_configured.sagemaker_notebook_instance_without_direct_internet_access_configured import ( sagemaker_notebook_instance_without_direct_internet_access_configured, @@ -52,15 +49,12 @@ class Test_sagemaker_notebook_instance_without_direct_internet_access_configured aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_without_direct_internet_access_configured.sagemaker_notebook_instance_without_direct_internet_access_configured.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_without_direct_internet_access_configured.sagemaker_notebook_instance_without_direct_internet_access_configured.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_without_direct_internet_access_configured.sagemaker_notebook_instance_without_direct_internet_access_configured import ( sagemaker_notebook_instance_without_direct_internet_access_configured, @@ -93,15 +87,12 @@ class Test_sagemaker_notebook_instance_without_direct_internet_access_configured aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_without_direct_internet_access_configured.sagemaker_notebook_instance_without_direct_internet_access_configured.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_without_direct_internet_access_configured.sagemaker_notebook_instance_without_direct_internet_access_configured.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_notebook_instance_without_direct_internet_access_configured.sagemaker_notebook_instance_without_direct_internet_access_configured import ( sagemaker_notebook_instance_without_direct_internet_access_configured, diff --git a/tests/providers/aws/services/sagemaker/sagemaker_training_jobs_intercontainer_encryption_enabled/sagemaker_training_jobs_intercontainer_encryption_enabled_test.py b/tests/providers/aws/services/sagemaker/sagemaker_training_jobs_intercontainer_encryption_enabled/sagemaker_training_jobs_intercontainer_encryption_enabled_test.py index 327c787267..80814733f9 100644 --- a/tests/providers/aws/services/sagemaker/sagemaker_training_jobs_intercontainer_encryption_enabled/sagemaker_training_jobs_intercontainer_encryption_enabled_test.py +++ b/tests/providers/aws/services/sagemaker/sagemaker_training_jobs_intercontainer_encryption_enabled/sagemaker_training_jobs_intercontainer_encryption_enabled_test.py @@ -18,15 +18,12 @@ class Test_sagemaker_training_jobs_intercontainer_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_intercontainer_encryption_enabled.sagemaker_training_jobs_intercontainer_encryption_enabled.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_intercontainer_encryption_enabled.sagemaker_training_jobs_intercontainer_encryption_enabled.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_intercontainer_encryption_enabled.sagemaker_training_jobs_intercontainer_encryption_enabled import ( sagemaker_training_jobs_intercontainer_encryption_enabled, @@ -50,15 +47,12 @@ class Test_sagemaker_training_jobs_intercontainer_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_intercontainer_encryption_enabled.sagemaker_training_jobs_intercontainer_encryption_enabled.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_intercontainer_encryption_enabled.sagemaker_training_jobs_intercontainer_encryption_enabled.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_intercontainer_encryption_enabled.sagemaker_training_jobs_intercontainer_encryption_enabled import ( sagemaker_training_jobs_intercontainer_encryption_enabled, @@ -88,15 +82,12 @@ class Test_sagemaker_training_jobs_intercontainer_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_intercontainer_encryption_enabled.sagemaker_training_jobs_intercontainer_encryption_enabled.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_intercontainer_encryption_enabled.sagemaker_training_jobs_intercontainer_encryption_enabled.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_intercontainer_encryption_enabled.sagemaker_training_jobs_intercontainer_encryption_enabled import ( sagemaker_training_jobs_intercontainer_encryption_enabled, diff --git a/tests/providers/aws/services/sagemaker/sagemaker_training_jobs_network_isolation_enabled/sagemaker_training_jobs_network_isolation_enabled_test.py b/tests/providers/aws/services/sagemaker/sagemaker_training_jobs_network_isolation_enabled/sagemaker_training_jobs_network_isolation_enabled_test.py index 79fd0fbe64..c0b9409347 100644 --- a/tests/providers/aws/services/sagemaker/sagemaker_training_jobs_network_isolation_enabled/sagemaker_training_jobs_network_isolation_enabled_test.py +++ b/tests/providers/aws/services/sagemaker/sagemaker_training_jobs_network_isolation_enabled/sagemaker_training_jobs_network_isolation_enabled_test.py @@ -20,15 +20,12 @@ class Test_sagemaker_training_jobs_network_isolation_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_network_isolation_enabled.sagemaker_training_jobs_network_isolation_enabled.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_network_isolation_enabled.sagemaker_training_jobs_network_isolation_enabled.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_network_isolation_enabled.sagemaker_training_jobs_network_isolation_enabled import ( sagemaker_training_jobs_network_isolation_enabled, @@ -52,15 +49,12 @@ class Test_sagemaker_training_jobs_network_isolation_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_network_isolation_enabled.sagemaker_training_jobs_network_isolation_enabled.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_network_isolation_enabled.sagemaker_training_jobs_network_isolation_enabled.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_network_isolation_enabled.sagemaker_training_jobs_network_isolation_enabled import ( sagemaker_training_jobs_network_isolation_enabled, @@ -90,15 +84,12 @@ class Test_sagemaker_training_jobs_network_isolation_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_network_isolation_enabled.sagemaker_training_jobs_network_isolation_enabled.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_network_isolation_enabled.sagemaker_training_jobs_network_isolation_enabled.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_network_isolation_enabled.sagemaker_training_jobs_network_isolation_enabled import ( sagemaker_training_jobs_network_isolation_enabled, diff --git a/tests/providers/aws/services/sagemaker/sagemaker_training_jobs_volume_and_output_encryption_enabled/sagemaker_training_jobs_volume_and_output_encryption_enabled_test.py b/tests/providers/aws/services/sagemaker/sagemaker_training_jobs_volume_and_output_encryption_enabled/sagemaker_training_jobs_volume_and_output_encryption_enabled_test.py index ce105c2b2b..365efa0960 100644 --- a/tests/providers/aws/services/sagemaker/sagemaker_training_jobs_volume_and_output_encryption_enabled/sagemaker_training_jobs_volume_and_output_encryption_enabled_test.py +++ b/tests/providers/aws/services/sagemaker/sagemaker_training_jobs_volume_and_output_encryption_enabled/sagemaker_training_jobs_volume_and_output_encryption_enabled_test.py @@ -20,15 +20,12 @@ class Test_sagemaker_training_jobs_volume_and_output_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_volume_and_output_encryption_enabled.sagemaker_training_jobs_volume_and_output_encryption_enabled.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_volume_and_output_encryption_enabled.sagemaker_training_jobs_volume_and_output_encryption_enabled.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_volume_and_output_encryption_enabled.sagemaker_training_jobs_volume_and_output_encryption_enabled import ( sagemaker_training_jobs_volume_and_output_encryption_enabled, @@ -52,15 +49,12 @@ class Test_sagemaker_training_jobs_volume_and_output_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_volume_and_output_encryption_enabled.sagemaker_training_jobs_volume_and_output_encryption_enabled.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_volume_and_output_encryption_enabled.sagemaker_training_jobs_volume_and_output_encryption_enabled.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_volume_and_output_encryption_enabled.sagemaker_training_jobs_volume_and_output_encryption_enabled import ( sagemaker_training_jobs_volume_and_output_encryption_enabled, @@ -90,15 +84,12 @@ class Test_sagemaker_training_jobs_volume_and_output_encryption_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_volume_and_output_encryption_enabled.sagemaker_training_jobs_volume_and_output_encryption_enabled.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_volume_and_output_encryption_enabled.sagemaker_training_jobs_volume_and_output_encryption_enabled.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_volume_and_output_encryption_enabled.sagemaker_training_jobs_volume_and_output_encryption_enabled import ( sagemaker_training_jobs_volume_and_output_encryption_enabled, diff --git a/tests/providers/aws/services/sagemaker/sagemaker_training_jobs_vpc_settings_configured/sagemaker_training_jobs_vpc_settings_configured_test.py b/tests/providers/aws/services/sagemaker/sagemaker_training_jobs_vpc_settings_configured/sagemaker_training_jobs_vpc_settings_configured_test.py index 4dc54f99b8..c4a75a3f5b 100644 --- a/tests/providers/aws/services/sagemaker/sagemaker_training_jobs_vpc_settings_configured/sagemaker_training_jobs_vpc_settings_configured_test.py +++ b/tests/providers/aws/services/sagemaker/sagemaker_training_jobs_vpc_settings_configured/sagemaker_training_jobs_vpc_settings_configured_test.py @@ -20,15 +20,12 @@ class Test_sagemaker_training_jobs_vpc_settings_configured: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_vpc_settings_configured.sagemaker_training_jobs_vpc_settings_configured.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_vpc_settings_configured.sagemaker_training_jobs_vpc_settings_configured.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_vpc_settings_configured.sagemaker_training_jobs_vpc_settings_configured import ( sagemaker_training_jobs_vpc_settings_configured, @@ -52,15 +49,12 @@ class Test_sagemaker_training_jobs_vpc_settings_configured: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_vpc_settings_configured.sagemaker_training_jobs_vpc_settings_configured.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_vpc_settings_configured.sagemaker_training_jobs_vpc_settings_configured.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_vpc_settings_configured.sagemaker_training_jobs_vpc_settings_configured import ( sagemaker_training_jobs_vpc_settings_configured, @@ -90,15 +84,12 @@ class Test_sagemaker_training_jobs_vpc_settings_configured: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_vpc_settings_configured.sagemaker_training_jobs_vpc_settings_configured.sagemaker_client", - sagemaker_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_vpc_settings_configured.sagemaker_training_jobs_vpc_settings_configured.sagemaker_client", + sagemaker_client, ): from prowler.providers.aws.services.sagemaker.sagemaker_training_jobs_vpc_settings_configured.sagemaker_training_jobs_vpc_settings_configured import ( sagemaker_training_jobs_vpc_settings_configured, diff --git a/tests/providers/aws/services/secretsmanager/secretsmanager_automatic_rotation_enabled/secretsmanager_automatic_rotation_enabled_test.py b/tests/providers/aws/services/secretsmanager/secretsmanager_automatic_rotation_enabled/secretsmanager_automatic_rotation_enabled_test.py index e953ca1d9d..6376b4fffb 100644 --- a/tests/providers/aws/services/secretsmanager/secretsmanager_automatic_rotation_enabled/secretsmanager_automatic_rotation_enabled_test.py +++ b/tests/providers/aws/services/secretsmanager/secretsmanager_automatic_rotation_enabled/secretsmanager_automatic_rotation_enabled_test.py @@ -10,15 +10,12 @@ class Test_secretsmanager_automatic_rotation_enabled: secretsmanager_client = mock.MagicMock secretsmanager_client.secrets = {} - with ( - mock.patch( - "prowler.providers.aws.services.secretsmanager.secretsmanager_service.SecretsManager", - new=secretsmanager_client, - ), - mock.patch( - "prowler.providers.aws.services.secretsmanager.secretsmanager_client.secretsmanager_client", - new=secretsmanager_client, - ), + with mock.patch( + "prowler.providers.aws.services.secretsmanager.secretsmanager_service.SecretsManager", + new=secretsmanager_client, + ), mock.patch( + "prowler.providers.aws.services.secretsmanager.secretsmanager_client.secretsmanager_client", + new=secretsmanager_client, ): # Test Check from prowler.providers.aws.services.secretsmanager.secretsmanager_automatic_rotation_enabled.secretsmanager_automatic_rotation_enabled import ( @@ -44,15 +41,12 @@ class Test_secretsmanager_automatic_rotation_enabled: last_rotated_date=datetime.min, ) } - with ( - mock.patch( - "prowler.providers.aws.services.secretsmanager.secretsmanager_service.SecretsManager", - new=secretsmanager_client, - ), - mock.patch( - "prowler.providers.aws.services.secretsmanager.secretsmanager_client.secretsmanager_client", - new=secretsmanager_client, - ), + with mock.patch( + "prowler.providers.aws.services.secretsmanager.secretsmanager_service.SecretsManager", + new=secretsmanager_client, + ), mock.patch( + "prowler.providers.aws.services.secretsmanager.secretsmanager_client.secretsmanager_client", + new=secretsmanager_client, ): # Test Check from prowler.providers.aws.services.secretsmanager.secretsmanager_automatic_rotation_enabled.secretsmanager_automatic_rotation_enabled import ( @@ -86,15 +80,12 @@ class Test_secretsmanager_automatic_rotation_enabled: last_rotated_date=datetime.min, ) } - with ( - mock.patch( - "prowler.providers.aws.services.secretsmanager.secretsmanager_service.SecretsManager", - new=secretsmanager_client, - ), - mock.patch( - "prowler.providers.aws.services.secretsmanager.secretsmanager_client.secretsmanager_client", - new=secretsmanager_client, - ), + with mock.patch( + "prowler.providers.aws.services.secretsmanager.secretsmanager_service.SecretsManager", + new=secretsmanager_client, + ), mock.patch( + "prowler.providers.aws.services.secretsmanager.secretsmanager_client.secretsmanager_client", + new=secretsmanager_client, ): # Test Check from prowler.providers.aws.services.secretsmanager.secretsmanager_automatic_rotation_enabled.secretsmanager_automatic_rotation_enabled import ( diff --git a/tests/providers/aws/services/secretsmanager/secretsmanager_not_publicly_accessible/secretsmanager_not_publicly_accessible_test.py b/tests/providers/aws/services/secretsmanager/secretsmanager_not_publicly_accessible/secretsmanager_not_publicly_accessible_test.py index 88ecb8ab18..92c6a23063 100644 --- a/tests/providers/aws/services/secretsmanager/secretsmanager_not_publicly_accessible/secretsmanager_not_publicly_accessible_test.py +++ b/tests/providers/aws/services/secretsmanager/secretsmanager_not_publicly_accessible/secretsmanager_not_publicly_accessible_test.py @@ -15,15 +15,12 @@ class Test_secretsmanager_not_publicly_accessible: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.secretsmanager.secretsmanager_not_publicly_accessible.secretsmanager_not_publicly_accessible.secretsmanager_client", - new=SecretsManager(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.secretsmanager.secretsmanager_not_publicly_accessible.secretsmanager_not_publicly_accessible.secretsmanager_client", + new=SecretsManager(aws_provider), ): # Test Check from prowler.providers.aws.services.secretsmanager.secretsmanager_not_publicly_accessible.secretsmanager_not_publicly_accessible import ( @@ -50,15 +47,12 @@ class Test_secretsmanager_not_publicly_accessible: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.secretsmanager.secretsmanager_not_publicly_accessible.secretsmanager_not_publicly_accessible.secretsmanager_client", - new=SecretsManager(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.secretsmanager.secretsmanager_not_publicly_accessible.secretsmanager_not_publicly_accessible.secretsmanager_client", + new=SecretsManager(aws_provider), ): # Test Check from prowler.providers.aws.services.secretsmanager.secretsmanager_not_publicly_accessible.secretsmanager_not_publicly_accessible import ( @@ -93,15 +87,12 @@ class Test_secretsmanager_not_publicly_accessible: aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.secretsmanager.secretsmanager_not_publicly_accessible.secretsmanager_not_publicly_accessible.secretsmanager_client", - new=SecretsManager(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.secretsmanager.secretsmanager_not_publicly_accessible.secretsmanager_not_publicly_accessible.secretsmanager_client", + new=SecretsManager(aws_provider), ): # Test Check from prowler.providers.aws.services.secretsmanager.secretsmanager_not_publicly_accessible.secretsmanager_not_publicly_accessible import ( diff --git a/tests/providers/aws/services/secretsmanager/secretsmanager_secret_rotated_periodically/secretsmanager_secret_rotated_periodically_test.py b/tests/providers/aws/services/secretsmanager/secretsmanager_secret_rotated_periodically/secretsmanager_secret_rotated_periodically_test.py index 8dcf4b436f..f3053c396c 100644 --- a/tests/providers/aws/services/secretsmanager/secretsmanager_secret_rotated_periodically/secretsmanager_secret_rotated_periodically_test.py +++ b/tests/providers/aws/services/secretsmanager/secretsmanager_secret_rotated_periodically/secretsmanager_secret_rotated_periodically_test.py @@ -61,15 +61,12 @@ class Test_secretsmanager_secret_rotated_periodically: SecretsManager, ) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.secretsmanager.secretsmanager_secret_rotated_periodically.secretsmanager_secret_rotated_periodically.secretsmanager_client", - new=SecretsManager(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.secretsmanager.secretsmanager_secret_rotated_periodically.secretsmanager_secret_rotated_periodically.secretsmanager_client", + new=SecretsManager(aws_provider), ): # Test Check from prowler.providers.aws.services.secretsmanager.secretsmanager_secret_rotated_periodically.secretsmanager_secret_rotated_periodically import ( @@ -100,15 +97,12 @@ class Test_secretsmanager_secret_rotated_periodically: SecretsManager, ) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.secretsmanager.secretsmanager_secret_rotated_periodically.secretsmanager_secret_rotated_periodically.secretsmanager_client", - new=SecretsManager(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.secretsmanager.secretsmanager_secret_rotated_periodically.secretsmanager_secret_rotated_periodically.secretsmanager_client", + new=SecretsManager(aws_provider), ): from prowler.providers.aws.services.secretsmanager.secretsmanager_secret_rotated_periodically.secretsmanager_secret_rotated_periodically import ( secretsmanager_secret_rotated_periodically, @@ -141,15 +135,12 @@ class Test_secretsmanager_secret_rotated_periodically: SecretsManager, ) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.secretsmanager.secretsmanager_secret_rotated_periodically.secretsmanager_secret_rotated_periodically.secretsmanager_client", - new=SecretsManager(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.secretsmanager.secretsmanager_secret_rotated_periodically.secretsmanager_secret_rotated_periodically.secretsmanager_client", + new=SecretsManager(aws_provider), ): from prowler.providers.aws.services.secretsmanager.secretsmanager_secret_rotated_periodically.secretsmanager_secret_rotated_periodically import ( secretsmanager_secret_rotated_periodically, @@ -185,15 +176,12 @@ class Test_secretsmanager_secret_rotated_periodically: SecretsManager, ) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.secretsmanager.secretsmanager_secret_rotated_periodically.secretsmanager_secret_rotated_periodically.secretsmanager_client", - new=SecretsManager(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.secretsmanager.secretsmanager_secret_rotated_periodically.secretsmanager_secret_rotated_periodically.secretsmanager_client", + new=SecretsManager(aws_provider), ): from prowler.providers.aws.services.secretsmanager.secretsmanager_secret_rotated_periodically.secretsmanager_secret_rotated_periodically import ( secretsmanager_secret_rotated_periodically, diff --git a/tests/providers/aws/services/secretsmanager/secretsmanager_secret_unused/secretsmanager_secret_unused_test.py b/tests/providers/aws/services/secretsmanager/secretsmanager_secret_unused/secretsmanager_secret_unused_test.py index a92082d5c5..fdee11e253 100644 --- a/tests/providers/aws/services/secretsmanager/secretsmanager_secret_unused/secretsmanager_secret_unused_test.py +++ b/tests/providers/aws/services/secretsmanager/secretsmanager_secret_unused/secretsmanager_secret_unused_test.py @@ -61,15 +61,12 @@ class Test_secretsmanager_secret_unused: SecretsManager, ) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.secretsmanager.secretsmanager_secret_unused.secretsmanager_secret_unused.secretsmanager_client", - new=SecretsManager(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.secretsmanager.secretsmanager_secret_unused.secretsmanager_secret_unused.secretsmanager_client", + new=SecretsManager(aws_provider), ): # Test Check from prowler.providers.aws.services.secretsmanager.secretsmanager_secret_unused.secretsmanager_secret_unused import ( @@ -100,15 +97,12 @@ class Test_secretsmanager_secret_unused: SecretsManager, ) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.secretsmanager.secretsmanager_secret_unused.secretsmanager_secret_unused.secretsmanager_client", - new=SecretsManager(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.secretsmanager.secretsmanager_secret_unused.secretsmanager_secret_unused.secretsmanager_client", + new=SecretsManager(aws_provider), ): from prowler.providers.aws.services.secretsmanager.secretsmanager_secret_unused.secretsmanager_secret_unused import ( secretsmanager_secret_unused, @@ -141,15 +135,12 @@ class Test_secretsmanager_secret_unused: SecretsManager, ) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.secretsmanager.secretsmanager_secret_unused.secretsmanager_secret_unused.secretsmanager_client", - new=SecretsManager(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.secretsmanager.secretsmanager_secret_unused.secretsmanager_secret_unused.secretsmanager_client", + new=SecretsManager(aws_provider), ): from prowler.providers.aws.services.secretsmanager.secretsmanager_secret_unused.secretsmanager_secret_unused import ( secretsmanager_secret_unused, @@ -187,15 +178,12 @@ class Test_secretsmanager_secret_unused: SecretsManager, ) - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - patch( - "prowler.providers.aws.services.secretsmanager.secretsmanager_secret_unused.secretsmanager_secret_unused.secretsmanager_client", - new=SecretsManager(aws_provider), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), patch( + "prowler.providers.aws.services.secretsmanager.secretsmanager_secret_unused.secretsmanager_secret_unused.secretsmanager_client", + new=SecretsManager(aws_provider), ): from prowler.providers.aws.services.secretsmanager.secretsmanager_secret_unused.secretsmanager_secret_unused import ( secretsmanager_secret_unused, diff --git a/tests/providers/aws/services/securityhub/securityhub_enabled/securityhub_enabled_test.py b/tests/providers/aws/services/securityhub/securityhub_enabled/securityhub_enabled_test.py index 030aa7fdad..a34e784ac3 100644 --- a/tests/providers/aws/services/securityhub/securityhub_enabled/securityhub_enabled_test.py +++ b/tests/providers/aws/services/securityhub/securityhub_enabled/securityhub_enabled_test.py @@ -24,15 +24,12 @@ class Test_securityhub_enabled: tags=[{"test_key": "test_value"}], ) ] - with ( - mock.patch( - "prowler.providers.aws.services.securityhub.securityhub_service.SecurityHub", - new=securityhub_client, - ), - mock.patch( - "prowler.providers.aws.services.securityhub.securityhub_service.SecurityHub.get_unknown_arn", - return_value="arn:aws:securityhub:eu-west-1:0123456789012:hub/unknown", - ), + with mock.patch( + "prowler.providers.aws.services.securityhub.securityhub_service.SecurityHub", + new=securityhub_client, + ), mock.patch( + "prowler.providers.aws.services.securityhub.securityhub_service.SecurityHub.get_unknown_arn", + return_value="arn:aws:securityhub:eu-west-1:0123456789012:hub/unknown", ): # Test Check from prowler.providers.aws.services.securityhub.securityhub_enabled.securityhub_enabled import ( diff --git a/tests/providers/aws/services/ses/ses_identity_not_publicly_accessible/ses_identity_not_publicly_accessible_test.py b/tests/providers/aws/services/ses/ses_identity_not_publicly_accessible/ses_identity_not_publicly_accessible_test.py index 15b8a9dd32..19d524726b 100644 --- a/tests/providers/aws/services/ses/ses_identity_not_publicly_accessible/ses_identity_not_publicly_accessible_test.py +++ b/tests/providers/aws/services/ses/ses_identity_not_publicly_accessible/ses_identity_not_publicly_accessible_test.py @@ -60,15 +60,12 @@ class Test_ses_identities_not_publicly_accessible: client("sesv2", region_name=AWS_REGION_EU_WEST_1) aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ses.ses_identity_not_publicly_accessible.ses_identity_not_publicly_accessible.ses_client", - new=SES(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ses.ses_identity_not_publicly_accessible.ses_identity_not_publicly_accessible.ses_client", + new=SES(aws_provider), ): from prowler.providers.aws.services.ses.ses_identity_not_publicly_accessible.ses_identity_not_publicly_accessible import ( ses_identity_not_publicly_accessible, @@ -84,15 +81,12 @@ class Test_ses_identities_not_publicly_accessible: client("sesv2", region_name=AWS_REGION_EU_WEST_1) aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ses.ses_identity_not_publicly_accessible.ses_identity_not_publicly_accessible.ses_client", - new=SES(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ses.ses_identity_not_publicly_accessible.ses_identity_not_publicly_accessible.ses_client", + new=SES(aws_provider), ): from prowler.providers.aws.services.ses.ses_identity_not_publicly_accessible.ses_identity_not_publicly_accessible import ( ses_identity_not_publicly_accessible, @@ -120,15 +114,12 @@ class Test_ses_identities_not_publicly_accessible: client("sesv2", region_name=AWS_REGION_EU_WEST_1) aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.ses.ses_identity_not_publicly_accessible.ses_identity_not_publicly_accessible.ses_client", - new=SES(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.ses.ses_identity_not_publicly_accessible.ses_identity_not_publicly_accessible.ses_client", + new=SES(aws_provider), ): from prowler.providers.aws.services.ses.ses_identity_not_publicly_accessible.ses_identity_not_publicly_accessible import ( ses_identity_not_publicly_accessible, diff --git a/tests/providers/aws/services/shield/shield_advanced_protection_in_cloudfront_distributions/shield_advanced_protection_in_cloudfront_distributions_test.py b/tests/providers/aws/services/shield/shield_advanced_protection_in_cloudfront_distributions/shield_advanced_protection_in_cloudfront_distributions_test.py index f712093af1..3d7883db61 100644 --- a/tests/providers/aws/services/shield/shield_advanced_protection_in_cloudfront_distributions/shield_advanced_protection_in_cloudfront_distributions_test.py +++ b/tests/providers/aws/services/shield/shield_advanced_protection_in_cloudfront_distributions/shield_advanced_protection_in_cloudfront_distributions_test.py @@ -12,15 +12,12 @@ class Test_shield_advanced_protection_in_cloudfront_distributions: shield_client.enabled = False # CloudFront Client cloudfront_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.aws.services.shield.shield_service.Shield", - new=shield_client, - ), - mock.patch( - "prowler.providers.aws.services.cloudfront.cloudfront_service.CloudFront", - new=cloudfront_client, - ), + with mock.patch( + "prowler.providers.aws.services.shield.shield_service.Shield", + new=shield_client, + ), mock.patch( + "prowler.providers.aws.services.cloudfront.cloudfront_service.CloudFront", + new=cloudfront_client, ): # Test Check from prowler.providers.aws.services.shield.shield_advanced_protection_in_cloudfront_distributions.shield_advanced_protection_in_cloudfront_distributions import ( @@ -63,15 +60,12 @@ class Test_shield_advanced_protection_in_cloudfront_distributions: ) } - with ( - mock.patch( - "prowler.providers.aws.services.shield.shield_service.Shield", - new=shield_client, - ), - mock.patch( - "prowler.providers.aws.services.cloudfront.cloudfront_service.CloudFront", - new=cloudfront_client, - ), + with mock.patch( + "prowler.providers.aws.services.shield.shield_service.Shield", + new=shield_client, + ), mock.patch( + "prowler.providers.aws.services.cloudfront.cloudfront_service.CloudFront", + new=cloudfront_client, ): # Test Check from prowler.providers.aws.services.shield.shield_advanced_protection_in_cloudfront_distributions.shield_advanced_protection_in_cloudfront_distributions import ( @@ -113,15 +107,12 @@ class Test_shield_advanced_protection_in_cloudfront_distributions: shield_client.region = AWS_REGION_EU_WEST_1 shield_client.protections = {} - with ( - mock.patch( - "prowler.providers.aws.services.shield.shield_service.Shield", - new=shield_client, - ), - mock.patch( - "prowler.providers.aws.services.cloudfront.cloudfront_service.CloudFront", - new=cloudfront_client, - ), + with mock.patch( + "prowler.providers.aws.services.shield.shield_service.Shield", + new=shield_client, + ), mock.patch( + "prowler.providers.aws.services.cloudfront.cloudfront_service.CloudFront", + new=cloudfront_client, ): # Test Check from prowler.providers.aws.services.shield.shield_advanced_protection_in_cloudfront_distributions.shield_advanced_protection_in_cloudfront_distributions import ( @@ -163,15 +154,12 @@ class Test_shield_advanced_protection_in_cloudfront_distributions: shield_client.region = AWS_REGION_EU_WEST_1 shield_client.protections = {} - with ( - mock.patch( - "prowler.providers.aws.services.shield.shield_service.Shield", - new=shield_client, - ), - mock.patch( - "prowler.providers.aws.services.cloudfront.cloudfront_service.CloudFront", - new=cloudfront_client, - ), + with mock.patch( + "prowler.providers.aws.services.shield.shield_service.Shield", + new=shield_client, + ), mock.patch( + "prowler.providers.aws.services.cloudfront.cloudfront_service.CloudFront", + new=cloudfront_client, ): # Test Check from prowler.providers.aws.services.shield.shield_advanced_protection_in_cloudfront_distributions.shield_advanced_protection_in_cloudfront_distributions import ( diff --git a/tests/providers/aws/services/shield/shield_advanced_protection_in_global_accelerators/shield_advanced_protection_in_global_accelerators_test.py b/tests/providers/aws/services/shield/shield_advanced_protection_in_global_accelerators/shield_advanced_protection_in_global_accelerators_test.py index be6a03d921..eb1a04bbf1 100644 --- a/tests/providers/aws/services/shield/shield_advanced_protection_in_global_accelerators/shield_advanced_protection_in_global_accelerators_test.py +++ b/tests/providers/aws/services/shield/shield_advanced_protection_in_global_accelerators/shield_advanced_protection_in_global_accelerators_test.py @@ -14,15 +14,12 @@ class Test_shield_advanced_protection_in_global_accelerators: shield_client.enabled = False # GlobalAccelerator Client globalaccelerator_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.aws.services.shield.shield_service.Shield", - new=shield_client, - ), - mock.patch( - "prowler.providers.aws.services.globalaccelerator.globalaccelerator_service.GlobalAccelerator", - new=globalaccelerator_client, - ), + with mock.patch( + "prowler.providers.aws.services.shield.shield_service.Shield", + new=shield_client, + ), mock.patch( + "prowler.providers.aws.services.globalaccelerator.globalaccelerator_service.GlobalAccelerator", + new=globalaccelerator_client, ): # Test Check from prowler.providers.aws.services.shield.shield_advanced_protection_in_global_accelerators.shield_advanced_protection_in_global_accelerators import ( @@ -65,15 +62,12 @@ class Test_shield_advanced_protection_in_global_accelerators: ) } - with ( - mock.patch( - "prowler.providers.aws.services.shield.shield_service.Shield", - new=shield_client, - ), - mock.patch( - "prowler.providers.aws.services.globalaccelerator.globalaccelerator_service.GlobalAccelerator", - new=globalaccelerator_client, - ), + with mock.patch( + "prowler.providers.aws.services.shield.shield_service.Shield", + new=shield_client, + ), mock.patch( + "prowler.providers.aws.services.globalaccelerator.globalaccelerator_service.GlobalAccelerator", + new=globalaccelerator_client, ): # Test Check from prowler.providers.aws.services.shield.shield_advanced_protection_in_global_accelerators.shield_advanced_protection_in_global_accelerators import ( @@ -118,15 +112,12 @@ class Test_shield_advanced_protection_in_global_accelerators: shield_client.region = AWS_REGION_EU_WEST_1 shield_client.protections = {} - with ( - mock.patch( - "prowler.providers.aws.services.shield.shield_service.Shield", - new=shield_client, - ), - mock.patch( - "prowler.providers.aws.services.globalaccelerator.globalaccelerator_service.GlobalAccelerator", - new=globalaccelerator_client, - ), + with mock.patch( + "prowler.providers.aws.services.shield.shield_service.Shield", + new=shield_client, + ), mock.patch( + "prowler.providers.aws.services.globalaccelerator.globalaccelerator_service.GlobalAccelerator", + new=globalaccelerator_client, ): # Test Check from prowler.providers.aws.services.shield.shield_advanced_protection_in_global_accelerators.shield_advanced_protection_in_global_accelerators import ( @@ -170,15 +161,12 @@ class Test_shield_advanced_protection_in_global_accelerators: shield_client.region = AWS_REGION_EU_WEST_1 shield_client.protections = {} - with ( - mock.patch( - "prowler.providers.aws.services.shield.shield_service.Shield", - new=shield_client, - ), - mock.patch( - "prowler.providers.aws.services.globalaccelerator.globalaccelerator_service.GlobalAccelerator", - new=globalaccelerator_client, - ), + with mock.patch( + "prowler.providers.aws.services.shield.shield_service.Shield", + new=shield_client, + ), mock.patch( + "prowler.providers.aws.services.globalaccelerator.globalaccelerator_service.GlobalAccelerator", + new=globalaccelerator_client, ): # Test Check from prowler.providers.aws.services.shield.shield_advanced_protection_in_global_accelerators.shield_advanced_protection_in_global_accelerators import ( diff --git a/tests/providers/aws/services/shield/shield_advanced_protection_in_route53_hosted_zones/shield_advanced_protection_in_route53_hosted_zones_test.py b/tests/providers/aws/services/shield/shield_advanced_protection_in_route53_hosted_zones/shield_advanced_protection_in_route53_hosted_zones_test.py index b56b0f572a..e3f460eda2 100644 --- a/tests/providers/aws/services/shield/shield_advanced_protection_in_route53_hosted_zones/shield_advanced_protection_in_route53_hosted_zones_test.py +++ b/tests/providers/aws/services/shield/shield_advanced_protection_in_route53_hosted_zones/shield_advanced_protection_in_route53_hosted_zones_test.py @@ -12,19 +12,15 @@ class Test_shield_advanced_protection_in_route53_hosted_zones: shield_client.enabled = False # Route53 Client route53_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.aws.services.shield.shield_service.Shield", - new=shield_client, - ), - mock.patch( - "prowler.providers.aws.services.route53.route53_service.Route53", - new=shield_client, - ), - mock.patch( - "prowler.providers.aws.services.shield.shield_advanced_protection_in_route53_hosted_zones.shield_advanced_protection_in_route53_hosted_zones.route53_client", - new=route53_client, - ), + with mock.patch( + "prowler.providers.aws.services.shield.shield_service.Shield", + new=shield_client, + ), mock.patch( + "prowler.providers.aws.services.route53.route53_service.Route53", + new=shield_client, + ), mock.patch( + "prowler.providers.aws.services.shield.shield_advanced_protection_in_route53_hosted_zones.shield_advanced_protection_in_route53_hosted_zones.route53_client", + new=route53_client, ): # Test Check from prowler.providers.aws.services.shield.shield_advanced_protection_in_route53_hosted_zones.shield_advanced_protection_in_route53_hosted_zones import ( @@ -69,19 +65,15 @@ class Test_shield_advanced_protection_in_route53_hosted_zones: ) } - with ( - mock.patch( - "prowler.providers.aws.services.shield.shield_service.Shield", - new=shield_client, - ), - mock.patch( - "prowler.providers.aws.services.route53.route53_service.Route53", - new=shield_client, - ), - mock.patch( - "prowler.providers.aws.services.shield.shield_advanced_protection_in_route53_hosted_zones.shield_advanced_protection_in_route53_hosted_zones.route53_client", - new=route53_client, - ), + with mock.patch( + "prowler.providers.aws.services.shield.shield_service.Shield", + new=shield_client, + ), mock.patch( + "prowler.providers.aws.services.route53.route53_service.Route53", + new=shield_client, + ), mock.patch( + "prowler.providers.aws.services.shield.shield_advanced_protection_in_route53_hosted_zones.shield_advanced_protection_in_route53_hosted_zones.route53_client", + new=route53_client, ): # Test Check from prowler.providers.aws.services.shield.shield_advanced_protection_in_route53_hosted_zones.shield_advanced_protection_in_route53_hosted_zones import ( @@ -125,19 +117,15 @@ class Test_shield_advanced_protection_in_route53_hosted_zones: shield_client.region = AWS_REGION_EU_WEST_1 shield_client.protections = {} - with ( - mock.patch( - "prowler.providers.aws.services.shield.shield_service.Shield", - new=shield_client, - ), - mock.patch( - "prowler.providers.aws.services.route53.route53_service.Route53", - new=shield_client, - ), - mock.patch( - "prowler.providers.aws.services.shield.shield_advanced_protection_in_route53_hosted_zones.shield_advanced_protection_in_route53_hosted_zones.route53_client", - new=route53_client, - ), + with mock.patch( + "prowler.providers.aws.services.shield.shield_service.Shield", + new=shield_client, + ), mock.patch( + "prowler.providers.aws.services.route53.route53_service.Route53", + new=shield_client, + ), mock.patch( + "prowler.providers.aws.services.shield.shield_advanced_protection_in_route53_hosted_zones.shield_advanced_protection_in_route53_hosted_zones.route53_client", + new=route53_client, ): # Test Check from prowler.providers.aws.services.shield.shield_advanced_protection_in_route53_hosted_zones.shield_advanced_protection_in_route53_hosted_zones import ( @@ -181,19 +169,15 @@ class Test_shield_advanced_protection_in_route53_hosted_zones: shield_client.region = AWS_REGION_EU_WEST_1 shield_client.protections = {} - with ( - mock.patch( - "prowler.providers.aws.services.shield.shield_service.Shield", - new=shield_client, - ), - mock.patch( - "prowler.providers.aws.services.route53.route53_service.Route53", - new=shield_client, - ), - mock.patch( - "prowler.providers.aws.services.shield.shield_advanced_protection_in_route53_hosted_zones.shield_advanced_protection_in_route53_hosted_zones.route53_client", - new=route53_client, - ), + with mock.patch( + "prowler.providers.aws.services.shield.shield_service.Shield", + new=shield_client, + ), mock.patch( + "prowler.providers.aws.services.route53.route53_service.Route53", + new=shield_client, + ), mock.patch( + "prowler.providers.aws.services.shield.shield_advanced_protection_in_route53_hosted_zones.shield_advanced_protection_in_route53_hosted_zones.route53_client", + new=route53_client, ): # Test Check from prowler.providers.aws.services.shield.shield_advanced_protection_in_route53_hosted_zones.shield_advanced_protection_in_route53_hosted_zones import ( diff --git a/tests/providers/aws/services/sqs/sqs_queues_not_publicly_accessible/sqs_queues_not_publicly_accessible_fixer_test.py b/tests/providers/aws/services/sqs/sqs_queues_not_publicly_accessible/sqs_queues_not_publicly_accessible_fixer_test.py index dfef1e6952..8603573e3a 100644 --- a/tests/providers/aws/services/sqs/sqs_queues_not_publicly_accessible/sqs_queues_not_publicly_accessible_fixer_test.py +++ b/tests/providers/aws/services/sqs/sqs_queues_not_publicly_accessible/sqs_queues_not_publicly_accessible_fixer_test.py @@ -25,15 +25,12 @@ class Test_sqs_queues_not_publicly_accessible_fixer: from prowler.providers.aws.services.sqs.sqs_service import SQS - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sqs.sqs_queues_not_publicly_accessible.sqs_queues_not_publicly_accessible_fixer.sqs_client", - new=SQS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sqs.sqs_queues_not_publicly_accessible.sqs_queues_not_publicly_accessible_fixer.sqs_client", + new=SQS(aws_provider), ): # Test Fixer from prowler.providers.aws.services.sqs.sqs_queues_not_publicly_accessible.sqs_queues_not_publicly_accessible_fixer import ( @@ -61,15 +58,12 @@ class Test_sqs_queues_not_publicly_accessible_fixer: from prowler.providers.aws.services.sqs.sqs_service import SQS - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sqs.sqs_queues_not_publicly_accessible.sqs_queues_not_publicly_accessible_fixer.sqs_client", - new=SQS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sqs.sqs_queues_not_publicly_accessible.sqs_queues_not_publicly_accessible_fixer.sqs_client", + new=SQS(aws_provider), ): # Test Fixer from prowler.providers.aws.services.sqs.sqs_queues_not_publicly_accessible.sqs_queues_not_publicly_accessible_fixer import ( @@ -97,15 +91,12 @@ class Test_sqs_queues_not_publicly_accessible_fixer: from prowler.providers.aws.services.sqs.sqs_service import SQS - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.sqs.sqs_queues_not_publicly_accessible.sqs_queues_not_publicly_accessible_fixer.sqs_client", - new=SQS(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.sqs.sqs_queues_not_publicly_accessible.sqs_queues_not_publicly_accessible_fixer.sqs_client", + new=SQS(aws_provider), ): # Test Fixer from prowler.providers.aws.services.sqs.sqs_queues_not_publicly_accessible.sqs_queues_not_publicly_accessible_fixer import ( diff --git a/tests/providers/aws/services/sqs/sqs_queues_not_publicly_accessible/sqs_queues_not_publicly_accessible_test.py b/tests/providers/aws/services/sqs/sqs_queues_not_publicly_accessible/sqs_queues_not_publicly_accessible_test.py index 8c694a5b19..508f1725a9 100644 --- a/tests/providers/aws/services/sqs/sqs_queues_not_publicly_accessible/sqs_queues_not_publicly_accessible_test.py +++ b/tests/providers/aws/services/sqs/sqs_queues_not_publicly_accessible/sqs_queues_not_publicly_accessible_test.py @@ -108,15 +108,12 @@ class Test_sqs_queues_not_publicly_accessible: def test_no_queues(self): sqs_client = mock.MagicMock sqs_client.queues = [] - with ( - mock.patch( - "prowler.providers.aws.services.sqs.sqs_service.SQS", - new=sqs_client, - ), - mock.patch( - "prowler.providers.aws.services.sqs.sqs_client.sqs_client", - new=sqs_client, - ), + with mock.patch( + "prowler.providers.aws.services.sqs.sqs_service.SQS", + new=sqs_client, + ), mock.patch( + "prowler.providers.aws.services.sqs.sqs_client.sqs_client", + new=sqs_client, ): from prowler.providers.aws.services.sqs.sqs_queues_not_publicly_accessible.sqs_queues_not_publicly_accessible import ( sqs_queues_not_publicly_accessible, @@ -138,15 +135,12 @@ class Test_sqs_queues_not_publicly_accessible: arn=test_queue_arn, ) ) - with ( - mock.patch( - "prowler.providers.aws.services.sqs.sqs_service.SQS", - new=sqs_client, - ), - mock.patch( - "prowler.providers.aws.services.sqs.sqs_client.sqs_client", - new=sqs_client, - ), + with mock.patch( + "prowler.providers.aws.services.sqs.sqs_service.SQS", + new=sqs_client, + ), mock.patch( + "prowler.providers.aws.services.sqs.sqs_client.sqs_client", + new=sqs_client, ): from prowler.providers.aws.services.sqs.sqs_queues_not_publicly_accessible.sqs_queues_not_publicly_accessible import ( sqs_queues_not_publicly_accessible, @@ -177,15 +171,12 @@ class Test_sqs_queues_not_publicly_accessible: arn=test_queue_arn, ) ) - with ( - mock.patch( - "prowler.providers.aws.services.sqs.sqs_service.SQS", - new=sqs_client, - ), - mock.patch( - "prowler.providers.aws.services.sqs.sqs_client.sqs_client", - new=sqs_client, - ), + with mock.patch( + "prowler.providers.aws.services.sqs.sqs_service.SQS", + new=sqs_client, + ), mock.patch( + "prowler.providers.aws.services.sqs.sqs_client.sqs_client", + new=sqs_client, ): from prowler.providers.aws.services.sqs.sqs_queues_not_publicly_accessible.sqs_queues_not_publicly_accessible import ( sqs_queues_not_publicly_accessible, @@ -217,15 +208,12 @@ class Test_sqs_queues_not_publicly_accessible: arn=test_queue_arn, ) ) - with ( - mock.patch( - "prowler.providers.aws.services.sqs.sqs_service.SQS", - new=sqs_client, - ), - mock.patch( - "prowler.providers.aws.services.sqs.sqs_client.sqs_client", - new=sqs_client, - ), + with mock.patch( + "prowler.providers.aws.services.sqs.sqs_service.SQS", + new=sqs_client, + ), mock.patch( + "prowler.providers.aws.services.sqs.sqs_client.sqs_client", + new=sqs_client, ): from prowler.providers.aws.services.sqs.sqs_queues_not_publicly_accessible.sqs_queues_not_publicly_accessible import ( sqs_queues_not_publicly_accessible, @@ -257,15 +245,12 @@ class Test_sqs_queues_not_publicly_accessible: arn=test_queue_arn, ) ) - with ( - mock.patch( - "prowler.providers.aws.services.sqs.sqs_service.SQS", - new=sqs_client, - ), - mock.patch( - "prowler.providers.aws.services.sqs.sqs_client.sqs_client", - new=sqs_client, - ), + with mock.patch( + "prowler.providers.aws.services.sqs.sqs_service.SQS", + new=sqs_client, + ), mock.patch( + "prowler.providers.aws.services.sqs.sqs_client.sqs_client", + new=sqs_client, ): from prowler.providers.aws.services.sqs.sqs_queues_not_publicly_accessible.sqs_queues_not_publicly_accessible import ( sqs_queues_not_publicly_accessible, @@ -297,15 +282,12 @@ class Test_sqs_queues_not_publicly_accessible: arn=test_queue_arn, ) ) - with ( - mock.patch( - "prowler.providers.aws.services.sqs.sqs_service.SQS", - new=sqs_client, - ), - mock.patch( - "prowler.providers.aws.services.sqs.sqs_client.sqs_client", - new=sqs_client, - ), + with mock.patch( + "prowler.providers.aws.services.sqs.sqs_service.SQS", + new=sqs_client, + ), mock.patch( + "prowler.providers.aws.services.sqs.sqs_client.sqs_client", + new=sqs_client, ): from prowler.providers.aws.services.sqs.sqs_queues_not_publicly_accessible.sqs_queues_not_publicly_accessible import ( sqs_queues_not_publicly_accessible, @@ -337,15 +319,12 @@ class Test_sqs_queues_not_publicly_accessible: arn=test_queue_arn, ) ) - with ( - mock.patch( - "prowler.providers.aws.services.sqs.sqs_service.SQS", - new=sqs_client, - ), - mock.patch( - "prowler.providers.aws.services.sqs.sqs_client.sqs_client", - new=sqs_client, - ), + with mock.patch( + "prowler.providers.aws.services.sqs.sqs_service.SQS", + new=sqs_client, + ), mock.patch( + "prowler.providers.aws.services.sqs.sqs_client.sqs_client", + new=sqs_client, ): from prowler.providers.aws.services.sqs.sqs_queues_not_publicly_accessible.sqs_queues_not_publicly_accessible import ( sqs_queues_not_publicly_accessible, diff --git a/tests/providers/aws/services/sqs/sqs_queues_server_side_encryption_enabled/sqs_queues_server_side_encryption_enabled_test.py b/tests/providers/aws/services/sqs/sqs_queues_server_side_encryption_enabled/sqs_queues_server_side_encryption_enabled_test.py index 297c99c02c..29d7f5ed05 100644 --- a/tests/providers/aws/services/sqs/sqs_queues_server_side_encryption_enabled/sqs_queues_server_side_encryption_enabled_test.py +++ b/tests/providers/aws/services/sqs/sqs_queues_server_side_encryption_enabled/sqs_queues_server_side_encryption_enabled_test.py @@ -16,15 +16,12 @@ class Test_sqs_queues_server_side_encryption_enabled: def test_no_queues(self): sqs_client = mock.MagicMock sqs_client.queues = [] - with ( - mock.patch( - "prowler.providers.aws.services.sqs.sqs_service.SQS", - new=sqs_client, - ), - mock.patch( - "prowler.providers.aws.services.sqs.sqs_client.sqs_client", - new=sqs_client, - ), + with mock.patch( + "prowler.providers.aws.services.sqs.sqs_service.SQS", + new=sqs_client, + ), mock.patch( + "prowler.providers.aws.services.sqs.sqs_client.sqs_client", + new=sqs_client, ): from prowler.providers.aws.services.sqs.sqs_queues_server_side_encryption_enabled.sqs_queues_server_side_encryption_enabled import ( sqs_queues_server_side_encryption_enabled, @@ -46,15 +43,12 @@ class Test_sqs_queues_server_side_encryption_enabled: arn=test_queue_arn, ) ) - with ( - mock.patch( - "prowler.providers.aws.services.sqs.sqs_service.SQS", - new=sqs_client, - ), - mock.patch( - "prowler.providers.aws.services.sqs.sqs_client.sqs_client", - new=sqs_client, - ), + with mock.patch( + "prowler.providers.aws.services.sqs.sqs_service.SQS", + new=sqs_client, + ), mock.patch( + "prowler.providers.aws.services.sqs.sqs_client.sqs_client", + new=sqs_client, ): from prowler.providers.aws.services.sqs.sqs_queues_server_side_encryption_enabled.sqs_queues_server_side_encryption_enabled import ( sqs_queues_server_side_encryption_enabled, @@ -82,15 +76,12 @@ class Test_sqs_queues_server_side_encryption_enabled: arn=test_queue_arn, ) ) - with ( - mock.patch( - "prowler.providers.aws.services.sqs.sqs_service.SQS", - new=sqs_client, - ), - mock.patch( - "prowler.providers.aws.services.sqs.sqs_client.sqs_client", - new=sqs_client, - ), + with mock.patch( + "prowler.providers.aws.services.sqs.sqs_service.SQS", + new=sqs_client, + ), mock.patch( + "prowler.providers.aws.services.sqs.sqs_client.sqs_client", + new=sqs_client, ): from prowler.providers.aws.services.sqs.sqs_queues_server_side_encryption_enabled.sqs_queues_server_side_encryption_enabled import ( sqs_queues_server_side_encryption_enabled, diff --git a/tests/providers/aws/services/ssm/ssm_managed_compliant_patching/ssm_managed_compliant_patching_test.py b/tests/providers/aws/services/ssm/ssm_managed_compliant_patching/ssm_managed_compliant_patching_test.py index af55449e88..dda2bce64e 100644 --- a/tests/providers/aws/services/ssm/ssm_managed_compliant_patching/ssm_managed_compliant_patching_test.py +++ b/tests/providers/aws/services/ssm/ssm_managed_compliant_patching/ssm_managed_compliant_patching_test.py @@ -17,23 +17,18 @@ class Test_ssm_managed_compliant_patching: ssm_client.compliance_resources = {} ec2_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.aws.services.ssm.ssm_service.SSM", - new=ssm_client, - ), - mock.patch( - "prowler.providers.aws.services.ssm.ssm_client.ssm_client", - new=ssm_client, - ), - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.ssm.ssm_managed_compliant_patching.ssm_managed_compliant_patching.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.aws.services.ssm.ssm_service.SSM", + new=ssm_client, + ), mock.patch( + "prowler.providers.aws.services.ssm.ssm_client.ssm_client", + new=ssm_client, + ), mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.ssm.ssm_managed_compliant_patching.ssm_managed_compliant_patching.ec2_client", + new=ec2_client, ): # Test Check from prowler.providers.aws.services.ssm.ssm_managed_compliant_patching.ssm_managed_compliant_patching import ( @@ -68,23 +63,18 @@ class Test_ssm_managed_compliant_patching: ) ] - with ( - mock.patch( - "prowler.providers.aws.services.ssm.ssm_service.SSM", - new=ssm_client, - ), - mock.patch( - "prowler.providers.aws.services.ssm.ssm_client.ssm_client", - new=ssm_client, - ), - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.ssm.ssm_managed_compliant_patching.ssm_managed_compliant_patching.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.aws.services.ssm.ssm_service.SSM", + new=ssm_client, + ), mock.patch( + "prowler.providers.aws.services.ssm.ssm_client.ssm_client", + new=ssm_client, + ), mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.ssm.ssm_managed_compliant_patching.ssm_managed_compliant_patching.ec2_client", + new=ec2_client, ): # Test Check from prowler.providers.aws.services.ssm.ssm_managed_compliant_patching.ssm_managed_compliant_patching import ( @@ -134,23 +124,18 @@ class Test_ssm_managed_compliant_patching: ) ] - with ( - mock.patch( - "prowler.providers.aws.services.ssm.ssm_service.SSM", - new=ssm_client, - ), - mock.patch( - "prowler.providers.aws.services.ssm.ssm_client.ssm_client", - new=ssm_client, - ), - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), - ), - mock.patch( - "prowler.providers.aws.services.ssm.ssm_managed_compliant_patching.ssm_managed_compliant_patching.ec2_client", - new=ec2_client, - ), + with mock.patch( + "prowler.providers.aws.services.ssm.ssm_service.SSM", + new=ssm_client, + ), mock.patch( + "prowler.providers.aws.services.ssm.ssm_client.ssm_client", + new=ssm_client, + ), mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_aws_provider([AWS_REGION_US_EAST_1]), + ), mock.patch( + "prowler.providers.aws.services.ssm.ssm_managed_compliant_patching.ssm_managed_compliant_patching.ec2_client", + new=ec2_client, ): # Test Check from prowler.providers.aws.services.ssm.ssm_managed_compliant_patching.ssm_managed_compliant_patching import ( diff --git a/tests/providers/aws/services/vpc/vpc_flow_logs_enabled/vpc_flow_logs_enabled_test.py b/tests/providers/aws/services/vpc/vpc_flow_logs_enabled/vpc_flow_logs_enabled_test.py index e43159adcd..80d53faa17 100644 --- a/tests/providers/aws/services/vpc/vpc_flow_logs_enabled/vpc_flow_logs_enabled_test.py +++ b/tests/providers/aws/services/vpc/vpc_flow_logs_enabled/vpc_flow_logs_enabled_test.py @@ -20,15 +20,12 @@ class Test_vpc_flow_logs_enabled: [AWS_REGION_US_EAST_1, AWS_REGION_EU_WEST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_flow_logs_enabled.vpc_flow_logs_enabled.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_flow_logs_enabled.vpc_flow_logs_enabled.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.vpc.vpc_flow_logs_enabled.vpc_flow_logs_enabled import ( @@ -74,15 +71,12 @@ class Test_vpc_flow_logs_enabled: [AWS_REGION_US_EAST_1, AWS_REGION_EU_WEST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_flow_logs_enabled.vpc_flow_logs_enabled.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_flow_logs_enabled.vpc_flow_logs_enabled.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.vpc.vpc_flow_logs_enabled.vpc_flow_logs_enabled import ( @@ -114,15 +108,12 @@ class Test_vpc_flow_logs_enabled: [AWS_REGION_US_EAST_1, AWS_REGION_EU_WEST_1] ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_flow_logs_enabled.vpc_flow_logs_enabled.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_flow_logs_enabled.vpc_flow_logs_enabled.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.vpc.vpc_flow_logs_enabled.vpc_flow_logs_enabled import ( @@ -156,15 +147,12 @@ class Test_vpc_flow_logs_enabled: ) aws_provider._scan_unused_services = False - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_flow_logs_enabled.vpc_flow_logs_enabled.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_flow_logs_enabled.vpc_flow_logs_enabled.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.vpc.vpc_flow_logs_enabled.vpc_flow_logs_enabled import ( @@ -191,15 +179,12 @@ class Test_vpc_flow_logs_enabled: ) aws_provider._scan_unused_services = False - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_flow_logs_enabled.vpc_flow_logs_enabled.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_flow_logs_enabled.vpc_flow_logs_enabled.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.vpc.vpc_flow_logs_enabled.vpc_flow_logs_enabled import ( diff --git a/tests/providers/aws/services/vpc/vpc_vpn_connection_tunnels_up/vpc_vpn_connection_tunnels_up_test.py b/tests/providers/aws/services/vpc/vpc_vpn_connection_tunnels_up/vpc_vpn_connection_tunnels_up_test.py index 5b70a70809..2248242ed7 100644 --- a/tests/providers/aws/services/vpc/vpc_vpn_connection_tunnels_up/vpc_vpn_connection_tunnels_up_test.py +++ b/tests/providers/aws/services/vpc/vpc_vpn_connection_tunnels_up/vpc_vpn_connection_tunnels_up_test.py @@ -74,15 +74,12 @@ class Test_vpc_vpn_connection_tunnels_up: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_vpn_connection_tunnels_up.vpc_vpn_connection_tunnels_up.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_vpn_connection_tunnels_up.vpc_vpn_connection_tunnels_up.vpc_client", + new=VPC(aws_provider), ): # Test Check from prowler.providers.aws.services.vpc.vpc_vpn_connection_tunnels_up.vpc_vpn_connection_tunnels_up import ( @@ -100,15 +97,12 @@ class Test_vpc_vpn_connection_tunnels_up: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_vpn_connection_tunnels_up.vpc_vpn_connection_tunnels_up.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_vpn_connection_tunnels_up.vpc_vpn_connection_tunnels_up.vpc_client", + new=VPC(aws_provider), ): from prowler.providers.aws.services.vpc.vpc_vpn_connection_tunnels_up.vpc_vpn_connection_tunnels_up import ( vpc_vpn_connection_tunnels_up, @@ -138,15 +132,12 @@ class Test_vpc_vpn_connection_tunnels_up: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.vpc.vpc_vpn_connection_tunnels_up.vpc_vpn_connection_tunnels_up.vpc_client", - new=VPC(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.vpc.vpc_vpn_connection_tunnels_up.vpc_vpn_connection_tunnels_up.vpc_client", + new=VPC(aws_provider), ): from prowler.providers.aws.services.vpc.vpc_vpn_connection_tunnels_up.vpc_vpn_connection_tunnels_up import ( vpc_vpn_connection_tunnels_up, diff --git a/tests/providers/aws/services/wafv2/wafv2_webacl_logging_enabled/wafv2_webacl_logging_enabled_test.py b/tests/providers/aws/services/wafv2/wafv2_webacl_logging_enabled/wafv2_webacl_logging_enabled_test.py index 1f0e529731..5bc95cd04a 100644 --- a/tests/providers/aws/services/wafv2/wafv2_webacl_logging_enabled/wafv2_webacl_logging_enabled_test.py +++ b/tests/providers/aws/services/wafv2/wafv2_webacl_logging_enabled/wafv2_webacl_logging_enabled_test.py @@ -13,15 +13,12 @@ class Test_wafv2_webacl_rule_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", - new=WAFv2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", + new=WAFv2(aws_provider), ): from prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled import ( wafv2_webacl_rule_logging_enabled, @@ -73,15 +70,12 @@ class Test_wafv2_webacl_rule_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", - new=WAFv2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", + new=WAFv2(aws_provider), ): from prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled import ( wafv2_webacl_rule_logging_enabled, @@ -142,15 +136,12 @@ class Test_wafv2_webacl_rule_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", - new=WAFv2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", + new=WAFv2(aws_provider), ): from prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled import ( wafv2_webacl_rule_logging_enabled, @@ -214,15 +205,12 @@ class Test_wafv2_webacl_rule_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", - new=WAFv2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", + new=WAFv2(aws_provider), ): from prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled import ( wafv2_webacl_rule_logging_enabled, @@ -286,15 +274,12 @@ class Test_wafv2_webacl_rule_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", - new=WAFv2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", + new=WAFv2(aws_provider), ): from prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled import ( wafv2_webacl_rule_logging_enabled, @@ -375,15 +360,12 @@ class Test_wafv2_webacl_rule_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", - new=WAFv2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", + new=WAFv2(aws_provider), ): from prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled import ( wafv2_webacl_rule_logging_enabled, @@ -464,15 +446,12 @@ class Test_wafv2_webacl_rule_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", - new=WAFv2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", + new=WAFv2(aws_provider), ): from prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled import ( wafv2_webacl_rule_logging_enabled, diff --git a/tests/providers/aws/services/wafv2/wafv2_webacl_rule_logging_enabled/wafv2_webacl_rule_logging_enabled_test.py b/tests/providers/aws/services/wafv2/wafv2_webacl_rule_logging_enabled/wafv2_webacl_rule_logging_enabled_test.py index dcf784be1c..f594a21949 100644 --- a/tests/providers/aws/services/wafv2/wafv2_webacl_rule_logging_enabled/wafv2_webacl_rule_logging_enabled_test.py +++ b/tests/providers/aws/services/wafv2/wafv2_webacl_rule_logging_enabled/wafv2_webacl_rule_logging_enabled_test.py @@ -13,15 +13,12 @@ class Test_wafv2_webacl_rule_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", - new=WAFv2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", + new=WAFv2(aws_provider), ): # Test Check from prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled import ( @@ -50,15 +47,12 @@ class Test_wafv2_webacl_rule_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", - new=WAFv2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", + new=WAFv2(aws_provider), ): # Test Check from prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled import ( @@ -110,15 +104,12 @@ class Test_wafv2_webacl_rule_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", - new=WAFv2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", + new=WAFv2(aws_provider), ): # Test Check from prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled import ( @@ -179,15 +170,12 @@ class Test_wafv2_webacl_rule_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", - new=WAFv2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", + new=WAFv2(aws_provider), ): # Test Check from prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled import ( @@ -245,15 +233,12 @@ class Test_wafv2_webacl_rule_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", - new=WAFv2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", + new=WAFv2(aws_provider), ): # Test Check from prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled import ( @@ -311,15 +296,12 @@ class Test_wafv2_webacl_rule_logging_enabled: aws_provider = set_mocked_aws_provider([AWS_REGION_US_EAST_1]) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=aws_provider, - ), - mock.patch( - "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", - new=WAFv2(aws_provider), - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=aws_provider, + ), mock.patch( + "prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled.wafv2_client", + new=WAFv2(aws_provider), ): # Test Check from prowler.providers.aws.services.wafv2.wafv2_webacl_rule_logging_enabled.wafv2_webacl_rule_logging_enabled import ( diff --git a/tests/providers/aws/services/wellarchitected/wellarchitected_workload_no_high_or_medium_risks/wellarchitected_workload_no_high_or_medium_risks_test.py b/tests/providers/aws/services/wellarchitected/wellarchitected_workload_no_high_or_medium_risks/wellarchitected_workload_no_high_or_medium_risks_test.py index 895b6c2c9e..884f25e113 100644 --- a/tests/providers/aws/services/wellarchitected/wellarchitected_workload_no_high_or_medium_risks/wellarchitected_workload_no_high_or_medium_risks_test.py +++ b/tests/providers/aws/services/wellarchitected/wellarchitected_workload_no_high_or_medium_risks/wellarchitected_workload_no_high_or_medium_risks_test.py @@ -13,15 +13,12 @@ class Test_wellarchitected_workload_no_high_or_medium_risks: def test_no_wellarchitected(self): wellarchitected_client = mock.MagicMock wellarchitected_client.workloads = [] - with ( - mock.patch( - "prowler.providers.aws.services.wellarchitected.wellarchitected_service.WellArchitected", - wellarchitected_client, - ), - mock.patch( - "prowler.providers.aws.services.wellarchitected.wellarchitected_client.wellarchitected_client", - wellarchitected_client, - ), + with mock.patch( + "prowler.providers.aws.services.wellarchitected.wellarchitected_service.WellArchitected", + wellarchitected_client, + ), mock.patch( + "prowler.providers.aws.services.wellarchitected.wellarchitected_client.wellarchitected_client", + wellarchitected_client, ): from prowler.providers.aws.services.wellarchitected.wellarchitected_workload_no_high_or_medium_risks.wellarchitected_workload_no_high_or_medium_risks import ( wellarchitected_workload_no_high_or_medium_risks, @@ -45,15 +42,12 @@ class Test_wellarchitected_workload_no_high_or_medium_risks: region=AWS_REGION_EU_WEST_1, ) ) - with ( - mock.patch( - "prowler.providers.aws.services.wellarchitected.wellarchitected_service.WellArchitected", - wellarchitected_client, - ), - mock.patch( - "prowler.providers.aws.services.wellarchitected.wellarchitected_client.wellarchitected_client", - wellarchitected_client, - ), + with mock.patch( + "prowler.providers.aws.services.wellarchitected.wellarchitected_service.WellArchitected", + wellarchitected_client, + ), mock.patch( + "prowler.providers.aws.services.wellarchitected.wellarchitected_client.wellarchitected_client", + wellarchitected_client, ): from prowler.providers.aws.services.wellarchitected.wellarchitected_workload_no_high_or_medium_risks.wellarchitected_workload_no_high_or_medium_risks import ( wellarchitected_workload_no_high_or_medium_risks, @@ -90,15 +84,12 @@ class Test_wellarchitected_workload_no_high_or_medium_risks: region=AWS_REGION_EU_WEST_1, ) ) - with ( - mock.patch( - "prowler.providers.aws.services.wellarchitected.wellarchitected_service.WellArchitected", - wellarchitected_client, - ), - mock.patch( - "prowler.providers.aws.services.wellarchitected.wellarchitected_client.wellarchitected_client", - wellarchitected_client, - ), + with mock.patch( + "prowler.providers.aws.services.wellarchitected.wellarchitected_service.WellArchitected", + wellarchitected_client, + ), mock.patch( + "prowler.providers.aws.services.wellarchitected.wellarchitected_client.wellarchitected_client", + wellarchitected_client, ): from prowler.providers.aws.services.wellarchitected.wellarchitected_workload_no_high_or_medium_risks.wellarchitected_workload_no_high_or_medium_risks import ( wellarchitected_workload_no_high_or_medium_risks, @@ -137,15 +128,12 @@ class Test_wellarchitected_workload_no_high_or_medium_risks: region=AWS_REGION_EU_WEST_1, ) ) - with ( - mock.patch( - "prowler.providers.aws.services.wellarchitected.wellarchitected_service.WellArchitected", - wellarchitected_client, - ), - mock.patch( - "prowler.providers.aws.services.wellarchitected.wellarchitected_client.wellarchitected_client", - wellarchitected_client, - ), + with mock.patch( + "prowler.providers.aws.services.wellarchitected.wellarchitected_service.WellArchitected", + wellarchitected_client, + ), mock.patch( + "prowler.providers.aws.services.wellarchitected.wellarchitected_client.wellarchitected_client", + wellarchitected_client, ): from prowler.providers.aws.services.wellarchitected.wellarchitected_workload_no_high_or_medium_risks.wellarchitected_workload_no_high_or_medium_risks import ( wellarchitected_workload_no_high_or_medium_risks, diff --git a/tests/providers/aws/services/workspaces/workspaces_volume_encryption_enabled/workspaces_volume_encryption_enabled_test.py b/tests/providers/aws/services/workspaces/workspaces_volume_encryption_enabled/workspaces_volume_encryption_enabled_test.py index bc9731b75f..f6d15e5913 100644 --- a/tests/providers/aws/services/workspaces/workspaces_volume_encryption_enabled/workspaces_volume_encryption_enabled_test.py +++ b/tests/providers/aws/services/workspaces/workspaces_volume_encryption_enabled/workspaces_volume_encryption_enabled_test.py @@ -12,15 +12,12 @@ class Test_workspaces_volume_encryption_enabled: def test_no_workspaces(self): workspaces_client = mock.MagicMock workspaces_client.workspaces = [] - with ( - mock.patch( - "prowler.providers.aws.services.workspaces.workspaces_service.WorkSpaces", - workspaces_client, - ), - mock.patch( - "prowler.providers.aws.services.workspaces.workspaces_client.workspaces_client", - workspaces_client, - ), + with mock.patch( + "prowler.providers.aws.services.workspaces.workspaces_service.WorkSpaces", + workspaces_client, + ), mock.patch( + "prowler.providers.aws.services.workspaces.workspaces_client.workspaces_client", + workspaces_client, ): from prowler.providers.aws.services.workspaces.workspaces_volume_encryption_enabled.workspaces_volume_encryption_enabled import ( workspaces_volume_encryption_enabled, @@ -43,15 +40,12 @@ class Test_workspaces_volume_encryption_enabled: subnet_id="subnet-12345678", ) ) - with ( - mock.patch( - "prowler.providers.aws.services.workspaces.workspaces_service.WorkSpaces", - workspaces_client, - ), - mock.patch( - "prowler.providers.aws.services.workspaces.workspaces_client.workspaces_client", - workspaces_client, - ), + with mock.patch( + "prowler.providers.aws.services.workspaces.workspaces_service.WorkSpaces", + workspaces_client, + ), mock.patch( + "prowler.providers.aws.services.workspaces.workspaces_client.workspaces_client", + workspaces_client, ): from prowler.providers.aws.services.workspaces.workspaces_volume_encryption_enabled.workspaces_volume_encryption_enabled import ( workspaces_volume_encryption_enabled, @@ -82,15 +76,12 @@ class Test_workspaces_volume_encryption_enabled: subnet_id="subnet-12345678", ) ) - with ( - mock.patch( - "prowler.providers.aws.services.workspaces.workspaces_service.WorkSpaces", - workspaces_client, - ), - mock.patch( - "prowler.providers.aws.services.workspaces.workspaces_client.workspaces_client", - workspaces_client, - ), + with mock.patch( + "prowler.providers.aws.services.workspaces.workspaces_service.WorkSpaces", + workspaces_client, + ), mock.patch( + "prowler.providers.aws.services.workspaces.workspaces_client.workspaces_client", + workspaces_client, ): from prowler.providers.aws.services.workspaces.workspaces_volume_encryption_enabled.workspaces_volume_encryption_enabled import ( workspaces_volume_encryption_enabled, @@ -121,15 +112,12 @@ class Test_workspaces_volume_encryption_enabled: subnet_id="subnet-12345678", ) ) - with ( - mock.patch( - "prowler.providers.aws.services.workspaces.workspaces_service.WorkSpaces", - workspaces_client, - ), - mock.patch( - "prowler.providers.aws.services.workspaces.workspaces_client.workspaces_client", - workspaces_client, - ), + with mock.patch( + "prowler.providers.aws.services.workspaces.workspaces_service.WorkSpaces", + workspaces_client, + ), mock.patch( + "prowler.providers.aws.services.workspaces.workspaces_client.workspaces_client", + workspaces_client, ): from prowler.providers.aws.services.workspaces.workspaces_volume_encryption_enabled.workspaces_volume_encryption_enabled import ( workspaces_volume_encryption_enabled, @@ -160,15 +148,12 @@ class Test_workspaces_volume_encryption_enabled: subnet_id="subnet-12345678", ) ) - with ( - mock.patch( - "prowler.providers.aws.services.workspaces.workspaces_service.WorkSpaces", - workspaces_client, - ), - mock.patch( - "prowler.providers.aws.services.workspaces.workspaces_client.workspaces_client", - workspaces_client, - ), + with mock.patch( + "prowler.providers.aws.services.workspaces.workspaces_service.WorkSpaces", + workspaces_client, + ), mock.patch( + "prowler.providers.aws.services.workspaces.workspaces_client.workspaces_client", + workspaces_client, ): from prowler.providers.aws.services.workspaces.workspaces_volume_encryption_enabled.workspaces_volume_encryption_enabled import ( workspaces_volume_encryption_enabled, diff --git a/tests/providers/azure/azure_provider_test.py b/tests/providers/azure/azure_provider_test.py index e5e93bad6b..e754964e39 100644 --- a/tests/providers/azure/azure_provider_test.py +++ b/tests/providers/azure/azure_provider_test.py @@ -40,15 +40,12 @@ class TestAzureProvider: ) azure_region = "AzureCloud" - with ( - patch( - "prowler.providers.azure.azure_provider.AzureProvider.setup_identity", - return_value=AzureIdentityInfo(), - ), - patch( - "prowler.providers.azure.azure_provider.AzureProvider.get_locations", - return_value={}, - ), + with patch( + "prowler.providers.azure.azure_provider.AzureProvider.setup_identity", + return_value=AzureIdentityInfo(), + ), patch( + "prowler.providers.azure.azure_provider.AzureProvider.get_locations", + return_value={}, ): azure_provider = AzureProvider( az_cli_auth, @@ -100,15 +97,12 @@ class TestAzureProvider: fixer_config = default_fixer_config_file_path azure_region = "AzureCloud" - with ( - patch( - "prowler.providers.azure.azure_provider.AzureProvider.setup_identity", - return_value=AzureIdentityInfo(), - ), - patch( - "prowler.providers.azure.azure_provider.AzureProvider.get_locations", - return_value={}, - ), + with patch( + "prowler.providers.azure.azure_provider.AzureProvider.setup_identity", + return_value=AzureIdentityInfo(), + ), patch( + "prowler.providers.azure.azure_provider.AzureProvider.get_locations", + return_value={}, ): with pytest.raises(AzureNoAuthenticationMethodError) as exception: @@ -141,15 +135,12 @@ class TestAzureProvider: fixer_config = default_fixer_config_file_path azure_region = "AzureCloud" - with ( - patch( - "prowler.providers.azure.azure_provider.AzureProvider.setup_identity", - return_value=AzureIdentityInfo(), - ), - patch( - "prowler.providers.azure.azure_provider.AzureProvider.get_locations", - return_value={}, - ), + with patch( + "prowler.providers.azure.azure_provider.AzureProvider.setup_identity", + return_value=AzureIdentityInfo(), + ), patch( + "prowler.providers.azure.azure_provider.AzureProvider.get_locations", + return_value={}, ): with pytest.raises(AzureBrowserAuthNoTenantIDError) as exception: @@ -183,15 +174,12 @@ class TestAzureProvider: fixer_config = default_fixer_config_file_path azure_region = "AzureCloud" - with ( - patch( - "prowler.providers.azure.azure_provider.AzureProvider.setup_identity", - return_value=AzureIdentityInfo(), - ), - patch( - "prowler.providers.azure.azure_provider.AzureProvider.get_locations", - return_value={}, - ), + with patch( + "prowler.providers.azure.azure_provider.AzureProvider.setup_identity", + return_value=AzureIdentityInfo(), + ), patch( + "prowler.providers.azure.azure_provider.AzureProvider.get_locations", + return_value={}, ): with pytest.raises(AzureTenantIDNoBrowserAuthError) as exception: @@ -213,17 +201,13 @@ class TestAzureProvider: ) def test_test_connection_browser_auth(self): - with ( - patch( - "prowler.providers.azure.azure_provider.DefaultAzureCredential" - ) as mock_default_credential, - patch( - "prowler.providers.azure.azure_provider.AzureProvider.setup_session" - ) as mock_setup_session, - patch( - "prowler.providers.azure.azure_provider.SubscriptionClient" - ) as mock_resource_client, - ): + with patch( + "prowler.providers.azure.azure_provider.DefaultAzureCredential" + ) as mock_default_credential, patch( + "prowler.providers.azure.azure_provider.AzureProvider.setup_session" + ) as mock_setup_session, patch( + "prowler.providers.azure.azure_provider.SubscriptionClient" + ) as mock_resource_client: # Mock the return value of DefaultAzureCredential mock_credentials = MagicMock() @@ -252,20 +236,15 @@ class TestAzureProvider: assert test_connection.error is None def test_test_connection_tenant_id_client_id_client_secret(self): - with ( - patch( - "prowler.providers.azure.azure_provider.DefaultAzureCredential" - ) as mock_default_credential, - patch( - "prowler.providers.azure.azure_provider.AzureProvider.setup_session" - ) as mock_setup_session, - patch( - "prowler.providers.azure.azure_provider.SubscriptionClient" - ) as mock_resource_client, - patch( - "prowler.providers.azure.azure_provider.AzureProvider.validate_static_credentials" - ) as mock_validate_static_credentials, - ): + with patch( + "prowler.providers.azure.azure_provider.DefaultAzureCredential" + ) as mock_default_credential, patch( + "prowler.providers.azure.azure_provider.AzureProvider.setup_session" + ) as mock_setup_session, patch( + "prowler.providers.azure.azure_provider.SubscriptionClient" + ) as mock_resource_client, patch( + "prowler.providers.azure.azure_provider.AzureProvider.validate_static_credentials" + ) as mock_validate_static_credentials: # Mock the return value of DefaultAzureCredential mock_credentials = MagicMock() @@ -303,20 +282,15 @@ class TestAzureProvider: assert test_connection.error is None def test_test_connection_provider_validation(self): - with ( - patch( - "prowler.providers.azure.azure_provider.DefaultAzureCredential" - ) as mock_default_credential, - patch( - "prowler.providers.azure.azure_provider.AzureProvider.setup_session" - ) as mock_setup_session, - patch( - "prowler.providers.azure.azure_provider.SubscriptionClient" - ) as mock_resource_client, - patch( - "prowler.providers.azure.azure_provider.AzureProvider.validate_static_credentials" - ) as mock_validate_static_credentials, - ): + with patch( + "prowler.providers.azure.azure_provider.DefaultAzureCredential" + ) as mock_default_credential, patch( + "prowler.providers.azure.azure_provider.AzureProvider.setup_session" + ) as mock_setup_session, patch( + "prowler.providers.azure.azure_provider.SubscriptionClient" + ) as mock_resource_client, patch( + "prowler.providers.azure.azure_provider.AzureProvider.validate_static_credentials" + ) as mock_validate_static_credentials: # Mock the return value of DefaultAzureCredential mock_default_credential.return_value = { @@ -354,20 +328,15 @@ class TestAzureProvider: assert test_connection.error is None def test_test_connection_provider_validation_error(self): - with ( - patch( - "prowler.providers.azure.azure_provider.DefaultAzureCredential" - ) as mock_default_credential, - patch( - "prowler.providers.azure.azure_provider.AzureProvider.setup_session" - ) as mock_setup_session, - patch( - "prowler.providers.azure.azure_provider.SubscriptionClient" - ) as mock_resource_client, - patch( - "prowler.providers.azure.azure_provider.AzureProvider.validate_static_credentials" - ) as mock_validate_static_credentials, - ): + with patch( + "prowler.providers.azure.azure_provider.DefaultAzureCredential" + ) as mock_default_credential, patch( + "prowler.providers.azure.azure_provider.AzureProvider.setup_session" + ) as mock_setup_session, patch( + "prowler.providers.azure.azure_provider.SubscriptionClient" + ) as mock_resource_client, patch( + "prowler.providers.azure.azure_provider.AzureProvider.validate_static_credentials" + ) as mock_validate_static_credentials: # Mock the return value of DefaultAzureCredential mock_default_credential.return_value = { @@ -433,15 +402,12 @@ class TestAzureProvider: ) def test_test_connection_with_httpresponseerror(self): - with ( - patch( - "prowler.providers.azure.azure_provider.AzureProvider.get_locations", - return_value={}, - ), - patch( - "prowler.providers.azure.azure_provider.AzureProvider.setup_session" - ) as mock_setup_session, - ): + with patch( + "prowler.providers.azure.azure_provider.AzureProvider.get_locations", + return_value={}, + ), patch( + "prowler.providers.azure.azure_provider.AzureProvider.setup_session" + ) as mock_setup_session: mock_setup_session.side_effect = AzureHTTPResponseError( file="test_file", original_exception="Simulated HttpResponseError" diff --git a/tests/providers/azure/services/aisearch/aisearch_service_public_access_level_is_disabled/aisearch_service_public_access_level_is_disabled_test.py b/tests/providers/azure/services/aisearch/aisearch_service_public_access_level_is_disabled/aisearch_service_public_access_level_is_disabled_test.py index 7e35a2ca36..c1cac0546e 100644 --- a/tests/providers/azure/services/aisearch/aisearch_service_public_access_level_is_disabled/aisearch_service_public_access_level_is_disabled_test.py +++ b/tests/providers/azure/services/aisearch/aisearch_service_public_access_level_is_disabled/aisearch_service_public_access_level_is_disabled_test.py @@ -13,15 +13,12 @@ class Test_AISearch_service_not_publicly_accessible: aisearch_client = mock.MagicMock aisearch_client.aisearch_services = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aisearch.aisearch_service_not_publicly_accessible.aisearch_service_not_publicly_accessible.aisearch_client", - new=aisearch_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aisearch.aisearch_service_not_publicly_accessible.aisearch_service_not_publicly_accessible.aisearch_client", + new=aisearch_client, ): from prowler.providers.azure.services.aisearch.aisearch_service_not_publicly_accessible.aisearch_service_not_publicly_accessible import ( aisearch_service_not_publicly_accessible, @@ -46,15 +43,12 @@ class Test_AISearch_service_not_publicly_accessible: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aisearch.aisearch_service_not_publicly_accessible.aisearch_service_not_publicly_accessible.aisearch_client", - new=aisearch_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aisearch.aisearch_service_not_publicly_accessible.aisearch_service_not_publicly_accessible.aisearch_client", + new=aisearch_client, ): from prowler.providers.azure.services.aisearch.aisearch_service_not_publicly_accessible.aisearch_service_not_publicly_accessible import ( aisearch_service_not_publicly_accessible, @@ -88,15 +82,12 @@ class Test_AISearch_service_not_publicly_accessible: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aisearch.aisearch_service_not_publicly_accessible.aisearch_service_not_publicly_accessible.aisearch_client", - new=aisearch_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aisearch.aisearch_service_not_publicly_accessible.aisearch_service_not_publicly_accessible.aisearch_client", + new=aisearch_client, ): from prowler.providers.azure.services.aisearch.aisearch_service_not_publicly_accessible.aisearch_service_not_publicly_accessible import ( aisearch_service_not_publicly_accessible, diff --git a/tests/providers/azure/services/aks/aks_cluster_rbac_enabled/aks_cluster_rbac_enabled_test.py b/tests/providers/azure/services/aks/aks_cluster_rbac_enabled/aks_cluster_rbac_enabled_test.py index a4706f013a..5eca4a8d31 100644 --- a/tests/providers/azure/services/aks/aks_cluster_rbac_enabled/aks_cluster_rbac_enabled_test.py +++ b/tests/providers/azure/services/aks/aks_cluster_rbac_enabled/aks_cluster_rbac_enabled_test.py @@ -13,15 +13,12 @@ class Test_aks_cluster_rbac_enabled: aks_client = mock.MagicMock aks_client.clusters = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aks.aks_cluster_rbac_enabled.aks_cluster_rbac_enabled.aks_client", - new=aks_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aks.aks_cluster_rbac_enabled.aks_cluster_rbac_enabled.aks_client", + new=aks_client, ): from prowler.providers.azure.services.aks.aks_cluster_rbac_enabled.aks_cluster_rbac_enabled import ( aks_cluster_rbac_enabled, @@ -35,15 +32,12 @@ class Test_aks_cluster_rbac_enabled: aks_client = mock.MagicMock aks_client.clusters = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aks.aks_cluster_rbac_enabled.aks_cluster_rbac_enabled.aks_client", - new=aks_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aks.aks_cluster_rbac_enabled.aks_cluster_rbac_enabled.aks_client", + new=aks_client, ): from prowler.providers.azure.services.aks.aks_cluster_rbac_enabled.aks_cluster_rbac_enabled import ( aks_cluster_rbac_enabled, @@ -71,15 +65,12 @@ class Test_aks_cluster_rbac_enabled: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aks.aks_cluster_rbac_enabled.aks_cluster_rbac_enabled.aks_client", - new=aks_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aks.aks_cluster_rbac_enabled.aks_cluster_rbac_enabled.aks_client", + new=aks_client, ): from prowler.providers.azure.services.aks.aks_cluster_rbac_enabled.aks_cluster_rbac_enabled import ( aks_cluster_rbac_enabled, @@ -116,15 +107,12 @@ class Test_aks_cluster_rbac_enabled: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aks.aks_cluster_rbac_enabled.aks_cluster_rbac_enabled.aks_client", - new=aks_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aks.aks_cluster_rbac_enabled.aks_cluster_rbac_enabled.aks_client", + new=aks_client, ): from prowler.providers.azure.services.aks.aks_cluster_rbac_enabled.aks_cluster_rbac_enabled import ( aks_cluster_rbac_enabled, diff --git a/tests/providers/azure/services/aks/aks_clusters_created_with_private_nodes/aks_clusters_created_with_private_nodes_test.py b/tests/providers/azure/services/aks/aks_clusters_created_with_private_nodes/aks_clusters_created_with_private_nodes_test.py index 74ee1e7206..7c8bc6e9b9 100644 --- a/tests/providers/azure/services/aks/aks_clusters_created_with_private_nodes/aks_clusters_created_with_private_nodes_test.py +++ b/tests/providers/azure/services/aks/aks_clusters_created_with_private_nodes/aks_clusters_created_with_private_nodes_test.py @@ -13,15 +13,12 @@ class Test_aks_clusters_created_with_private_nodes: aks_client = mock.MagicMock aks_client.clusters = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aks.aks_clusters_created_with_private_nodes.aks_clusters_created_with_private_nodes.aks_client", - new=aks_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aks.aks_clusters_created_with_private_nodes.aks_clusters_created_with_private_nodes.aks_client", + new=aks_client, ): from prowler.providers.azure.services.aks.aks_clusters_created_with_private_nodes.aks_clusters_created_with_private_nodes import ( aks_clusters_created_with_private_nodes, @@ -35,15 +32,12 @@ class Test_aks_clusters_created_with_private_nodes: aks_client = mock.MagicMock aks_client.clusters = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aks.aks_clusters_created_with_private_nodes.aks_clusters_created_with_private_nodes.aks_client", - new=aks_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aks.aks_clusters_created_with_private_nodes.aks_clusters_created_with_private_nodes.aks_client", + new=aks_client, ): from prowler.providers.azure.services.aks.aks_clusters_created_with_private_nodes.aks_clusters_created_with_private_nodes import ( aks_clusters_created_with_private_nodes, @@ -71,15 +65,12 @@ class Test_aks_clusters_created_with_private_nodes: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aks.aks_clusters_created_with_private_nodes.aks_clusters_created_with_private_nodes.aks_client", - new=aks_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aks.aks_clusters_created_with_private_nodes.aks_clusters_created_with_private_nodes.aks_client", + new=aks_client, ): from prowler.providers.azure.services.aks.aks_clusters_created_with_private_nodes.aks_clusters_created_with_private_nodes import ( aks_clusters_created_with_private_nodes, @@ -116,15 +107,12 @@ class Test_aks_clusters_created_with_private_nodes: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aks.aks_clusters_created_with_private_nodes.aks_clusters_created_with_private_nodes.aks_client", - new=aks_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aks.aks_clusters_created_with_private_nodes.aks_clusters_created_with_private_nodes.aks_client", + new=aks_client, ): from prowler.providers.azure.services.aks.aks_clusters_created_with_private_nodes.aks_clusters_created_with_private_nodes import ( aks_clusters_created_with_private_nodes, @@ -165,15 +153,12 @@ class Test_aks_clusters_created_with_private_nodes: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aks.aks_clusters_created_with_private_nodes.aks_clusters_created_with_private_nodes.aks_client", - new=aks_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aks.aks_clusters_created_with_private_nodes.aks_clusters_created_with_private_nodes.aks_client", + new=aks_client, ): from prowler.providers.azure.services.aks.aks_clusters_created_with_private_nodes.aks_clusters_created_with_private_nodes import ( aks_clusters_created_with_private_nodes, diff --git a/tests/providers/azure/services/aks/aks_clusters_public_access_disabled/aks_clusters_public_access_disabled_test.py b/tests/providers/azure/services/aks/aks_clusters_public_access_disabled/aks_clusters_public_access_disabled_test.py index 6d012c39e5..2a733115ae 100644 --- a/tests/providers/azure/services/aks/aks_clusters_public_access_disabled/aks_clusters_public_access_disabled_test.py +++ b/tests/providers/azure/services/aks/aks_clusters_public_access_disabled/aks_clusters_public_access_disabled_test.py @@ -13,15 +13,12 @@ class Test_aks_clusters_public_access_disabled: aks_client = mock.MagicMock aks_client.clusters = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aks.aks_clusters_public_access_disabled.aks_clusters_public_access_disabled.aks_client", - new=aks_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aks.aks_clusters_public_access_disabled.aks_clusters_public_access_disabled.aks_client", + new=aks_client, ): from prowler.providers.azure.services.aks.aks_clusters_public_access_disabled.aks_clusters_public_access_disabled import ( aks_clusters_public_access_disabled, @@ -35,15 +32,12 @@ class Test_aks_clusters_public_access_disabled: aks_client = mock.MagicMock aks_client.clusters = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aks.aks_clusters_public_access_disabled.aks_clusters_public_access_disabled.aks_client", - new=aks_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aks.aks_clusters_public_access_disabled.aks_clusters_public_access_disabled.aks_client", + new=aks_client, ): from prowler.providers.azure.services.aks.aks_clusters_public_access_disabled.aks_clusters_public_access_disabled import ( aks_clusters_public_access_disabled, @@ -71,15 +65,12 @@ class Test_aks_clusters_public_access_disabled: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aks.aks_clusters_public_access_disabled.aks_clusters_public_access_disabled.aks_client", - new=aks_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aks.aks_clusters_public_access_disabled.aks_clusters_public_access_disabled.aks_client", + new=aks_client, ): from prowler.providers.azure.services.aks.aks_clusters_public_access_disabled.aks_clusters_public_access_disabled import ( aks_clusters_public_access_disabled, @@ -116,15 +107,12 @@ class Test_aks_clusters_public_access_disabled: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aks.aks_clusters_public_access_disabled.aks_clusters_public_access_disabled.aks_client", - new=aks_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aks.aks_clusters_public_access_disabled.aks_clusters_public_access_disabled.aks_client", + new=aks_client, ): from prowler.providers.azure.services.aks.aks_clusters_public_access_disabled.aks_clusters_public_access_disabled import ( aks_clusters_public_access_disabled, @@ -161,15 +149,12 @@ class Test_aks_clusters_public_access_disabled: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aks.aks_clusters_public_access_disabled.aks_clusters_public_access_disabled.aks_client", - new=aks_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aks.aks_clusters_public_access_disabled.aks_clusters_public_access_disabled.aks_client", + new=aks_client, ): from prowler.providers.azure.services.aks.aks_clusters_public_access_disabled.aks_clusters_public_access_disabled import ( aks_clusters_public_access_disabled, diff --git a/tests/providers/azure/services/aks/aks_network_policy_enabled/aks_network_policy_enabled_test.py b/tests/providers/azure/services/aks/aks_network_policy_enabled/aks_network_policy_enabled_test.py index a172e94151..97392defa4 100644 --- a/tests/providers/azure/services/aks/aks_network_policy_enabled/aks_network_policy_enabled_test.py +++ b/tests/providers/azure/services/aks/aks_network_policy_enabled/aks_network_policy_enabled_test.py @@ -13,15 +13,12 @@ class Test_aks_network_policy_enabled: aks_client = mock.MagicMock aks_client.clusters = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aks.aks_network_policy_enabled.aks_network_policy_enabled.aks_client", - new=aks_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aks.aks_network_policy_enabled.aks_network_policy_enabled.aks_client", + new=aks_client, ): from prowler.providers.azure.services.aks.aks_network_policy_enabled.aks_network_policy_enabled import ( aks_network_policy_enabled, @@ -35,15 +32,12 @@ class Test_aks_network_policy_enabled: aks_client = mock.MagicMock aks_client.clusters = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aks.aks_network_policy_enabled.aks_network_policy_enabled.aks_client", - new=aks_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aks.aks_network_policy_enabled.aks_network_policy_enabled.aks_client", + new=aks_client, ): from prowler.providers.azure.services.aks.aks_network_policy_enabled.aks_network_policy_enabled import ( aks_network_policy_enabled, @@ -71,15 +65,12 @@ class Test_aks_network_policy_enabled: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aks.aks_network_policy_enabled.aks_network_policy_enabled.aks_client", - new=aks_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aks.aks_network_policy_enabled.aks_network_policy_enabled.aks_client", + new=aks_client, ): from prowler.providers.azure.services.aks.aks_network_policy_enabled.aks_network_policy_enabled import ( aks_network_policy_enabled, @@ -116,15 +107,12 @@ class Test_aks_network_policy_enabled: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.aks.aks_network_policy_enabled.aks_network_policy_enabled.aks_client", - new=aks_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.aks.aks_network_policy_enabled.aks_network_policy_enabled.aks_client", + new=aks_client, ): from prowler.providers.azure.services.aks.aks_network_policy_enabled.aks_network_policy_enabled import ( aks_network_policy_enabled, diff --git a/tests/providers/azure/services/app/app_client_certificates_on/app_client_certificates_on_test.py b/tests/providers/azure/services/app/app_client_certificates_on/app_client_certificates_on_test.py index 581b3be994..309be1c0da 100644 --- a/tests/providers/azure/services/app/app_client_certificates_on/app_client_certificates_on_test.py +++ b/tests/providers/azure/services/app/app_client_certificates_on/app_client_certificates_on_test.py @@ -12,15 +12,12 @@ class Test_app_client_certificates_on: app_client = mock.MagicMock app_client.apps = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_client_certificates_on.app_client_certificates_on.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_client_certificates_on.app_client_certificates_on.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_client_certificates_on.app_client_certificates_on import ( app_client_certificates_on, @@ -34,15 +31,12 @@ class Test_app_client_certificates_on: app_client = mock.MagicMock app_client.apps = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_client_certificates_on.app_client_certificates_on.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_client_certificates_on.app_client_certificates_on.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_client_certificates_on.app_client_certificates_on import ( app_client_certificates_on, @@ -56,15 +50,12 @@ class Test_app_client_certificates_on: resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_client_certificates_on.app_client_certificates_on.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_client_certificates_on.app_client_certificates_on.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_client_certificates_on.app_client_certificates_on import ( app_client_certificates_on, @@ -102,15 +93,12 @@ class Test_app_client_certificates_on: resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_client_certificates_on.app_client_certificates_on.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_client_certificates_on.app_client_certificates_on.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_client_certificates_on.app_client_certificates_on import ( app_client_certificates_on, diff --git a/tests/providers/azure/services/app/app_ensure_auth_is_set_up/app_ensure_auth_is_set_up_test.py b/tests/providers/azure/services/app/app_ensure_auth_is_set_up/app_ensure_auth_is_set_up_test.py index 79bf195d80..7d94418f00 100644 --- a/tests/providers/azure/services/app/app_ensure_auth_is_set_up/app_ensure_auth_is_set_up_test.py +++ b/tests/providers/azure/services/app/app_ensure_auth_is_set_up/app_ensure_auth_is_set_up_test.py @@ -12,15 +12,12 @@ class Test_app_ensure_auth_is_set_up: app_client = mock.MagicMock app_client.apps = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_auth_is_set_up.app_ensure_auth_is_set_up.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_auth_is_set_up.app_ensure_auth_is_set_up.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_auth_is_set_up.app_ensure_auth_is_set_up import ( app_ensure_auth_is_set_up, @@ -34,15 +31,12 @@ class Test_app_ensure_auth_is_set_up: app_client = mock.MagicMock app_client.apps = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_auth_is_set_up.app_ensure_auth_is_set_up.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_auth_is_set_up.app_ensure_auth_is_set_up.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_auth_is_set_up.app_ensure_auth_is_set_up import ( app_ensure_auth_is_set_up, @@ -56,15 +50,12 @@ class Test_app_ensure_auth_is_set_up: resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_auth_is_set_up.app_ensure_auth_is_set_up.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_auth_is_set_up.app_ensure_auth_is_set_up.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_auth_is_set_up.app_ensure_auth_is_set_up import ( app_ensure_auth_is_set_up, @@ -102,15 +93,12 @@ class Test_app_ensure_auth_is_set_up: resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_auth_is_set_up.app_ensure_auth_is_set_up.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_auth_is_set_up.app_ensure_auth_is_set_up.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_auth_is_set_up.app_ensure_auth_is_set_up import ( app_ensure_auth_is_set_up, diff --git a/tests/providers/azure/services/app/app_ensure_http_is_redirected_to_https/app_ensure_http_is_redirected_to_https_test.py b/tests/providers/azure/services/app/app_ensure_http_is_redirected_to_https/app_ensure_http_is_redirected_to_https_test.py index 08743bbd79..5d039bed5c 100644 --- a/tests/providers/azure/services/app/app_ensure_http_is_redirected_to_https/app_ensure_http_is_redirected_to_https_test.py +++ b/tests/providers/azure/services/app/app_ensure_http_is_redirected_to_https/app_ensure_http_is_redirected_to_https_test.py @@ -12,15 +12,12 @@ class Test_app_ensure_http_is_redirected_to_https: app_client = mock.MagicMock app_client.apps = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_http_is_redirected_to_https.app_ensure_http_is_redirected_to_https.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_http_is_redirected_to_https.app_ensure_http_is_redirected_to_https.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_http_is_redirected_to_https.app_ensure_http_is_redirected_to_https import ( app_ensure_http_is_redirected_to_https, @@ -34,15 +31,12 @@ class Test_app_ensure_http_is_redirected_to_https: app_client = mock.MagicMock app_client.apps = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_http_is_redirected_to_https.app_ensure_http_is_redirected_to_https.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_http_is_redirected_to_https.app_ensure_http_is_redirected_to_https.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_http_is_redirected_to_https.app_ensure_http_is_redirected_to_https import ( app_ensure_http_is_redirected_to_https, @@ -56,15 +50,12 @@ class Test_app_ensure_http_is_redirected_to_https: resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_http_is_redirected_to_https.app_ensure_http_is_redirected_to_https.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_http_is_redirected_to_https.app_ensure_http_is_redirected_to_https.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_http_is_redirected_to_https.app_ensure_http_is_redirected_to_https import ( app_ensure_http_is_redirected_to_https, @@ -102,15 +93,12 @@ class Test_app_ensure_http_is_redirected_to_https: resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_http_is_redirected_to_https.app_ensure_http_is_redirected_to_https.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_http_is_redirected_to_https.app_ensure_http_is_redirected_to_https.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_http_is_redirected_to_https.app_ensure_http_is_redirected_to_https import ( app_ensure_http_is_redirected_to_https, diff --git a/tests/providers/azure/services/app/app_ensure_java_version_is_latest/app_ensure_java_version_is_latest_test.py b/tests/providers/azure/services/app/app_ensure_java_version_is_latest/app_ensure_java_version_is_latest_test.py index 22a9d3771a..1ada777b82 100644 --- a/tests/providers/azure/services/app/app_ensure_java_version_is_latest/app_ensure_java_version_is_latest_test.py +++ b/tests/providers/azure/services/app/app_ensure_java_version_is_latest/app_ensure_java_version_is_latest_test.py @@ -12,15 +12,12 @@ class Test_app_ensure_java_version_is_latest: app_client = mock.MagicMock app_client.apps = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest import ( app_ensure_java_version_is_latest, @@ -34,15 +31,12 @@ class Test_app_ensure_java_version_is_latest: app_client = mock.MagicMock app_client.apps = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest import ( app_ensure_java_version_is_latest, @@ -56,15 +50,12 @@ class Test_app_ensure_java_version_is_latest: resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest import ( app_ensure_java_version_is_latest, @@ -95,15 +86,12 @@ class Test_app_ensure_java_version_is_latest: app_client.audit_config = {"java_latest_version": "17"} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest import ( app_ensure_java_version_is_latest, @@ -145,15 +133,12 @@ class Test_app_ensure_java_version_is_latest: app_client.audit_config = {"java_latest_version": "17"} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest import ( app_ensure_java_version_is_latest, @@ -195,15 +180,12 @@ class Test_app_ensure_java_version_is_latest: app_client.audit_config = {"java_latest_version": "17"} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest import ( app_ensure_java_version_is_latest, @@ -244,15 +226,12 @@ class Test_app_ensure_java_version_is_latest: app_client = mock.MagicMock app_client.audit_config = {"java_latest_version": "17"} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest import ( app_ensure_java_version_is_latest, @@ -294,15 +273,12 @@ class Test_app_ensure_java_version_is_latest: app_client.audit_config = {"java_latest_version": "17"} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_java_version_is_latest.app_ensure_java_version_is_latest import ( app_ensure_java_version_is_latest, diff --git a/tests/providers/azure/services/app/app_ensure_php_version_is_latest/app_ensure_php_version_is_latest_test.py b/tests/providers/azure/services/app/app_ensure_php_version_is_latest/app_ensure_php_version_is_latest_test.py index 3db89439a3..9fcdb0523d 100644 --- a/tests/providers/azure/services/app/app_ensure_php_version_is_latest/app_ensure_php_version_is_latest_test.py +++ b/tests/providers/azure/services/app/app_ensure_php_version_is_latest/app_ensure_php_version_is_latest_test.py @@ -12,15 +12,12 @@ class Test_app_ensure_php_version_is_latest: app_client = mock.MagicMock app_client.apps = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_php_version_is_latest.app_ensure_php_version_is_latest.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_php_version_is_latest.app_ensure_php_version_is_latest.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_php_version_is_latest.app_ensure_php_version_is_latest import ( app_ensure_php_version_is_latest, @@ -34,15 +31,12 @@ class Test_app_ensure_php_version_is_latest: app_client = mock.MagicMock app_client.apps = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_php_version_is_latest.app_ensure_php_version_is_latest.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_php_version_is_latest.app_ensure_php_version_is_latest.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_php_version_is_latest.app_ensure_php_version_is_latest import ( app_ensure_php_version_is_latest, @@ -56,15 +50,12 @@ class Test_app_ensure_php_version_is_latest: resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_php_version_is_latest.app_ensure_php_version_is_latest.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_php_version_is_latest.app_ensure_php_version_is_latest.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_php_version_is_latest.app_ensure_php_version_is_latest import ( app_ensure_php_version_is_latest, @@ -95,15 +86,12 @@ class Test_app_ensure_php_version_is_latest: app_client.audit_config = {"php_latest_version": "8.2"} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_php_version_is_latest.app_ensure_php_version_is_latest.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_php_version_is_latest.app_ensure_php_version_is_latest.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_php_version_is_latest.app_ensure_php_version_is_latest import ( app_ensure_php_version_is_latest, @@ -143,15 +131,12 @@ class Test_app_ensure_php_version_is_latest: app_client.audit_config = {"php_latest_version": "8.2"} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_php_version_is_latest.app_ensure_php_version_is_latest.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_php_version_is_latest.app_ensure_php_version_is_latest.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_php_version_is_latest.app_ensure_php_version_is_latest import ( app_ensure_php_version_is_latest, diff --git a/tests/providers/azure/services/app/app_ensure_python_version_is_latest/app_ensure_python_version_is_latest_test.py b/tests/providers/azure/services/app/app_ensure_python_version_is_latest/app_ensure_python_version_is_latest_test.py index 899d3615b5..c1301b5a84 100644 --- a/tests/providers/azure/services/app/app_ensure_python_version_is_latest/app_ensure_python_version_is_latest_test.py +++ b/tests/providers/azure/services/app/app_ensure_python_version_is_latest/app_ensure_python_version_is_latest_test.py @@ -12,15 +12,12 @@ class Test_app_ensure_python_version_is_latest: app_client = mock.MagicMock app_client.apps = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_python_version_is_latest.app_ensure_python_version_is_latest.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_python_version_is_latest.app_ensure_python_version_is_latest.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_python_version_is_latest.app_ensure_python_version_is_latest import ( app_ensure_python_version_is_latest, @@ -34,15 +31,12 @@ class Test_app_ensure_python_version_is_latest: app_client = mock.MagicMock app_client.apps = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_python_version_is_latest.app_ensure_python_version_is_latest.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_python_version_is_latest.app_ensure_python_version_is_latest.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_python_version_is_latest.app_ensure_python_version_is_latest import ( app_ensure_python_version_is_latest, @@ -55,15 +49,12 @@ class Test_app_ensure_python_version_is_latest: def test_app_configurations_none(self): resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_python_version_is_latest.app_ensure_python_version_is_latest.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_python_version_is_latest.app_ensure_python_version_is_latest.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_python_version_is_latest.app_ensure_python_version_is_latest import ( app_ensure_python_version_is_latest, @@ -94,15 +85,12 @@ class Test_app_ensure_python_version_is_latest: app_client.audit_config = {"python_latest_version": "3.12"} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_python_version_is_latest.app_ensure_python_version_is_latest.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_python_version_is_latest.app_ensure_python_version_is_latest.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_python_version_is_latest.app_ensure_python_version_is_latest import ( app_ensure_python_version_is_latest, @@ -142,15 +130,12 @@ class Test_app_ensure_python_version_is_latest: app_client.audit_config = {"python_latest_version": "3.12"} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_python_version_is_latest.app_ensure_python_version_is_latest.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_python_version_is_latest.app_ensure_python_version_is_latest.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_python_version_is_latest.app_ensure_python_version_is_latest import ( app_ensure_python_version_is_latest, diff --git a/tests/providers/azure/services/app/app_ensure_using_http20/app_ensure_using_http20_test.py b/tests/providers/azure/services/app/app_ensure_using_http20/app_ensure_using_http20_test.py index d392d4d3b0..10bf8f7942 100644 --- a/tests/providers/azure/services/app/app_ensure_using_http20/app_ensure_using_http20_test.py +++ b/tests/providers/azure/services/app/app_ensure_using_http20/app_ensure_using_http20_test.py @@ -12,15 +12,12 @@ class Test_app_ensure_using_http20: app_client = mock.MagicMock app_client.apps = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_using_http20.app_ensure_using_http20.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_using_http20.app_ensure_using_http20.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_using_http20.app_ensure_using_http20 import ( app_ensure_using_http20, @@ -34,15 +31,12 @@ class Test_app_ensure_using_http20: app_client = mock.MagicMock app_client.apps = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_using_http20.app_ensure_using_http20.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_using_http20.app_ensure_using_http20.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_using_http20.app_ensure_using_http20 import ( app_ensure_using_http20, @@ -56,15 +50,12 @@ class Test_app_ensure_using_http20: resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_using_http20.app_ensure_using_http20.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_using_http20.app_ensure_using_http20.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_using_http20.app_ensure_using_http20 import ( app_ensure_using_http20, @@ -102,15 +93,12 @@ class Test_app_ensure_using_http20: resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_using_http20.app_ensure_using_http20.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_using_http20.app_ensure_using_http20.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_using_http20.app_ensure_using_http20 import ( app_ensure_using_http20, @@ -148,15 +136,12 @@ class Test_app_ensure_using_http20: resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ensure_using_http20.app_ensure_using_http20.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ensure_using_http20.app_ensure_using_http20.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ensure_using_http20.app_ensure_using_http20 import ( app_ensure_using_http20, diff --git a/tests/providers/azure/services/app/app_ftp_deployment_disabled/app_ftp_deployment_disabled_test.py b/tests/providers/azure/services/app/app_ftp_deployment_disabled/app_ftp_deployment_disabled_test.py index 6d0633fef3..58875566d7 100644 --- a/tests/providers/azure/services/app/app_ftp_deployment_disabled/app_ftp_deployment_disabled_test.py +++ b/tests/providers/azure/services/app/app_ftp_deployment_disabled/app_ftp_deployment_disabled_test.py @@ -12,15 +12,12 @@ class Test_app_ftp_deployment_disabled: app_client = mock.MagicMock app_client.apps = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ftp_deployment_disabled.app_ftp_deployment_disabled.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ftp_deployment_disabled.app_ftp_deployment_disabled.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ftp_deployment_disabled.app_ftp_deployment_disabled import ( app_ftp_deployment_disabled, @@ -34,15 +31,12 @@ class Test_app_ftp_deployment_disabled: app_client = mock.MagicMock app_client.apps = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ftp_deployment_disabled.app_ftp_deployment_disabled.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ftp_deployment_disabled.app_ftp_deployment_disabled.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ftp_deployment_disabled.app_ftp_deployment_disabled import ( app_ftp_deployment_disabled, @@ -56,15 +50,12 @@ class Test_app_ftp_deployment_disabled: resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ftp_deployment_disabled.app_ftp_deployment_disabled.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ftp_deployment_disabled.app_ftp_deployment_disabled.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ftp_deployment_disabled.app_ftp_deployment_disabled import ( app_ftp_deployment_disabled, @@ -102,15 +93,12 @@ class Test_app_ftp_deployment_disabled: resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ftp_deployment_disabled.app_ftp_deployment_disabled.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ftp_deployment_disabled.app_ftp_deployment_disabled.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ftp_deployment_disabled.app_ftp_deployment_disabled import ( app_ftp_deployment_disabled, @@ -148,15 +136,12 @@ class Test_app_ftp_deployment_disabled: resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_ftp_deployment_disabled.app_ftp_deployment_disabled.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_ftp_deployment_disabled.app_ftp_deployment_disabled.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_ftp_deployment_disabled.app_ftp_deployment_disabled import ( app_ftp_deployment_disabled, diff --git a/tests/providers/azure/services/app/app_function_access_keys_configured/app_function_access_keys_configured_test.py b/tests/providers/azure/services/app/app_function_access_keys_configured/app_function_access_keys_configured_test.py index 42d8360ee1..c81e8392d9 100644 --- a/tests/providers/azure/services/app/app_function_access_keys_configured/app_function_access_keys_configured_test.py +++ b/tests/providers/azure/services/app/app_function_access_keys_configured/app_function_access_keys_configured_test.py @@ -11,15 +11,12 @@ class Test_app_function_access_keys_configured: def test_app_no_subscriptions(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_access_keys_configured.app_function_access_keys_configured.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_access_keys_configured.app_function_access_keys_configured.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_access_keys_configured.app_function_access_keys_configured import ( app_function_access_keys_configured, @@ -35,15 +32,12 @@ class Test_app_function_access_keys_configured: app_client = mock.MagicMock app_client.functions = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_access_keys_configured.app_function_access_keys_configured.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_access_keys_configured.app_function_access_keys_configured.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_access_keys_configured.app_function_access_keys_configured import ( app_function_access_keys_configured, @@ -57,15 +51,12 @@ class Test_app_function_access_keys_configured: app_client = mock.MagicMock app_client.functions = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_access_keys_configured.app_function_access_keys_configured.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_access_keys_configured.app_function_access_keys_configured.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_access_keys_configured.app_function_access_keys_configured import ( app_function_access_keys_configured, @@ -109,15 +100,12 @@ class Test_app_function_access_keys_configured: app_client = mock.MagicMock app_client.functions = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_access_keys_configured.app_function_access_keys_configured.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_access_keys_configured.app_function_access_keys_configured.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_access_keys_configured.app_function_access_keys_configured import ( app_function_access_keys_configured, diff --git a/tests/providers/azure/services/app/app_function_application_insights_enabled/app_function_application_insights_enabled_test.py b/tests/providers/azure/services/app/app_function_application_insights_enabled/app_function_application_insights_enabled_test.py index 4f934f1977..ee624f29ed 100644 --- a/tests/providers/azure/services/app/app_function_application_insights_enabled/app_function_application_insights_enabled_test.py +++ b/tests/providers/azure/services/app/app_function_application_insights_enabled/app_function_application_insights_enabled_test.py @@ -11,15 +11,12 @@ class Test_app_function_application_insights_enabled: def test_app_no_subscriptions(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled import ( app_function_application_insights_enabled, @@ -34,15 +31,12 @@ class Test_app_function_application_insights_enabled: def test_app_subscription_empty(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled import ( app_function_application_insights_enabled, @@ -58,19 +52,15 @@ class Test_app_function_application_insights_enabled: app_client = mock.MagicMock app_insights = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled.app_client", - new=app_client, - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled.appinsights_client", - new=app_insights, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled.app_client", + new=app_client, + ), mock.patch( + "prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled.appinsights_client", + new=app_insights, ): from prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled import ( app_function_application_insights_enabled, @@ -127,19 +117,15 @@ class Test_app_function_application_insights_enabled: app_client = mock.MagicMock app_insights = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled.app_client", - new=app_client, - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled.appinsights_client", - new=app_insights, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled.app_client", + new=app_client, + ), mock.patch( + "prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled.appinsights_client", + new=app_insights, ): from prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled import ( app_function_application_insights_enabled, @@ -196,19 +182,15 @@ class Test_app_function_application_insights_enabled: app_client = mock.MagicMock app_insights = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled.app_client", - new=app_client, - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled.appinsights_client", - new=app_insights, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled.app_client", + new=app_client, + ), mock.patch( + "prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled.appinsights_client", + new=app_insights, ): from prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled import ( app_function_application_insights_enabled, @@ -265,19 +247,15 @@ class Test_app_function_application_insights_enabled: app_client = mock.MagicMock app_insights = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled.app_client", - new=app_client, - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled.appinsights_client", - new=app_insights, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled.app_client", + new=app_client, + ), mock.patch( + "prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled.appinsights_client", + new=app_insights, ): from prowler.providers.azure.services.app.app_function_application_insights_enabled.app_function_application_insights_enabled import ( app_function_application_insights_enabled, diff --git a/tests/providers/azure/services/app/app_function_ftps_deployment_disabled/app_function_ftps_deployment_disabled_test.py b/tests/providers/azure/services/app/app_function_ftps_deployment_disabled/app_function_ftps_deployment_disabled_test.py index 2546ba2f44..90bd22cb64 100644 --- a/tests/providers/azure/services/app/app_function_ftps_deployment_disabled/app_function_ftps_deployment_disabled_test.py +++ b/tests/providers/azure/services/app/app_function_ftps_deployment_disabled/app_function_ftps_deployment_disabled_test.py @@ -11,15 +11,12 @@ class Test_app_function_ftps_deployment_disabled: def test_no_subscriptions(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_ftps_deployment_disabled.app_function_ftps_deployment_disabled.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_ftps_deployment_disabled.app_function_ftps_deployment_disabled.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_ftps_deployment_disabled.app_function_ftps_deployment_disabled import ( app_function_ftps_deployment_disabled, @@ -34,15 +31,12 @@ class Test_app_function_ftps_deployment_disabled: def test_subscription_empty(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_ftps_deployment_disabled.app_function_ftps_deployment_disabled.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_ftps_deployment_disabled.app_function_ftps_deployment_disabled.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_ftps_deployment_disabled.app_function_ftps_deployment_disabled import ( app_function_ftps_deployment_disabled, @@ -57,15 +51,12 @@ class Test_app_function_ftps_deployment_disabled: def test_function_ftp_deployment_enabled(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_ftps_deployment_disabled.app_function_ftps_deployment_disabled.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_ftps_deployment_disabled.app_function_ftps_deployment_disabled.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_ftps_deployment_disabled.app_function_ftps_deployment_disabled import ( app_function_ftps_deployment_disabled, @@ -107,15 +98,12 @@ class Test_app_function_ftps_deployment_disabled: def test_function_ftps_deployment_enabled(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_ftps_deployment_disabled.app_function_ftps_deployment_disabled.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_ftps_deployment_disabled.app_function_ftps_deployment_disabled.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_ftps_deployment_disabled.app_function_ftps_deployment_disabled import ( app_function_ftps_deployment_disabled, @@ -157,15 +145,12 @@ class Test_app_function_ftps_deployment_disabled: def test_function_ftp_and_ftps_deployment_disabled(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_ftps_deployment_disabled.app_function_ftps_deployment_disabled.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_ftps_deployment_disabled.app_function_ftps_deployment_disabled.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_ftps_deployment_disabled.app_function_ftps_deployment_disabled import ( app_function_ftps_deployment_disabled, diff --git a/tests/providers/azure/services/app/app_function_identity_is_configured/app_function_identity_is_configured_test.py b/tests/providers/azure/services/app/app_function_identity_is_configured/app_function_identity_is_configured_test.py index ffa4a0e743..c300fcead4 100644 --- a/tests/providers/azure/services/app/app_function_identity_is_configured/app_function_identity_is_configured_test.py +++ b/tests/providers/azure/services/app/app_function_identity_is_configured/app_function_identity_is_configured_test.py @@ -11,15 +11,12 @@ class Test_app_function_identity_is_configured: def test_app_no_subscriptions(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_identity_is_configured.app_function_identity_is_configured.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_identity_is_configured.app_function_identity_is_configured.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_identity_is_configured.app_function_identity_is_configured import ( app_function_identity_is_configured, @@ -34,15 +31,12 @@ class Test_app_function_identity_is_configured: def test_app_subscription_empty(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_identity_is_configured.app_function_identity_is_configured.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_identity_is_configured.app_function_identity_is_configured.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_identity_is_configured.app_function_identity_is_configured import ( app_function_identity_is_configured, @@ -57,15 +51,12 @@ class Test_app_function_identity_is_configured: def test_app_function_no_identity(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_identity_is_configured.app_function_identity_is_configured.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_identity_is_configured.app_function_identity_is_configured.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_identity_is_configured.app_function_identity_is_configured import ( app_function_identity_is_configured, @@ -107,15 +98,12 @@ class Test_app_function_identity_is_configured: def test_app_function_identity_configured(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_identity_is_configured.app_function_identity_is_configured.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_identity_is_configured.app_function_identity_is_configured.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_identity_is_configured.app_function_identity_is_configured import ( app_function_identity_is_configured, diff --git a/tests/providers/azure/services/app/app_function_identity_without_admin_privileges/app_function_identity_without_admin_privileges_test.py b/tests/providers/azure/services/app/app_function_identity_without_admin_privileges/app_function_identity_without_admin_privileges_test.py index 68dc5293a7..480eb5cb2f 100644 --- a/tests/providers/azure/services/app/app_function_identity_without_admin_privileges/app_function_identity_without_admin_privileges_test.py +++ b/tests/providers/azure/services/app/app_function_identity_without_admin_privileges/app_function_identity_without_admin_privileges_test.py @@ -12,15 +12,12 @@ class Test_app_function_identity_without_admin_privileges: def test_app_no_subscriptions(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_identity_without_admin_privileges.app_function_identity_without_admin_privileges.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_identity_without_admin_privileges.app_function_identity_without_admin_privileges.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_identity_without_admin_privileges.app_function_identity_without_admin_privileges import ( app_function_identity_without_admin_privileges, @@ -35,15 +32,12 @@ class Test_app_function_identity_without_admin_privileges: def test_app_subscription_empty(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_identity_without_admin_privileges.app_function_identity_without_admin_privileges.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_identity_without_admin_privileges.app_function_identity_without_admin_privileges.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_identity_without_admin_privileges.app_function_identity_without_admin_privileges import ( app_function_identity_without_admin_privileges, @@ -58,15 +52,12 @@ class Test_app_function_identity_without_admin_privileges: def test_app_function_no_identity(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_identity_without_admin_privileges.app_function_identity_without_admin_privileges.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_identity_without_admin_privileges.app_function_identity_without_admin_privileges.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_identity_without_admin_privileges.app_function_identity_without_admin_privileges import ( app_function_identity_without_admin_privileges, @@ -100,19 +91,15 @@ class Test_app_function_identity_without_admin_privileges: app_client = mock.MagicMock iam_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_identity_without_admin_privileges.app_function_identity_without_admin_privileges.app_client", - new=app_client, - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_identity_without_admin_privileges.app_function_identity_without_admin_privileges.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_identity_without_admin_privileges.app_function_identity_without_admin_privileges.app_client", + new=app_client, + ), mock.patch( + "prowler.providers.azure.services.app.app_function_identity_without_admin_privileges.app_function_identity_without_admin_privileges.iam_client", + new=iam_client, ): from prowler.providers.azure.services.app.app_function_identity_without_admin_privileges.app_function_identity_without_admin_privileges import ( app_function_identity_without_admin_privileges, @@ -181,19 +168,15 @@ class Test_app_function_identity_without_admin_privileges: app_client = mock.MagicMock iam_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_identity_without_admin_privileges.app_function_identity_without_admin_privileges.app_client", - new=app_client, - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_identity_without_admin_privileges.app_function_identity_without_admin_privileges.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_identity_without_admin_privileges.app_function_identity_without_admin_privileges.app_client", + new=app_client, + ), mock.patch( + "prowler.providers.azure.services.app.app_function_identity_without_admin_privileges.app_function_identity_without_admin_privileges.iam_client", + new=iam_client, ): from prowler.providers.azure.services.app.app_function_identity_without_admin_privileges.app_function_identity_without_admin_privileges import ( app_function_identity_without_admin_privileges, diff --git a/tests/providers/azure/services/app/app_function_latest_runtime_version/app_function_latest_runtime_version_test.py b/tests/providers/azure/services/app/app_function_latest_runtime_version/app_function_latest_runtime_version_test.py index 533874141d..581458c7a1 100644 --- a/tests/providers/azure/services/app/app_function_latest_runtime_version/app_function_latest_runtime_version_test.py +++ b/tests/providers/azure/services/app/app_function_latest_runtime_version/app_function_latest_runtime_version_test.py @@ -11,15 +11,12 @@ class Test_app_function_latest_runtime_version: def test_app_no_subscriptions(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_latest_runtime_version.app_function_latest_runtime_version.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_latest_runtime_version.app_function_latest_runtime_version.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_latest_runtime_version.app_function_latest_runtime_version import ( app_function_latest_runtime_version, @@ -34,15 +31,12 @@ class Test_app_function_latest_runtime_version: def test_app_subscription_empty(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_latest_runtime_version.app_function_latest_runtime_version.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_latest_runtime_version.app_function_latest_runtime_version.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_latest_runtime_version.app_function_latest_runtime_version import ( app_function_latest_runtime_version, @@ -57,15 +51,12 @@ class Test_app_function_latest_runtime_version: def test_app_function_runtime_is_latest(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_latest_runtime_version.app_function_latest_runtime_version.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_latest_runtime_version.app_function_latest_runtime_version.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_latest_runtime_version.app_function_latest_runtime_version import ( app_function_latest_runtime_version, @@ -106,15 +97,12 @@ class Test_app_function_latest_runtime_version: def test_app_function_runtime_is_not_latest(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_latest_runtime_version.app_function_latest_runtime_version.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_latest_runtime_version.app_function_latest_runtime_version.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_latest_runtime_version.app_function_latest_runtime_version import ( app_function_latest_runtime_version, diff --git a/tests/providers/azure/services/app/app_function_not_publicly_accessible/app_function_not_publicly_accessible_test.py b/tests/providers/azure/services/app/app_function_not_publicly_accessible/app_function_not_publicly_accessible_test.py index 749a094e65..326f110a30 100644 --- a/tests/providers/azure/services/app/app_function_not_publicly_accessible/app_function_not_publicly_accessible_test.py +++ b/tests/providers/azure/services/app/app_function_not_publicly_accessible/app_function_not_publicly_accessible_test.py @@ -11,15 +11,12 @@ class Test_app_function_not_publicly_accessible: def test_app_no_subscriptions(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_not_publicly_accessible.app_function_not_publicly_accessible.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_not_publicly_accessible.app_function_not_publicly_accessible.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_not_publicly_accessible.app_function_not_publicly_accessible import ( app_function_not_publicly_accessible, @@ -34,15 +31,12 @@ class Test_app_function_not_publicly_accessible: def test_app_subscription_empty(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_not_publicly_accessible.app_function_not_publicly_accessible.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_not_publicly_accessible.app_function_not_publicly_accessible.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_not_publicly_accessible.app_function_not_publicly_accessible import ( app_function_not_publicly_accessible, @@ -57,15 +51,12 @@ class Test_app_function_not_publicly_accessible: def test_app_function_not_publicly_accessible(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_not_publicly_accessible.app_function_not_publicly_accessible.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_not_publicly_accessible.app_function_not_publicly_accessible.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_not_publicly_accessible.app_function_not_publicly_accessible import ( app_function_not_publicly_accessible, @@ -107,15 +98,12 @@ class Test_app_function_not_publicly_accessible: def test_app_function_publicly_accessible(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_not_publicly_accessible.app_function_not_publicly_accessible.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_not_publicly_accessible.app_function_not_publicly_accessible.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_not_publicly_accessible.app_function_not_publicly_accessible import ( app_function_not_publicly_accessible, diff --git a/tests/providers/azure/services/app/app_function_vnet_integration_enabled/app_function_vnet_integration_enabled_test.py b/tests/providers/azure/services/app/app_function_vnet_integration_enabled/app_function_vnet_integration_enabled_test.py index 33a99d0086..28820837ed 100644 --- a/tests/providers/azure/services/app/app_function_vnet_integration_enabled/app_function_vnet_integration_enabled_test.py +++ b/tests/providers/azure/services/app/app_function_vnet_integration_enabled/app_function_vnet_integration_enabled_test.py @@ -11,15 +11,12 @@ class Test_app_function_vnet_integration_enabled: def test_app_no_subscriptions(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_vnet_integration_enabled.app_function_vnet_integration_enabled.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_vnet_integration_enabled.app_function_vnet_integration_enabled.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_vnet_integration_enabled.app_function_vnet_integration_enabled import ( app_function_vnet_integration_enabled, @@ -34,15 +31,12 @@ class Test_app_function_vnet_integration_enabled: def test_app_subscription_empty(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_vnet_integration_enabled.app_function_vnet_integration_enabled.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_vnet_integration_enabled.app_function_vnet_integration_enabled.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_vnet_integration_enabled.app_function_vnet_integration_enabled import ( app_function_vnet_integration_enabled, @@ -57,15 +51,12 @@ class Test_app_function_vnet_integration_enabled: def test_app_function_vnet_integration_enabled(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_vnet_integration_enabled.app_function_vnet_integration_enabled.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_vnet_integration_enabled.app_function_vnet_integration_enabled.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_vnet_integration_enabled.app_function_vnet_integration_enabled import ( app_function_vnet_integration_enabled, @@ -106,15 +97,12 @@ class Test_app_function_vnet_integration_enabled: def test_app_function_vnet_integration_disabled(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_function_vnet_integration_enabled.app_function_vnet_integration_enabled.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_function_vnet_integration_enabled.app_function_vnet_integration_enabled.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_function_vnet_integration_enabled.app_function_vnet_integration_enabled import ( app_function_vnet_integration_enabled, diff --git a/tests/providers/azure/services/app/app_http_logs_enabled/app_http_logs_enabled_test.py b/tests/providers/azure/services/app/app_http_logs_enabled/app_http_logs_enabled_test.py index d318175a22..b139c67f79 100644 --- a/tests/providers/azure/services/app/app_http_logs_enabled/app_http_logs_enabled_test.py +++ b/tests/providers/azure/services/app/app_http_logs_enabled/app_http_logs_enabled_test.py @@ -12,15 +12,12 @@ class Test_app_http_logs_enabled: app_client = mock.MagicMock app_client.apps = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_http_logs_enabled.app_http_logs_enabled.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_http_logs_enabled.app_http_logs_enabled.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_http_logs_enabled.app_http_logs_enabled import ( @@ -35,15 +32,12 @@ class Test_app_http_logs_enabled: app_client = mock.MagicMock app_client.apps = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_register_with_identity.app_register_with_identity.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_register_with_identity.app_register_with_identity.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_register_with_identity.app_register_with_identity import ( app_register_with_identity, @@ -55,15 +49,12 @@ class Test_app_http_logs_enabled: def test_no_diagnostics_settings(self): app_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_http_logs_enabled.app_http_logs_enabled.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_http_logs_enabled.app_http_logs_enabled.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_http_logs_enabled.app_http_logs_enabled import ( app_http_logs_enabled, @@ -100,15 +91,12 @@ class Test_app_http_logs_enabled: def test_diagnostic_setting_configured(self): app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_http_logs_enabled.app_http_logs_enabled.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_http_logs_enabled.app_http_logs_enabled.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_http_logs_enabled.app_http_logs_enabled import ( app_http_logs_enabled, diff --git a/tests/providers/azure/services/app/app_minimum_tls_version_12/app_minimum_tls_version_12_test.py b/tests/providers/azure/services/app/app_minimum_tls_version_12/app_minimum_tls_version_12_test.py index a3055c855a..e554867049 100644 --- a/tests/providers/azure/services/app/app_minimum_tls_version_12/app_minimum_tls_version_12_test.py +++ b/tests/providers/azure/services/app/app_minimum_tls_version_12/app_minimum_tls_version_12_test.py @@ -12,15 +12,12 @@ class Test_app_minimum_tls_version_12: app_client = mock.MagicMock app_client.apps = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_minimum_tls_version_12.app_minimum_tls_version_12.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_minimum_tls_version_12.app_minimum_tls_version_12.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_minimum_tls_version_12.app_minimum_tls_version_12 import ( app_minimum_tls_version_12, @@ -34,15 +31,12 @@ class Test_app_minimum_tls_version_12: app_client = mock.MagicMock app_client.apps = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_minimum_tls_version_12.app_minimum_tls_version_12.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_minimum_tls_version_12.app_minimum_tls_version_12.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_minimum_tls_version_12.app_minimum_tls_version_12 import ( app_minimum_tls_version_12, @@ -56,15 +50,12 @@ class Test_app_minimum_tls_version_12: resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_minimum_tls_version_12.app_minimum_tls_version_12.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_minimum_tls_version_12.app_minimum_tls_version_12.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_minimum_tls_version_12.app_minimum_tls_version_12 import ( app_minimum_tls_version_12, @@ -102,15 +93,12 @@ class Test_app_minimum_tls_version_12: resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_minimum_tls_version_12.app_minimum_tls_version_12.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_minimum_tls_version_12.app_minimum_tls_version_12.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_minimum_tls_version_12.app_minimum_tls_version_12 import ( app_minimum_tls_version_12, @@ -148,15 +136,12 @@ class Test_app_minimum_tls_version_12: resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_minimum_tls_version_12.app_minimum_tls_version_12.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_minimum_tls_version_12.app_minimum_tls_version_12.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_minimum_tls_version_12.app_minimum_tls_version_12 import ( app_minimum_tls_version_12, @@ -194,15 +179,12 @@ class Test_app_minimum_tls_version_12: resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_minimum_tls_version_12.app_minimum_tls_version_12.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_minimum_tls_version_12.app_minimum_tls_version_12.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_minimum_tls_version_12.app_minimum_tls_version_12 import ( app_minimum_tls_version_12, diff --git a/tests/providers/azure/services/app/app_register_with_identity/app_register_with_identity_test.py b/tests/providers/azure/services/app/app_register_with_identity/app_register_with_identity_test.py index 785439fe0e..866f0623a1 100644 --- a/tests/providers/azure/services/app/app_register_with_identity/app_register_with_identity_test.py +++ b/tests/providers/azure/services/app/app_register_with_identity/app_register_with_identity_test.py @@ -12,15 +12,12 @@ class Test_app_register_with_identity: app_client = mock.MagicMock app_client.apps = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_register_with_identity.app_register_with_identity.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_register_with_identity.app_register_with_identity.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_register_with_identity.app_register_with_identity import ( app_register_with_identity, @@ -34,15 +31,12 @@ class Test_app_register_with_identity: app_client = mock.MagicMock app_client.apps = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_register_with_identity.app_register_with_identity.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_register_with_identity.app_register_with_identity.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_register_with_identity.app_register_with_identity import ( app_register_with_identity, @@ -56,15 +50,12 @@ class Test_app_register_with_identity: resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_register_with_identity.app_register_with_identity.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_register_with_identity.app_register_with_identity.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_register_with_identity.app_register_with_identity import ( app_register_with_identity, @@ -102,15 +93,12 @@ class Test_app_register_with_identity: resource_id = f"/subscriptions/{uuid4()}" app_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.app.app_register_with_identity.app_register_with_identity.app_client", - new=app_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.app.app_register_with_identity.app_register_with_identity.app_client", + new=app_client, ): from prowler.providers.azure.services.app.app_register_with_identity.app_register_with_identity import ( app_register_with_identity, diff --git a/tests/providers/azure/services/appinsights/appinsights_ensure_is_configured/appinsights_ensure_is_configured_test.py b/tests/providers/azure/services/appinsights/appinsights_ensure_is_configured/appinsights_ensure_is_configured_test.py index d948d78c18..eabd441f92 100644 --- a/tests/providers/azure/services/appinsights/appinsights_ensure_is_configured/appinsights_ensure_is_configured_test.py +++ b/tests/providers/azure/services/appinsights/appinsights_ensure_is_configured/appinsights_ensure_is_configured_test.py @@ -12,15 +12,12 @@ class Test_appinsights_ensure_is_configured: appinsights_client = mock.MagicMock appinsights_client.components = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.appinsights.appinsights_ensure_is_configured.appinsights_ensure_is_configured.appinsights_client", - new=appinsights_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.appinsights.appinsights_ensure_is_configured.appinsights_ensure_is_configured.appinsights_client", + new=appinsights_client, ): from prowler.providers.azure.services.appinsights.appinsights_ensure_is_configured.appinsights_ensure_is_configured import ( appinsights_ensure_is_configured, @@ -34,15 +31,12 @@ class Test_appinsights_ensure_is_configured: appinsights_client = mock.MagicMock appinsights_client.components = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.appinsights.appinsights_ensure_is_configured.appinsights_ensure_is_configured.appinsights_client", - new=appinsights_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.appinsights.appinsights_ensure_is_configured.appinsights_ensure_is_configured.appinsights_client", + new=appinsights_client, ): from prowler.providers.azure.services.appinsights.appinsights_ensure_is_configured.appinsights_ensure_is_configured import ( appinsights_ensure_is_configured, @@ -74,15 +68,12 @@ class Test_appinsights_ensure_is_configured: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.appinsights.appinsights_ensure_is_configured.appinsights_ensure_is_configured.appinsights_client", - new=appinsights_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.appinsights.appinsights_ensure_is_configured.appinsights_ensure_is_configured.appinsights_client", + new=appinsights_client, ): from prowler.providers.azure.services.appinsights.appinsights_ensure_is_configured.appinsights_ensure_is_configured import ( appinsights_ensure_is_configured, diff --git a/tests/providers/azure/services/containerregistry/containerregistry_admin_user_disabled/containerregistry_admin_user_disabled_test.py b/tests/providers/azure/services/containerregistry/containerregistry_admin_user_disabled/containerregistry_admin_user_disabled_test.py index a2295b6d06..6f4b3e6c5c 100644 --- a/tests/providers/azure/services/containerregistry/containerregistry_admin_user_disabled/containerregistry_admin_user_disabled_test.py +++ b/tests/providers/azure/services/containerregistry/containerregistry_admin_user_disabled/containerregistry_admin_user_disabled_test.py @@ -13,15 +13,12 @@ class TestContainerRegistryAdminUserDisabled: containerregistry_client = MagicMock() containerregistry_client.registries = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.containerregistry.containerregistry_admin_user_disabled.containerregistry_admin_user_disabled.containerregistry_client", - new=containerregistry_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.containerregistry.containerregistry_admin_user_disabled.containerregistry_admin_user_disabled.containerregistry_client", + new=containerregistry_client, ): from prowler.providers.azure.services.containerregistry.containerregistry_admin_user_disabled.containerregistry_admin_user_disabled import ( containerregistry_admin_user_disabled, @@ -35,15 +32,12 @@ class TestContainerRegistryAdminUserDisabled: containerregistry_client = MagicMock() registry_id = str(uuid4()) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.containerregistry.containerregistry_admin_user_disabled.containerregistry_admin_user_disabled.containerregistry_client", - new=containerregistry_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.containerregistry.containerregistry_admin_user_disabled.containerregistry_admin_user_disabled.containerregistry_client", + new=containerregistry_client, ): from prowler.providers.azure.services.containerregistry.containerregistry_admin_user_disabled.containerregistry_admin_user_disabled import ( containerregistry_admin_user_disabled, @@ -92,15 +86,12 @@ class TestContainerRegistryAdminUserDisabled: containerregistry_client = mock.MagicMock() containerregistry_client.registries = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.containerregistry.containerregistry_admin_user_disabled.containerregistry_admin_user_disabled.containerregistry_client", - new=containerregistry_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.containerregistry.containerregistry_admin_user_disabled.containerregistry_admin_user_disabled.containerregistry_client", + new=containerregistry_client, ): from prowler.providers.azure.services.containerregistry.containerregistry_admin_user_disabled.containerregistry_admin_user_disabled import ( containerregistry_admin_user_disabled, diff --git a/tests/providers/azure/services/containerregistry/containerregistry_not_publicly_accessible/containerregistry_not_publicly_accessible_test.py b/tests/providers/azure/services/containerregistry/containerregistry_not_publicly_accessible/containerregistry_not_publicly_accessible_test.py index 683552daca..a54e5b8554 100644 --- a/tests/providers/azure/services/containerregistry/containerregistry_not_publicly_accessible/containerregistry_not_publicly_accessible_test.py +++ b/tests/providers/azure/services/containerregistry/containerregistry_not_publicly_accessible/containerregistry_not_publicly_accessible_test.py @@ -13,15 +13,12 @@ class Test_containerregistry_not_publicly_accessible: containerregistry_client = MagicMock() containerregistry_client.registries = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.containerregistry.containerregistry_not_publicly_accessible.containerregistry_not_publicly_accessible.containerregistry_client", - new=containerregistry_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.containerregistry.containerregistry_not_publicly_accessible.containerregistry_not_publicly_accessible.containerregistry_client", + new=containerregistry_client, ): from prowler.providers.azure.services.containerregistry.containerregistry_not_publicly_accessible.containerregistry_not_publicly_accessible import ( containerregistry_not_publicly_accessible, @@ -35,15 +32,12 @@ class Test_containerregistry_not_publicly_accessible: containerregistry_client = MagicMock() registry_id = str(uuid4()) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.containerregistry.containerregistry_not_publicly_accessible.containerregistry_not_publicly_accessible.containerregistry_client", - new=containerregistry_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.containerregistry.containerregistry_not_publicly_accessible.containerregistry_not_publicly_accessible.containerregistry_client", + new=containerregistry_client, ): from prowler.providers.azure.services.containerregistry.containerregistry_not_publicly_accessible.containerregistry_not_publicly_accessible import ( containerregistry_not_publicly_accessible, @@ -109,15 +103,12 @@ class Test_containerregistry_not_publicly_accessible: containerregistry_client = mock.MagicMock() containerregistry_client.registries = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.containerregistry.containerregistry_not_publicly_accessible.containerregistry_not_publicly_accessible.containerregistry_client", - new=containerregistry_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.containerregistry.containerregistry_not_publicly_accessible.containerregistry_not_publicly_accessible.containerregistry_client", + new=containerregistry_client, ): from prowler.providers.azure.services.containerregistry.containerregistry_not_publicly_accessible.containerregistry_not_publicly_accessible import ( containerregistry_not_publicly_accessible, diff --git a/tests/providers/azure/services/containerregistry/containerregistry_service_test.py b/tests/providers/azure/services/containerregistry/containerregistry_service_test.py index 3e4a02406e..744459a75f 100644 --- a/tests/providers/azure/services/containerregistry/containerregistry_service_test.py +++ b/tests/providers/azure/services/containerregistry/containerregistry_service_test.py @@ -9,15 +9,12 @@ from tests.providers.azure.azure_fixtures import ( class TestContainerRegistryService: def test_get_container_registry(self): - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - patch( - "prowler.providers.azure.services.monitor.monitor_service.Monitor", - new=MagicMock(), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), patch( + "prowler.providers.azure.services.monitor.monitor_service.Monitor", + new=MagicMock(), ): from prowler.providers.azure.services.containerregistry.containerregistry_service import ( ContainerRegistryInfo, diff --git a/tests/providers/azure/services/containerregistry/containerregistry_uses_private_link/containerregistry_uses_private_link_test.py b/tests/providers/azure/services/containerregistry/containerregistry_uses_private_link/containerregistry_uses_private_link_test.py index c49da40f98..422aebdbe4 100644 --- a/tests/providers/azure/services/containerregistry/containerregistry_uses_private_link/containerregistry_uses_private_link_test.py +++ b/tests/providers/azure/services/containerregistry/containerregistry_uses_private_link/containerregistry_uses_private_link_test.py @@ -19,15 +19,12 @@ class Test_containerregistry_uses_private_link: containerregistry_client = MagicMock() containerregistry_client.registries = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.containerregistry.containerregistry_uses_private_link.containerregistry_uses_private_link.containerregistry_client", - new=containerregistry_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.containerregistry.containerregistry_uses_private_link.containerregistry_uses_private_link.containerregistry_client", + new=containerregistry_client, ): from prowler.providers.azure.services.containerregistry.containerregistry_uses_private_link.containerregistry_uses_private_link import ( containerregistry_uses_private_link, @@ -41,15 +38,12 @@ class Test_containerregistry_uses_private_link: containerregistry_client = MagicMock() registry_id = str(uuid4()) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.containerregistry.containerregistry_uses_private_link.containerregistry_uses_private_link.containerregistry_client", - new=containerregistry_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.containerregistry.containerregistry_uses_private_link.containerregistry_uses_private_link.containerregistry_client", + new=containerregistry_client, ): from prowler.providers.azure.services.containerregistry.containerregistry_service import ( ContainerRegistryInfo, @@ -98,15 +92,12 @@ class Test_containerregistry_uses_private_link: containerregistry_client = mock.MagicMock() containerregistry_client.registries = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.containerregistry.containerregistry_uses_private_link.containerregistry_uses_private_link.containerregistry_client", - new=containerregistry_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.containerregistry.containerregistry_uses_private_link.containerregistry_uses_private_link.containerregistry_client", + new=containerregistry_client, ): from prowler.providers.azure.services.containerregistry.containerregistry_service import ( ContainerRegistryInfo, diff --git a/tests/providers/azure/services/cosmosdb/cosmosdb_account_firewall_use_selected_networks/cosmosdb_account_firewall_use_selected_networks_test.py b/tests/providers/azure/services/cosmosdb/cosmosdb_account_firewall_use_selected_networks/cosmosdb_account_firewall_use_selected_networks_test.py index 7d7b793954..347f65e393 100644 --- a/tests/providers/azure/services/cosmosdb/cosmosdb_account_firewall_use_selected_networks/cosmosdb_account_firewall_use_selected_networks_test.py +++ b/tests/providers/azure/services/cosmosdb/cosmosdb_account_firewall_use_selected_networks/cosmosdb_account_firewall_use_selected_networks_test.py @@ -13,15 +13,12 @@ class Test_cosmosdb_account_firewall_use_selected_networks: cosmosdb_client = mock.MagicMock cosmosdb_client.accounts = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.cosmosdb.cosmosdb_account_firewall_use_selected_networks.cosmosdb_account_firewall_use_selected_networks.cosmosdb_client", - new=cosmosdb_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.cosmosdb.cosmosdb_account_firewall_use_selected_networks.cosmosdb_account_firewall_use_selected_networks.cosmosdb_client", + new=cosmosdb_client, ): from prowler.providers.azure.services.cosmosdb.cosmosdb_account_firewall_use_selected_networks.cosmosdb_account_firewall_use_selected_networks import ( cosmosdb_account_firewall_use_selected_networks, @@ -51,15 +48,12 @@ class Test_cosmosdb_account_firewall_use_selected_networks: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.cosmosdb.cosmosdb_account_firewall_use_selected_networks.cosmosdb_account_firewall_use_selected_networks.cosmosdb_client", - new=cosmosdb_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.cosmosdb.cosmosdb_account_firewall_use_selected_networks.cosmosdb_account_firewall_use_selected_networks.cosmosdb_client", + new=cosmosdb_client, ): from prowler.providers.azure.services.cosmosdb.cosmosdb_account_firewall_use_selected_networks.cosmosdb_account_firewall_use_selected_networks import ( cosmosdb_account_firewall_use_selected_networks, @@ -98,15 +92,12 @@ class Test_cosmosdb_account_firewall_use_selected_networks: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.cosmosdb.cosmosdb_account_firewall_use_selected_networks.cosmosdb_account_firewall_use_selected_networks.cosmosdb_client", - new=cosmosdb_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.cosmosdb.cosmosdb_account_firewall_use_selected_networks.cosmosdb_account_firewall_use_selected_networks.cosmosdb_client", + new=cosmosdb_client, ): from prowler.providers.azure.services.cosmosdb.cosmosdb_account_firewall_use_selected_networks.cosmosdb_account_firewall_use_selected_networks import ( cosmosdb_account_firewall_use_selected_networks, diff --git a/tests/providers/azure/services/cosmosdb/cosmosdb_account_use_aad_and_rbac/cosmosdb_account_use_aad_and_rbac_test.py b/tests/providers/azure/services/cosmosdb/cosmosdb_account_use_aad_and_rbac/cosmosdb_account_use_aad_and_rbac_test.py index 8fcdf99fa9..071fbd042e 100644 --- a/tests/providers/azure/services/cosmosdb/cosmosdb_account_use_aad_and_rbac/cosmosdb_account_use_aad_and_rbac_test.py +++ b/tests/providers/azure/services/cosmosdb/cosmosdb_account_use_aad_and_rbac/cosmosdb_account_use_aad_and_rbac_test.py @@ -13,15 +13,12 @@ class Test_cosmosdb_account_use_aad_and_rbac: cosmosdb_client = mock.MagicMock cosmosdb_client.accounts = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.cosmosdb.cosmosdb_account_use_aad_and_rbac.cosmosdb_account_use_aad_and_rbac.cosmosdb_client", - new=cosmosdb_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.cosmosdb.cosmosdb_account_use_aad_and_rbac.cosmosdb_account_use_aad_and_rbac.cosmosdb_client", + new=cosmosdb_client, ): from prowler.providers.azure.services.cosmosdb.cosmosdb_account_use_aad_and_rbac.cosmosdb_account_use_aad_and_rbac import ( cosmosdb_account_use_aad_and_rbac, @@ -51,15 +48,12 @@ class Test_cosmosdb_account_use_aad_and_rbac: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.cosmosdb.cosmosdb_account_use_aad_and_rbac.cosmosdb_account_use_aad_and_rbac.cosmosdb_client", - new=cosmosdb_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.cosmosdb.cosmosdb_account_use_aad_and_rbac.cosmosdb_account_use_aad_and_rbac.cosmosdb_client", + new=cosmosdb_client, ): from prowler.providers.azure.services.cosmosdb.cosmosdb_account_use_aad_and_rbac.cosmosdb_account_use_aad_and_rbac import ( cosmosdb_account_use_aad_and_rbac, @@ -98,15 +92,12 @@ class Test_cosmosdb_account_use_aad_and_rbac: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.cosmosdb.cosmosdb_account_use_aad_and_rbac.cosmosdb_account_use_aad_and_rbac.cosmosdb_client", - new=cosmosdb_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.cosmosdb.cosmosdb_account_use_aad_and_rbac.cosmosdb_account_use_aad_and_rbac.cosmosdb_client", + new=cosmosdb_client, ): from prowler.providers.azure.services.cosmosdb.cosmosdb_account_use_aad_and_rbac.cosmosdb_account_use_aad_and_rbac import ( cosmosdb_account_use_aad_and_rbac, diff --git a/tests/providers/azure/services/cosmosdb/cosmosdb_account_use_private_endpoints/cosmosdb_account_use_private_endpoints_test.py b/tests/providers/azure/services/cosmosdb/cosmosdb_account_use_private_endpoints/cosmosdb_account_use_private_endpoints_test.py index d3827dee0b..84eee1eca4 100644 --- a/tests/providers/azure/services/cosmosdb/cosmosdb_account_use_private_endpoints/cosmosdb_account_use_private_endpoints_test.py +++ b/tests/providers/azure/services/cosmosdb/cosmosdb_account_use_private_endpoints/cosmosdb_account_use_private_endpoints_test.py @@ -15,15 +15,12 @@ class Test_cosmosdb_account_use_private_endpoints: cosmosdb_client = mock.MagicMock cosmosdb_client.accounts = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.cosmosdb.cosmosdb_account_use_private_endpoints.cosmosdb_account_use_private_endpoints.cosmosdb_client", - new=cosmosdb_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.cosmosdb.cosmosdb_account_use_private_endpoints.cosmosdb_account_use_private_endpoints.cosmosdb_client", + new=cosmosdb_client, ): from prowler.providers.azure.services.cosmosdb.cosmosdb_account_use_private_endpoints.cosmosdb_account_use_private_endpoints import ( cosmosdb_account_use_private_endpoints, @@ -53,15 +50,12 @@ class Test_cosmosdb_account_use_private_endpoints: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.cosmosdb.cosmosdb_account_use_private_endpoints.cosmosdb_account_use_private_endpoints.cosmosdb_client", - new=cosmosdb_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.cosmosdb.cosmosdb_account_use_private_endpoints.cosmosdb_account_use_private_endpoints.cosmosdb_client", + new=cosmosdb_client, ): from prowler.providers.azure.services.cosmosdb.cosmosdb_account_use_private_endpoints.cosmosdb_account_use_private_endpoints import ( cosmosdb_account_use_private_endpoints, @@ -104,15 +98,12 @@ class Test_cosmosdb_account_use_private_endpoints: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.cosmosdb.cosmosdb_account_use_private_endpoints.cosmosdb_account_use_private_endpoints.cosmosdb_client", - new=cosmosdb_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.cosmosdb.cosmosdb_account_use_private_endpoints.cosmosdb_account_use_private_endpoints.cosmosdb_client", + new=cosmosdb_client, ): from prowler.providers.azure.services.cosmosdb.cosmosdb_account_use_private_endpoints.cosmosdb_account_use_private_endpoints import ( cosmosdb_account_use_private_endpoints, diff --git a/tests/providers/azure/services/defender/defender_additional_email_configured_with_a_security_contact/defender_additional_email_configured_with_a_security_contact_test.py b/tests/providers/azure/services/defender/defender_additional_email_configured_with_a_security_contact/defender_additional_email_configured_with_a_security_contact_test.py index bb63fd7c5b..8bb0eacfed 100644 --- a/tests/providers/azure/services/defender/defender_additional_email_configured_with_a_security_contact/defender_additional_email_configured_with_a_security_contact_test.py +++ b/tests/providers/azure/services/defender/defender_additional_email_configured_with_a_security_contact/defender_additional_email_configured_with_a_security_contact_test.py @@ -13,15 +13,12 @@ class Test_defender_additional_email_configured_with_a_security_contact: defender_client = mock.MagicMock defender_client.security_contacts = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact import ( defender_additional_email_configured_with_a_security_contact, @@ -49,15 +46,12 @@ class Test_defender_additional_email_configured_with_a_security_contact: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact import ( defender_additional_email_configured_with_a_security_contact, @@ -93,15 +87,12 @@ class Test_defender_additional_email_configured_with_a_security_contact: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact import ( defender_additional_email_configured_with_a_security_contact, @@ -137,15 +128,12 @@ class Test_defender_additional_email_configured_with_a_security_contact: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact import ( defender_additional_email_configured_with_a_security_contact, @@ -181,15 +169,12 @@ class Test_defender_additional_email_configured_with_a_security_contact: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact import ( defender_additional_email_configured_with_a_security_contact, @@ -225,15 +210,12 @@ class Test_defender_additional_email_configured_with_a_security_contact: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact import ( defender_additional_email_configured_with_a_security_contact, @@ -268,15 +250,12 @@ class Test_defender_additional_email_configured_with_a_security_contact: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_additional_email_configured_with_a_security_contact.defender_additional_email_configured_with_a_security_contact import ( defender_additional_email_configured_with_a_security_contact, diff --git a/tests/providers/azure/services/defender/defender_assessments_vm_endpoint_protection_installed/defender_assessments_vm_endpoint_protection_installed_test.py b/tests/providers/azure/services/defender/defender_assessments_vm_endpoint_protection_installed/defender_assessments_vm_endpoint_protection_installed_test.py index dd16c7571c..c5ca349648 100644 --- a/tests/providers/azure/services/defender/defender_assessments_vm_endpoint_protection_installed/defender_assessments_vm_endpoint_protection_installed_test.py +++ b/tests/providers/azure/services/defender/defender_assessments_vm_endpoint_protection_installed/defender_assessments_vm_endpoint_protection_installed_test.py @@ -13,15 +13,12 @@ class Test_defender_assessments_vm_endpoint_protection_installed: defender_client = mock.MagicMock defender_client.assessments = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_assessments_vm_endpoint_protection_installed.defender_assessments_vm_endpoint_protection_installed.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_assessments_vm_endpoint_protection_installed.defender_assessments_vm_endpoint_protection_installed.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_assessments_vm_endpoint_protection_installed.defender_assessments_vm_endpoint_protection_installed import ( defender_assessments_vm_endpoint_protection_installed, @@ -35,15 +32,12 @@ class Test_defender_assessments_vm_endpoint_protection_installed: defender_client = mock.MagicMock defender_client.assessments = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_assessments_vm_endpoint_protection_installed.defender_assessments_vm_endpoint_protection_installed.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_assessments_vm_endpoint_protection_installed.defender_assessments_vm_endpoint_protection_installed.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_assessments_vm_endpoint_protection_installed.defender_assessments_vm_endpoint_protection_installed import ( defender_assessments_vm_endpoint_protection_installed, @@ -66,15 +60,12 @@ class Test_defender_assessments_vm_endpoint_protection_installed: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_assessments_vm_endpoint_protection_installed.defender_assessments_vm_endpoint_protection_installed.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_assessments_vm_endpoint_protection_installed.defender_assessments_vm_endpoint_protection_installed.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_assessments_vm_endpoint_protection_installed.defender_assessments_vm_endpoint_protection_installed import ( defender_assessments_vm_endpoint_protection_installed, @@ -104,15 +95,12 @@ class Test_defender_assessments_vm_endpoint_protection_installed: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_assessments_vm_endpoint_protection_installed.defender_assessments_vm_endpoint_protection_installed.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_assessments_vm_endpoint_protection_installed.defender_assessments_vm_endpoint_protection_installed.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_assessments_vm_endpoint_protection_installed.defender_assessments_vm_endpoint_protection_installed import ( defender_assessments_vm_endpoint_protection_installed, diff --git a/tests/providers/azure/services/defender/defender_auto_provisioning_log_analytics_agent_vms_on/defender_auto_provisioning_log_analytics_agent_vms_on_test.py b/tests/providers/azure/services/defender/defender_auto_provisioning_log_analytics_agent_vms_on/defender_auto_provisioning_log_analytics_agent_vms_on_test.py index 3f39654c2b..0b87647d11 100644 --- a/tests/providers/azure/services/defender/defender_auto_provisioning_log_analytics_agent_vms_on/defender_auto_provisioning_log_analytics_agent_vms_on_test.py +++ b/tests/providers/azure/services/defender/defender_auto_provisioning_log_analytics_agent_vms_on/defender_auto_provisioning_log_analytics_agent_vms_on_test.py @@ -15,15 +15,12 @@ class Test_defender_auto_provisioning_log_analytics_agent_vms_on: defender_client = mock.MagicMock defender_client.auto_provisioning_settings = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_auto_provisioning_log_analytics_agent_vms_on.defender_auto_provisioning_log_analytics_agent_vms_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_auto_provisioning_log_analytics_agent_vms_on.defender_auto_provisioning_log_analytics_agent_vms_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_auto_provisioning_log_analytics_agent_vms_on.defender_auto_provisioning_log_analytics_agent_vms_on import ( defender_auto_provisioning_log_analytics_agent_vms_on, @@ -47,15 +44,12 @@ class Test_defender_auto_provisioning_log_analytics_agent_vms_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_auto_provisioning_log_analytics_agent_vms_on.defender_auto_provisioning_log_analytics_agent_vms_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_auto_provisioning_log_analytics_agent_vms_on.defender_auto_provisioning_log_analytics_agent_vms_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_auto_provisioning_log_analytics_agent_vms_on.defender_auto_provisioning_log_analytics_agent_vms_on import ( defender_auto_provisioning_log_analytics_agent_vms_on, @@ -87,15 +81,12 @@ class Test_defender_auto_provisioning_log_analytics_agent_vms_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_auto_provisioning_log_analytics_agent_vms_on.defender_auto_provisioning_log_analytics_agent_vms_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_auto_provisioning_log_analytics_agent_vms_on.defender_auto_provisioning_log_analytics_agent_vms_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_auto_provisioning_log_analytics_agent_vms_on.defender_auto_provisioning_log_analytics_agent_vms_on import ( defender_auto_provisioning_log_analytics_agent_vms_on, @@ -133,15 +124,12 @@ class Test_defender_auto_provisioning_log_analytics_agent_vms_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_auto_provisioning_log_analytics_agent_vms_on.defender_auto_provisioning_log_analytics_agent_vms_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_auto_provisioning_log_analytics_agent_vms_on.defender_auto_provisioning_log_analytics_agent_vms_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_auto_provisioning_log_analytics_agent_vms_on.defender_auto_provisioning_log_analytics_agent_vms_on import ( defender_auto_provisioning_log_analytics_agent_vms_on, diff --git a/tests/providers/azure/services/defender/defender_auto_provisioning_vulnerabilty_assessments_machines_on/defender_auto_provisioning_vulnerabilty_assessments_machines_on_test.py b/tests/providers/azure/services/defender/defender_auto_provisioning_vulnerabilty_assessments_machines_on/defender_auto_provisioning_vulnerabilty_assessments_machines_on_test.py index bb78eb8fae..ba5a75a951 100644 --- a/tests/providers/azure/services/defender/defender_auto_provisioning_vulnerabilty_assessments_machines_on/defender_auto_provisioning_vulnerabilty_assessments_machines_on_test.py +++ b/tests/providers/azure/services/defender/defender_auto_provisioning_vulnerabilty_assessments_machines_on/defender_auto_provisioning_vulnerabilty_assessments_machines_on_test.py @@ -13,15 +13,12 @@ class Test_defender_auto_provisioning_vulnerabilty_assessments_machines_on: defender_client = mock.MagicMock defender_client.assessments = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_auto_provisioning_vulnerabilty_assessments_machines_on.defender_auto_provisioning_vulnerabilty_assessments_machines_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_auto_provisioning_vulnerabilty_assessments_machines_on.defender_auto_provisioning_vulnerabilty_assessments_machines_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_auto_provisioning_vulnerabilty_assessments_machines_on.defender_auto_provisioning_vulnerabilty_assessments_machines_on import ( defender_auto_provisioning_vulnerabilty_assessments_machines_on, @@ -44,15 +41,12 @@ class Test_defender_auto_provisioning_vulnerabilty_assessments_machines_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_auto_provisioning_vulnerabilty_assessments_machines_on.defender_auto_provisioning_vulnerabilty_assessments_machines_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_auto_provisioning_vulnerabilty_assessments_machines_on.defender_auto_provisioning_vulnerabilty_assessments_machines_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_auto_provisioning_vulnerabilty_assessments_machines_on.defender_auto_provisioning_vulnerabilty_assessments_machines_on import ( defender_auto_provisioning_vulnerabilty_assessments_machines_on, @@ -83,15 +77,12 @@ class Test_defender_auto_provisioning_vulnerabilty_assessments_machines_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_auto_provisioning_vulnerabilty_assessments_machines_on.defender_auto_provisioning_vulnerabilty_assessments_machines_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_auto_provisioning_vulnerabilty_assessments_machines_on.defender_auto_provisioning_vulnerabilty_assessments_machines_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_auto_provisioning_vulnerabilty_assessments_machines_on.defender_auto_provisioning_vulnerabilty_assessments_machines_on import ( defender_auto_provisioning_vulnerabilty_assessments_machines_on, diff --git a/tests/providers/azure/services/defender/defender_container_images_resolved_vulnerabilities/defender_container_images_resolved_vulnerabilities_test.py b/tests/providers/azure/services/defender/defender_container_images_resolved_vulnerabilities/defender_container_images_resolved_vulnerabilities_test.py index bfaabbce9a..073e6431e2 100644 --- a/tests/providers/azure/services/defender/defender_container_images_resolved_vulnerabilities/defender_container_images_resolved_vulnerabilities_test.py +++ b/tests/providers/azure/services/defender/defender_container_images_resolved_vulnerabilities/defender_container_images_resolved_vulnerabilities_test.py @@ -13,15 +13,12 @@ class Test_defender_container_images_resolved_vulnerabilities: defender_client = mock.MagicMock defender_client.assessments = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_container_images_resolved_vulnerabilities.defender_container_images_resolved_vulnerabilities.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_container_images_resolved_vulnerabilities.defender_container_images_resolved_vulnerabilities.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_container_images_resolved_vulnerabilities.defender_container_images_resolved_vulnerabilities import ( defender_container_images_resolved_vulnerabilities, @@ -35,15 +32,12 @@ class Test_defender_container_images_resolved_vulnerabilities: defender_client = mock.MagicMock defender_client.assessments = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_container_images_resolved_vulnerabilities.defender_container_images_resolved_vulnerabilities.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_container_images_resolved_vulnerabilities.defender_container_images_resolved_vulnerabilities.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_container_images_resolved_vulnerabilities.defender_container_images_resolved_vulnerabilities import ( defender_container_images_resolved_vulnerabilities, @@ -65,15 +59,12 @@ class Test_defender_container_images_resolved_vulnerabilities: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_container_images_resolved_vulnerabilities.defender_container_images_resolved_vulnerabilities.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_container_images_resolved_vulnerabilities.defender_container_images_resolved_vulnerabilities.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_container_images_resolved_vulnerabilities.defender_container_images_resolved_vulnerabilities import ( defender_container_images_resolved_vulnerabilities, @@ -95,15 +86,12 @@ class Test_defender_container_images_resolved_vulnerabilities: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_container_images_resolved_vulnerabilities.defender_container_images_resolved_vulnerabilities.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_container_images_resolved_vulnerabilities.defender_container_images_resolved_vulnerabilities.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_container_images_resolved_vulnerabilities.defender_container_images_resolved_vulnerabilities import ( defender_container_images_resolved_vulnerabilities, @@ -143,15 +131,12 @@ class Test_defender_container_images_resolved_vulnerabilities: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_container_images_resolved_vulnerabilities.defender_container_images_resolved_vulnerabilities.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_container_images_resolved_vulnerabilities.defender_container_images_resolved_vulnerabilities.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_container_images_resolved_vulnerabilities.defender_container_images_resolved_vulnerabilities import ( defender_container_images_resolved_vulnerabilities, @@ -191,15 +176,12 @@ class Test_defender_container_images_resolved_vulnerabilities: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_container_images_resolved_vulnerabilities.defender_container_images_resolved_vulnerabilities.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_container_images_resolved_vulnerabilities.defender_container_images_resolved_vulnerabilities.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_container_images_resolved_vulnerabilities.defender_container_images_resolved_vulnerabilities import ( defender_container_images_resolved_vulnerabilities, diff --git a/tests/providers/azure/services/defender/defender_container_images_scan_enabled/defender_container_images_scan_enabled_test.py b/tests/providers/azure/services/defender/defender_container_images_scan_enabled/defender_container_images_scan_enabled_test.py index 13bbd57838..964d704889 100644 --- a/tests/providers/azure/services/defender/defender_container_images_scan_enabled/defender_container_images_scan_enabled_test.py +++ b/tests/providers/azure/services/defender/defender_container_images_scan_enabled/defender_container_images_scan_enabled_test.py @@ -14,15 +14,12 @@ class Test_defender_container_images_scan_enabled: defender_client = mock.MagicMock defender_client.pricings = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_container_images_scan_enabled.defender_container_images_scan_enabled.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_container_images_scan_enabled.defender_container_images_scan_enabled.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_container_images_scan_enabled.defender_container_images_scan_enabled import ( defender_container_images_scan_enabled, @@ -36,15 +33,12 @@ class Test_defender_container_images_scan_enabled: defender_client = mock.MagicMock defender_client.pricings = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_container_images_scan_enabled.defender_container_images_scan_enabled.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_container_images_scan_enabled.defender_container_images_scan_enabled.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_container_images_scan_enabled.defender_container_images_scan_enabled import ( defender_container_images_scan_enabled, @@ -66,15 +60,12 @@ class Test_defender_container_images_scan_enabled: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_container_images_scan_enabled.defender_container_images_scan_enabled.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_container_images_scan_enabled.defender_container_images_scan_enabled.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_container_images_scan_enabled.defender_container_images_scan_enabled import ( defender_container_images_scan_enabled, @@ -97,15 +88,12 @@ class Test_defender_container_images_scan_enabled: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_container_images_scan_enabled.defender_container_images_scan_enabled.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_container_images_scan_enabled.defender_container_images_scan_enabled.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_container_images_scan_enabled.defender_container_images_scan_enabled import ( defender_container_images_scan_enabled, @@ -140,15 +128,12 @@ class Test_defender_container_images_scan_enabled: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_container_images_scan_enabled.defender_container_images_scan_enabled.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_container_images_scan_enabled.defender_container_images_scan_enabled.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_container_images_scan_enabled.defender_container_images_scan_enabled import ( defender_container_images_scan_enabled, @@ -183,15 +168,12 @@ class Test_defender_container_images_scan_enabled: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_container_images_scan_enabled.defender_container_images_scan_enabled.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_container_images_scan_enabled.defender_container_images_scan_enabled.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_container_images_scan_enabled.defender_container_images_scan_enabled import ( defender_container_images_scan_enabled, diff --git a/tests/providers/azure/services/defender/defender_ensure_defender_for_app_services_is_on/defender_ensure_defender_for_app_services_is_on_test.py b/tests/providers/azure/services/defender/defender_ensure_defender_for_app_services_is_on/defender_ensure_defender_for_app_services_is_on_test.py index e3c83045f7..aff46c4ee9 100644 --- a/tests/providers/azure/services/defender/defender_ensure_defender_for_app_services_is_on/defender_ensure_defender_for_app_services_is_on_test.py +++ b/tests/providers/azure/services/defender/defender_ensure_defender_for_app_services_is_on/defender_ensure_defender_for_app_services_is_on_test.py @@ -13,15 +13,12 @@ class Test_defender_ensure_defender_for_app_services_is_on: defender_client = mock.MagicMock defender_client.pricings = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_app_services_is_on.defender_ensure_defender_for_app_services_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_app_services_is_on.defender_ensure_defender_for_app_services_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_app_services_is_on.defender_ensure_defender_for_app_services_is_on import ( defender_ensure_defender_for_app_services_is_on, @@ -44,15 +41,12 @@ class Test_defender_ensure_defender_for_app_services_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_app_services_is_on.defender_ensure_defender_for_app_services_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_app_services_is_on.defender_ensure_defender_for_app_services_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_app_services_is_on.defender_ensure_defender_for_app_services_is_on import ( defender_ensure_defender_for_app_services_is_on, @@ -83,15 +77,12 @@ class Test_defender_ensure_defender_for_app_services_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_app_services_is_on.defender_ensure_defender_for_app_services_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_app_services_is_on.defender_ensure_defender_for_app_services_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_app_services_is_on.defender_ensure_defender_for_app_services_is_on import ( defender_ensure_defender_for_app_services_is_on, diff --git a/tests/providers/azure/services/defender/defender_ensure_defender_for_arm_is_on/defender_ensure_defender_for_arm_is_on_test.py b/tests/providers/azure/services/defender/defender_ensure_defender_for_arm_is_on/defender_ensure_defender_for_arm_is_on_test.py index 4341ce9e08..d3d3aecccd 100644 --- a/tests/providers/azure/services/defender/defender_ensure_defender_for_arm_is_on/defender_ensure_defender_for_arm_is_on_test.py +++ b/tests/providers/azure/services/defender/defender_ensure_defender_for_arm_is_on/defender_ensure_defender_for_arm_is_on_test.py @@ -13,15 +13,12 @@ class Test_defender_ensure_defender_for_arm_is_on: defender_client = mock.MagicMock defender_client.pricings = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_arm_is_on.defender_ensure_defender_for_arm_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_arm_is_on.defender_ensure_defender_for_arm_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_arm_is_on.defender_ensure_defender_for_arm_is_on import ( defender_ensure_defender_for_arm_is_on, @@ -44,15 +41,12 @@ class Test_defender_ensure_defender_for_arm_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_arm_is_on.defender_ensure_defender_for_arm_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_arm_is_on.defender_ensure_defender_for_arm_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_arm_is_on.defender_ensure_defender_for_arm_is_on import ( defender_ensure_defender_for_arm_is_on, @@ -83,15 +77,12 @@ class Test_defender_ensure_defender_for_arm_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_arm_is_on.defender_ensure_defender_for_arm_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_arm_is_on.defender_ensure_defender_for_arm_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_arm_is_on.defender_ensure_defender_for_arm_is_on import ( defender_ensure_defender_for_arm_is_on, diff --git a/tests/providers/azure/services/defender/defender_ensure_defender_for_azure_sql_databases_is_on/defender_ensure_defender_for_azure_sql_databases_is_on_test.py b/tests/providers/azure/services/defender/defender_ensure_defender_for_azure_sql_databases_is_on/defender_ensure_defender_for_azure_sql_databases_is_on_test.py index 15269e71f1..7e5a4f87ae 100644 --- a/tests/providers/azure/services/defender/defender_ensure_defender_for_azure_sql_databases_is_on/defender_ensure_defender_for_azure_sql_databases_is_on_test.py +++ b/tests/providers/azure/services/defender/defender_ensure_defender_for_azure_sql_databases_is_on/defender_ensure_defender_for_azure_sql_databases_is_on_test.py @@ -13,15 +13,12 @@ class Test_defender_ensure_defender_for_azure_sql_databases_is_on: defender_client = mock.MagicMock defender_client.pricings = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_azure_sql_databases_is_on.defender_ensure_defender_for_azure_sql_databases_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_azure_sql_databases_is_on.defender_ensure_defender_for_azure_sql_databases_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_azure_sql_databases_is_on.defender_ensure_defender_for_azure_sql_databases_is_on import ( defender_ensure_defender_for_azure_sql_databases_is_on, @@ -44,15 +41,12 @@ class Test_defender_ensure_defender_for_azure_sql_databases_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_azure_sql_databases_is_on.defender_ensure_defender_for_azure_sql_databases_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_azure_sql_databases_is_on.defender_ensure_defender_for_azure_sql_databases_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_azure_sql_databases_is_on.defender_ensure_defender_for_azure_sql_databases_is_on import ( defender_ensure_defender_for_azure_sql_databases_is_on, @@ -83,15 +77,12 @@ class Test_defender_ensure_defender_for_azure_sql_databases_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_azure_sql_databases_is_on.defender_ensure_defender_for_azure_sql_databases_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_azure_sql_databases_is_on.defender_ensure_defender_for_azure_sql_databases_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_azure_sql_databases_is_on.defender_ensure_defender_for_azure_sql_databases_is_on import ( defender_ensure_defender_for_azure_sql_databases_is_on, diff --git a/tests/providers/azure/services/defender/defender_ensure_defender_for_containers_is_on/defender_ensure_defender_for_containers_is_on_test.py b/tests/providers/azure/services/defender/defender_ensure_defender_for_containers_is_on/defender_ensure_defender_for_containers_is_on_test.py index d40fbe1b1c..8da5e30299 100644 --- a/tests/providers/azure/services/defender/defender_ensure_defender_for_containers_is_on/defender_ensure_defender_for_containers_is_on_test.py +++ b/tests/providers/azure/services/defender/defender_ensure_defender_for_containers_is_on/defender_ensure_defender_for_containers_is_on_test.py @@ -13,15 +13,12 @@ class Test_defender_ensure_defender_for_containers_is_on: defender_client = mock.MagicMock defender_client.pricings = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_containers_is_on.defender_ensure_defender_for_containers_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_containers_is_on.defender_ensure_defender_for_containers_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_containers_is_on.defender_ensure_defender_for_containers_is_on import ( defender_ensure_defender_for_containers_is_on, @@ -44,15 +41,12 @@ class Test_defender_ensure_defender_for_containers_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_containers_is_on.defender_ensure_defender_for_containers_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_containers_is_on.defender_ensure_defender_for_containers_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_containers_is_on.defender_ensure_defender_for_containers_is_on import ( defender_ensure_defender_for_containers_is_on, @@ -83,15 +77,12 @@ class Test_defender_ensure_defender_for_containers_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_containers_is_on.defender_ensure_defender_for_containers_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_containers_is_on.defender_ensure_defender_for_containers_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_containers_is_on.defender_ensure_defender_for_containers_is_on import ( defender_ensure_defender_for_containers_is_on, diff --git a/tests/providers/azure/services/defender/defender_ensure_defender_for_cosmosdb_is_on/defender_ensure_defender_for_cosmosdb_is_on_test.py b/tests/providers/azure/services/defender/defender_ensure_defender_for_cosmosdb_is_on/defender_ensure_defender_for_cosmosdb_is_on_test.py index 5f7906e1c2..cc16ad59c7 100644 --- a/tests/providers/azure/services/defender/defender_ensure_defender_for_cosmosdb_is_on/defender_ensure_defender_for_cosmosdb_is_on_test.py +++ b/tests/providers/azure/services/defender/defender_ensure_defender_for_cosmosdb_is_on/defender_ensure_defender_for_cosmosdb_is_on_test.py @@ -13,15 +13,12 @@ class Test_defender_ensure_defender_for_cosmosdb_is_on: defender_client = mock.MagicMock defender_client.pricings = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_cosmosdb_is_on.defender_ensure_defender_for_cosmosdb_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_cosmosdb_is_on.defender_ensure_defender_for_cosmosdb_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_cosmosdb_is_on.defender_ensure_defender_for_cosmosdb_is_on import ( defender_ensure_defender_for_cosmosdb_is_on, @@ -44,15 +41,12 @@ class Test_defender_ensure_defender_for_cosmosdb_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_cosmosdb_is_on.defender_ensure_defender_for_cosmosdb_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_cosmosdb_is_on.defender_ensure_defender_for_cosmosdb_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_cosmosdb_is_on.defender_ensure_defender_for_cosmosdb_is_on import ( defender_ensure_defender_for_cosmosdb_is_on, @@ -83,15 +77,12 @@ class Test_defender_ensure_defender_for_cosmosdb_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_cosmosdb_is_on.defender_ensure_defender_for_cosmosdb_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_cosmosdb_is_on.defender_ensure_defender_for_cosmosdb_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_cosmosdb_is_on.defender_ensure_defender_for_cosmosdb_is_on import ( defender_ensure_defender_for_cosmosdb_is_on, diff --git a/tests/providers/azure/services/defender/defender_ensure_defender_for_databases_is_on/defender_ensure_defender_for_databases_is_on_test.py b/tests/providers/azure/services/defender/defender_ensure_defender_for_databases_is_on/defender_ensure_defender_for_databases_is_on_test.py index 755151da20..bb3cd8d7fd 100644 --- a/tests/providers/azure/services/defender/defender_ensure_defender_for_databases_is_on/defender_ensure_defender_for_databases_is_on_test.py +++ b/tests/providers/azure/services/defender/defender_ensure_defender_for_databases_is_on/defender_ensure_defender_for_databases_is_on_test.py @@ -13,15 +13,12 @@ class Test_defender_ensure_defender_for_databases_is_on: defender_client = mock.MagicMock defender_client.pricings = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on import ( defender_ensure_defender_for_databases_is_on, @@ -44,15 +41,12 @@ class Test_defender_ensure_defender_for_databases_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on import ( defender_ensure_defender_for_databases_is_on, @@ -75,15 +69,12 @@ class Test_defender_ensure_defender_for_databases_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on import ( defender_ensure_defender_for_databases_is_on, @@ -106,15 +97,12 @@ class Test_defender_ensure_defender_for_databases_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on import ( defender_ensure_defender_for_databases_is_on, @@ -137,15 +125,12 @@ class Test_defender_ensure_defender_for_databases_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on import ( defender_ensure_defender_for_databases_is_on, @@ -183,15 +168,12 @@ class Test_defender_ensure_defender_for_databases_is_on: }, } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on import ( defender_ensure_defender_for_databases_is_on, @@ -237,15 +219,12 @@ class Test_defender_ensure_defender_for_databases_is_on: }, } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_databases_is_on.defender_ensure_defender_for_databases_is_on import ( defender_ensure_defender_for_databases_is_on, diff --git a/tests/providers/azure/services/defender/defender_ensure_defender_for_dns_is_on/defender_ensure_defender_for_dns_is_on_test.py b/tests/providers/azure/services/defender/defender_ensure_defender_for_dns_is_on/defender_ensure_defender_for_dns_is_on_test.py index 64d81487f0..f489ace5f3 100644 --- a/tests/providers/azure/services/defender/defender_ensure_defender_for_dns_is_on/defender_ensure_defender_for_dns_is_on_test.py +++ b/tests/providers/azure/services/defender/defender_ensure_defender_for_dns_is_on/defender_ensure_defender_for_dns_is_on_test.py @@ -13,15 +13,12 @@ class Test_defender_ensure_defender_for_dns_is_on: defender_client = mock.MagicMock defender_client.pricings = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_dns_is_on.defender_ensure_defender_for_dns_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_dns_is_on.defender_ensure_defender_for_dns_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_dns_is_on.defender_ensure_defender_for_dns_is_on import ( defender_ensure_defender_for_dns_is_on, @@ -44,15 +41,12 @@ class Test_defender_ensure_defender_for_dns_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_dns_is_on.defender_ensure_defender_for_dns_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_dns_is_on.defender_ensure_defender_for_dns_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_dns_is_on.defender_ensure_defender_for_dns_is_on import ( defender_ensure_defender_for_dns_is_on, @@ -83,15 +77,12 @@ class Test_defender_ensure_defender_for_dns_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_dns_is_on.defender_ensure_defender_for_dns_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_dns_is_on.defender_ensure_defender_for_dns_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_dns_is_on.defender_ensure_defender_for_dns_is_on import ( defender_ensure_defender_for_dns_is_on, diff --git a/tests/providers/azure/services/defender/defender_ensure_defender_for_keyvault_is_on/defender_ensure_defender_for_keyvault_is_on_test.py b/tests/providers/azure/services/defender/defender_ensure_defender_for_keyvault_is_on/defender_ensure_defender_for_keyvault_is_on_test.py index 4c4315e75c..9616c696df 100644 --- a/tests/providers/azure/services/defender/defender_ensure_defender_for_keyvault_is_on/defender_ensure_defender_for_keyvault_is_on_test.py +++ b/tests/providers/azure/services/defender/defender_ensure_defender_for_keyvault_is_on/defender_ensure_defender_for_keyvault_is_on_test.py @@ -13,15 +13,12 @@ class Test_defender_ensure_defender_for_keyvault_is_on: defender_client = mock.MagicMock defender_client.pricings = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_keyvault_is_on.defender_ensure_defender_for_keyvault_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_keyvault_is_on.defender_ensure_defender_for_keyvault_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_keyvault_is_on.defender_ensure_defender_for_keyvault_is_on import ( defender_ensure_defender_for_keyvault_is_on, @@ -44,15 +41,12 @@ class Test_defender_ensure_defender_for_keyvault_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_keyvault_is_on.defender_ensure_defender_for_keyvault_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_keyvault_is_on.defender_ensure_defender_for_keyvault_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_keyvault_is_on.defender_ensure_defender_for_keyvault_is_on import ( defender_ensure_defender_for_keyvault_is_on, @@ -83,15 +77,12 @@ class Test_defender_ensure_defender_for_keyvault_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_keyvault_is_on.defender_ensure_defender_for_keyvault_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_keyvault_is_on.defender_ensure_defender_for_keyvault_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_keyvault_is_on.defender_ensure_defender_for_keyvault_is_on import ( defender_ensure_defender_for_keyvault_is_on, diff --git a/tests/providers/azure/services/defender/defender_ensure_defender_for_os_relational_databases_is_on/defender_ensure_defender_for_os_relational_databases_is_on_test.py b/tests/providers/azure/services/defender/defender_ensure_defender_for_os_relational_databases_is_on/defender_ensure_defender_for_os_relational_databases_is_on_test.py index 48ddb8b72d..e7ef603a89 100644 --- a/tests/providers/azure/services/defender/defender_ensure_defender_for_os_relational_databases_is_on/defender_ensure_defender_for_os_relational_databases_is_on_test.py +++ b/tests/providers/azure/services/defender/defender_ensure_defender_for_os_relational_databases_is_on/defender_ensure_defender_for_os_relational_databases_is_on_test.py @@ -13,15 +13,12 @@ class Test_defender_ensure_defender_for_os_relational_databases_is_on: defender_client = mock.MagicMock defender_client.pricings = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_os_relational_databases_is_on.defender_ensure_defender_for_os_relational_databases_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_os_relational_databases_is_on.defender_ensure_defender_for_os_relational_databases_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_os_relational_databases_is_on.defender_ensure_defender_for_os_relational_databases_is_on import ( defender_ensure_defender_for_os_relational_databases_is_on, @@ -44,15 +41,12 @@ class Test_defender_ensure_defender_for_os_relational_databases_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_os_relational_databases_is_on.defender_ensure_defender_for_os_relational_databases_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_os_relational_databases_is_on.defender_ensure_defender_for_os_relational_databases_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_os_relational_databases_is_on.defender_ensure_defender_for_os_relational_databases_is_on import ( defender_ensure_defender_for_os_relational_databases_is_on, @@ -86,15 +80,12 @@ class Test_defender_ensure_defender_for_os_relational_databases_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_os_relational_databases_is_on.defender_ensure_defender_for_os_relational_databases_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_os_relational_databases_is_on.defender_ensure_defender_for_os_relational_databases_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_os_relational_databases_is_on.defender_ensure_defender_for_os_relational_databases_is_on import ( defender_ensure_defender_for_os_relational_databases_is_on, diff --git a/tests/providers/azure/services/defender/defender_ensure_defender_for_server_is_on/defender_ensure_defender_for_server_is_on_test.py b/tests/providers/azure/services/defender/defender_ensure_defender_for_server_is_on/defender_ensure_defender_for_server_is_on_test.py index 17405d9f6d..3c183c1ee3 100644 --- a/tests/providers/azure/services/defender/defender_ensure_defender_for_server_is_on/defender_ensure_defender_for_server_is_on_test.py +++ b/tests/providers/azure/services/defender/defender_ensure_defender_for_server_is_on/defender_ensure_defender_for_server_is_on_test.py @@ -13,15 +13,12 @@ class Test_defender_ensure_defender_for_server_is_on: defender_client = mock.MagicMock defender_client.pricings = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_server_is_on.defender_ensure_defender_for_server_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_server_is_on.defender_ensure_defender_for_server_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_server_is_on.defender_ensure_defender_for_server_is_on import ( defender_ensure_defender_for_server_is_on, @@ -44,15 +41,12 @@ class Test_defender_ensure_defender_for_server_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_server_is_on.defender_ensure_defender_for_server_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_server_is_on.defender_ensure_defender_for_server_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_server_is_on.defender_ensure_defender_for_server_is_on import ( defender_ensure_defender_for_server_is_on, @@ -83,15 +77,12 @@ class Test_defender_ensure_defender_for_server_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_server_is_on.defender_ensure_defender_for_server_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_server_is_on.defender_ensure_defender_for_server_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_server_is_on.defender_ensure_defender_for_server_is_on import ( defender_ensure_defender_for_server_is_on, diff --git a/tests/providers/azure/services/defender/defender_ensure_defender_for_sql_servers_is_on/defender_ensure_defender_for_sql_servers_is_on_test.py b/tests/providers/azure/services/defender/defender_ensure_defender_for_sql_servers_is_on/defender_ensure_defender_for_sql_servers_is_on_test.py index 72e1f82257..5ce5aad101 100644 --- a/tests/providers/azure/services/defender/defender_ensure_defender_for_sql_servers_is_on/defender_ensure_defender_for_sql_servers_is_on_test.py +++ b/tests/providers/azure/services/defender/defender_ensure_defender_for_sql_servers_is_on/defender_ensure_defender_for_sql_servers_is_on_test.py @@ -13,15 +13,12 @@ class Test_defender_ensure_defender_for_sql_servers_is_on: defender_client = mock.MagicMock defender_client.pricings = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_sql_servers_is_on.defender_ensure_defender_for_sql_servers_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_sql_servers_is_on.defender_ensure_defender_for_sql_servers_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_sql_servers_is_on.defender_ensure_defender_for_sql_servers_is_on import ( defender_ensure_defender_for_sql_servers_is_on, @@ -44,15 +41,12 @@ class Test_defender_ensure_defender_for_sql_servers_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_sql_servers_is_on.defender_ensure_defender_for_sql_servers_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_sql_servers_is_on.defender_ensure_defender_for_sql_servers_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_sql_servers_is_on.defender_ensure_defender_for_sql_servers_is_on import ( defender_ensure_defender_for_sql_servers_is_on, @@ -83,15 +77,12 @@ class Test_defender_ensure_defender_for_sql_servers_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_sql_servers_is_on.defender_ensure_defender_for_sql_servers_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_sql_servers_is_on.defender_ensure_defender_for_sql_servers_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_sql_servers_is_on.defender_ensure_defender_for_sql_servers_is_on import ( defender_ensure_defender_for_sql_servers_is_on, diff --git a/tests/providers/azure/services/defender/defender_ensure_defender_for_storage_is_on/defender_ensure_defender_for_storage_is_on_test.py b/tests/providers/azure/services/defender/defender_ensure_defender_for_storage_is_on/defender_ensure_defender_for_storage_is_on_test.py index 53db01231c..0aabf069d0 100644 --- a/tests/providers/azure/services/defender/defender_ensure_defender_for_storage_is_on/defender_ensure_defender_for_storage_is_on_test.py +++ b/tests/providers/azure/services/defender/defender_ensure_defender_for_storage_is_on/defender_ensure_defender_for_storage_is_on_test.py @@ -13,15 +13,12 @@ class Test_defender_ensure_defender_for_storage_is_on: defender_client = mock.MagicMock defender_client.pricings = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_storage_is_on.defender_ensure_defender_for_storage_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_storage_is_on.defender_ensure_defender_for_storage_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_storage_is_on.defender_ensure_defender_for_storage_is_on import ( defender_ensure_defender_for_storage_is_on, @@ -44,15 +41,12 @@ class Test_defender_ensure_defender_for_storage_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_storage_is_on.defender_ensure_defender_for_storage_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_storage_is_on.defender_ensure_defender_for_storage_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_storage_is_on.defender_ensure_defender_for_storage_is_on import ( defender_ensure_defender_for_storage_is_on, @@ -83,15 +77,12 @@ class Test_defender_ensure_defender_for_storage_is_on: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_defender_for_storage_is_on.defender_ensure_defender_for_storage_is_on.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_defender_for_storage_is_on.defender_ensure_defender_for_storage_is_on.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_defender_for_storage_is_on.defender_ensure_defender_for_storage_is_on import ( defender_ensure_defender_for_storage_is_on, diff --git a/tests/providers/azure/services/defender/defender_ensure_mcas_is_enabled/defender_ensure_mcas_is_enabled_test.py b/tests/providers/azure/services/defender/defender_ensure_mcas_is_enabled/defender_ensure_mcas_is_enabled_test.py index 26f7bb880a..aba618d0eb 100644 --- a/tests/providers/azure/services/defender/defender_ensure_mcas_is_enabled/defender_ensure_mcas_is_enabled_test.py +++ b/tests/providers/azure/services/defender/defender_ensure_mcas_is_enabled/defender_ensure_mcas_is_enabled_test.py @@ -13,15 +13,12 @@ class Test_defender_ensure_mcas_is_enabled: defender_client = mock.MagicMock defender_client.settings = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_mcas_is_enabled.defender_ensure_mcas_is_enabled.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_mcas_is_enabled.defender_ensure_mcas_is_enabled.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_mcas_is_enabled.defender_ensure_mcas_is_enabled import ( defender_ensure_mcas_is_enabled, @@ -45,15 +42,12 @@ class Test_defender_ensure_mcas_is_enabled: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_mcas_is_enabled.defender_ensure_mcas_is_enabled.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_mcas_is_enabled.defender_ensure_mcas_is_enabled.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_mcas_is_enabled.defender_ensure_mcas_is_enabled import ( defender_ensure_mcas_is_enabled, @@ -85,15 +79,12 @@ class Test_defender_ensure_mcas_is_enabled: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_mcas_is_enabled.defender_ensure_mcas_is_enabled.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_mcas_is_enabled.defender_ensure_mcas_is_enabled.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_mcas_is_enabled.defender_ensure_mcas_is_enabled import ( defender_ensure_mcas_is_enabled, @@ -115,15 +106,12 @@ class Test_defender_ensure_mcas_is_enabled: defender_client = mock.MagicMock defender_client.settings = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_mcas_is_enabled.defender_ensure_mcas_is_enabled.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_mcas_is_enabled.defender_ensure_mcas_is_enabled.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_mcas_is_enabled.defender_ensure_mcas_is_enabled import ( defender_ensure_mcas_is_enabled, diff --git a/tests/providers/azure/services/defender/defender_ensure_notify_emails_to_owners/defender_ensure_notify_emails_to_owners_test.py b/tests/providers/azure/services/defender/defender_ensure_notify_emails_to_owners/defender_ensure_notify_emails_to_owners_test.py index 5eff665838..c33dab2788 100644 --- a/tests/providers/azure/services/defender/defender_ensure_notify_emails_to_owners/defender_ensure_notify_emails_to_owners_test.py +++ b/tests/providers/azure/services/defender/defender_ensure_notify_emails_to_owners/defender_ensure_notify_emails_to_owners_test.py @@ -13,15 +13,12 @@ class Test_defender_ensure_notify_emails_to_owners: defender_client = mock.MagicMock defender_client.security_contacts = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_notify_emails_to_owners.defender_ensure_notify_emails_to_owners.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_notify_emails_to_owners.defender_ensure_notify_emails_to_owners.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_notify_emails_to_owners.defender_ensure_notify_emails_to_owners import ( defender_ensure_notify_emails_to_owners, @@ -49,15 +46,12 @@ class Test_defender_ensure_notify_emails_to_owners: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_notify_emails_to_owners.defender_ensure_notify_emails_to_owners.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_notify_emails_to_owners.defender_ensure_notify_emails_to_owners.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_notify_emails_to_owners.defender_ensure_notify_emails_to_owners import ( defender_ensure_notify_emails_to_owners, @@ -93,15 +87,12 @@ class Test_defender_ensure_notify_emails_to_owners: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_notify_emails_to_owners.defender_ensure_notify_emails_to_owners.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_notify_emails_to_owners.defender_ensure_notify_emails_to_owners.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_notify_emails_to_owners.defender_ensure_notify_emails_to_owners import ( defender_ensure_notify_emails_to_owners, @@ -137,15 +128,12 @@ class Test_defender_ensure_notify_emails_to_owners: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_notify_emails_to_owners.defender_ensure_notify_emails_to_owners.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_notify_emails_to_owners.defender_ensure_notify_emails_to_owners.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_notify_emails_to_owners.defender_ensure_notify_emails_to_owners import ( defender_ensure_notify_emails_to_owners, @@ -180,15 +168,12 @@ class Test_defender_ensure_notify_emails_to_owners: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_notify_emails_to_owners.defender_ensure_notify_emails_to_owners.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_notify_emails_to_owners.defender_ensure_notify_emails_to_owners.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_notify_emails_to_owners.defender_ensure_notify_emails_to_owners import ( defender_ensure_notify_emails_to_owners, diff --git a/tests/providers/azure/services/defender/defender_ensure_system_updates_are_applied/defender_ensure_system_updates_are_applied_test.py b/tests/providers/azure/services/defender/defender_ensure_system_updates_are_applied/defender_ensure_system_updates_are_applied_test.py index 4a98f0bba6..8a9cd7a651 100644 --- a/tests/providers/azure/services/defender/defender_ensure_system_updates_are_applied/defender_ensure_system_updates_are_applied_test.py +++ b/tests/providers/azure/services/defender/defender_ensure_system_updates_are_applied/defender_ensure_system_updates_are_applied_test.py @@ -13,15 +13,12 @@ class Test_defender_ensure_system_updates_are_applied: defender_client = mock.MagicMock defender_client.assessments = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_system_updates_are_applied.defender_ensure_system_updates_are_applied.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_system_updates_are_applied.defender_ensure_system_updates_are_applied.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_system_updates_are_applied.defender_ensure_system_updates_are_applied import ( defender_ensure_system_updates_are_applied, @@ -54,15 +51,12 @@ class Test_defender_ensure_system_updates_are_applied: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_system_updates_are_applied.defender_ensure_system_updates_are_applied.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_system_updates_are_applied.defender_ensure_system_updates_are_applied.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_system_updates_are_applied.defender_ensure_system_updates_are_applied import ( defender_ensure_system_updates_are_applied, @@ -105,15 +99,12 @@ class Test_defender_ensure_system_updates_are_applied: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_system_updates_are_applied.defender_ensure_system_updates_are_applied.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_system_updates_are_applied.defender_ensure_system_updates_are_applied.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_system_updates_are_applied.defender_ensure_system_updates_are_applied import ( defender_ensure_system_updates_are_applied, @@ -154,15 +145,12 @@ class Test_defender_ensure_system_updates_are_applied: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_system_updates_are_applied.defender_ensure_system_updates_are_applied.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_system_updates_are_applied.defender_ensure_system_updates_are_applied.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_system_updates_are_applied.defender_ensure_system_updates_are_applied import ( defender_ensure_system_updates_are_applied, @@ -205,15 +193,12 @@ class Test_defender_ensure_system_updates_are_applied: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_system_updates_are_applied.defender_ensure_system_updates_are_applied.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_system_updates_are_applied.defender_ensure_system_updates_are_applied.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_system_updates_are_applied.defender_ensure_system_updates_are_applied import ( defender_ensure_system_updates_are_applied, diff --git a/tests/providers/azure/services/defender/defender_ensure_wdatp_is_enabled/defender_ensure_wdatp_is_enabled_test.py b/tests/providers/azure/services/defender/defender_ensure_wdatp_is_enabled/defender_ensure_wdatp_is_enabled_test.py index c7d7a120e9..d548095d8e 100644 --- a/tests/providers/azure/services/defender/defender_ensure_wdatp_is_enabled/defender_ensure_wdatp_is_enabled_test.py +++ b/tests/providers/azure/services/defender/defender_ensure_wdatp_is_enabled/defender_ensure_wdatp_is_enabled_test.py @@ -13,15 +13,12 @@ class Test_defender_ensure_wdatp_is_enabled: defender_client = mock.MagicMock defender_client.settings = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_wdatp_is_enabled.defender_ensure_wdatp_is_enabled.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_wdatp_is_enabled.defender_ensure_wdatp_is_enabled.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_wdatp_is_enabled.defender_ensure_wdatp_is_enabled import ( defender_ensure_wdatp_is_enabled, @@ -45,15 +42,12 @@ class Test_defender_ensure_wdatp_is_enabled: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_wdatp_is_enabled.defender_ensure_wdatp_is_enabled.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_wdatp_is_enabled.defender_ensure_wdatp_is_enabled.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_wdatp_is_enabled.defender_ensure_wdatp_is_enabled import ( defender_ensure_wdatp_is_enabled, @@ -85,15 +79,12 @@ class Test_defender_ensure_wdatp_is_enabled: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_wdatp_is_enabled.defender_ensure_wdatp_is_enabled.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_wdatp_is_enabled.defender_ensure_wdatp_is_enabled.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_wdatp_is_enabled.defender_ensure_wdatp_is_enabled import ( defender_ensure_wdatp_is_enabled, @@ -115,15 +106,12 @@ class Test_defender_ensure_wdatp_is_enabled: defender_client = mock.MagicMock defender_client.settings = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.defender.defender_ensure_wdatp_is_enabled.defender_ensure_wdatp_is_enabled.defender_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.defender.defender_ensure_wdatp_is_enabled.defender_ensure_wdatp_is_enabled.defender_client", + new=defender_client, ): from prowler.providers.azure.services.defender.defender_ensure_wdatp_is_enabled.defender_ensure_wdatp_is_enabled import ( defender_ensure_wdatp_is_enabled, diff --git a/tests/providers/azure/services/entra/entra_conditional_access_policy_require_mfa_for_management_api/entra_conditional_access_policy_require_mfa_for_management_api_test.py b/tests/providers/azure/services/entra/entra_conditional_access_policy_require_mfa_for_management_api/entra_conditional_access_policy_require_mfa_for_management_api_test.py index 3c880886ee..1a4fafc2b7 100644 --- a/tests/providers/azure/services/entra/entra_conditional_access_policy_require_mfa_for_management_api/entra_conditional_access_policy_require_mfa_for_management_api_test.py +++ b/tests/providers/azure/services/entra/entra_conditional_access_policy_require_mfa_for_management_api/entra_conditional_access_policy_require_mfa_for_management_api_test.py @@ -8,15 +8,12 @@ class Test_entra_conditional_access_policy_require_mfa_for_management_api: def test_entra_no_subscriptions(self): entra_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api.entra_client", + new=entra_client, ): from prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api import ( entra_conditional_access_policy_require_mfa_for_management_api, @@ -31,15 +28,12 @@ class Test_entra_conditional_access_policy_require_mfa_for_management_api: def test_entra_tenant_no_policies(self): entra_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api.entra_client", + new=entra_client, ): from prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api import ( entra_conditional_access_policy_require_mfa_for_management_api, @@ -63,15 +57,12 @@ class Test_entra_conditional_access_policy_require_mfa_for_management_api: entra_client = mock.MagicMock policy_id = str(uuid4()) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api.entra_client", + new=entra_client, ): from prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api import ( entra_conditional_access_policy_require_mfa_for_management_api, @@ -107,15 +98,12 @@ class Test_entra_conditional_access_policy_require_mfa_for_management_api: entra_client = mock.MagicMock policy_id = str(uuid4()) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api.entra_client", + new=entra_client, ): from prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api import ( entra_conditional_access_policy_require_mfa_for_management_api, @@ -151,15 +139,12 @@ class Test_entra_conditional_access_policy_require_mfa_for_management_api: entra_client = mock.MagicMock policy_id = str(uuid4()) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api.entra_client", + new=entra_client, ): from prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api import ( entra_conditional_access_policy_require_mfa_for_management_api, @@ -195,15 +180,12 @@ class Test_entra_conditional_access_policy_require_mfa_for_management_api: entra_client = mock.MagicMock policy_id = str(uuid4()) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api.entra_client", + new=entra_client, ): from prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api import ( entra_conditional_access_policy_require_mfa_for_management_api, @@ -239,15 +221,12 @@ class Test_entra_conditional_access_policy_require_mfa_for_management_api: entra_client = mock.MagicMock policy_id = str(uuid4()) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api.entra_client", + new=entra_client, ): from prowler.providers.azure.services.entra.entra_conditional_access_policy_require_mfa_for_management_api.entra_conditional_access_policy_require_mfa_for_management_api import ( entra_conditional_access_policy_require_mfa_for_management_api, diff --git a/tests/providers/azure/services/entra/entra_security_defaults_enabled/entra_security_defaults_enabled_test.py b/tests/providers/azure/services/entra/entra_security_defaults_enabled/entra_security_defaults_enabled_test.py index 562c008c52..33f3f5a839 100644 --- a/tests/providers/azure/services/entra/entra_security_defaults_enabled/entra_security_defaults_enabled_test.py +++ b/tests/providers/azure/services/entra/entra_security_defaults_enabled/entra_security_defaults_enabled_test.py @@ -8,15 +8,12 @@ class Test_entra_security_defaults_enabled: def test_entra_no_tenants(self): entra_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.entra.entra_security_defaults_enabled.entra_security_defaults_enabled.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.entra.entra_security_defaults_enabled.entra_security_defaults_enabled.entra_client", + new=entra_client, ): from prowler.providers.azure.services.entra.entra_security_defaults_enabled.entra_security_defaults_enabled import ( entra_security_defaults_enabled, @@ -31,15 +28,12 @@ class Test_entra_security_defaults_enabled: def test_entra_tenant_empty(self): entra_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.entra.entra_security_defaults_enabled.entra_security_defaults_enabled.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.entra.entra_security_defaults_enabled.entra_security_defaults_enabled.entra_client", + new=entra_client, ): from prowler.providers.azure.services.entra.entra_security_defaults_enabled.entra_security_defaults_enabled import ( entra_security_defaults_enabled, @@ -59,15 +53,12 @@ class Test_entra_security_defaults_enabled: def test_entra_security_default_enabled(self): entra_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.entra.entra_security_defaults_enabled.entra_security_defaults_enabled.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.entra.entra_security_defaults_enabled.entra_security_defaults_enabled.entra_client", + new=entra_client, ): from prowler.providers.azure.services.entra.entra_security_defaults_enabled.entra_security_defaults_enabled import ( entra_security_defaults_enabled, @@ -94,15 +85,12 @@ class Test_entra_security_defaults_enabled: def test_entra_security_default_disabled(self): entra_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.entra.entra_security_defaults_enabled.entra_security_defaults_enabled.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.entra.entra_security_defaults_enabled.entra_security_defaults_enabled.entra_client", + new=entra_client, ): from prowler.providers.azure.services.entra.entra_security_defaults_enabled.entra_security_defaults_enabled import ( entra_security_defaults_enabled, diff --git a/tests/providers/azure/services/entra/entra_users_cannot_create_microsoft_365_groups/entra_users_cannot_create_microsoft_365_groups_test.py b/tests/providers/azure/services/entra/entra_users_cannot_create_microsoft_365_groups/entra_users_cannot_create_microsoft_365_groups_test.py index 9e28397880..86c03a7a23 100644 --- a/tests/providers/azure/services/entra/entra_users_cannot_create_microsoft_365_groups/entra_users_cannot_create_microsoft_365_groups_test.py +++ b/tests/providers/azure/services/entra/entra_users_cannot_create_microsoft_365_groups/entra_users_cannot_create_microsoft_365_groups_test.py @@ -8,15 +8,12 @@ class Test_entra_users_cannot_create_microsoft_365_groups: def test_entra_no_tenant(self): entra_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.entra.entra_users_cannot_create_microsoft_365_groups.entra_users_cannot_create_microsoft_365_groups.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.entra.entra_users_cannot_create_microsoft_365_groups.entra_users_cannot_create_microsoft_365_groups.entra_client", + new=entra_client, ): from prowler.providers.azure.services.entra.entra_users_cannot_create_microsoft_365_groups.entra_users_cannot_create_microsoft_365_groups import ( entra_users_cannot_create_microsoft_365_groups, @@ -31,15 +28,12 @@ class Test_entra_users_cannot_create_microsoft_365_groups: def test_entra_tenant_empty(self): entra_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.entra.entra_users_cannot_create_microsoft_365_groups.entra_users_cannot_create_microsoft_365_groups.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.entra.entra_users_cannot_create_microsoft_365_groups.entra_users_cannot_create_microsoft_365_groups.entra_client", + new=entra_client, ): from prowler.providers.azure.services.entra.entra_users_cannot_create_microsoft_365_groups.entra_users_cannot_create_microsoft_365_groups import ( entra_users_cannot_create_microsoft_365_groups, @@ -59,15 +53,12 @@ class Test_entra_users_cannot_create_microsoft_365_groups: def test_entra_users_cannot_create_microsoft_365_groups(self): entra_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.entra.entra_users_cannot_create_microsoft_365_groups.entra_users_cannot_create_microsoft_365_groups.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.entra.entra_users_cannot_create_microsoft_365_groups.entra_users_cannot_create_microsoft_365_groups.entra_client", + new=entra_client, ): from prowler.providers.azure.services.entra.entra_service import ( GroupSetting, @@ -106,15 +97,12 @@ class Test_entra_users_cannot_create_microsoft_365_groups: def test_entra_users_can_create_microsoft_365_groups(self): entra_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.entra.entra_users_cannot_create_microsoft_365_groups.entra_users_cannot_create_microsoft_365_groups.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.entra.entra_users_cannot_create_microsoft_365_groups.entra_users_cannot_create_microsoft_365_groups.entra_client", + new=entra_client, ): from prowler.providers.azure.services.entra.entra_service import ( GroupSetting, @@ -151,15 +139,12 @@ class Test_entra_users_cannot_create_microsoft_365_groups: def test_entra_users_can_create_microsoft_365_groups_no_setting(self): entra_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.entra.entra_users_cannot_create_microsoft_365_groups.entra_users_cannot_create_microsoft_365_groups.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.entra.entra_users_cannot_create_microsoft_365_groups.entra_users_cannot_create_microsoft_365_groups.entra_client", + new=entra_client, ): from prowler.providers.azure.services.entra.entra_service import ( GroupSetting, diff --git a/tests/providers/azure/services/iam/iam_custom_role_has_permissions_to_administer_resource_locks/iam_custom_role_has_permissions_to_administer_resource_locks_test.py b/tests/providers/azure/services/iam/iam_custom_role_has_permissions_to_administer_resource_locks/iam_custom_role_has_permissions_to_administer_resource_locks_test.py index 90c838317d..bd454aafed 100644 --- a/tests/providers/azure/services/iam/iam_custom_role_has_permissions_to_administer_resource_locks/iam_custom_role_has_permissions_to_administer_resource_locks_test.py +++ b/tests/providers/azure/services/iam/iam_custom_role_has_permissions_to_administer_resource_locks/iam_custom_role_has_permissions_to_administer_resource_locks_test.py @@ -15,15 +15,12 @@ class Test_iam_custom_role_has_permissions_to_administer_resource_locks: defender_client = mock.MagicMock defender_client.custom_roles = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.iam.iam_custom_role_has_permissions_to_administer_resource_locks.iam_custom_role_has_permissions_to_administer_resource_locks.iam_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.iam.iam_custom_role_has_permissions_to_administer_resource_locks.iam_custom_role_has_permissions_to_administer_resource_locks.iam_client", + new=defender_client, ): from prowler.providers.azure.services.iam.iam_custom_role_has_permissions_to_administer_resource_locks.iam_custom_role_has_permissions_to_administer_resource_locks import ( iam_custom_role_has_permissions_to_administer_resource_locks, @@ -57,15 +54,12 @@ class Test_iam_custom_role_has_permissions_to_administer_resource_locks: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.iam.iam_custom_role_has_permissions_to_administer_resource_locks.iam_custom_role_has_permissions_to_administer_resource_locks.iam_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.iam.iam_custom_role_has_permissions_to_administer_resource_locks.iam_custom_role_has_permissions_to_administer_resource_locks.iam_client", + new=defender_client, ): from prowler.providers.azure.services.iam.iam_custom_role_has_permissions_to_administer_resource_locks.iam_custom_role_has_permissions_to_administer_resource_locks import ( iam_custom_role_has_permissions_to_administer_resource_locks, @@ -103,15 +97,12 @@ class Test_iam_custom_role_has_permissions_to_administer_resource_locks: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.iam.iam_custom_role_has_permissions_to_administer_resource_locks.iam_custom_role_has_permissions_to_administer_resource_locks.iam_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.iam.iam_custom_role_has_permissions_to_administer_resource_locks.iam_custom_role_has_permissions_to_administer_resource_locks.iam_client", + new=defender_client, ): from prowler.providers.azure.services.iam.iam_custom_role_has_permissions_to_administer_resource_locks.iam_custom_role_has_permissions_to_administer_resource_locks import ( iam_custom_role_has_permissions_to_administer_resource_locks, @@ -171,15 +162,12 @@ class Test_iam_custom_role_has_permissions_to_administer_resource_locks: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.iam.iam_custom_role_has_permissions_to_administer_resource_locks.iam_custom_role_has_permissions_to_administer_resource_locks.iam_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.iam.iam_custom_role_has_permissions_to_administer_resource_locks.iam_custom_role_has_permissions_to_administer_resource_locks.iam_client", + new=defender_client, ): from prowler.providers.azure.services.iam.iam_custom_role_has_permissions_to_administer_resource_locks.iam_custom_role_has_permissions_to_administer_resource_locks import ( iam_custom_role_has_permissions_to_administer_resource_locks, diff --git a/tests/providers/azure/services/iam/iam_subscription_roles_owner_custom_not_created/iam_subscription_roles_owner_custom_not_created_test.py b/tests/providers/azure/services/iam/iam_subscription_roles_owner_custom_not_created/iam_subscription_roles_owner_custom_not_created_test.py index 52f824829d..e5af1120c7 100644 --- a/tests/providers/azure/services/iam/iam_subscription_roles_owner_custom_not_created/iam_subscription_roles_owner_custom_not_created_test.py +++ b/tests/providers/azure/services/iam/iam_subscription_roles_owner_custom_not_created/iam_subscription_roles_owner_custom_not_created_test.py @@ -15,15 +15,12 @@ class Test_iam_subscription_roles_owner_custom_not_created: defender_client = mock.MagicMock defender_client.custom_roles = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.iam.iam_subscription_roles_owner_custom_not_created.iam_subscription_roles_owner_custom_not_created.iam_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.iam.iam_subscription_roles_owner_custom_not_created.iam_subscription_roles_owner_custom_not_created.iam_client", + new=defender_client, ): from prowler.providers.azure.services.iam.iam_subscription_roles_owner_custom_not_created.iam_subscription_roles_owner_custom_not_created import ( iam_subscription_roles_owner_custom_not_created, @@ -48,15 +45,12 @@ class Test_iam_subscription_roles_owner_custom_not_created: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.iam.iam_subscription_roles_owner_custom_not_created.iam_subscription_roles_owner_custom_not_created.iam_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.iam.iam_subscription_roles_owner_custom_not_created.iam_subscription_roles_owner_custom_not_created.iam_client", + new=defender_client, ): from prowler.providers.azure.services.iam.iam_subscription_roles_owner_custom_not_created.iam_subscription_roles_owner_custom_not_created import ( iam_subscription_roles_owner_custom_not_created, @@ -92,15 +86,12 @@ class Test_iam_subscription_roles_owner_custom_not_created: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.iam.iam_subscription_roles_owner_custom_not_created.iam_subscription_roles_owner_custom_not_created.iam_client", - new=defender_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.iam.iam_subscription_roles_owner_custom_not_created.iam_subscription_roles_owner_custom_not_created.iam_client", + new=defender_client, ): from prowler.providers.azure.services.iam.iam_subscription_roles_owner_custom_not_created.iam_subscription_roles_owner_custom_not_created import ( iam_subscription_roles_owner_custom_not_created, diff --git a/tests/providers/azure/services/keyvault/keyvault_key_expiration_set_in_non_rbac/keyvault_key_expiration_set_in_non_rbac_test.py b/tests/providers/azure/services/keyvault/keyvault_key_expiration_set_in_non_rbac/keyvault_key_expiration_set_in_non_rbac_test.py index 8c9901080b..dfabadd0e0 100644 --- a/tests/providers/azure/services/keyvault/keyvault_key_expiration_set_in_non_rbac/keyvault_key_expiration_set_in_non_rbac_test.py +++ b/tests/providers/azure/services/keyvault/keyvault_key_expiration_set_in_non_rbac/keyvault_key_expiration_set_in_non_rbac_test.py @@ -14,15 +14,12 @@ class Test_keyvault_key_expiration_set_in_non_rbac: keyvault_client = mock.MagicMock keyvault_client.key_vaults = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_key_expiration_set_in_non_rbac.keyvault_key_expiration_set_in_non_rbac.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_key_expiration_set_in_non_rbac.keyvault_key_expiration_set_in_non_rbac.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_key_expiration_set_in_non_rbac.keyvault_key_expiration_set_in_non_rbac import ( keyvault_key_expiration_set_in_non_rbac, @@ -35,15 +32,12 @@ class Test_keyvault_key_expiration_set_in_non_rbac: def test_no_keys(self): keyvault_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_key_expiration_set_in_non_rbac.keyvault_key_expiration_set_in_non_rbac.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_key_expiration_set_in_non_rbac.keyvault_key_expiration_set_in_non_rbac.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_key_expiration_set_in_non_rbac.keyvault_key_expiration_set_in_non_rbac import ( keyvault_key_expiration_set_in_non_rbac, @@ -79,15 +73,12 @@ class Test_keyvault_key_expiration_set_in_non_rbac: keyvault_id = str(uuid4()) key_name = "Key Name" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_key_expiration_set_in_non_rbac.keyvault_key_expiration_set_in_non_rbac.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_key_expiration_set_in_non_rbac.keyvault_key_expiration_set_in_non_rbac.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_key_expiration_set_in_non_rbac.keyvault_key_expiration_set_in_non_rbac import ( keyvault_key_expiration_set_in_non_rbac, @@ -139,15 +130,12 @@ class Test_keyvault_key_expiration_set_in_non_rbac: keyvault_name = "Keyvault Name" keyvault_id = str(uuid4()) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_key_expiration_set_in_non_rbac.keyvault_key_expiration_set_in_non_rbac.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_key_expiration_set_in_non_rbac.keyvault_key_expiration_set_in_non_rbac.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_key_expiration_set_in_non_rbac.keyvault_key_expiration_set_in_non_rbac import ( keyvault_key_expiration_set_in_non_rbac, diff --git a/tests/providers/azure/services/keyvault/keyvault_key_rotation_enabled/keyvault_key_rotation_enabled_test.py b/tests/providers/azure/services/keyvault/keyvault_key_rotation_enabled/keyvault_key_rotation_enabled_test.py index cb4224a12f..99998a359b 100644 --- a/tests/providers/azure/services/keyvault/keyvault_key_rotation_enabled/keyvault_key_rotation_enabled_test.py +++ b/tests/providers/azure/services/keyvault/keyvault_key_rotation_enabled/keyvault_key_rotation_enabled_test.py @@ -14,15 +14,12 @@ class Test_keyvault_key_rotation_enabled: keyvault_client = mock.MagicMock keyvault_client.key_vaults = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_key_rotation_enabled.keyvault_key_rotation_enabled.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_key_rotation_enabled.keyvault_key_rotation_enabled.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_key_rotation_enabled.keyvault_key_rotation_enabled import ( keyvault_key_rotation_enabled, @@ -35,15 +32,12 @@ class Test_keyvault_key_rotation_enabled: def test_no_keys(self): keyvault_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_key_rotation_enabled.keyvault_key_rotation_enabled.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_key_rotation_enabled.keyvault_key_rotation_enabled.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_key_rotation_enabled.keyvault_key_rotation_enabled import ( keyvault_key_rotation_enabled, @@ -78,15 +72,12 @@ class Test_keyvault_key_rotation_enabled: keyvault_name = "keyvault_name" key_name = "key_name" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_key_rotation_enabled.keyvault_key_rotation_enabled.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_key_rotation_enabled.keyvault_key_rotation_enabled.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_key_rotation_enabled.keyvault_key_rotation_enabled import ( keyvault_key_rotation_enabled, @@ -140,15 +131,12 @@ class Test_keyvault_key_rotation_enabled: keyvault_name = "keyvault_name" key_name = "key_name" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_key_rotation_enabled.keyvault_key_rotation_enabled.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_key_rotation_enabled.keyvault_key_rotation_enabled.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_key_rotation_enabled.keyvault_key_rotation_enabled import ( keyvault_key_rotation_enabled, diff --git a/tests/providers/azure/services/keyvault/keyvault_logging_enabled/keyvault_logging_enabled_test.py b/tests/providers/azure/services/keyvault/keyvault_logging_enabled/keyvault_logging_enabled_test.py index 1bbc078c08..43bfe8340c 100644 --- a/tests/providers/azure/services/keyvault/keyvault_logging_enabled/keyvault_logging_enabled_test.py +++ b/tests/providers/azure/services/keyvault/keyvault_logging_enabled/keyvault_logging_enabled_test.py @@ -13,19 +13,15 @@ class Test_keyvault_logging_enabled: keyvault_client = mock.MagicMock keyvault_client.key_vaults = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_service.Monitor", - new=mock.MagicMock(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_logging_enabled.keyvault_logging_enabled.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_service.Monitor", + new=mock.MagicMock(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_logging_enabled.keyvault_logging_enabled.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_logging_enabled.keyvault_logging_enabled import ( @@ -39,19 +35,15 @@ class Test_keyvault_logging_enabled: def test_no_diagnostic_settings(self): keyvault_client = mock.MagicMock keyvault_client.key_vaults = {AZURE_SUBSCRIPTION_ID: []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_service.Monitor", - new=mock.MagicMock(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_logging_enabled.keyvault_logging_enabled.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_service.Monitor", + new=mock.MagicMock(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_logging_enabled.keyvault_logging_enabled.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_logging_enabled.keyvault_logging_enabled import ( keyvault_logging_enabled, @@ -64,19 +56,15 @@ class Test_keyvault_logging_enabled: def test_diagnostic_settings_configured(self): keyvault_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_service.Monitor", - new=mock.MagicMock(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_logging_enabled.keyvault_logging_enabled.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_service.Monitor", + new=mock.MagicMock(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_logging_enabled.keyvault_logging_enabled.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_logging_enabled.keyvault_logging_enabled import ( keyvault_logging_enabled, diff --git a/tests/providers/azure/services/keyvault/keyvault_non_rbac_secret_expiration_set/keyvault_non_rbac_secret_expiration_set_test.py b/tests/providers/azure/services/keyvault/keyvault_non_rbac_secret_expiration_set/keyvault_non_rbac_secret_expiration_set_test.py index 9a799a1a06..e896972b6c 100644 --- a/tests/providers/azure/services/keyvault/keyvault_non_rbac_secret_expiration_set/keyvault_non_rbac_secret_expiration_set_test.py +++ b/tests/providers/azure/services/keyvault/keyvault_non_rbac_secret_expiration_set/keyvault_non_rbac_secret_expiration_set_test.py @@ -14,15 +14,12 @@ class Test_keyvault_non_rbac_secret_expiration_set: keyvault_client = mock.MagicMock keyvault_client.key_vaults = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_non_rbac_secret_expiration_set.keyvault_non_rbac_secret_expiration_set.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_non_rbac_secret_expiration_set.keyvault_non_rbac_secret_expiration_set.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_non_rbac_secret_expiration_set.keyvault_non_rbac_secret_expiration_set import ( keyvault_non_rbac_secret_expiration_set, @@ -35,15 +32,12 @@ class Test_keyvault_non_rbac_secret_expiration_set: def test_no_secrets(self): keyvault_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_non_rbac_secret_expiration_set.keyvault_non_rbac_secret_expiration_set.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_non_rbac_secret_expiration_set.keyvault_non_rbac_secret_expiration_set.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_non_rbac_secret_expiration_set.keyvault_non_rbac_secret_expiration_set import ( keyvault_non_rbac_secret_expiration_set, @@ -80,15 +74,12 @@ class Test_keyvault_non_rbac_secret_expiration_set: keyvault_id = str(uuid4()) secret_name = "Secret" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_non_rbac_secret_expiration_set.keyvault_non_rbac_secret_expiration_set.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_non_rbac_secret_expiration_set.keyvault_non_rbac_secret_expiration_set.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_non_rbac_secret_expiration_set.keyvault_non_rbac_secret_expiration_set import ( keyvault_non_rbac_secret_expiration_set, @@ -142,15 +133,12 @@ class Test_keyvault_non_rbac_secret_expiration_set: secret1_name = "Secret1" secret2_name = "Secret2" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_non_rbac_secret_expiration_set.keyvault_non_rbac_secret_expiration_set.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_non_rbac_secret_expiration_set.keyvault_non_rbac_secret_expiration_set.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_non_rbac_secret_expiration_set.keyvault_non_rbac_secret_expiration_set import ( keyvault_non_rbac_secret_expiration_set, @@ -209,15 +197,12 @@ class Test_keyvault_non_rbac_secret_expiration_set: keyvault_name = "Keyvault Name" keyvault_id = str(uuid4()) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_non_rbac_secret_expiration_set.keyvault_non_rbac_secret_expiration_set.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_non_rbac_secret_expiration_set.keyvault_non_rbac_secret_expiration_set.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_non_rbac_secret_expiration_set.keyvault_non_rbac_secret_expiration_set import ( keyvault_non_rbac_secret_expiration_set, diff --git a/tests/providers/azure/services/keyvault/keyvault_private_endpoints/keyvault_private_endpoints_test.py b/tests/providers/azure/services/keyvault/keyvault_private_endpoints/keyvault_private_endpoints_test.py index e071453485..81826fc862 100644 --- a/tests/providers/azure/services/keyvault/keyvault_private_endpoints/keyvault_private_endpoints_test.py +++ b/tests/providers/azure/services/keyvault/keyvault_private_endpoints/keyvault_private_endpoints_test.py @@ -17,15 +17,12 @@ class Test_keyvault_private_endpoints: keyvault_client = mock.MagicMock keyvault_client.key_vaults = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_private_endpoints.keyvault_private_endpoints.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_private_endpoints.keyvault_private_endpoints.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_private_endpoints.keyvault_private_endpoints import ( keyvault_private_endpoints, @@ -40,15 +37,12 @@ class Test_keyvault_private_endpoints: keyvault_name = "Keyvault Name" keyvault_id = str(uuid4()) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_private_endpoints.keyvault_private_endpoints.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_private_endpoints.keyvault_private_endpoints.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_private_endpoints.keyvault_private_endpoints import ( keyvault_private_endpoints, @@ -97,15 +91,12 @@ class Test_keyvault_private_endpoints: id="id", ) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_private_endpoints.keyvault_private_endpoints.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_private_endpoints.keyvault_private_endpoints.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_private_endpoints.keyvault_private_endpoints import ( keyvault_private_endpoints, diff --git a/tests/providers/azure/services/keyvault/keyvault_rbac_enabled/keyvault_rbac_enabled_test.py b/tests/providers/azure/services/keyvault/keyvault_rbac_enabled/keyvault_rbac_enabled_test.py index 00729ab0bf..449f4de9dc 100644 --- a/tests/providers/azure/services/keyvault/keyvault_rbac_enabled/keyvault_rbac_enabled_test.py +++ b/tests/providers/azure/services/keyvault/keyvault_rbac_enabled/keyvault_rbac_enabled_test.py @@ -14,15 +14,12 @@ class Test_keyvault_rbac_enabled: keyvault_client = mock.MagicMock keyvault_client.key_vaults = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_rbac_enabled.keyvault_rbac_enabled.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_rbac_enabled.keyvault_rbac_enabled.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_rbac_enabled.keyvault_rbac_enabled import ( keyvault_rbac_enabled, @@ -37,15 +34,12 @@ class Test_keyvault_rbac_enabled: keyvault_name = "Keyvault Name" keyvault_id = str(uuid4()) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_rbac_enabled.keyvault_rbac_enabled.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_rbac_enabled.keyvault_rbac_enabled.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_rbac_enabled.keyvault_rbac_enabled import ( keyvault_rbac_enabled, @@ -89,15 +83,12 @@ class Test_keyvault_rbac_enabled: keyvault_name = "Keyvault Name" keyvault_id = str(uuid4()) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_rbac_enabled.keyvault_rbac_enabled.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_rbac_enabled.keyvault_rbac_enabled.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_rbac_enabled.keyvault_rbac_enabled import ( keyvault_rbac_enabled, diff --git a/tests/providers/azure/services/keyvault/keyvault_rbac_key_expiration_set/keyvault_rbac_key_expiration_set_test.py b/tests/providers/azure/services/keyvault/keyvault_rbac_key_expiration_set/keyvault_rbac_key_expiration_set_test.py index c9034e7e87..32cb3752f1 100644 --- a/tests/providers/azure/services/keyvault/keyvault_rbac_key_expiration_set/keyvault_rbac_key_expiration_set_test.py +++ b/tests/providers/azure/services/keyvault/keyvault_rbac_key_expiration_set/keyvault_rbac_key_expiration_set_test.py @@ -14,15 +14,12 @@ class Test_keyvault_rbac_key_expiration_set: keyvault_client = mock.MagicMock keyvault_client.key_vaults = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_rbac_key_expiration_set.keyvault_rbac_key_expiration_set.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_rbac_key_expiration_set.keyvault_rbac_key_expiration_set.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_rbac_key_expiration_set.keyvault_rbac_key_expiration_set import ( keyvault_rbac_key_expiration_set, @@ -35,15 +32,12 @@ class Test_keyvault_rbac_key_expiration_set: def test_no_keys(self): keyvault_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_rbac_key_expiration_set.keyvault_rbac_key_expiration_set.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_rbac_key_expiration_set.keyvault_rbac_key_expiration_set.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_rbac_key_expiration_set.keyvault_rbac_key_expiration_set import ( keyvault_rbac_key_expiration_set, @@ -79,15 +73,12 @@ class Test_keyvault_rbac_key_expiration_set: keyvault_id = str(uuid4()) key_name = "Key Name" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_rbac_key_expiration_set.keyvault_rbac_key_expiration_set.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_rbac_key_expiration_set.keyvault_rbac_key_expiration_set.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_rbac_key_expiration_set.keyvault_rbac_key_expiration_set import ( keyvault_rbac_key_expiration_set, @@ -139,15 +130,12 @@ class Test_keyvault_rbac_key_expiration_set: keyvault_name = "Keyvault Name" keyvault_id = str(uuid4()) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_rbac_key_expiration_set.keyvault_rbac_key_expiration_set.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_rbac_key_expiration_set.keyvault_rbac_key_expiration_set.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_rbac_key_expiration_set.keyvault_rbac_key_expiration_set import ( keyvault_rbac_key_expiration_set, diff --git a/tests/providers/azure/services/keyvault/keyvault_rbac_secret_expiration_set/keyvault_rbac_secret_expiration_set_test.py b/tests/providers/azure/services/keyvault/keyvault_rbac_secret_expiration_set/keyvault_rbac_secret_expiration_set_test.py index 59b5285de5..7c17723865 100644 --- a/tests/providers/azure/services/keyvault/keyvault_rbac_secret_expiration_set/keyvault_rbac_secret_expiration_set_test.py +++ b/tests/providers/azure/services/keyvault/keyvault_rbac_secret_expiration_set/keyvault_rbac_secret_expiration_set_test.py @@ -14,15 +14,12 @@ class Test_keyvault_rbac_secret_expiration_set: keyvault_client = mock.MagicMock keyvault_client.key_vaults = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_rbac_secret_expiration_set.keyvault_rbac_secret_expiration_set.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_rbac_secret_expiration_set.keyvault_rbac_secret_expiration_set.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_rbac_secret_expiration_set.keyvault_rbac_secret_expiration_set import ( keyvault_rbac_secret_expiration_set, @@ -35,15 +32,12 @@ class Test_keyvault_rbac_secret_expiration_set: def test_no_secrets(self): keyvault_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_rbac_secret_expiration_set.keyvault_rbac_secret_expiration_set.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_rbac_secret_expiration_set.keyvault_rbac_secret_expiration_set.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_rbac_secret_expiration_set.keyvault_rbac_secret_expiration_set import ( keyvault_rbac_secret_expiration_set, @@ -79,15 +73,12 @@ class Test_keyvault_rbac_secret_expiration_set: keyvault_id = str(uuid4()) secret_name = "Secret" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_rbac_secret_expiration_set.keyvault_rbac_secret_expiration_set.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_rbac_secret_expiration_set.keyvault_rbac_secret_expiration_set.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_rbac_secret_expiration_set.keyvault_rbac_secret_expiration_set import ( keyvault_rbac_secret_expiration_set, @@ -141,15 +132,12 @@ class Test_keyvault_rbac_secret_expiration_set: secret1_name = "Secret1" secret2_name = "Secret2" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_rbac_secret_expiration_set.keyvault_rbac_secret_expiration_set.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_rbac_secret_expiration_set.keyvault_rbac_secret_expiration_set.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_rbac_secret_expiration_set.keyvault_rbac_secret_expiration_set import ( keyvault_rbac_secret_expiration_set, @@ -208,15 +196,12 @@ class Test_keyvault_rbac_secret_expiration_set: keyvault_name = "Keyvault Name" keyvault_id = str(uuid4()) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_rbac_secret_expiration_set.keyvault_rbac_secret_expiration_set.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_rbac_secret_expiration_set.keyvault_rbac_secret_expiration_set.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_rbac_secret_expiration_set.keyvault_rbac_secret_expiration_set import ( keyvault_rbac_secret_expiration_set, diff --git a/tests/providers/azure/services/keyvault/keyvault_recoverable/keyvault_recoverable_test.py b/tests/providers/azure/services/keyvault/keyvault_recoverable/keyvault_recoverable_test.py index 733683d7a1..613f8e0276 100644 --- a/tests/providers/azure/services/keyvault/keyvault_recoverable/keyvault_recoverable_test.py +++ b/tests/providers/azure/services/keyvault/keyvault_recoverable/keyvault_recoverable_test.py @@ -15,15 +15,12 @@ class Test_keyvault_recoverable: keyvault_client = mock.MagicMock keyvault_client.key_vaults = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_recoverable.keyvault_recoverable.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_recoverable.keyvault_recoverable.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_recoverable.keyvault_recoverable import ( keyvault_recoverable, @@ -38,15 +35,12 @@ class Test_keyvault_recoverable: keyvault_name = "Keyvault Name" keyvault_id = str(uuid4()) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_recoverable.keyvault_recoverable.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_recoverable.keyvault_recoverable.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_recoverable.keyvault_recoverable import ( keyvault_recoverable, @@ -92,15 +86,12 @@ class Test_keyvault_recoverable: keyvault_name = "Keyvault Name" keyvault_id = str(uuid4()) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_recoverable.keyvault_recoverable.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_recoverable.keyvault_recoverable.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_recoverable.keyvault_recoverable import ( keyvault_recoverable, @@ -161,15 +152,12 @@ class Test_keyvault_recoverable: keyvault_name = "Keyvault Name" keyvault_id = str(uuid4()) - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.keyvault.keyvault_recoverable.keyvault_recoverable.keyvault_client", - new=keyvault_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.keyvault.keyvault_recoverable.keyvault_recoverable.keyvault_client", + new=keyvault_client, ): from prowler.providers.azure.services.keyvault.keyvault_recoverable.keyvault_recoverable import ( keyvault_recoverable, diff --git a/tests/providers/azure/services/keyvault/keyvault_service_test.py b/tests/providers/azure/services/keyvault/keyvault_service_test.py index f0a73d9081..46a87c86cb 100644 --- a/tests/providers/azure/services/keyvault/keyvault_service_test.py +++ b/tests/providers/azure/services/keyvault/keyvault_service_test.py @@ -66,15 +66,12 @@ from tests.providers.azure.azure_fixtures import ( # ) class Test_keyvault_service: def test_keyvault_service_(self): - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - patch( - "prowler.providers.azure.services.monitor.monitor_service.Monitor", - new=MagicMock(), - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), patch( + "prowler.providers.azure.services.monitor.monitor_service.Monitor", + new=MagicMock(), ): from prowler.providers.azure.services.keyvault.keyvault_service import ( # KeyVault, DiagnosticSetting, diff --git a/tests/providers/azure/services/monitor/monitor_alert_create_policy_assignment/monitor_alert_create_policy_assignment_test.py b/tests/providers/azure/services/monitor/monitor_alert_create_policy_assignment/monitor_alert_create_policy_assignment_test.py index 9c91b432b0..f18acd9c01 100644 --- a/tests/providers/azure/services/monitor/monitor_alert_create_policy_assignment/monitor_alert_create_policy_assignment_test.py +++ b/tests/providers/azure/services/monitor/monitor_alert_create_policy_assignment/monitor_alert_create_policy_assignment_test.py @@ -13,15 +13,12 @@ class Test_monitor_alert_create_policy_assignment: monitor_client = mock.MagicMock monitor_client.alert_rules = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_create_policy_assignment.monitor_alert_create_policy_assignment.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_create_policy_assignment.monitor_alert_create_policy_assignment.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_create_policy_assignment.monitor_alert_create_policy_assignment import ( monitor_alert_create_policy_assignment, @@ -34,15 +31,12 @@ class Test_monitor_alert_create_policy_assignment: def test_no_alert_rules(self): monitor_client = mock.MagicMock monitor_client.alert_rules = {AZURE_SUBSCRIPTION_ID: []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_create_policy_assignment.monitor_alert_create_policy_assignment.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_create_policy_assignment.monitor_alert_create_policy_assignment.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_create_policy_assignment.monitor_alert_create_policy_assignment import ( monitor_alert_create_policy_assignment, @@ -63,15 +57,12 @@ class Test_monitor_alert_create_policy_assignment: def test_alert_rules_configured(self): monitor_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_create_policy_assignment.monitor_alert_create_policy_assignment.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_create_policy_assignment.monitor_alert_create_policy_assignment.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_create_policy_assignment.monitor_alert_create_policy_assignment import ( monitor_alert_create_policy_assignment, diff --git a/tests/providers/azure/services/monitor/monitor_alert_create_update_nsg/monitor_alert_create_update_nsg_test.py b/tests/providers/azure/services/monitor/monitor_alert_create_update_nsg/monitor_alert_create_update_nsg_test.py index b99e2dfd11..3cd1ef7155 100644 --- a/tests/providers/azure/services/monitor/monitor_alert_create_update_nsg/monitor_alert_create_update_nsg_test.py +++ b/tests/providers/azure/services/monitor/monitor_alert_create_update_nsg/monitor_alert_create_update_nsg_test.py @@ -12,15 +12,12 @@ class Test_monitor_alert_create_update_nsg: def test_monitor_alert_create_update_nsg_no_subscriptions(self): monitor_client = mock.MagicMock monitor_client.alert_rules = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_create_update_nsg.monitor_alert_create_update_nsg.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_create_update_nsg.monitor_alert_create_update_nsg.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_create_update_nsg.monitor_alert_create_update_nsg import ( monitor_alert_create_update_nsg, @@ -33,15 +30,12 @@ class Test_monitor_alert_create_update_nsg: def test_no_alert_rules(self): monitor_client = mock.MagicMock() monitor_client.alert_rules = {AZURE_SUBSCRIPTION_ID: []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_create_update_nsg.monitor_alert_create_update_nsg.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_create_update_nsg.monitor_alert_create_update_nsg.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_create_update_nsg.monitor_alert_create_update_nsg import ( monitor_alert_create_update_nsg, @@ -61,15 +55,12 @@ class Test_monitor_alert_create_update_nsg: def test_alert_rules_configured(self): monitor_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_create_update_nsg.monitor_alert_create_update_nsg.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_create_update_nsg.monitor_alert_create_update_nsg.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_create_update_nsg.monitor_alert_create_update_nsg import ( monitor_alert_create_update_nsg, diff --git a/tests/providers/azure/services/monitor/monitor_alert_create_update_public_ip_address_rule/monitor_alert_create_update_public_ip_address_rule_test.py b/tests/providers/azure/services/monitor/monitor_alert_create_update_public_ip_address_rule/monitor_alert_create_update_public_ip_address_rule_test.py index aa2625db35..f5b0027fdb 100644 --- a/tests/providers/azure/services/monitor/monitor_alert_create_update_public_ip_address_rule/monitor_alert_create_update_public_ip_address_rule_test.py +++ b/tests/providers/azure/services/monitor/monitor_alert_create_update_public_ip_address_rule/monitor_alert_create_update_public_ip_address_rule_test.py @@ -12,15 +12,12 @@ class Test_monitor_alert_create_update_security_solution: def test_monitor_alert_create_update_public_ip_address_rule_no_subscriptions(self): monitor_client = mock.MagicMock() monitor_client.alert_rules = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_create_update_public_ip_address_rule.monitor_alert_create_update_public_ip_address_rule.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_create_update_public_ip_address_rule.monitor_alert_create_update_public_ip_address_rule.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_create_update_public_ip_address_rule.monitor_alert_create_update_public_ip_address_rule import ( monitor_alert_create_update_public_ip_address_rule, @@ -33,15 +30,12 @@ class Test_monitor_alert_create_update_security_solution: def test_no_alert_rules(self): monitor_client = mock.MagicMock() monitor_client.alert_rules = {AZURE_SUBSCRIPTION_ID: []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_create_update_public_ip_address_rule.monitor_alert_create_update_public_ip_address_rule.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_create_update_public_ip_address_rule.monitor_alert_create_update_public_ip_address_rule.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_create_update_public_ip_address_rule.monitor_alert_create_update_public_ip_address_rule import ( monitor_alert_create_update_public_ip_address_rule, @@ -62,15 +56,12 @@ class Test_monitor_alert_create_update_security_solution: def test_alert_rules_configured(self): monitor_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_create_update_public_ip_address_rule.monitor_alert_create_update_public_ip_address_rule.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_create_update_public_ip_address_rule.monitor_alert_create_update_public_ip_address_rule.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_create_update_public_ip_address_rule.monitor_alert_create_update_public_ip_address_rule import ( monitor_alert_create_update_public_ip_address_rule, diff --git a/tests/providers/azure/services/monitor/monitor_alert_create_update_security_solution/monitor_alert_create_update_security_solution_test.py b/tests/providers/azure/services/monitor/monitor_alert_create_update_security_solution/monitor_alert_create_update_security_solution_test.py index b6b926e463..bb96296ed8 100644 --- a/tests/providers/azure/services/monitor/monitor_alert_create_update_security_solution/monitor_alert_create_update_security_solution_test.py +++ b/tests/providers/azure/services/monitor/monitor_alert_create_update_security_solution/monitor_alert_create_update_security_solution_test.py @@ -12,15 +12,12 @@ class Test_monitor_alert_create_update_security_solution: def test_monitor_alert_create_update_security_solution_no_subscriptions(self): monitor_client = mock.MagicMock() monitor_client.alert_rules = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_create_update_security_solution.monitor_alert_create_update_security_solution.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_create_update_security_solution.monitor_alert_create_update_security_solution.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_create_update_security_solution.monitor_alert_create_update_security_solution import ( monitor_alert_create_update_security_solution, @@ -33,15 +30,12 @@ class Test_monitor_alert_create_update_security_solution: def test_no_alert_rules(self): monitor_client = mock.MagicMock() monitor_client.alert_rules = {AZURE_SUBSCRIPTION_ID: []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_create_update_security_solution.monitor_alert_create_update_security_solution.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_create_update_security_solution.monitor_alert_create_update_security_solution.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_create_update_security_solution.monitor_alert_create_update_security_solution import ( monitor_alert_create_update_security_solution, @@ -62,15 +56,12 @@ class Test_monitor_alert_create_update_security_solution: def test_alert_rules_configured(self): monitor_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_create_update_security_solution.monitor_alert_create_update_security_solution.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_create_update_security_solution.monitor_alert_create_update_security_solution.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_create_update_security_solution.monitor_alert_create_update_security_solution import ( monitor_alert_create_update_security_solution, diff --git a/tests/providers/azure/services/monitor/monitor_alert_create_update_sqlserver_fr/monitor_alert_create_update_sqlserver_fr_test.py b/tests/providers/azure/services/monitor/monitor_alert_create_update_sqlserver_fr/monitor_alert_create_update_sqlserver_fr_test.py index e386357e62..6f0b015746 100644 --- a/tests/providers/azure/services/monitor/monitor_alert_create_update_sqlserver_fr/monitor_alert_create_update_sqlserver_fr_test.py +++ b/tests/providers/azure/services/monitor/monitor_alert_create_update_sqlserver_fr/monitor_alert_create_update_sqlserver_fr_test.py @@ -12,15 +12,12 @@ class Test_monitor_alert_create_update_sqlserver_fr: def test_monitor_alert_create_update_sqlserver_fr_no_subscriptions(self): monitor_client = mock.MagicMock() monitor_client.alert_rules = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_create_update_sqlserver_fr.monitor_alert_create_update_sqlserver_fr.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_create_update_sqlserver_fr.monitor_alert_create_update_sqlserver_fr.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_create_update_sqlserver_fr.monitor_alert_create_update_sqlserver_fr import ( monitor_alert_create_update_sqlserver_fr, @@ -33,15 +30,12 @@ class Test_monitor_alert_create_update_sqlserver_fr: def test_no_alert_rules(self): monitor_client = mock.MagicMock() monitor_client.alert_rules = {AZURE_SUBSCRIPTION_ID: []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_create_update_sqlserver_fr.monitor_alert_create_update_sqlserver_fr.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_create_update_sqlserver_fr.monitor_alert_create_update_sqlserver_fr.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_create_update_sqlserver_fr.monitor_alert_create_update_sqlserver_fr import ( monitor_alert_create_update_sqlserver_fr, @@ -62,15 +56,12 @@ class Test_monitor_alert_create_update_sqlserver_fr: def test_alert_rules_configured(self): monitor_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_create_update_sqlserver_fr.monitor_alert_create_update_sqlserver_fr.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_create_update_sqlserver_fr.monitor_alert_create_update_sqlserver_fr.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_create_update_sqlserver_fr.monitor_alert_create_update_sqlserver_fr import ( monitor_alert_create_update_sqlserver_fr, diff --git a/tests/providers/azure/services/monitor/monitor_alert_delete_nsg/monitor_alert_delete_nsg_test.py b/tests/providers/azure/services/monitor/monitor_alert_delete_nsg/monitor_alert_delete_nsg_test.py index 3525ab6a13..2a0c9a46e9 100644 --- a/tests/providers/azure/services/monitor/monitor_alert_delete_nsg/monitor_alert_delete_nsg_test.py +++ b/tests/providers/azure/services/monitor/monitor_alert_delete_nsg/monitor_alert_delete_nsg_test.py @@ -12,15 +12,12 @@ class Test_monitor_alert_delete_nsg: def test_monitor_alert_delete_nsg_no_subscriptions(self): monitor_client = mock.MagicMock() monitor_client.alert_rules = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_delete_nsg.monitor_alert_delete_nsg.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_delete_nsg.monitor_alert_delete_nsg.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_delete_nsg.monitor_alert_delete_nsg import ( monitor_alert_delete_nsg, @@ -33,15 +30,12 @@ class Test_monitor_alert_delete_nsg: def test_no_alert_rules(self): monitor_client = mock.MagicMock() monitor_client.alert_rules = {AZURE_SUBSCRIPTION_ID: []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_delete_nsg.monitor_alert_delete_nsg.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_delete_nsg.monitor_alert_delete_nsg.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_delete_nsg.monitor_alert_delete_nsg import ( monitor_alert_delete_nsg, @@ -62,15 +56,12 @@ class Test_monitor_alert_delete_nsg: def test_alert_rules_configured(self): monitor_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_delete_nsg.monitor_alert_delete_nsg.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_delete_nsg.monitor_alert_delete_nsg.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_delete_nsg.monitor_alert_delete_nsg import ( monitor_alert_delete_nsg, diff --git a/tests/providers/azure/services/monitor/monitor_alert_delete_policy_assignment/monitor_alert_delete_policy_assignment_test.py b/tests/providers/azure/services/monitor/monitor_alert_delete_policy_assignment/monitor_alert_delete_policy_assignment_test.py index 627874eed7..d46fafad93 100644 --- a/tests/providers/azure/services/monitor/monitor_alert_delete_policy_assignment/monitor_alert_delete_policy_assignment_test.py +++ b/tests/providers/azure/services/monitor/monitor_alert_delete_policy_assignment/monitor_alert_delete_policy_assignment_test.py @@ -13,15 +13,12 @@ class Test_monitor_alert_delete_policy_assignment: monitor_client = mock.MagicMock monitor_client.alert_rules = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_delete_policy_assignment.monitor_alert_delete_policy_assignment.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_delete_policy_assignment.monitor_alert_delete_policy_assignment.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_delete_policy_assignment.monitor_alert_delete_policy_assignment import ( monitor_alert_delete_policy_assignment, @@ -34,15 +31,12 @@ class Test_monitor_alert_delete_policy_assignment: def test_no_alert_rules(self): monitor_client = mock.MagicMock monitor_client.alert_rules = {AZURE_SUBSCRIPTION_ID: []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_delete_policy_assignment.monitor_alert_delete_policy_assignment.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_delete_policy_assignment.monitor_alert_delete_policy_assignment.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_delete_policy_assignment.monitor_alert_delete_policy_assignment import ( monitor_alert_delete_policy_assignment, @@ -63,15 +57,12 @@ class Test_monitor_alert_delete_policy_assignment: def test_alert_rules_configured(self): monitor_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_delete_policy_assignment.monitor_alert_delete_policy_assignment.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_delete_policy_assignment.monitor_alert_delete_policy_assignment.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_delete_policy_assignment.monitor_alert_delete_policy_assignment import ( monitor_alert_delete_policy_assignment, diff --git a/tests/providers/azure/services/monitor/monitor_alert_delete_public_ip_address_rule/monitor_alert_delete_public_ip_address_rule_test.py b/tests/providers/azure/services/monitor/monitor_alert_delete_public_ip_address_rule/monitor_alert_delete_public_ip_address_rule_test.py index ff217c21ab..d313be7e27 100644 --- a/tests/providers/azure/services/monitor/monitor_alert_delete_public_ip_address_rule/monitor_alert_delete_public_ip_address_rule_test.py +++ b/tests/providers/azure/services/monitor/monitor_alert_delete_public_ip_address_rule/monitor_alert_delete_public_ip_address_rule_test.py @@ -12,15 +12,12 @@ class Test_monitor_alert_create_update_security_solution: def test_monitor_alert_delete_public_ip_address_rule_no_subscriptions(self): monitor_client = mock.MagicMock() monitor_client.alert_rules = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_delete_public_ip_address_rule.monitor_alert_delete_public_ip_address_rule.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_delete_public_ip_address_rule.monitor_alert_delete_public_ip_address_rule.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_delete_public_ip_address_rule.monitor_alert_delete_public_ip_address_rule import ( monitor_alert_delete_public_ip_address_rule, @@ -33,15 +30,12 @@ class Test_monitor_alert_create_update_security_solution: def test_no_alert_rules(self): monitor_client = mock.MagicMock() monitor_client.alert_rules = {AZURE_SUBSCRIPTION_ID: []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_delete_public_ip_address_rule.monitor_alert_delete_public_ip_address_rule.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_delete_public_ip_address_rule.monitor_alert_delete_public_ip_address_rule.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_delete_public_ip_address_rule.monitor_alert_delete_public_ip_address_rule import ( monitor_alert_delete_public_ip_address_rule, @@ -62,15 +56,12 @@ class Test_monitor_alert_create_update_security_solution: def test_alert_rules_configured(self): monitor_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_delete_public_ip_address_rule.monitor_alert_delete_public_ip_address_rule.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_delete_public_ip_address_rule.monitor_alert_delete_public_ip_address_rule.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_delete_public_ip_address_rule.monitor_alert_delete_public_ip_address_rule import ( monitor_alert_delete_public_ip_address_rule, diff --git a/tests/providers/azure/services/monitor/monitor_alert_delete_security_solution/monitor_alert_delete_security_solution_test.py b/tests/providers/azure/services/monitor/monitor_alert_delete_security_solution/monitor_alert_delete_security_solution_test.py index b0da67d639..9c65023bf3 100644 --- a/tests/providers/azure/services/monitor/monitor_alert_delete_security_solution/monitor_alert_delete_security_solution_test.py +++ b/tests/providers/azure/services/monitor/monitor_alert_delete_security_solution/monitor_alert_delete_security_solution_test.py @@ -12,15 +12,12 @@ class Test_monitor_alert_create_update_security_solution: def test_monitor_alert_delete_security_solution_no_subscriptions(self): monitor_client = mock.MagicMock() monitor_client.alert_rules = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_delete_security_solution.monitor_alert_delete_security_solution.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_delete_security_solution.monitor_alert_delete_security_solution.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_delete_security_solution.monitor_alert_delete_security_solution import ( monitor_alert_delete_security_solution, @@ -33,15 +30,12 @@ class Test_monitor_alert_create_update_security_solution: def test_no_alert_rules(self): monitor_client = mock.MagicMock() monitor_client.alert_rules = {AZURE_SUBSCRIPTION_ID: []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_delete_security_solution.monitor_alert_delete_security_solution.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_delete_security_solution.monitor_alert_delete_security_solution.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_delete_security_solution.monitor_alert_delete_security_solution import ( monitor_alert_delete_security_solution, @@ -62,15 +56,12 @@ class Test_monitor_alert_create_update_security_solution: def test_alert_rules_configured(self): monitor_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_delete_security_solution.monitor_alert_delete_security_solution.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_delete_security_solution.monitor_alert_delete_security_solution.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_delete_security_solution.monitor_alert_delete_security_solution import ( monitor_alert_delete_security_solution, diff --git a/tests/providers/azure/services/monitor/monitor_alert_delete_sqlserver_fr/monitor_alert_delete_sqlserver_fr_test.py b/tests/providers/azure/services/monitor/monitor_alert_delete_sqlserver_fr/monitor_alert_delete_sqlserver_fr_test.py index 43cd769e13..151c2e004c 100644 --- a/tests/providers/azure/services/monitor/monitor_alert_delete_sqlserver_fr/monitor_alert_delete_sqlserver_fr_test.py +++ b/tests/providers/azure/services/monitor/monitor_alert_delete_sqlserver_fr/monitor_alert_delete_sqlserver_fr_test.py @@ -12,15 +12,12 @@ class Test_monitor_alert_delete_sqlserver_fr: def test_monitor_alert_delete_sqlserver_fr_no_subscriptions(self): monitor_client = mock.MagicMock() monitor_client.alert_rules = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_delete_sqlserver_fr.monitor_alert_delete_sqlserver_fr.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_delete_sqlserver_fr.monitor_alert_delete_sqlserver_fr.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_delete_sqlserver_fr.monitor_alert_delete_sqlserver_fr import ( monitor_alert_delete_sqlserver_fr, @@ -33,15 +30,12 @@ class Test_monitor_alert_delete_sqlserver_fr: def test_no_alert_rules(self): monitor_client = mock.MagicMock() monitor_client.alert_rules = {AZURE_SUBSCRIPTION_ID: []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_delete_sqlserver_fr.monitor_alert_delete_sqlserver_fr.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_delete_sqlserver_fr.monitor_alert_delete_sqlserver_fr.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_delete_sqlserver_fr.monitor_alert_delete_sqlserver_fr import ( monitor_alert_delete_sqlserver_fr, @@ -62,15 +56,12 @@ class Test_monitor_alert_delete_sqlserver_fr: def test_alert_rules_configured(self): monitor_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_alert_delete_sqlserver_fr.monitor_alert_delete_sqlserver_fr.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_alert_delete_sqlserver_fr.monitor_alert_delete_sqlserver_fr.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_alert_delete_sqlserver_fr.monitor_alert_delete_sqlserver_fr import ( monitor_alert_delete_sqlserver_fr, diff --git a/tests/providers/azure/services/monitor/monitor_diagnostic_setting_with_appropriate_categories/monitor_diagnostic_setting_with_appropriate_categories_test.py b/tests/providers/azure/services/monitor/monitor_diagnostic_setting_with_appropriate_categories/monitor_diagnostic_setting_with_appropriate_categories_test.py index fd3c44f961..26e1573735 100644 --- a/tests/providers/azure/services/monitor/monitor_diagnostic_setting_with_appropriate_categories/monitor_diagnostic_setting_with_appropriate_categories_test.py +++ b/tests/providers/azure/services/monitor/monitor_diagnostic_setting_with_appropriate_categories/monitor_diagnostic_setting_with_appropriate_categories_test.py @@ -13,15 +13,12 @@ class Test_monitor_diagnostic_setting_with_appropriate_categories: monitor_client = mock.MagicMock monitor_client.diagnostics_settings = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_diagnostic_setting_with_appropriate_categories.monitor_diagnostic_setting_with_appropriate_categories.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_diagnostic_setting_with_appropriate_categories.monitor_diagnostic_setting_with_appropriate_categories.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_diagnostic_setting_with_appropriate_categories.monitor_diagnostic_setting_with_appropriate_categories import ( @@ -35,15 +32,12 @@ class Test_monitor_diagnostic_setting_with_appropriate_categories: def test_no_diagnostic_settings(self): monitor_client = mock.MagicMock monitor_client.diagnostics_settings = {AZURE_SUBSCRIPTION_ID: []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_diagnostic_setting_with_appropriate_categories.monitor_diagnostic_setting_with_appropriate_categories.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_diagnostic_setting_with_appropriate_categories.monitor_diagnostic_setting_with_appropriate_categories.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_diagnostic_setting_with_appropriate_categories.monitor_diagnostic_setting_with_appropriate_categories import ( monitor_diagnostic_setting_with_appropriate_categories, @@ -64,15 +58,12 @@ class Test_monitor_diagnostic_setting_with_appropriate_categories: def test_diagnostic_settings_configured(self): monitor_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_diagnostic_setting_with_appropriate_categories.monitor_diagnostic_setting_with_appropriate_categories.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_diagnostic_setting_with_appropriate_categories.monitor_diagnostic_setting_with_appropriate_categories.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_diagnostic_setting_with_appropriate_categories.monitor_diagnostic_setting_with_appropriate_categories import ( monitor_diagnostic_setting_with_appropriate_categories, diff --git a/tests/providers/azure/services/monitor/monitor_diagnostic_settings_exists/monitor_diagnostic_settings_exists_test.py b/tests/providers/azure/services/monitor/monitor_diagnostic_settings_exists/monitor_diagnostic_settings_exists_test.py index 9aca95c44e..21910d7699 100644 --- a/tests/providers/azure/services/monitor/monitor_diagnostic_settings_exists/monitor_diagnostic_settings_exists_test.py +++ b/tests/providers/azure/services/monitor/monitor_diagnostic_settings_exists/monitor_diagnostic_settings_exists_test.py @@ -14,15 +14,12 @@ class Test_monitor_diagnostic_settings_exists: monitor_client = mock.MagicMock monitor_client.diagnostics_settings = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_diagnostic_settings_exists.monitor_diagnostic_settings_exists.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_diagnostic_settings_exists.monitor_diagnostic_settings_exists.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_diagnostic_settings_exists.monitor_diagnostic_settings_exists import ( monitor_diagnostic_settings_exists, @@ -35,15 +32,12 @@ class Test_monitor_diagnostic_settings_exists: def test_no_diagnostic_settings(self): monitor_client = mock.MagicMock monitor_client.diagnostics_settings = {AZURE_SUBSCRIPTION_ID: []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_diagnostic_settings_exists.monitor_diagnostic_settings_exists.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_diagnostic_settings_exists.monitor_diagnostic_settings_exists.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_diagnostic_settings_exists.monitor_diagnostic_settings_exists import ( monitor_diagnostic_settings_exists, @@ -63,25 +57,19 @@ class Test_monitor_diagnostic_settings_exists: monitor_client = mock.MagicMock storage_client = mock.MagicMock - with ( - mock.patch( + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_diagnostic_settings_exists.monitor_diagnostic_settings_exists.monitor_client", + new=monitor_client, + ): + with mock.patch( "prowler.providers.common.provider.Provider.get_global_provider", return_value=set_mocked_azure_provider(), - ), - mock.patch( + ), mock.patch( "prowler.providers.azure.services.monitor.monitor_diagnostic_settings_exists.monitor_diagnostic_settings_exists.monitor_client", new=monitor_client, - ), - ): - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_diagnostic_settings_exists.monitor_diagnostic_settings_exists.monitor_client", - new=monitor_client, - ), ): from prowler.providers.azure.services.monitor.monitor_diagnostic_settings_exists.monitor_diagnostic_settings_exists import ( monitor_diagnostic_settings_exists, diff --git a/tests/providers/azure/services/monitor/monitor_storage_account_with_activity_logs_cmk_encrypted/monitor_storage_account_with_activity_logs_cmk_encrypted_test.py b/tests/providers/azure/services/monitor/monitor_storage_account_with_activity_logs_cmk_encrypted/monitor_storage_account_with_activity_logs_cmk_encrypted_test.py index ea1a90638c..cc5819387b 100644 --- a/tests/providers/azure/services/monitor/monitor_storage_account_with_activity_logs_cmk_encrypted/monitor_storage_account_with_activity_logs_cmk_encrypted_test.py +++ b/tests/providers/azure/services/monitor/monitor_storage_account_with_activity_logs_cmk_encrypted/monitor_storage_account_with_activity_logs_cmk_encrypted_test.py @@ -13,15 +13,12 @@ class Test_monitor_storage_account_with_activity_logs_cmk_encrypted: monitor_client = mock.MagicMock monitor_client.diagnostics_settings = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_storage_account_with_activity_logs_cmk_encrypted.monitor_storage_account_with_activity_logs_cmk_encrypted.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_storage_account_with_activity_logs_cmk_encrypted.monitor_storage_account_with_activity_logs_cmk_encrypted.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_storage_account_with_activity_logs_cmk_encrypted.monitor_storage_account_with_activity_logs_cmk_encrypted import ( monitor_storage_account_with_activity_logs_cmk_encrypted, @@ -34,15 +31,12 @@ class Test_monitor_storage_account_with_activity_logs_cmk_encrypted: def test_no_diagnostic_settings(self): monitor_client = mock.MagicMock monitor_client.diagnostics_settings = {AZURE_SUBSCRIPTION_ID: []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_storage_account_with_activity_logs_cmk_encrypted.monitor_storage_account_with_activity_logs_cmk_encrypted.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_storage_account_with_activity_logs_cmk_encrypted.monitor_storage_account_with_activity_logs_cmk_encrypted.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_storage_account_with_activity_logs_cmk_encrypted.monitor_storage_account_with_activity_logs_cmk_encrypted import ( monitor_storage_account_with_activity_logs_cmk_encrypted, @@ -56,15 +50,12 @@ class Test_monitor_storage_account_with_activity_logs_cmk_encrypted: monitor_client = mock.MagicMock storage_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_storage_account_with_activity_logs_cmk_encrypted.monitor_storage_account_with_activity_logs_cmk_encrypted.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_storage_account_with_activity_logs_cmk_encrypted.monitor_storage_account_with_activity_logs_cmk_encrypted.monitor_client", + new=monitor_client, ): with mock.patch( "prowler.providers.azure.services.monitor.monitor_storage_account_with_activity_logs_cmk_encrypted.monitor_storage_account_with_activity_logs_cmk_encrypted.storage_client", diff --git a/tests/providers/azure/services/monitor/monitor_storage_account_with_activity_logs_is_private/monitor_storage_account_with_activity_logs_is_private_test.py b/tests/providers/azure/services/monitor/monitor_storage_account_with_activity_logs_is_private/monitor_storage_account_with_activity_logs_is_private_test.py index 2e71cae298..a9a1633584 100644 --- a/tests/providers/azure/services/monitor/monitor_storage_account_with_activity_logs_is_private/monitor_storage_account_with_activity_logs_is_private_test.py +++ b/tests/providers/azure/services/monitor/monitor_storage_account_with_activity_logs_is_private/monitor_storage_account_with_activity_logs_is_private_test.py @@ -13,15 +13,12 @@ class Test_monitor_storage_account_with_activity_logs_is_private: monitor_client = mock.MagicMock monitor_client.diagnostics_settings = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_storage_account_with_activity_logs_is_private.monitor_storage_account_with_activity_logs_is_private.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_storage_account_with_activity_logs_is_private.monitor_storage_account_with_activity_logs_is_private.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_storage_account_with_activity_logs_is_private.monitor_storage_account_with_activity_logs_is_private import ( monitor_storage_account_with_activity_logs_is_private, @@ -34,15 +31,12 @@ class Test_monitor_storage_account_with_activity_logs_is_private: def test_no_diagnostic_settings(self): monitor_client = mock.MagicMock monitor_client.diagnostics_settings = {AZURE_SUBSCRIPTION_ID: []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_storage_account_with_activity_logs_is_private.monitor_storage_account_with_activity_logs_is_private.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_storage_account_with_activity_logs_is_private.monitor_storage_account_with_activity_logs_is_private.monitor_client", + new=monitor_client, ): from prowler.providers.azure.services.monitor.monitor_storage_account_with_activity_logs_is_private.monitor_storage_account_with_activity_logs_is_private import ( monitor_storage_account_with_activity_logs_is_private, @@ -56,15 +50,12 @@ class Test_monitor_storage_account_with_activity_logs_is_private: monitor_client = mock.MagicMock storage_client = mock.MagicMock - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.monitor.monitor_storage_account_with_activity_logs_is_private.monitor_storage_account_with_activity_logs_is_private.monitor_client", - new=monitor_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.monitor.monitor_storage_account_with_activity_logs_is_private.monitor_storage_account_with_activity_logs_is_private.monitor_client", + new=monitor_client, ): with mock.patch( "prowler.providers.azure.services.monitor.monitor_storage_account_with_activity_logs_is_private.monitor_storage_account_with_activity_logs_is_private.storage_client", diff --git a/tests/providers/azure/services/mysql/mysql_flexible_server_audit_log_connection_activated/mysql_flexible_server_audit_log_connection_activated_test.py b/tests/providers/azure/services/mysql/mysql_flexible_server_audit_log_connection_activated/mysql_flexible_server_audit_log_connection_activated_test.py index 84daaa758c..7fcdfccd1e 100644 --- a/tests/providers/azure/services/mysql/mysql_flexible_server_audit_log_connection_activated/mysql_flexible_server_audit_log_connection_activated_test.py +++ b/tests/providers/azure/services/mysql/mysql_flexible_server_audit_log_connection_activated/mysql_flexible_server_audit_log_connection_activated_test.py @@ -16,15 +16,12 @@ class Test_mysql_flexible_server_audit_log_connection_activated: mysql_client = mock.MagicMock mysql_client.flexible_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.mysql.mysql_flexible_server_audit_log_connection_activated.mysql_flexible_server_audit_log_connection_activated.mysql_client", - new=mysql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.mysql.mysql_flexible_server_audit_log_connection_activated.mysql_flexible_server_audit_log_connection_activated.mysql_client", + new=mysql_client, ): from prowler.providers.azure.services.mysql.mysql_flexible_server_audit_log_connection_activated.mysql_flexible_server_audit_log_connection_activated import ( mysql_flexible_server_audit_log_connection_activated, @@ -38,15 +35,12 @@ class Test_mysql_flexible_server_audit_log_connection_activated: mysql_client = mock.MagicMock mysql_client.flexible_servers = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.mysql.mysql_flexible_server_audit_log_connection_activated.mysql_flexible_server_audit_log_connection_activated.mysql_client", - new=mysql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.mysql.mysql_flexible_server_audit_log_connection_activated.mysql_flexible_server_audit_log_connection_activated.mysql_client", + new=mysql_client, ): from prowler.providers.azure.services.mysql.mysql_flexible_server_audit_log_connection_activated.mysql_flexible_server_audit_log_connection_activated import ( mysql_flexible_server_audit_log_connection_activated, @@ -77,15 +71,12 @@ class Test_mysql_flexible_server_audit_log_connection_activated: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.mysql.mysql_flexible_server_audit_log_connection_activated.mysql_flexible_server_audit_log_connection_activated.mysql_client", - new=mysql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.mysql.mysql_flexible_server_audit_log_connection_activated.mysql_flexible_server_audit_log_connection_activated.mysql_client", + new=mysql_client, ): from prowler.providers.azure.services.mysql.mysql_flexible_server_audit_log_connection_activated.mysql_flexible_server_audit_log_connection_activated import ( mysql_flexible_server_audit_log_connection_activated, @@ -128,15 +119,12 @@ class Test_mysql_flexible_server_audit_log_connection_activated: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.mysql.mysql_flexible_server_audit_log_connection_activated.mysql_flexible_server_audit_log_connection_activated.mysql_client", - new=mysql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.mysql.mysql_flexible_server_audit_log_connection_activated.mysql_flexible_server_audit_log_connection_activated.mysql_client", + new=mysql_client, ): from prowler.providers.azure.services.mysql.mysql_flexible_server_audit_log_connection_activated.mysql_flexible_server_audit_log_connection_activated import ( mysql_flexible_server_audit_log_connection_activated, @@ -179,15 +167,12 @@ class Test_mysql_flexible_server_audit_log_connection_activated: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.mysql.mysql_flexible_server_audit_log_connection_activated.mysql_flexible_server_audit_log_connection_activated.mysql_client", - new=mysql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.mysql.mysql_flexible_server_audit_log_connection_activated.mysql_flexible_server_audit_log_connection_activated.mysql_client", + new=mysql_client, ): from prowler.providers.azure.services.mysql.mysql_flexible_server_audit_log_connection_activated.mysql_flexible_server_audit_log_connection_activated import ( mysql_flexible_server_audit_log_connection_activated, diff --git a/tests/providers/azure/services/mysql/mysql_flexible_server_minimum_tls_version_12/mysql_flexible_server_minimum_tls_version_12_test.py b/tests/providers/azure/services/mysql/mysql_flexible_server_minimum_tls_version_12/mysql_flexible_server_minimum_tls_version_12_test.py index 8d277dad77..04ed12ecc9 100644 --- a/tests/providers/azure/services/mysql/mysql_flexible_server_minimum_tls_version_12/mysql_flexible_server_minimum_tls_version_12_test.py +++ b/tests/providers/azure/services/mysql/mysql_flexible_server_minimum_tls_version_12/mysql_flexible_server_minimum_tls_version_12_test.py @@ -16,15 +16,12 @@ class Test_mysql_flexible_server_minimum_tls_version_12: mysql_client = mock.MagicMock mysql_client.flexible_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12.mysql_client", - new=mysql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12.mysql_client", + new=mysql_client, ): from prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12 import ( mysql_flexible_server_minimum_tls_version_12, @@ -38,15 +35,12 @@ class Test_mysql_flexible_server_minimum_tls_version_12: mysql_client = mock.MagicMock mysql_client.flexible_servers = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12.mysql_client", - new=mysql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12.mysql_client", + new=mysql_client, ): from prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12 import ( mysql_flexible_server_minimum_tls_version_12, @@ -70,15 +64,12 @@ class Test_mysql_flexible_server_minimum_tls_version_12: ) } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12.mysql_client", - new=mysql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12.mysql_client", + new=mysql_client, ): from prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12 import ( mysql_flexible_server_minimum_tls_version_12, @@ -117,15 +108,12 @@ class Test_mysql_flexible_server_minimum_tls_version_12: ) } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12.mysql_client", - new=mysql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12.mysql_client", + new=mysql_client, ): from prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12 import ( mysql_flexible_server_minimum_tls_version_12, @@ -167,15 +155,12 @@ class Test_mysql_flexible_server_minimum_tls_version_12: ) } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12.mysql_client", - new=mysql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12.mysql_client", + new=mysql_client, ): from prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12 import ( mysql_flexible_server_minimum_tls_version_12, @@ -217,15 +202,12 @@ class Test_mysql_flexible_server_minimum_tls_version_12: ) } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12.mysql_client", - new=mysql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12.mysql_client", + new=mysql_client, ): from prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12 import ( mysql_flexible_server_minimum_tls_version_12, @@ -267,15 +249,12 @@ class Test_mysql_flexible_server_minimum_tls_version_12: ) } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12.mysql_client", - new=mysql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12.mysql_client", + new=mysql_client, ): from prowler.providers.azure.services.mysql.mysql_flexible_server_minimum_tls_version_12.mysql_flexible_server_minimum_tls_version_12 import ( mysql_flexible_server_minimum_tls_version_12, diff --git a/tests/providers/azure/services/mysql/mysql_flexible_server_ssl_connection_enabled/mysql_flexible_server_ssl_connection_enabled_test.py b/tests/providers/azure/services/mysql/mysql_flexible_server_ssl_connection_enabled/mysql_flexible_server_ssl_connection_enabled_test.py index f540fe4865..2cdd1cf4cb 100644 --- a/tests/providers/azure/services/mysql/mysql_flexible_server_ssl_connection_enabled/mysql_flexible_server_ssl_connection_enabled_test.py +++ b/tests/providers/azure/services/mysql/mysql_flexible_server_ssl_connection_enabled/mysql_flexible_server_ssl_connection_enabled_test.py @@ -16,15 +16,12 @@ class Test_mysql_flexible_server_ssl_connection_enabled: mysql_client = mock.MagicMock mysql_client.flexible_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.mysql.mysql_flexible_server_ssl_connection_enabled.mysql_flexible_server_ssl_connection_enabled.mysql_client", - new=mysql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.mysql.mysql_flexible_server_ssl_connection_enabled.mysql_flexible_server_ssl_connection_enabled.mysql_client", + new=mysql_client, ): from prowler.providers.azure.services.mysql.mysql_flexible_server_ssl_connection_enabled.mysql_flexible_server_ssl_connection_enabled import ( mysql_flexible_server_ssl_connection_enabled, @@ -38,15 +35,12 @@ class Test_mysql_flexible_server_ssl_connection_enabled: mysql_client = mock.MagicMock mysql_client.flexible_servers = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.mysql.mysql_flexible_server_ssl_connection_enabled.mysql_flexible_server_ssl_connection_enabled.mysql_client", - new=mysql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.mysql.mysql_flexible_server_ssl_connection_enabled.mysql_flexible_server_ssl_connection_enabled.mysql_client", + new=mysql_client, ): from prowler.providers.azure.services.mysql.mysql_flexible_server_ssl_connection_enabled.mysql_flexible_server_ssl_connection_enabled import ( mysql_flexible_server_ssl_connection_enabled, @@ -77,15 +71,12 @@ class Test_mysql_flexible_server_ssl_connection_enabled: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.mysql.mysql_flexible_server_ssl_connection_enabled.mysql_flexible_server_ssl_connection_enabled.mysql_client", - new=mysql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.mysql.mysql_flexible_server_ssl_connection_enabled.mysql_flexible_server_ssl_connection_enabled.mysql_client", + new=mysql_client, ): from prowler.providers.azure.services.mysql.mysql_flexible_server_ssl_connection_enabled.mysql_flexible_server_ssl_connection_enabled import ( mysql_flexible_server_ssl_connection_enabled, @@ -128,15 +119,12 @@ class Test_mysql_flexible_server_ssl_connection_enabled: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.mysql.mysql_flexible_server_ssl_connection_enabled.mysql_flexible_server_ssl_connection_enabled.mysql_client", - new=mysql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.mysql.mysql_flexible_server_ssl_connection_enabled.mysql_flexible_server_ssl_connection_enabled.mysql_client", + new=mysql_client, ): from prowler.providers.azure.services.mysql.mysql_flexible_server_ssl_connection_enabled.mysql_flexible_server_ssl_connection_enabled import ( mysql_flexible_server_ssl_connection_enabled, @@ -173,15 +161,12 @@ class Test_mysql_flexible_server_ssl_connection_enabled: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.mysql.mysql_flexible_server_ssl_connection_enabled.mysql_flexible_server_ssl_connection_enabled.mysql_client", - new=mysql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.mysql.mysql_flexible_server_ssl_connection_enabled.mysql_flexible_server_ssl_connection_enabled.mysql_client", + new=mysql_client, ): from prowler.providers.azure.services.mysql.mysql_flexible_server_ssl_connection_enabled.mysql_flexible_server_ssl_connection_enabled import ( mysql_flexible_server_ssl_connection_enabled, @@ -235,15 +220,12 @@ class Test_mysql_flexible_server_ssl_connection_enabled: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.mysql.mysql_flexible_server_ssl_connection_enabled.mysql_flexible_server_ssl_connection_enabled.mysql_client", - new=mysql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.mysql.mysql_flexible_server_ssl_connection_enabled.mysql_flexible_server_ssl_connection_enabled.mysql_client", + new=mysql_client, ): from prowler.providers.azure.services.mysql.mysql_flexible_server_ssl_connection_enabled.mysql_flexible_server_ssl_connection_enabled import ( mysql_flexible_server_ssl_connection_enabled, diff --git a/tests/providers/azure/services/network/network_bastion_host_exists/network_bastion_host_exists_test.py b/tests/providers/azure/services/network/network_bastion_host_exists/network_bastion_host_exists_test.py index cae45b65c0..9965bb88dc 100644 --- a/tests/providers/azure/services/network/network_bastion_host_exists/network_bastion_host_exists_test.py +++ b/tests/providers/azure/services/network/network_bastion_host_exists/network_bastion_host_exists_test.py @@ -13,19 +13,15 @@ class Test_network_bastion_host_exists: network_client = mock.MagicMock network_client.bastion_hosts = {AZURE_SUBSCRIPTION_ID: []} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_bastion_host_exists.network_bastion_host_exists import ( network_bastion_host_exists, @@ -58,19 +54,15 @@ class Test_network_bastion_host_exists: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_bastion_host_exists.network_bastion_host_exists import ( network_bastion_host_exists, diff --git a/tests/providers/azure/services/network/network_flow_log_captured_sent/network_flow_log_captured_sent_test.py b/tests/providers/azure/services/network/network_flow_log_captured_sent/network_flow_log_captured_sent_test.py index 24ac2c76d5..5506fa7740 100644 --- a/tests/providers/azure/services/network/network_flow_log_captured_sent/network_flow_log_captured_sent_test.py +++ b/tests/providers/azure/services/network/network_flow_log_captured_sent/network_flow_log_captured_sent_test.py @@ -12,15 +12,12 @@ class Test_network_flow_log_captured_sent: network_client = mock.MagicMock network_client.network_watchers = {} - with ( - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_flow_log_captured_sent.network_flow_log_captured_sent import ( network_flow_log_captured_sent, @@ -46,15 +43,12 @@ class Test_network_flow_log_captured_sent: ] } - with ( - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_flow_log_captured_sent.network_flow_log_captured_sent import ( network_flow_log_captured_sent, @@ -94,15 +88,12 @@ class Test_network_flow_log_captured_sent: ] } - with ( - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_flow_log_captured_sent.network_flow_log_captured_sent import ( network_flow_log_captured_sent, @@ -142,15 +133,12 @@ class Test_network_flow_log_captured_sent: ] } - with ( - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_flow_log_captured_sent.network_flow_log_captured_sent import ( network_flow_log_captured_sent, diff --git a/tests/providers/azure/services/network/network_flow_log_more_than_90_days/network_flow_log_more_than_90_days_test.py b/tests/providers/azure/services/network/network_flow_log_more_than_90_days/network_flow_log_more_than_90_days_test.py index 44ce72a127..c27b02de79 100644 --- a/tests/providers/azure/services/network/network_flow_log_more_than_90_days/network_flow_log_more_than_90_days_test.py +++ b/tests/providers/azure/services/network/network_flow_log_more_than_90_days/network_flow_log_more_than_90_days_test.py @@ -15,19 +15,15 @@ class Test_network_flow_log_more_than_90_days: network_client = mock.MagicMock network_client.network_watchers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_flow_log_more_than_90_days.network_flow_log_more_than_90_days import ( network_flow_log_more_than_90_days, @@ -53,19 +49,15 @@ class Test_network_flow_log_more_than_90_days: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_flow_log_more_than_90_days.network_flow_log_more_than_90_days import ( network_flow_log_more_than_90_days, @@ -105,19 +97,15 @@ class Test_network_flow_log_more_than_90_days: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_flow_log_more_than_90_days.network_flow_log_more_than_90_days import ( network_flow_log_more_than_90_days, @@ -157,19 +145,15 @@ class Test_network_flow_log_more_than_90_days: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_flow_log_more_than_90_days.network_flow_log_more_than_90_days import ( network_flow_log_more_than_90_days, @@ -209,19 +193,15 @@ class Test_network_flow_log_more_than_90_days: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_flow_log_more_than_90_days.network_flow_log_more_than_90_days import ( network_flow_log_more_than_90_days, diff --git a/tests/providers/azure/services/network/network_http_internet_access_restricted/network_http_internet_access_restricted_test.py b/tests/providers/azure/services/network/network_http_internet_access_restricted/network_http_internet_access_restricted_test.py index 31cd18c2a3..082e96e1b9 100644 --- a/tests/providers/azure/services/network/network_http_internet_access_restricted/network_http_internet_access_restricted_test.py +++ b/tests/providers/azure/services/network/network_http_internet_access_restricted/network_http_internet_access_restricted_test.py @@ -15,19 +15,15 @@ class Test_network_http_internet_access_restricted: network_client = mock.MagicMock network_client.security_groups = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_http_internet_access_restricted.network_http_internet_access_restricted import ( network_http_internet_access_restricted, @@ -53,19 +49,15 @@ class Test_network_http_internet_access_restricted: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_http_internet_access_restricted.network_http_internet_access_restricted import ( network_http_internet_access_restricted, @@ -108,19 +100,15 @@ class Test_network_http_internet_access_restricted: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_http_internet_access_restricted.network_http_internet_access_restricted import ( network_http_internet_access_restricted, @@ -163,19 +151,15 @@ class Test_network_http_internet_access_restricted: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_http_internet_access_restricted.network_http_internet_access_restricted import ( network_http_internet_access_restricted, @@ -218,19 +202,15 @@ class Test_network_http_internet_access_restricted: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_http_internet_access_restricted.network_http_internet_access_restricted import ( network_http_internet_access_restricted, diff --git a/tests/providers/azure/services/network/network_public_ip_shodan/network_public_ip_shodan_test.py b/tests/providers/azure/services/network/network_public_ip_shodan/network_public_ip_shodan_test.py index a2d7f3753a..5b74ebf70c 100644 --- a/tests/providers/azure/services/network/network_public_ip_shodan/network_public_ip_shodan_test.py +++ b/tests/providers/azure/services/network/network_public_ip_shodan/network_public_ip_shodan_test.py @@ -12,19 +12,15 @@ class Test_network_public_ip_shodan: network_client = mock.MagicMock network_client.public_ip_addresses = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_public_ip_shodan.network_public_ip_shodan import ( network_public_ip_shodan, @@ -58,23 +54,18 @@ class Test_network_public_ip_shodan: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), - mock.patch( - "prowler.providers.azure.services.network.network_public_ip_shodan.network_public_ip_shodan.shodan.Shodan.host", - return_value=shodan_info, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, + ), mock.patch( + "prowler.providers.azure.services.network.network_public_ip_shodan.network_public_ip_shodan.shodan.Shodan.host", + return_value=shodan_info, ): from prowler.providers.azure.services.network.network_public_ip_shodan.network_public_ip_shodan import ( network_public_ip_shodan, diff --git a/tests/providers/azure/services/network/network_rdp_internet_access_restricted/network_rdp_internet_access_restricted_test.py b/tests/providers/azure/services/network/network_rdp_internet_access_restricted/network_rdp_internet_access_restricted_test.py index f1410e1c7c..ce03204158 100644 --- a/tests/providers/azure/services/network/network_rdp_internet_access_restricted/network_rdp_internet_access_restricted_test.py +++ b/tests/providers/azure/services/network/network_rdp_internet_access_restricted/network_rdp_internet_access_restricted_test.py @@ -15,19 +15,15 @@ class Test_network_rdp_internet_access_restricted: network_client = mock.MagicMock network_client.security_groups = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_rdp_internet_access_restricted.network_rdp_internet_access_restricted import ( network_rdp_internet_access_restricted, @@ -53,19 +49,15 @@ class Test_network_rdp_internet_access_restricted: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_rdp_internet_access_restricted.network_rdp_internet_access_restricted import ( network_rdp_internet_access_restricted, @@ -108,19 +100,15 @@ class Test_network_rdp_internet_access_restricted: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_rdp_internet_access_restricted.network_rdp_internet_access_restricted import ( network_rdp_internet_access_restricted, @@ -163,19 +151,15 @@ class Test_network_rdp_internet_access_restricted: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_rdp_internet_access_restricted.network_rdp_internet_access_restricted import ( network_rdp_internet_access_restricted, diff --git a/tests/providers/azure/services/network/network_ssh_internet_access_restricted/network_ssh_internet_access_restricted_test.py b/tests/providers/azure/services/network/network_ssh_internet_access_restricted/network_ssh_internet_access_restricted_test.py index be567f6ae3..5070e7c29b 100644 --- a/tests/providers/azure/services/network/network_ssh_internet_access_restricted/network_ssh_internet_access_restricted_test.py +++ b/tests/providers/azure/services/network/network_ssh_internet_access_restricted/network_ssh_internet_access_restricted_test.py @@ -15,19 +15,15 @@ class Test_network_ssh_internet_access_restricted: network_client = mock.MagicMock network_client.security_groups = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_ssh_internet_access_restricted.network_ssh_internet_access_restricted import ( network_ssh_internet_access_restricted, @@ -53,19 +49,15 @@ class Test_network_ssh_internet_access_restricted: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_ssh_internet_access_restricted.network_ssh_internet_access_restricted import ( network_ssh_internet_access_restricted, @@ -108,19 +100,15 @@ class Test_network_ssh_internet_access_restricted: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_ssh_internet_access_restricted.network_ssh_internet_access_restricted import ( network_ssh_internet_access_restricted, @@ -163,19 +151,15 @@ class Test_network_ssh_internet_access_restricted: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_ssh_internet_access_restricted.network_ssh_internet_access_restricted import ( network_ssh_internet_access_restricted, @@ -218,19 +202,15 @@ class Test_network_ssh_internet_access_restricted: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_ssh_internet_access_restricted.network_ssh_internet_access_restricted import ( network_ssh_internet_access_restricted, diff --git a/tests/providers/azure/services/network/network_udp_internet_access_restricted/network_udp_internet_access_restricted_test.py b/tests/providers/azure/services/network/network_udp_internet_access_restricted/network_udp_internet_access_restricted_test.py index eea305e77b..f6c7a68f13 100644 --- a/tests/providers/azure/services/network/network_udp_internet_access_restricted/network_udp_internet_access_restricted_test.py +++ b/tests/providers/azure/services/network/network_udp_internet_access_restricted/network_udp_internet_access_restricted_test.py @@ -15,19 +15,15 @@ class Test_network_udp_internet_access_restricted: network_client = mock.MagicMock network_client.security_groups = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_udp_internet_access_restricted.network_udp_internet_access_restricted import ( network_udp_internet_access_restricted, @@ -53,19 +49,15 @@ class Test_network_udp_internet_access_restricted: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_udp_internet_access_restricted.network_udp_internet_access_restricted import ( network_udp_internet_access_restricted, @@ -107,19 +99,15 @@ class Test_network_udp_internet_access_restricted: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_udp_internet_access_restricted.network_udp_internet_access_restricted import ( network_udp_internet_access_restricted, @@ -161,19 +149,15 @@ class Test_network_udp_internet_access_restricted: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_udp_internet_access_restricted.network_udp_internet_access_restricted import ( network_udp_internet_access_restricted, diff --git a/tests/providers/azure/services/network/network_watcher_enabled/network_watcher_enabled_test.py b/tests/providers/azure/services/network/network_watcher_enabled/network_watcher_enabled_test.py index c6189c5cf4..24a7f1ef4d 100644 --- a/tests/providers/azure/services/network/network_watcher_enabled/network_watcher_enabled_test.py +++ b/tests/providers/azure/services/network/network_watcher_enabled/network_watcher_enabled_test.py @@ -16,19 +16,15 @@ class Test_network_watcher_enabled: network_client.security_groups = {} network_client.network_watchers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_watcher_enabled.network_watcher_enabled import ( network_watcher_enabled, @@ -57,19 +53,15 @@ class Test_network_watcher_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_watcher_enabled.network_watcher_enabled import ( network_watcher_enabled, @@ -107,19 +99,15 @@ class Test_network_watcher_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.network.network_service.Network", - new=network_client, - ) as service_client, - mock.patch( - "prowler.providers.azure.services.network.network_client.network_client", - new=service_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.network.network_service.Network", + new=network_client, + ) as service_client, mock.patch( + "prowler.providers.azure.services.network.network_client.network_client", + new=service_client, ): from prowler.providers.azure.services.network.network_watcher_enabled.network_watcher_enabled import ( network_watcher_enabled, diff --git a/tests/providers/azure/services/postgresql/postgresql_flexible_server_allow_access_services_disabled/postgresql_flexible_server_allow_access_services_disabled_test.py b/tests/providers/azure/services/postgresql/postgresql_flexible_server_allow_access_services_disabled/postgresql_flexible_server_allow_access_services_disabled_test.py index 442371d524..12c33b2306 100644 --- a/tests/providers/azure/services/postgresql/postgresql_flexible_server_allow_access_services_disabled/postgresql_flexible_server_allow_access_services_disabled_test.py +++ b/tests/providers/azure/services/postgresql/postgresql_flexible_server_allow_access_services_disabled/postgresql_flexible_server_allow_access_services_disabled_test.py @@ -16,15 +16,12 @@ class Test_postgresql_flexible_server_allow_access_services_disabled: postgresql_client = mock.MagicMock postgresql_client.flexible_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_allow_access_services_disabled.postgresql_flexible_server_allow_access_services_disabled.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_allow_access_services_disabled.postgresql_flexible_server_allow_access_services_disabled.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_allow_access_services_disabled.postgresql_flexible_server_allow_access_services_disabled import ( postgresql_flexible_server_allow_access_services_disabled, @@ -62,15 +59,12 @@ class Test_postgresql_flexible_server_allow_access_services_disabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_allow_access_services_disabled.postgresql_flexible_server_allow_access_services_disabled.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_allow_access_services_disabled.postgresql_flexible_server_allow_access_services_disabled.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_allow_access_services_disabled.postgresql_flexible_server_allow_access_services_disabled import ( postgresql_flexible_server_allow_access_services_disabled, @@ -117,15 +111,12 @@ class Test_postgresql_flexible_server_allow_access_services_disabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_allow_access_services_disabled.postgresql_flexible_server_allow_access_services_disabled.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_allow_access_services_disabled.postgresql_flexible_server_allow_access_services_disabled.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_allow_access_services_disabled.postgresql_flexible_server_allow_access_services_disabled import ( postgresql_flexible_server_allow_access_services_disabled, diff --git a/tests/providers/azure/services/postgresql/postgresql_flexible_server_connection_throttling_on/postgresql_flexible_server_connection_throttling_on_test.py b/tests/providers/azure/services/postgresql/postgresql_flexible_server_connection_throttling_on/postgresql_flexible_server_connection_throttling_on_test.py index 160f18c8d3..0631567de1 100644 --- a/tests/providers/azure/services/postgresql/postgresql_flexible_server_connection_throttling_on/postgresql_flexible_server_connection_throttling_on_test.py +++ b/tests/providers/azure/services/postgresql/postgresql_flexible_server_connection_throttling_on/postgresql_flexible_server_connection_throttling_on_test.py @@ -13,15 +13,12 @@ class Test_postgresql_flexible_server_connection_throttling_on: postgresql_client = mock.MagicMock postgresql_client.flexible_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_connection_throttling_on.postgresql_flexible_server_connection_throttling_on.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_connection_throttling_on.postgresql_flexible_server_connection_throttling_on.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_connection_throttling_on.postgresql_flexible_server_connection_throttling_on import ( postgresql_flexible_server_connection_throttling_on, @@ -53,15 +50,12 @@ class Test_postgresql_flexible_server_connection_throttling_on: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_connection_throttling_on.postgresql_flexible_server_connection_throttling_on.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_connection_throttling_on.postgresql_flexible_server_connection_throttling_on.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_connection_throttling_on.postgresql_flexible_server_connection_throttling_on import ( postgresql_flexible_server_connection_throttling_on, @@ -102,15 +96,12 @@ class Test_postgresql_flexible_server_connection_throttling_on: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_connection_throttling_on.postgresql_flexible_server_connection_throttling_on.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_connection_throttling_on.postgresql_flexible_server_connection_throttling_on.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_connection_throttling_on.postgresql_flexible_server_connection_throttling_on import ( postgresql_flexible_server_connection_throttling_on, diff --git a/tests/providers/azure/services/postgresql/postgresql_flexible_server_enforce_ssl_enabled/postgresql_flexible_server_enforce_ssl_enabled_test.py b/tests/providers/azure/services/postgresql/postgresql_flexible_server_enforce_ssl_enabled/postgresql_flexible_server_enforce_ssl_enabled_test.py index 62013f16e3..0a019b2cca 100644 --- a/tests/providers/azure/services/postgresql/postgresql_flexible_server_enforce_ssl_enabled/postgresql_flexible_server_enforce_ssl_enabled_test.py +++ b/tests/providers/azure/services/postgresql/postgresql_flexible_server_enforce_ssl_enabled/postgresql_flexible_server_enforce_ssl_enabled_test.py @@ -13,15 +13,12 @@ class Test_postgresql_flexible_server_enforce_ssl_enabled: postgresql_client = mock.MagicMock postgresql_client.flexible_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_enforce_ssl_enabled.postgresql_flexible_server_enforce_ssl_enabled.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_enforce_ssl_enabled.postgresql_flexible_server_enforce_ssl_enabled.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_enforce_ssl_enabled.postgresql_flexible_server_enforce_ssl_enabled import ( postgresql_flexible_server_enforce_ssl_enabled, @@ -53,15 +50,12 @@ class Test_postgresql_flexible_server_enforce_ssl_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_enforce_ssl_enabled.postgresql_flexible_server_enforce_ssl_enabled.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_enforce_ssl_enabled.postgresql_flexible_server_enforce_ssl_enabled.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_enforce_ssl_enabled.postgresql_flexible_server_enforce_ssl_enabled import ( postgresql_flexible_server_enforce_ssl_enabled, @@ -102,15 +96,12 @@ class Test_postgresql_flexible_server_enforce_ssl_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_enforce_ssl_enabled.postgresql_flexible_server_enforce_ssl_enabled.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_enforce_ssl_enabled.postgresql_flexible_server_enforce_ssl_enabled.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_enforce_ssl_enabled.postgresql_flexible_server_enforce_ssl_enabled import ( postgresql_flexible_server_enforce_ssl_enabled, diff --git a/tests/providers/azure/services/postgresql/postgresql_flexible_server_log_checkpoints_on/postgresql_flexible_server_log_checkpoints_on_test.py b/tests/providers/azure/services/postgresql/postgresql_flexible_server_log_checkpoints_on/postgresql_flexible_server_log_checkpoints_on_test.py index 1d44a1db65..760dd16777 100644 --- a/tests/providers/azure/services/postgresql/postgresql_flexible_server_log_checkpoints_on/postgresql_flexible_server_log_checkpoints_on_test.py +++ b/tests/providers/azure/services/postgresql/postgresql_flexible_server_log_checkpoints_on/postgresql_flexible_server_log_checkpoints_on_test.py @@ -13,15 +13,12 @@ class Test_postgresql_flexible_server_log_checkpoints_on: postgresql_client = mock.MagicMock postgresql_client.flexible_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_checkpoints_on.postgresql_flexible_server_log_checkpoints_on.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_checkpoints_on.postgresql_flexible_server_log_checkpoints_on.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_checkpoints_on.postgresql_flexible_server_log_checkpoints_on import ( postgresql_flexible_server_log_checkpoints_on, @@ -53,15 +50,12 @@ class Test_postgresql_flexible_server_log_checkpoints_on: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_checkpoints_on.postgresql_flexible_server_log_checkpoints_on.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_checkpoints_on.postgresql_flexible_server_log_checkpoints_on.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_checkpoints_on.postgresql_flexible_server_log_checkpoints_on import ( postgresql_flexible_server_log_checkpoints_on, @@ -102,15 +96,12 @@ class Test_postgresql_flexible_server_log_checkpoints_on: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_checkpoints_on.postgresql_flexible_server_log_checkpoints_on.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_checkpoints_on.postgresql_flexible_server_log_checkpoints_on.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_checkpoints_on.postgresql_flexible_server_log_checkpoints_on import ( postgresql_flexible_server_log_checkpoints_on, diff --git a/tests/providers/azure/services/postgresql/postgresql_flexible_server_log_connections_on/postgresql_flexible_server_log_connections_on_test.py b/tests/providers/azure/services/postgresql/postgresql_flexible_server_log_connections_on/postgresql_flexible_server_log_connections_on_test.py index bc58aa0fe7..4e4a777362 100644 --- a/tests/providers/azure/services/postgresql/postgresql_flexible_server_log_connections_on/postgresql_flexible_server_log_connections_on_test.py +++ b/tests/providers/azure/services/postgresql/postgresql_flexible_server_log_connections_on/postgresql_flexible_server_log_connections_on_test.py @@ -13,15 +13,12 @@ class Test_postgresql_flexible_server_log_connections_on: postgresql_client = mock.MagicMock postgresql_client.flexible_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_connections_on.postgresql_flexible_server_log_connections_on.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_connections_on.postgresql_flexible_server_log_connections_on.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_connections_on.postgresql_flexible_server_log_connections_on import ( postgresql_flexible_server_log_connections_on, @@ -53,15 +50,12 @@ class Test_postgresql_flexible_server_log_connections_on: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_connections_on.postgresql_flexible_server_log_connections_on.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_connections_on.postgresql_flexible_server_log_connections_on.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_connections_on.postgresql_flexible_server_log_connections_on import ( postgresql_flexible_server_log_connections_on, @@ -102,15 +96,12 @@ class Test_postgresql_flexible_server_log_connections_on: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_connections_on.postgresql_flexible_server_log_connections_on.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_connections_on.postgresql_flexible_server_log_connections_on.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_connections_on.postgresql_flexible_server_log_connections_on import ( postgresql_flexible_server_log_connections_on, diff --git a/tests/providers/azure/services/postgresql/postgresql_flexible_server_log_disconnections_on/postgresql_flexible_server_log_disconnections_on_test.py b/tests/providers/azure/services/postgresql/postgresql_flexible_server_log_disconnections_on/postgresql_flexible_server_log_disconnections_on_test.py index 75932b4342..2afe5401c5 100644 --- a/tests/providers/azure/services/postgresql/postgresql_flexible_server_log_disconnections_on/postgresql_flexible_server_log_disconnections_on_test.py +++ b/tests/providers/azure/services/postgresql/postgresql_flexible_server_log_disconnections_on/postgresql_flexible_server_log_disconnections_on_test.py @@ -13,15 +13,12 @@ class Test_postgresql_flexible_server_log_disconnections_on: postgresql_client = mock.MagicMock postgresql_client.flexible_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_disconnections_on.postgresql_flexible_server_log_disconnections_on.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_disconnections_on.postgresql_flexible_server_log_disconnections_on.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_disconnections_on.postgresql_flexible_server_log_disconnections_on import ( postgresql_flexible_server_log_disconnections_on, @@ -53,15 +50,12 @@ class Test_postgresql_flexible_server_log_disconnections_on: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_disconnections_on.postgresql_flexible_server_log_disconnections_on.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_disconnections_on.postgresql_flexible_server_log_disconnections_on.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_disconnections_on.postgresql_flexible_server_log_disconnections_on import ( postgresql_flexible_server_log_disconnections_on, @@ -102,15 +96,12 @@ class Test_postgresql_flexible_server_log_disconnections_on: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_disconnections_on.postgresql_flexible_server_log_disconnections_on.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_disconnections_on.postgresql_flexible_server_log_disconnections_on.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_disconnections_on.postgresql_flexible_server_log_disconnections_on import ( postgresql_flexible_server_log_disconnections_on, diff --git a/tests/providers/azure/services/postgresql/postgresql_flexible_server_log_retention_days_greater_3/postgresql_flexible_server_log_retention_days_greater_3_test.py b/tests/providers/azure/services/postgresql/postgresql_flexible_server_log_retention_days_greater_3/postgresql_flexible_server_log_retention_days_greater_3_test.py index 2823d000ab..ecd7ab0388 100644 --- a/tests/providers/azure/services/postgresql/postgresql_flexible_server_log_retention_days_greater_3/postgresql_flexible_server_log_retention_days_greater_3_test.py +++ b/tests/providers/azure/services/postgresql/postgresql_flexible_server_log_retention_days_greater_3/postgresql_flexible_server_log_retention_days_greater_3_test.py @@ -13,15 +13,12 @@ class Test_postgresql_flexible_server_log_retention_days_greater_3: postgresql_client = mock.MagicMock postgresql_client.flexible_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_retention_days_greater_3.postgresql_flexible_server_log_retention_days_greater_3.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_retention_days_greater_3.postgresql_flexible_server_log_retention_days_greater_3.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_retention_days_greater_3.postgresql_flexible_server_log_retention_days_greater_3 import ( postgresql_flexible_server_log_retention_days_greater_3, @@ -53,15 +50,12 @@ class Test_postgresql_flexible_server_log_retention_days_greater_3: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_retention_days_greater_3.postgresql_flexible_server_log_retention_days_greater_3.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_retention_days_greater_3.postgresql_flexible_server_log_retention_days_greater_3.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_retention_days_greater_3.postgresql_flexible_server_log_retention_days_greater_3 import ( postgresql_flexible_server_log_retention_days_greater_3, @@ -103,15 +97,12 @@ class Test_postgresql_flexible_server_log_retention_days_greater_3: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_retention_days_greater_3.postgresql_flexible_server_log_retention_days_greater_3.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_retention_days_greater_3.postgresql_flexible_server_log_retention_days_greater_3.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_retention_days_greater_3.postgresql_flexible_server_log_retention_days_greater_3 import ( postgresql_flexible_server_log_retention_days_greater_3, @@ -153,15 +144,12 @@ class Test_postgresql_flexible_server_log_retention_days_greater_3: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_retention_days_greater_3.postgresql_flexible_server_log_retention_days_greater_3.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_retention_days_greater_3.postgresql_flexible_server_log_retention_days_greater_3.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_retention_days_greater_3.postgresql_flexible_server_log_retention_days_greater_3 import ( postgresql_flexible_server_log_retention_days_greater_3, @@ -203,15 +191,12 @@ class Test_postgresql_flexible_server_log_retention_days_greater_3: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_retention_days_greater_3.postgresql_flexible_server_log_retention_days_greater_3.postgresql_client", - new=postgresql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_retention_days_greater_3.postgresql_flexible_server_log_retention_days_greater_3.postgresql_client", + new=postgresql_client, ): from prowler.providers.azure.services.postgresql.postgresql_flexible_server_log_retention_days_greater_3.postgresql_flexible_server_log_retention_days_greater_3 import ( postgresql_flexible_server_log_retention_days_greater_3, diff --git a/tests/providers/azure/services/sqlserver/sqlserver_auditing_enabled/sqlserver_auditing_enabled_test.py b/tests/providers/azure/services/sqlserver/sqlserver_auditing_enabled/sqlserver_auditing_enabled_test.py index 20c18ea998..e9cb45796e 100644 --- a/tests/providers/azure/services/sqlserver/sqlserver_auditing_enabled/sqlserver_auditing_enabled_test.py +++ b/tests/providers/azure/services/sqlserver/sqlserver_auditing_enabled/sqlserver_auditing_enabled_test.py @@ -19,15 +19,12 @@ class Test_sqlserver_auditing_enabled: sqlserver_client = mock.MagicMock sqlserver_client.sql_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_auditing_enabled.sqlserver_auditing_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_auditing_enabled.sqlserver_auditing_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_auditing_enabled.sqlserver_auditing_enabled import ( sqlserver_auditing_enabled, @@ -56,15 +53,12 @@ class Test_sqlserver_auditing_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_auditing_enabled.sqlserver_auditing_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_auditing_enabled.sqlserver_auditing_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_auditing_enabled.sqlserver_auditing_enabled import ( sqlserver_auditing_enabled, @@ -102,15 +96,12 @@ class Test_sqlserver_auditing_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_auditing_enabled.sqlserver_auditing_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_auditing_enabled.sqlserver_auditing_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_auditing_enabled.sqlserver_auditing_enabled import ( sqlserver_auditing_enabled, diff --git a/tests/providers/azure/services/sqlserver/sqlserver_auditing_retention_90_days/sqlserver_auditing_retention_90_days_test.py b/tests/providers/azure/services/sqlserver/sqlserver_auditing_retention_90_days/sqlserver_auditing_retention_90_days_test.py index d74632d357..1d222d33ae 100644 --- a/tests/providers/azure/services/sqlserver/sqlserver_auditing_retention_90_days/sqlserver_auditing_retention_90_days_test.py +++ b/tests/providers/azure/services/sqlserver/sqlserver_auditing_retention_90_days/sqlserver_auditing_retention_90_days_test.py @@ -15,15 +15,12 @@ class Test_sqlserver_auditing_retention_90_days: sqlserver_client = mock.MagicMock sqlserver_client.sql_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_auditing_retention_90_days.sqlserver_auditing_retention_90_days.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_auditing_retention_90_days.sqlserver_auditing_retention_90_days.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_auditing_retention_90_days.sqlserver_auditing_retention_90_days import ( sqlserver_auditing_retention_90_days, @@ -54,15 +51,12 @@ class Test_sqlserver_auditing_retention_90_days: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_auditing_retention_90_days.sqlserver_auditing_retention_90_days.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_auditing_retention_90_days.sqlserver_auditing_retention_90_days.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_auditing_retention_90_days.sqlserver_auditing_retention_90_days import ( sqlserver_auditing_retention_90_days, @@ -104,15 +98,12 @@ class Test_sqlserver_auditing_retention_90_days: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_auditing_retention_90_days.sqlserver_auditing_retention_90_days.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_auditing_retention_90_days.sqlserver_auditing_retention_90_days.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_auditing_retention_90_days.sqlserver_auditing_retention_90_days import ( sqlserver_auditing_retention_90_days, @@ -154,15 +145,12 @@ class Test_sqlserver_auditing_retention_90_days: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_auditing_retention_90_days.sqlserver_auditing_retention_90_days.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_auditing_retention_90_days.sqlserver_auditing_retention_90_days.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_auditing_retention_90_days.sqlserver_auditing_retention_90_days import ( sqlserver_auditing_retention_90_days, @@ -207,15 +195,12 @@ class Test_sqlserver_auditing_retention_90_days: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_auditing_retention_90_days.sqlserver_auditing_retention_90_days.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_auditing_retention_90_days.sqlserver_auditing_retention_90_days.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_auditing_retention_90_days.sqlserver_auditing_retention_90_days import ( sqlserver_auditing_retention_90_days, @@ -260,15 +245,12 @@ class Test_sqlserver_auditing_retention_90_days: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_auditing_retention_90_days.sqlserver_auditing_retention_90_days.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_auditing_retention_90_days.sqlserver_auditing_retention_90_days.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_auditing_retention_90_days.sqlserver_auditing_retention_90_days import ( sqlserver_auditing_retention_90_days, diff --git a/tests/providers/azure/services/sqlserver/sqlserver_azuread_administrator_enabled/sqlserver_azuread_administrator_enabled_test.py b/tests/providers/azure/services/sqlserver/sqlserver_azuread_administrator_enabled/sqlserver_azuread_administrator_enabled_test.py index 823455d385..6688b1a09a 100644 --- a/tests/providers/azure/services/sqlserver/sqlserver_azuread_administrator_enabled/sqlserver_azuread_administrator_enabled_test.py +++ b/tests/providers/azure/services/sqlserver/sqlserver_azuread_administrator_enabled/sqlserver_azuread_administrator_enabled_test.py @@ -15,15 +15,12 @@ class Test_sqlserver_azuread_administrator_enabled: sqlserver_client = mock.MagicMock sqlserver_client.sql_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_azuread_administrator_enabled.sqlserver_azuread_administrator_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_azuread_administrator_enabled.sqlserver_azuread_administrator_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_azuread_administrator_enabled.sqlserver_azuread_administrator_enabled import ( sqlserver_azuread_administrator_enabled, @@ -52,15 +49,12 @@ class Test_sqlserver_azuread_administrator_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_azuread_administrator_enabled.sqlserver_azuread_administrator_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_azuread_administrator_enabled.sqlserver_azuread_administrator_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_azuread_administrator_enabled.sqlserver_azuread_administrator_enabled import ( sqlserver_azuread_administrator_enabled, @@ -100,15 +94,12 @@ class Test_sqlserver_azuread_administrator_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_azuread_administrator_enabled.sqlserver_azuread_administrator_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_azuread_administrator_enabled.sqlserver_azuread_administrator_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_azuread_administrator_enabled.sqlserver_azuread_administrator_enabled import ( sqlserver_azuread_administrator_enabled, @@ -148,15 +139,12 @@ class Test_sqlserver_azuread_administrator_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_azuread_administrator_enabled.sqlserver_azuread_administrator_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_azuread_administrator_enabled.sqlserver_azuread_administrator_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_azuread_administrator_enabled.sqlserver_azuread_administrator_enabled import ( sqlserver_azuread_administrator_enabled, diff --git a/tests/providers/azure/services/sqlserver/sqlserver_microsoft_defender_enabled/sqlserver_microsoft_defender_enabled_test.py b/tests/providers/azure/services/sqlserver/sqlserver_microsoft_defender_enabled/sqlserver_microsoft_defender_enabled_test.py index 41bf400be6..b2768aa9b5 100644 --- a/tests/providers/azure/services/sqlserver/sqlserver_microsoft_defender_enabled/sqlserver_microsoft_defender_enabled_test.py +++ b/tests/providers/azure/services/sqlserver/sqlserver_microsoft_defender_enabled/sqlserver_microsoft_defender_enabled_test.py @@ -15,15 +15,12 @@ class Test_sqlserver_microsoft_defender_enabled: sqlserver_client = mock.MagicMock sqlserver_client.sql_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_microsoft_defender_enabled.sqlserver_microsoft_defender_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_microsoft_defender_enabled.sqlserver_microsoft_defender_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_microsoft_defender_enabled.sqlserver_microsoft_defender_enabled import ( sqlserver_microsoft_defender_enabled, @@ -53,15 +50,12 @@ class Test_sqlserver_microsoft_defender_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_microsoft_defender_enabled.sqlserver_microsoft_defender_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_microsoft_defender_enabled.sqlserver_microsoft_defender_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_microsoft_defender_enabled.sqlserver_microsoft_defender_enabled import ( sqlserver_microsoft_defender_enabled, @@ -91,15 +85,12 @@ class Test_sqlserver_microsoft_defender_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_microsoft_defender_enabled.sqlserver_microsoft_defender_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_microsoft_defender_enabled.sqlserver_microsoft_defender_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_microsoft_defender_enabled.sqlserver_microsoft_defender_enabled import ( sqlserver_microsoft_defender_enabled, @@ -138,15 +129,12 @@ class Test_sqlserver_microsoft_defender_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_microsoft_defender_enabled.sqlserver_microsoft_defender_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_microsoft_defender_enabled.sqlserver_microsoft_defender_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_microsoft_defender_enabled.sqlserver_microsoft_defender_enabled import ( sqlserver_microsoft_defender_enabled, diff --git a/tests/providers/azure/services/sqlserver/sqlserver_recommended_minimal_tls_version/sqlserver_recommended_minimal_tls_version_test.py b/tests/providers/azure/services/sqlserver/sqlserver_recommended_minimal_tls_version/sqlserver_recommended_minimal_tls_version_test.py index 0c6a7649b5..2b80067468 100644 --- a/tests/providers/azure/services/sqlserver/sqlserver_recommended_minimal_tls_version/sqlserver_recommended_minimal_tls_version_test.py +++ b/tests/providers/azure/services/sqlserver/sqlserver_recommended_minimal_tls_version/sqlserver_recommended_minimal_tls_version_test.py @@ -18,15 +18,12 @@ class Test_sqlserver_recommended_minimal_tls_version: sqlserver_client = mock.MagicMock sqlserver_client.sql_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_recommended_minimal_tls_version.sqlserver_recommended_minimal_tls_version.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_recommended_minimal_tls_version.sqlserver_recommended_minimal_tls_version.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_recommended_minimal_tls_version.sqlserver_recommended_minimal_tls_version import ( sqlserver_recommended_minimal_tls_version, @@ -71,15 +68,12 @@ class Test_sqlserver_recommended_minimal_tls_version: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_recommended_minimal_tls_version.sqlserver_recommended_minimal_tls_version.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_recommended_minimal_tls_version.sqlserver_recommended_minimal_tls_version.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_recommended_minimal_tls_version.sqlserver_recommended_minimal_tls_version import ( sqlserver_recommended_minimal_tls_version, @@ -133,15 +127,12 @@ class Test_sqlserver_recommended_minimal_tls_version: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_recommended_minimal_tls_version.sqlserver_recommended_minimal_tls_version.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_recommended_minimal_tls_version.sqlserver_recommended_minimal_tls_version.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_recommended_minimal_tls_version.sqlserver_recommended_minimal_tls_version import ( sqlserver_recommended_minimal_tls_version, @@ -195,15 +186,12 @@ class Test_sqlserver_recommended_minimal_tls_version: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_recommended_minimal_tls_version.sqlserver_recommended_minimal_tls_version.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_recommended_minimal_tls_version.sqlserver_recommended_minimal_tls_version.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_recommended_minimal_tls_version.sqlserver_recommended_minimal_tls_version import ( sqlserver_recommended_minimal_tls_version, diff --git a/tests/providers/azure/services/sqlserver/sqlserver_tde_encrypted_with_cmk/sqlserver_tde_encrypted_with_cmk_test.py b/tests/providers/azure/services/sqlserver/sqlserver_tde_encrypted_with_cmk/sqlserver_tde_encrypted_with_cmk_test.py index 73c9046940..5a881d005d 100644 --- a/tests/providers/azure/services/sqlserver/sqlserver_tde_encrypted_with_cmk/sqlserver_tde_encrypted_with_cmk_test.py +++ b/tests/providers/azure/services/sqlserver/sqlserver_tde_encrypted_with_cmk/sqlserver_tde_encrypted_with_cmk_test.py @@ -18,15 +18,12 @@ class Test_sqlserver_tde_encrypted_with_cmk: sqlserver_client = mock.MagicMock sqlserver_client.sql_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_tde_encrypted_with_cmk.sqlserver_tde_encrypted_with_cmk.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_tde_encrypted_with_cmk.sqlserver_tde_encrypted_with_cmk.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_tde_encrypted_with_cmk.sqlserver_tde_encrypted_with_cmk import ( sqlserver_tde_encrypted_with_cmk, @@ -56,15 +53,12 @@ class Test_sqlserver_tde_encrypted_with_cmk: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_tde_encrypted_with_cmk.sqlserver_tde_encrypted_with_cmk.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_tde_encrypted_with_cmk.sqlserver_tde_encrypted_with_cmk.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_tde_encrypted_with_cmk.sqlserver_tde_encrypted_with_cmk import ( sqlserver_tde_encrypted_with_cmk, @@ -105,15 +99,12 @@ class Test_sqlserver_tde_encrypted_with_cmk: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_tde_encrypted_with_cmk.sqlserver_tde_encrypted_with_cmk.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_tde_encrypted_with_cmk.sqlserver_tde_encrypted_with_cmk.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_tde_encrypted_with_cmk.sqlserver_tde_encrypted_with_cmk import ( sqlserver_tde_encrypted_with_cmk, @@ -163,15 +154,12 @@ class Test_sqlserver_tde_encrypted_with_cmk: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_tde_encrypted_with_cmk.sqlserver_tde_encrypted_with_cmk.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_tde_encrypted_with_cmk.sqlserver_tde_encrypted_with_cmk.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_tde_encrypted_with_cmk.sqlserver_tde_encrypted_with_cmk import ( sqlserver_tde_encrypted_with_cmk, @@ -221,15 +209,12 @@ class Test_sqlserver_tde_encrypted_with_cmk: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_tde_encrypted_with_cmk.sqlserver_tde_encrypted_with_cmk.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_tde_encrypted_with_cmk.sqlserver_tde_encrypted_with_cmk.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_tde_encrypted_with_cmk.sqlserver_tde_encrypted_with_cmk import ( sqlserver_tde_encrypted_with_cmk, diff --git a/tests/providers/azure/services/sqlserver/sqlserver_tde_encryption_enabled/sqlserver_tde_encryption_enabled_test.py b/tests/providers/azure/services/sqlserver/sqlserver_tde_encryption_enabled/sqlserver_tde_encryption_enabled_test.py index ff782535e5..9ea404f8ec 100644 --- a/tests/providers/azure/services/sqlserver/sqlserver_tde_encryption_enabled/sqlserver_tde_encryption_enabled_test.py +++ b/tests/providers/azure/services/sqlserver/sqlserver_tde_encryption_enabled/sqlserver_tde_encryption_enabled_test.py @@ -18,15 +18,12 @@ class Test_sqlserver_tde_encryption_enabled: sqlserver_client = mock.MagicMock sqlserver_client.sql_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_tde_encryption_enabled.sqlserver_tde_encryption_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_tde_encryption_enabled.sqlserver_tde_encryption_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_tde_encryption_enabled.sqlserver_tde_encryption_enabled import ( sqlserver_tde_encryption_enabled, @@ -56,15 +53,12 @@ class Test_sqlserver_tde_encryption_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_tde_encryption_enabled.sqlserver_tde_encryption_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_tde_encryption_enabled.sqlserver_tde_encryption_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_tde_encryption_enabled.sqlserver_tde_encryption_enabled import ( sqlserver_tde_encryption_enabled, @@ -105,15 +99,12 @@ class Test_sqlserver_tde_encryption_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_tde_encryption_enabled.sqlserver_tde_encryption_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_tde_encryption_enabled.sqlserver_tde_encryption_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_tde_encryption_enabled.sqlserver_tde_encryption_enabled import ( sqlserver_tde_encryption_enabled, @@ -163,15 +154,12 @@ class Test_sqlserver_tde_encryption_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_tde_encryption_enabled.sqlserver_tde_encryption_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_tde_encryption_enabled.sqlserver_tde_encryption_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_tde_encryption_enabled.sqlserver_tde_encryption_enabled import ( sqlserver_tde_encryption_enabled, @@ -231,15 +219,12 @@ class Test_sqlserver_tde_encryption_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_tde_encryption_enabled.sqlserver_tde_encryption_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_tde_encryption_enabled.sqlserver_tde_encryption_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_tde_encryption_enabled.sqlserver_tde_encryption_enabled import ( sqlserver_tde_encryption_enabled, diff --git a/tests/providers/azure/services/sqlserver/sqlserver_unrestricted_inbound_access/sqlserver_unrestricted_inbound_access_test.py b/tests/providers/azure/services/sqlserver/sqlserver_unrestricted_inbound_access/sqlserver_unrestricted_inbound_access_test.py index d3c951e6b3..c8164ce9fb 100644 --- a/tests/providers/azure/services/sqlserver/sqlserver_unrestricted_inbound_access/sqlserver_unrestricted_inbound_access_test.py +++ b/tests/providers/azure/services/sqlserver/sqlserver_unrestricted_inbound_access/sqlserver_unrestricted_inbound_access_test.py @@ -15,15 +15,12 @@ class Test_sqlserver_unrestricted_inbound_access: sqlserver_client = mock.MagicMock sqlserver_client.sql_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_unrestricted_inbound_access.sqlserver_unrestricted_inbound_access.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_unrestricted_inbound_access.sqlserver_unrestricted_inbound_access.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_unrestricted_inbound_access.sqlserver_unrestricted_inbound_access import ( sqlserver_unrestricted_inbound_access, @@ -56,15 +53,12 @@ class Test_sqlserver_unrestricted_inbound_access: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_unrestricted_inbound_access.sqlserver_unrestricted_inbound_access.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_unrestricted_inbound_access.sqlserver_unrestricted_inbound_access.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_unrestricted_inbound_access.sqlserver_unrestricted_inbound_access import ( sqlserver_unrestricted_inbound_access, @@ -106,15 +100,12 @@ class Test_sqlserver_unrestricted_inbound_access: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_unrestricted_inbound_access.sqlserver_unrestricted_inbound_access.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_unrestricted_inbound_access.sqlserver_unrestricted_inbound_access.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_unrestricted_inbound_access.sqlserver_unrestricted_inbound_access import ( sqlserver_unrestricted_inbound_access, diff --git a/tests/providers/azure/services/sqlserver/sqlserver_va_emails_notifications_admins_enabled/sqlserver_va_emails_notifications_admins_enabled_test.py b/tests/providers/azure/services/sqlserver/sqlserver_va_emails_notifications_admins_enabled/sqlserver_va_emails_notifications_admins_enabled_test.py index 917c7c9651..e787393df0 100644 --- a/tests/providers/azure/services/sqlserver/sqlserver_va_emails_notifications_admins_enabled/sqlserver_va_emails_notifications_admins_enabled_test.py +++ b/tests/providers/azure/services/sqlserver/sqlserver_va_emails_notifications_admins_enabled/sqlserver_va_emails_notifications_admins_enabled_test.py @@ -18,15 +18,12 @@ class Test_sqlserver_va_emails_notifications_admins_enabled: sqlserver_client = mock.MagicMock sqlserver_client.sql_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_va_emails_notifications_admins_enabled.sqlserver_va_emails_notifications_admins_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_va_emails_notifications_admins_enabled.sqlserver_va_emails_notifications_admins_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_va_emails_notifications_admins_enabled.sqlserver_va_emails_notifications_admins_enabled import ( sqlserver_va_emails_notifications_admins_enabled, @@ -58,15 +55,12 @@ class Test_sqlserver_va_emails_notifications_admins_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_va_emails_notifications_admins_enabled.sqlserver_va_emails_notifications_admins_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_va_emails_notifications_admins_enabled.sqlserver_va_emails_notifications_admins_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_va_emails_notifications_admins_enabled.sqlserver_va_emails_notifications_admins_enabled import ( sqlserver_va_emails_notifications_admins_enabled, @@ -112,15 +106,12 @@ class Test_sqlserver_va_emails_notifications_admins_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_va_emails_notifications_admins_enabled.sqlserver_va_emails_notifications_admins_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_va_emails_notifications_admins_enabled.sqlserver_va_emails_notifications_admins_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_va_emails_notifications_admins_enabled.sqlserver_va_emails_notifications_admins_enabled import ( sqlserver_va_emails_notifications_admins_enabled, @@ -166,15 +157,12 @@ class Test_sqlserver_va_emails_notifications_admins_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_va_emails_notifications_admins_enabled.sqlserver_va_emails_notifications_admins_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_va_emails_notifications_admins_enabled.sqlserver_va_emails_notifications_admins_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_va_emails_notifications_admins_enabled.sqlserver_va_emails_notifications_admins_enabled import ( sqlserver_va_emails_notifications_admins_enabled, @@ -220,15 +208,12 @@ class Test_sqlserver_va_emails_notifications_admins_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_va_emails_notifications_admins_enabled.sqlserver_va_emails_notifications_admins_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_va_emails_notifications_admins_enabled.sqlserver_va_emails_notifications_admins_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_va_emails_notifications_admins_enabled.sqlserver_va_emails_notifications_admins_enabled import ( sqlserver_va_emails_notifications_admins_enabled, diff --git a/tests/providers/azure/services/sqlserver/sqlserver_va_periodic_recurring_scans_enabled/sqlserver_va_periodic_recurring_scans_enabled_test.py b/tests/providers/azure/services/sqlserver/sqlserver_va_periodic_recurring_scans_enabled/sqlserver_va_periodic_recurring_scans_enabled_test.py index e9af2d8b23..dcefbe087e 100644 --- a/tests/providers/azure/services/sqlserver/sqlserver_va_periodic_recurring_scans_enabled/sqlserver_va_periodic_recurring_scans_enabled_test.py +++ b/tests/providers/azure/services/sqlserver/sqlserver_va_periodic_recurring_scans_enabled/sqlserver_va_periodic_recurring_scans_enabled_test.py @@ -18,15 +18,12 @@ class Test_sqlserver_va_periodic_recurring_scans_enabled: sqlserver_client = mock.MagicMock sqlserver_client.sql_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_va_periodic_recurring_scans_enabled import ( sqlserver_va_periodic_recurring_scans_enabled, @@ -58,15 +55,12 @@ class Test_sqlserver_va_periodic_recurring_scans_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_va_periodic_recurring_scans_enabled import ( sqlserver_va_periodic_recurring_scans_enabled, @@ -109,15 +103,12 @@ class Test_sqlserver_va_periodic_recurring_scans_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_va_periodic_recurring_scans_enabled import ( sqlserver_va_periodic_recurring_scans_enabled, @@ -163,15 +154,12 @@ class Test_sqlserver_va_periodic_recurring_scans_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_va_periodic_recurring_scans_enabled import ( sqlserver_va_periodic_recurring_scans_enabled, @@ -217,15 +205,12 @@ class Test_sqlserver_va_periodic_recurring_scans_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_va_periodic_recurring_scans_enabled.sqlserver_va_periodic_recurring_scans_enabled import ( sqlserver_va_periodic_recurring_scans_enabled, diff --git a/tests/providers/azure/services/sqlserver/sqlserver_va_scan_reports_configured/sqlserver_va_scan_reports_configured_test.py b/tests/providers/azure/services/sqlserver/sqlserver_va_scan_reports_configured/sqlserver_va_scan_reports_configured_test.py index ee1c15cc68..a1cafa120d 100644 --- a/tests/providers/azure/services/sqlserver/sqlserver_va_scan_reports_configured/sqlserver_va_scan_reports_configured_test.py +++ b/tests/providers/azure/services/sqlserver/sqlserver_va_scan_reports_configured/sqlserver_va_scan_reports_configured_test.py @@ -18,15 +18,12 @@ class Test_sqlserver_va_scan_reports_configured: sqlserver_client = mock.MagicMock sqlserver_client.sql_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_va_scan_reports_configured.sqlserver_va_scan_reports_configured.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_va_scan_reports_configured.sqlserver_va_scan_reports_configured.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_va_scan_reports_configured.sqlserver_va_scan_reports_configured import ( sqlserver_va_scan_reports_configured, @@ -58,15 +55,12 @@ class Test_sqlserver_va_scan_reports_configured: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_va_scan_reports_configured.sqlserver_va_scan_reports_configured.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_va_scan_reports_configured.sqlserver_va_scan_reports_configured.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_va_scan_reports_configured.sqlserver_va_scan_reports_configured import ( sqlserver_va_scan_reports_configured, @@ -112,15 +106,12 @@ class Test_sqlserver_va_scan_reports_configured: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_va_scan_reports_configured.sqlserver_va_scan_reports_configured.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_va_scan_reports_configured.sqlserver_va_scan_reports_configured.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_va_scan_reports_configured.sqlserver_va_scan_reports_configured import ( sqlserver_va_scan_reports_configured, @@ -166,15 +157,12 @@ class Test_sqlserver_va_scan_reports_configured: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_va_scan_reports_configured.sqlserver_va_scan_reports_configured.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_va_scan_reports_configured.sqlserver_va_scan_reports_configured.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_va_scan_reports_configured.sqlserver_va_scan_reports_configured import ( sqlserver_va_scan_reports_configured, @@ -220,15 +208,12 @@ class Test_sqlserver_va_scan_reports_configured: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_va_scan_reports_configured.sqlserver_va_scan_reports_configured.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_va_scan_reports_configured.sqlserver_va_scan_reports_configured.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_va_scan_reports_configured.sqlserver_va_scan_reports_configured import ( sqlserver_va_scan_reports_configured, @@ -274,15 +259,12 @@ class Test_sqlserver_va_scan_reports_configured: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_va_scan_reports_configured.sqlserver_va_scan_reports_configured.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_va_scan_reports_configured.sqlserver_va_scan_reports_configured.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_va_scan_reports_configured.sqlserver_va_scan_reports_configured import ( sqlserver_va_scan_reports_configured, diff --git a/tests/providers/azure/services/sqlserver/sqlserver_vulnerability_assessment_enabled/sqlserver_vulnerability_assessment_enabled_test.py b/tests/providers/azure/services/sqlserver/sqlserver_vulnerability_assessment_enabled/sqlserver_vulnerability_assessment_enabled_test.py index cd0f881d0e..9ac90ebad4 100644 --- a/tests/providers/azure/services/sqlserver/sqlserver_vulnerability_assessment_enabled/sqlserver_vulnerability_assessment_enabled_test.py +++ b/tests/providers/azure/services/sqlserver/sqlserver_vulnerability_assessment_enabled/sqlserver_vulnerability_assessment_enabled_test.py @@ -22,15 +22,12 @@ class Test_sqlserver_vulnerability_assessment_enabled: sqlserver_client = mock.MagicMock sqlserver_client.sql_servers = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_vulnerability_assessment_enabled.sqlserver_vulnerability_assessment_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_vulnerability_assessment_enabled.sqlserver_vulnerability_assessment_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_vulnerability_assessment_enabled.sqlserver_vulnerability_assessment_enabled import ( sqlserver_vulnerability_assessment_enabled, @@ -72,15 +69,12 @@ class Test_sqlserver_vulnerability_assessment_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_vulnerability_assessment_enabled.sqlserver_vulnerability_assessment_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_vulnerability_assessment_enabled.sqlserver_vulnerability_assessment_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_vulnerability_assessment_enabled.sqlserver_vulnerability_assessment_enabled import ( sqlserver_vulnerability_assessment_enabled, @@ -133,15 +127,12 @@ class Test_sqlserver_vulnerability_assessment_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_vulnerability_assessment_enabled.sqlserver_vulnerability_assessment_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_vulnerability_assessment_enabled.sqlserver_vulnerability_assessment_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_vulnerability_assessment_enabled.sqlserver_vulnerability_assessment_enabled import ( sqlserver_vulnerability_assessment_enabled, @@ -194,15 +185,12 @@ class Test_sqlserver_vulnerability_assessment_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.sqlserver.sqlserver_vulnerability_assessment_enabled.sqlserver_vulnerability_assessment_enabled.sqlserver_client", - new=sqlserver_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.sqlserver.sqlserver_vulnerability_assessment_enabled.sqlserver_vulnerability_assessment_enabled.sqlserver_client", + new=sqlserver_client, ): from prowler.providers.azure.services.sqlserver.sqlserver_vulnerability_assessment_enabled.sqlserver_vulnerability_assessment_enabled import ( sqlserver_vulnerability_assessment_enabled, diff --git a/tests/providers/azure/services/storage/storage_blob_public_access_level_is_disabled/storage_blob_public_access_level_is_disabled_test.py b/tests/providers/azure/services/storage/storage_blob_public_access_level_is_disabled/storage_blob_public_access_level_is_disabled_test.py index bba18c77dd..e0f83b3e1d 100644 --- a/tests/providers/azure/services/storage/storage_blob_public_access_level_is_disabled/storage_blob_public_access_level_is_disabled_test.py +++ b/tests/providers/azure/services/storage/storage_blob_public_access_level_is_disabled/storage_blob_public_access_level_is_disabled_test.py @@ -13,15 +13,12 @@ class Test_storage_blob_public_access_level_is_disabled: storage_client = mock.MagicMock storage_client.storage_accounts = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_blob_public_access_level_is_disabled.storage_blob_public_access_level_is_disabled.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_blob_public_access_level_is_disabled.storage_blob_public_access_level_is_disabled.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_blob_public_access_level_is_disabled.storage_blob_public_access_level_is_disabled import ( storage_blob_public_access_level_is_disabled, @@ -54,15 +51,12 @@ class Test_storage_blob_public_access_level_is_disabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_blob_public_access_level_is_disabled.storage_blob_public_access_level_is_disabled.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_blob_public_access_level_is_disabled.storage_blob_public_access_level_is_disabled.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_blob_public_access_level_is_disabled.storage_blob_public_access_level_is_disabled import ( storage_blob_public_access_level_is_disabled, @@ -104,15 +98,12 @@ class Test_storage_blob_public_access_level_is_disabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_blob_public_access_level_is_disabled.storage_blob_public_access_level_is_disabled.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_blob_public_access_level_is_disabled.storage_blob_public_access_level_is_disabled.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_blob_public_access_level_is_disabled.storage_blob_public_access_level_is_disabled import ( storage_blob_public_access_level_is_disabled, diff --git a/tests/providers/azure/services/storage/storage_default_network_access_rule_is_denied/storage_default_network_access_rule_is_denied_test.py b/tests/providers/azure/services/storage/storage_default_network_access_rule_is_denied/storage_default_network_access_rule_is_denied_test.py index ad80b1e80a..1bfc123b1e 100644 --- a/tests/providers/azure/services/storage/storage_default_network_access_rule_is_denied/storage_default_network_access_rule_is_denied_test.py +++ b/tests/providers/azure/services/storage/storage_default_network_access_rule_is_denied/storage_default_network_access_rule_is_denied_test.py @@ -15,15 +15,12 @@ class Test_storage_default_network_access_rule_is_denied: storage_client = mock.MagicMock storage_client.storage_accounts = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_default_network_access_rule_is_denied.storage_default_network_access_rule_is_denied.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_default_network_access_rule_is_denied.storage_default_network_access_rule_is_denied.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_default_network_access_rule_is_denied.storage_default_network_access_rule_is_denied import ( storage_default_network_access_rule_is_denied, @@ -56,15 +53,12 @@ class Test_storage_default_network_access_rule_is_denied: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_default_network_access_rule_is_denied.storage_default_network_access_rule_is_denied.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_default_network_access_rule_is_denied.storage_default_network_access_rule_is_denied.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_default_network_access_rule_is_denied.storage_default_network_access_rule_is_denied import ( storage_default_network_access_rule_is_denied, @@ -106,15 +100,12 @@ class Test_storage_default_network_access_rule_is_denied: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_default_network_access_rule_is_denied.storage_default_network_access_rule_is_denied.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_default_network_access_rule_is_denied.storage_default_network_access_rule_is_denied.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_default_network_access_rule_is_denied.storage_default_network_access_rule_is_denied import ( storage_default_network_access_rule_is_denied, diff --git a/tests/providers/azure/services/storage/storage_ensure_azure_services_are_trusted_to_access_is_enabled/storage_ensure_azure_services_are_trusted_to_access_is_enabled_test.py b/tests/providers/azure/services/storage/storage_ensure_azure_services_are_trusted_to_access_is_enabled/storage_ensure_azure_services_are_trusted_to_access_is_enabled_test.py index 7b2780eb89..6cd58bbc00 100644 --- a/tests/providers/azure/services/storage/storage_ensure_azure_services_are_trusted_to_access_is_enabled/storage_ensure_azure_services_are_trusted_to_access_is_enabled_test.py +++ b/tests/providers/azure/services/storage/storage_ensure_azure_services_are_trusted_to_access_is_enabled/storage_ensure_azure_services_are_trusted_to_access_is_enabled_test.py @@ -15,15 +15,12 @@ class Test_storage_ensure_azure_services_are_trusted_to_access_is_enabled: storage_client = mock.MagicMock storage_client.storage_accounts = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_ensure_azure_services_are_trusted_to_access_is_enabled.storage_ensure_azure_services_are_trusted_to_access_is_enabled.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_ensure_azure_services_are_trusted_to_access_is_enabled.storage_ensure_azure_services_are_trusted_to_access_is_enabled.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_ensure_azure_services_are_trusted_to_access_is_enabled.storage_ensure_azure_services_are_trusted_to_access_is_enabled import ( storage_ensure_azure_services_are_trusted_to_access_is_enabled, @@ -56,15 +53,12 @@ class Test_storage_ensure_azure_services_are_trusted_to_access_is_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_ensure_azure_services_are_trusted_to_access_is_enabled.storage_ensure_azure_services_are_trusted_to_access_is_enabled.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_ensure_azure_services_are_trusted_to_access_is_enabled.storage_ensure_azure_services_are_trusted_to_access_is_enabled.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_ensure_azure_services_are_trusted_to_access_is_enabled.storage_ensure_azure_services_are_trusted_to_access_is_enabled import ( storage_ensure_azure_services_are_trusted_to_access_is_enabled, @@ -106,15 +100,12 @@ class Test_storage_ensure_azure_services_are_trusted_to_access_is_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_ensure_azure_services_are_trusted_to_access_is_enabled.storage_ensure_azure_services_are_trusted_to_access_is_enabled.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_ensure_azure_services_are_trusted_to_access_is_enabled.storage_ensure_azure_services_are_trusted_to_access_is_enabled.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_ensure_azure_services_are_trusted_to_access_is_enabled.storage_ensure_azure_services_are_trusted_to_access_is_enabled import ( storage_ensure_azure_services_are_trusted_to_access_is_enabled, diff --git a/tests/providers/azure/services/storage/storage_ensure_encryption_with_customer_managed_keys/storage_ensure_encryption_with_customer_managed_keys_test.py b/tests/providers/azure/services/storage/storage_ensure_encryption_with_customer_managed_keys/storage_ensure_encryption_with_customer_managed_keys_test.py index edec8a84a4..4e906703b1 100644 --- a/tests/providers/azure/services/storage/storage_ensure_encryption_with_customer_managed_keys/storage_ensure_encryption_with_customer_managed_keys_test.py +++ b/tests/providers/azure/services/storage/storage_ensure_encryption_with_customer_managed_keys/storage_ensure_encryption_with_customer_managed_keys_test.py @@ -13,15 +13,12 @@ class Test_storage_ensure_encryption_with_customer_managed_keys: storage_client = mock.MagicMock storage_client.storage_accounts = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_ensure_encryption_with_customer_managed_keys.storage_ensure_encryption_with_customer_managed_keys.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_ensure_encryption_with_customer_managed_keys.storage_ensure_encryption_with_customer_managed_keys.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_ensure_encryption_with_customer_managed_keys.storage_ensure_encryption_with_customer_managed_keys import ( storage_ensure_encryption_with_customer_managed_keys, @@ -54,15 +51,12 @@ class Test_storage_ensure_encryption_with_customer_managed_keys: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_ensure_encryption_with_customer_managed_keys.storage_ensure_encryption_with_customer_managed_keys.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_ensure_encryption_with_customer_managed_keys.storage_ensure_encryption_with_customer_managed_keys.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_ensure_encryption_with_customer_managed_keys.storage_ensure_encryption_with_customer_managed_keys import ( storage_ensure_encryption_with_customer_managed_keys, @@ -104,15 +98,12 @@ class Test_storage_ensure_encryption_with_customer_managed_keys: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_ensure_encryption_with_customer_managed_keys.storage_ensure_encryption_with_customer_managed_keys.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_ensure_encryption_with_customer_managed_keys.storage_ensure_encryption_with_customer_managed_keys.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_ensure_encryption_with_customer_managed_keys.storage_ensure_encryption_with_customer_managed_keys import ( storage_ensure_encryption_with_customer_managed_keys, diff --git a/tests/providers/azure/services/storage/storage_ensure_minimum_tls_version_12/storage_ensure_minimum_tls_version_12_test.py b/tests/providers/azure/services/storage/storage_ensure_minimum_tls_version_12/storage_ensure_minimum_tls_version_12_test.py index 0db80267ba..7671e01dac 100644 --- a/tests/providers/azure/services/storage/storage_ensure_minimum_tls_version_12/storage_ensure_minimum_tls_version_12_test.py +++ b/tests/providers/azure/services/storage/storage_ensure_minimum_tls_version_12/storage_ensure_minimum_tls_version_12_test.py @@ -13,15 +13,12 @@ class Test_storage_ensure_minimum_tls_version_12: storage_client = mock.MagicMock storage_client.storage_accounts = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_ensure_minimum_tls_version_12.storage_ensure_minimum_tls_version_12.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_ensure_minimum_tls_version_12.storage_ensure_minimum_tls_version_12.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_ensure_minimum_tls_version_12.storage_ensure_minimum_tls_version_12 import ( storage_ensure_minimum_tls_version_12, @@ -54,15 +51,12 @@ class Test_storage_ensure_minimum_tls_version_12: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_ensure_minimum_tls_version_12.storage_ensure_minimum_tls_version_12.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_ensure_minimum_tls_version_12.storage_ensure_minimum_tls_version_12.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_ensure_minimum_tls_version_12.storage_ensure_minimum_tls_version_12 import ( storage_ensure_minimum_tls_version_12, @@ -104,15 +98,12 @@ class Test_storage_ensure_minimum_tls_version_12: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_ensure_minimum_tls_version_12.storage_ensure_minimum_tls_version_12.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_ensure_minimum_tls_version_12.storage_ensure_minimum_tls_version_12.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_ensure_minimum_tls_version_12.storage_ensure_minimum_tls_version_12 import ( storage_ensure_minimum_tls_version_12, diff --git a/tests/providers/azure/services/storage/storage_ensure_private_endpoints_in_storage_accounts/storage_ensure_private_endpoints_in_storage_accounts_test.py b/tests/providers/azure/services/storage/storage_ensure_private_endpoints_in_storage_accounts/storage_ensure_private_endpoints_in_storage_accounts_test.py index a446cb2d57..58ae1e0456 100644 --- a/tests/providers/azure/services/storage/storage_ensure_private_endpoints_in_storage_accounts/storage_ensure_private_endpoints_in_storage_accounts_test.py +++ b/tests/providers/azure/services/storage/storage_ensure_private_endpoints_in_storage_accounts/storage_ensure_private_endpoints_in_storage_accounts_test.py @@ -15,15 +15,12 @@ class Test_storage_ensure_private_endpoints_in_storage_accounts: storage_client = mock.MagicMock storage_client.storage_accounts = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_ensure_private_endpoints_in_storage_accounts.storage_ensure_private_endpoints_in_storage_accounts.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_ensure_private_endpoints_in_storage_accounts.storage_ensure_private_endpoints_in_storage_accounts.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_ensure_private_endpoints_in_storage_accounts.storage_ensure_private_endpoints_in_storage_accounts import ( storage_ensure_private_endpoints_in_storage_accounts, @@ -58,15 +55,12 @@ class Test_storage_ensure_private_endpoints_in_storage_accounts: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_ensure_private_endpoints_in_storage_accounts.storage_ensure_private_endpoints_in_storage_accounts.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_ensure_private_endpoints_in_storage_accounts.storage_ensure_private_endpoints_in_storage_accounts.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_ensure_private_endpoints_in_storage_accounts.storage_ensure_private_endpoints_in_storage_accounts import ( storage_ensure_private_endpoints_in_storage_accounts, @@ -110,15 +104,12 @@ class Test_storage_ensure_private_endpoints_in_storage_accounts: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_ensure_private_endpoints_in_storage_accounts.storage_ensure_private_endpoints_in_storage_accounts.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_ensure_private_endpoints_in_storage_accounts.storage_ensure_private_endpoints_in_storage_accounts.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_ensure_private_endpoints_in_storage_accounts.storage_ensure_private_endpoints_in_storage_accounts import ( storage_ensure_private_endpoints_in_storage_accounts, diff --git a/tests/providers/azure/services/storage/storage_ensure_soft_delete_is_enabled/storage_ensure_soft_delete_is_enabled_test.py b/tests/providers/azure/services/storage/storage_ensure_soft_delete_is_enabled/storage_ensure_soft_delete_is_enabled_test.py index 6992a523c9..c048251706 100644 --- a/tests/providers/azure/services/storage/storage_ensure_soft_delete_is_enabled/storage_ensure_soft_delete_is_enabled_test.py +++ b/tests/providers/azure/services/storage/storage_ensure_soft_delete_is_enabled/storage_ensure_soft_delete_is_enabled_test.py @@ -18,15 +18,12 @@ class Test_storage_ensure_soft_delete_is_enabled: storage_client = mock.MagicMock storage_client.storage_accounts = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_ensure_soft_delete_is_enabled.storage_ensure_soft_delete_is_enabled.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_ensure_soft_delete_is_enabled.storage_ensure_soft_delete_is_enabled.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_ensure_soft_delete_is_enabled.storage_ensure_soft_delete_is_enabled import ( storage_ensure_soft_delete_is_enabled, @@ -61,15 +58,12 @@ class Test_storage_ensure_soft_delete_is_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_ensure_soft_delete_is_enabled.storage_ensure_soft_delete_is_enabled.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_ensure_soft_delete_is_enabled.storage_ensure_soft_delete_is_enabled.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_ensure_soft_delete_is_enabled.storage_ensure_soft_delete_is_enabled import ( storage_ensure_soft_delete_is_enabled, @@ -112,15 +106,12 @@ class Test_storage_ensure_soft_delete_is_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_ensure_soft_delete_is_enabled.storage_ensure_soft_delete_is_enabled.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_ensure_soft_delete_is_enabled.storage_ensure_soft_delete_is_enabled.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_ensure_soft_delete_is_enabled.storage_ensure_soft_delete_is_enabled import ( storage_ensure_soft_delete_is_enabled, @@ -172,15 +163,12 @@ class Test_storage_ensure_soft_delete_is_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_ensure_soft_delete_is_enabled.storage_ensure_soft_delete_is_enabled.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_ensure_soft_delete_is_enabled.storage_ensure_soft_delete_is_enabled.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_ensure_soft_delete_is_enabled.storage_ensure_soft_delete_is_enabled import ( storage_ensure_soft_delete_is_enabled, diff --git a/tests/providers/azure/services/storage/storage_infrastructure_encryption_is_enabled/storage_infrastructure_encryption_is_enabled_test.py b/tests/providers/azure/services/storage/storage_infrastructure_encryption_is_enabled/storage_infrastructure_encryption_is_enabled_test.py index c354a92ded..9310f6638d 100644 --- a/tests/providers/azure/services/storage/storage_infrastructure_encryption_is_enabled/storage_infrastructure_encryption_is_enabled_test.py +++ b/tests/providers/azure/services/storage/storage_infrastructure_encryption_is_enabled/storage_infrastructure_encryption_is_enabled_test.py @@ -13,15 +13,12 @@ class Test_storage_infrastructure_encryption_is_enabled: storage_client = mock.MagicMock storage_client.storage_accounts = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_infrastructure_encryption_is_enabled.storage_infrastructure_encryption_is_enabled.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_infrastructure_encryption_is_enabled.storage_infrastructure_encryption_is_enabled.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_infrastructure_encryption_is_enabled.storage_infrastructure_encryption_is_enabled import ( storage_infrastructure_encryption_is_enabled, @@ -54,15 +51,12 @@ class Test_storage_infrastructure_encryption_is_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_infrastructure_encryption_is_enabled.storage_infrastructure_encryption_is_enabled.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_infrastructure_encryption_is_enabled.storage_infrastructure_encryption_is_enabled.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_infrastructure_encryption_is_enabled.storage_infrastructure_encryption_is_enabled import ( storage_infrastructure_encryption_is_enabled, @@ -104,15 +98,12 @@ class Test_storage_infrastructure_encryption_is_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_infrastructure_encryption_is_enabled.storage_infrastructure_encryption_is_enabled.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_infrastructure_encryption_is_enabled.storage_infrastructure_encryption_is_enabled.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_infrastructure_encryption_is_enabled.storage_infrastructure_encryption_is_enabled import ( storage_infrastructure_encryption_is_enabled, diff --git a/tests/providers/azure/services/storage/storage_key_rotation_90_days/storage_key_rotation_90_days_test.py b/tests/providers/azure/services/storage/storage_key_rotation_90_days/storage_key_rotation_90_days_test.py index c2c9d03e82..f5f1c2d207 100644 --- a/tests/providers/azure/services/storage/storage_key_rotation_90_days/storage_key_rotation_90_days_test.py +++ b/tests/providers/azure/services/storage/storage_key_rotation_90_days/storage_key_rotation_90_days_test.py @@ -13,15 +13,12 @@ class Test_storage_key_rotation_90_dayss: storage_client = mock.MagicMock storage_client.storage_accounts = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_key_rotation_90_days.storage_key_rotation_90_days.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_key_rotation_90_days.storage_key_rotation_90_days.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_key_rotation_90_days.storage_key_rotation_90_days import ( storage_key_rotation_90_days, @@ -55,15 +52,12 @@ class Test_storage_key_rotation_90_dayss: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_key_rotation_90_days.storage_key_rotation_90_days.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_key_rotation_90_days.storage_key_rotation_90_days.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_key_rotation_90_days.storage_key_rotation_90_days import ( storage_key_rotation_90_days, @@ -106,15 +100,12 @@ class Test_storage_key_rotation_90_dayss: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_key_rotation_90_days.storage_key_rotation_90_days.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_key_rotation_90_days.storage_key_rotation_90_days.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_key_rotation_90_days.storage_key_rotation_90_days import ( storage_key_rotation_90_days, @@ -156,15 +147,12 @@ class Test_storage_key_rotation_90_dayss: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_key_rotation_90_days.storage_key_rotation_90_days.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_key_rotation_90_days.storage_key_rotation_90_days.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_key_rotation_90_days.storage_key_rotation_90_days import ( storage_key_rotation_90_days, diff --git a/tests/providers/azure/services/storage/storage_secure_transfer_required_is_enabled/storage_secure_transfer_required_is_enabled_test.py b/tests/providers/azure/services/storage/storage_secure_transfer_required_is_enabled/storage_secure_transfer_required_is_enabled_test.py index d7a4f469e2..fc70988b8f 100644 --- a/tests/providers/azure/services/storage/storage_secure_transfer_required_is_enabled/storage_secure_transfer_required_is_enabled_test.py +++ b/tests/providers/azure/services/storage/storage_secure_transfer_required_is_enabled/storage_secure_transfer_required_is_enabled_test.py @@ -13,15 +13,12 @@ class Test_storage_secure_transfer_required_is_enabled: storage_client = mock.MagicMock storage_client.storage_accounts = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_secure_transfer_required_is_enabled.storage_secure_transfer_required_is_enabled.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_secure_transfer_required_is_enabled.storage_secure_transfer_required_is_enabled.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_secure_transfer_required_is_enabled.storage_secure_transfer_required_is_enabled import ( storage_secure_transfer_required_is_enabled, @@ -54,15 +51,12 @@ class Test_storage_secure_transfer_required_is_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_secure_transfer_required_is_enabled.storage_secure_transfer_required_is_enabled.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_secure_transfer_required_is_enabled.storage_secure_transfer_required_is_enabled.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_secure_transfer_required_is_enabled.storage_secure_transfer_required_is_enabled import ( storage_secure_transfer_required_is_enabled, @@ -104,15 +98,12 @@ class Test_storage_secure_transfer_required_is_enabled: ] } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.storage.storage_secure_transfer_required_is_enabled.storage_secure_transfer_required_is_enabled.storage_client", - new=storage_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.storage.storage_secure_transfer_required_is_enabled.storage_secure_transfer_required_is_enabled.storage_client", + new=storage_client, ): from prowler.providers.azure.services.storage.storage_secure_transfer_required_is_enabled.storage_secure_transfer_required_is_enabled import ( storage_secure_transfer_required_is_enabled, diff --git a/tests/providers/azure/services/vm/vm_ensure_attached_disks_encrypted_with_cmk/vm_ensure_attached_disks_encrypted_with_cmk_test.py b/tests/providers/azure/services/vm/vm_ensure_attached_disks_encrypted_with_cmk/vm_ensure_attached_disks_encrypted_with_cmk_test.py index ccf6e8f175..c4cfb11161 100644 --- a/tests/providers/azure/services/vm/vm_ensure_attached_disks_encrypted_with_cmk/vm_ensure_attached_disks_encrypted_with_cmk_test.py +++ b/tests/providers/azure/services/vm/vm_ensure_attached_disks_encrypted_with_cmk/vm_ensure_attached_disks_encrypted_with_cmk_test.py @@ -13,15 +13,12 @@ class Test_vm_ensure_attached_disks_encrypted_with_cmk: vm_client = mock.MagicMock vm_client.disks = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.vm.vm_ensure_attached_disks_encrypted_with_cmk.vm_ensure_attached_disks_encrypted_with_cmk.vm_client", - new=vm_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.vm.vm_ensure_attached_disks_encrypted_with_cmk.vm_ensure_attached_disks_encrypted_with_cmk.vm_client", + new=vm_client, ): from prowler.providers.azure.services.vm.vm_ensure_attached_disks_encrypted_with_cmk.vm_ensure_attached_disks_encrypted_with_cmk import ( vm_ensure_attached_disks_encrypted_with_cmk, @@ -35,15 +32,12 @@ class Test_vm_ensure_attached_disks_encrypted_with_cmk: vm_client = mock.MagicMock vm_client.disks = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.vm.vm_ensure_attached_disks_encrypted_with_cmk.vm_ensure_attached_disks_encrypted_with_cmk.vm_client", - new=vm_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.vm.vm_ensure_attached_disks_encrypted_with_cmk.vm_ensure_attached_disks_encrypted_with_cmk.vm_client", + new=vm_client, ): from prowler.providers.azure.services.vm.vm_ensure_attached_disks_encrypted_with_cmk.vm_ensure_attached_disks_encrypted_with_cmk import ( vm_ensure_attached_disks_encrypted_with_cmk, @@ -69,15 +63,12 @@ class Test_vm_ensure_attached_disks_encrypted_with_cmk: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.vm.vm_ensure_attached_disks_encrypted_with_cmk.vm_ensure_attached_disks_encrypted_with_cmk.vm_client", - new=vm_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.vm.vm_ensure_attached_disks_encrypted_with_cmk.vm_ensure_attached_disks_encrypted_with_cmk.vm_client", + new=vm_client, ): from prowler.providers.azure.services.vm.vm_ensure_attached_disks_encrypted_with_cmk.vm_ensure_attached_disks_encrypted_with_cmk import ( vm_ensure_attached_disks_encrypted_with_cmk, @@ -112,15 +103,12 @@ class Test_vm_ensure_attached_disks_encrypted_with_cmk: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.vm.vm_ensure_attached_disks_encrypted_with_cmk.vm_ensure_attached_disks_encrypted_with_cmk.vm_client", - new=vm_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.vm.vm_ensure_attached_disks_encrypted_with_cmk.vm_ensure_attached_disks_encrypted_with_cmk.vm_client", + new=vm_client, ): from prowler.providers.azure.services.vm.vm_ensure_attached_disks_encrypted_with_cmk.vm_ensure_attached_disks_encrypted_with_cmk import ( vm_ensure_attached_disks_encrypted_with_cmk, @@ -164,15 +152,12 @@ class Test_vm_ensure_attached_disks_encrypted_with_cmk: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.vm.vm_ensure_attached_disks_encrypted_with_cmk.vm_ensure_attached_disks_encrypted_with_cmk.vm_client", - new=vm_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.vm.vm_ensure_attached_disks_encrypted_with_cmk.vm_ensure_attached_disks_encrypted_with_cmk.vm_client", + new=vm_client, ): from prowler.providers.azure.services.vm.vm_ensure_attached_disks_encrypted_with_cmk.vm_ensure_attached_disks_encrypted_with_cmk import ( vm_ensure_attached_disks_encrypted_with_cmk, @@ -215,15 +200,12 @@ class Test_vm_ensure_attached_disks_encrypted_with_cmk: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.vm.vm_ensure_attached_disks_encrypted_with_cmk.vm_ensure_attached_disks_encrypted_with_cmk.vm_client", - new=vm_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.vm.vm_ensure_attached_disks_encrypted_with_cmk.vm_ensure_attached_disks_encrypted_with_cmk.vm_client", + new=vm_client, ): from prowler.providers.azure.services.vm.vm_ensure_attached_disks_encrypted_with_cmk.vm_ensure_attached_disks_encrypted_with_cmk import ( vm_ensure_attached_disks_encrypted_with_cmk, diff --git a/tests/providers/azure/services/vm/vm_ensure_unattached_disks_encrypted_with_cmk/vm_ensure_unattached_disks_encrypted_with_cmk_test.py b/tests/providers/azure/services/vm/vm_ensure_unattached_disks_encrypted_with_cmk/vm_ensure_unattached_disks_encrypted_with_cmk_test.py index 44ec78b4ed..f4e861c89d 100644 --- a/tests/providers/azure/services/vm/vm_ensure_unattached_disks_encrypted_with_cmk/vm_ensure_unattached_disks_encrypted_with_cmk_test.py +++ b/tests/providers/azure/services/vm/vm_ensure_unattached_disks_encrypted_with_cmk/vm_ensure_unattached_disks_encrypted_with_cmk_test.py @@ -13,15 +13,12 @@ class Test_vm_ensure_unattached_disks_encrypted_with_cmk: vm_client = mock.MagicMock vm_client.disks = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.vm.vm_ensure_unattached_disks_encrypted_with_cmk.vm_ensure_unattached_disks_encrypted_with_cmk.vm_client", - new=vm_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.vm.vm_ensure_unattached_disks_encrypted_with_cmk.vm_ensure_unattached_disks_encrypted_with_cmk.vm_client", + new=vm_client, ): from prowler.providers.azure.services.vm.vm_ensure_unattached_disks_encrypted_with_cmk.vm_ensure_unattached_disks_encrypted_with_cmk import ( vm_ensure_unattached_disks_encrypted_with_cmk, @@ -35,15 +32,12 @@ class Test_vm_ensure_unattached_disks_encrypted_with_cmk: vm_client = mock.MagicMock vm_client.disks = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.vm.vm_ensure_unattached_disks_encrypted_with_cmk.vm_ensure_unattached_disks_encrypted_with_cmk.vm_client", - new=vm_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.vm.vm_ensure_unattached_disks_encrypted_with_cmk.vm_ensure_unattached_disks_encrypted_with_cmk.vm_client", + new=vm_client, ): from prowler.providers.azure.services.vm.vm_ensure_unattached_disks_encrypted_with_cmk.vm_ensure_unattached_disks_encrypted_with_cmk import ( vm_ensure_unattached_disks_encrypted_with_cmk, @@ -69,15 +63,12 @@ class Test_vm_ensure_unattached_disks_encrypted_with_cmk: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.vm.vm_ensure_unattached_disks_encrypted_with_cmk.vm_ensure_unattached_disks_encrypted_with_cmk.vm_client", - new=vm_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.vm.vm_ensure_unattached_disks_encrypted_with_cmk.vm_ensure_unattached_disks_encrypted_with_cmk.vm_client", + new=vm_client, ): from prowler.providers.azure.services.vm.vm_ensure_unattached_disks_encrypted_with_cmk.vm_ensure_unattached_disks_encrypted_with_cmk import ( vm_ensure_unattached_disks_encrypted_with_cmk, @@ -112,15 +103,12 @@ class Test_vm_ensure_unattached_disks_encrypted_with_cmk: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.vm.vm_ensure_unattached_disks_encrypted_with_cmk.vm_ensure_unattached_disks_encrypted_with_cmk.vm_client", - new=vm_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.vm.vm_ensure_unattached_disks_encrypted_with_cmk.vm_ensure_unattached_disks_encrypted_with_cmk.vm_client", + new=vm_client, ): from prowler.providers.azure.services.vm.vm_ensure_unattached_disks_encrypted_with_cmk.vm_ensure_unattached_disks_encrypted_with_cmk import ( vm_ensure_unattached_disks_encrypted_with_cmk, @@ -164,15 +152,12 @@ class Test_vm_ensure_unattached_disks_encrypted_with_cmk: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.vm.vm_ensure_unattached_disks_encrypted_with_cmk.vm_ensure_unattached_disks_encrypted_with_cmk.vm_client", - new=vm_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.vm.vm_ensure_unattached_disks_encrypted_with_cmk.vm_ensure_unattached_disks_encrypted_with_cmk.vm_client", + new=vm_client, ): from prowler.providers.azure.services.vm.vm_ensure_unattached_disks_encrypted_with_cmk.vm_ensure_unattached_disks_encrypted_with_cmk import ( vm_ensure_unattached_disks_encrypted_with_cmk, @@ -215,15 +200,12 @@ class Test_vm_ensure_unattached_disks_encrypted_with_cmk: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.vm.vm_ensure_unattached_disks_encrypted_with_cmk.vm_ensure_unattached_disks_encrypted_with_cmk.vm_client", - new=vm_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.vm.vm_ensure_unattached_disks_encrypted_with_cmk.vm_ensure_unattached_disks_encrypted_with_cmk.vm_client", + new=vm_client, ): from prowler.providers.azure.services.vm.vm_ensure_unattached_disks_encrypted_with_cmk.vm_ensure_unattached_disks_encrypted_with_cmk import ( vm_ensure_unattached_disks_encrypted_with_cmk, diff --git a/tests/providers/azure/services/vm/vm_ensure_using_managed_disks/vm_ensure_using_managed_disks_test.py b/tests/providers/azure/services/vm/vm_ensure_using_managed_disks/vm_ensure_using_managed_disks_test.py index dbc04dd53f..de9b2ddd54 100644 --- a/tests/providers/azure/services/vm/vm_ensure_using_managed_disks/vm_ensure_using_managed_disks_test.py +++ b/tests/providers/azure/services/vm/vm_ensure_using_managed_disks/vm_ensure_using_managed_disks_test.py @@ -13,15 +13,12 @@ class Test_vm_ensure_using_managed_disks: vm_client = mock.MagicMock vm_client.virtual_machines = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.vm.vm_ensure_using_managed_disks.vm_ensure_using_managed_disks.vm_client", - new=vm_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.vm.vm_ensure_using_managed_disks.vm_ensure_using_managed_disks.vm_client", + new=vm_client, ): from prowler.providers.azure.services.vm.vm_ensure_using_managed_disks.vm_ensure_using_managed_disks import ( vm_ensure_using_managed_disks, @@ -35,15 +32,12 @@ class Test_vm_ensure_using_managed_disks: vm_client = mock.MagicMock vm_client.virtual_machines = {AZURE_SUBSCRIPTION_ID: {}} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.vm.vm_ensure_using_managed_disks.vm_ensure_using_managed_disks.vm_client", - new=vm_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.vm.vm_ensure_using_managed_disks.vm_ensure_using_managed_disks.vm_client", + new=vm_client, ): from prowler.providers.azure.services.vm.vm_ensure_using_managed_disks.vm_ensure_using_managed_disks import ( vm_ensure_using_managed_disks, @@ -81,15 +75,12 @@ class Test_vm_ensure_using_managed_disks: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.vm.vm_ensure_using_managed_disks.vm_ensure_using_managed_disks.vm_client", - new=vm_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.vm.vm_ensure_using_managed_disks.vm_ensure_using_managed_disks.vm_client", + new=vm_client, ): from prowler.providers.azure.services.vm.vm_ensure_using_managed_disks.vm_ensure_using_managed_disks import ( vm_ensure_using_managed_disks, @@ -136,15 +127,12 @@ class Test_vm_ensure_using_managed_disks: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.vm.vm_ensure_using_managed_disks.vm_ensure_using_managed_disks.vm_client", - new=vm_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.vm.vm_ensure_using_managed_disks.vm_ensure_using_managed_disks.vm_client", + new=vm_client, ): from prowler.providers.azure.services.vm.vm_ensure_using_managed_disks.vm_ensure_using_managed_disks import ( vm_ensure_using_managed_disks, @@ -191,15 +179,12 @@ class Test_vm_ensure_using_managed_disks: } } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_azure_provider(), - ), - mock.patch( - "prowler.providers.azure.services.vm.vm_ensure_using_managed_disks.vm_ensure_using_managed_disks.vm_client", - new=vm_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_azure_provider(), + ), mock.patch( + "prowler.providers.azure.services.vm.vm_ensure_using_managed_disks.vm_ensure_using_managed_disks.vm_client", + new=vm_client, ): from prowler.providers.azure.services.vm.vm_ensure_using_managed_disks.vm_ensure_using_managed_disks import ( vm_ensure_using_managed_disks, diff --git a/tests/providers/gcp/services/apikeys/apikeys_api_restrictions_configured/apikeys_api_restrictions_configured_test.py b/tests/providers/gcp/services/apikeys/apikeys_api_restrictions_configured/apikeys_api_restrictions_configured_test.py index 9427199486..40a9ac052f 100644 --- a/tests/providers/gcp/services/apikeys/apikeys_api_restrictions_configured/apikeys_api_restrictions_configured_test.py +++ b/tests/providers/gcp/services/apikeys/apikeys_api_restrictions_configured/apikeys_api_restrictions_configured_test.py @@ -9,15 +9,12 @@ class Test_apikeys_api_restrictions_configured: apikeys_client = mock.MagicMock() apikeys_client.keys = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.apikeys.apikeys_api_restrictions_configured.apikeys_api_restrictions_configured.apikeys_client", - new=apikeys_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.apikeys.apikeys_api_restrictions_configured.apikeys_api_restrictions_configured.apikeys_client", + new=apikeys_client, ): from prowler.providers.gcp.services.apikeys.apikeys_api_restrictions_configured.apikeys_api_restrictions_configured import ( apikeys_api_restrictions_configured, @@ -48,15 +45,12 @@ class Test_apikeys_api_restrictions_configured: apikeys_client.keys = [key] apikeys_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.apikeys.apikeys_api_restrictions_configured.apikeys_api_restrictions_configured.apikeys_client", - new=apikeys_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.apikeys.apikeys_api_restrictions_configured.apikeys_api_restrictions_configured.apikeys_client", + new=apikeys_client, ): from prowler.providers.gcp.services.apikeys.apikeys_api_restrictions_configured.apikeys_api_restrictions_configured import ( apikeys_api_restrictions_configured, @@ -89,15 +83,12 @@ class Test_apikeys_api_restrictions_configured: apikeys_client.keys = [key] apikeys_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.apikeys.apikeys_api_restrictions_configured.apikeys_api_restrictions_configured.apikeys_client", - new=apikeys_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.apikeys.apikeys_api_restrictions_configured.apikeys_api_restrictions_configured.apikeys_client", + new=apikeys_client, ): from prowler.providers.gcp.services.apikeys.apikeys_api_restrictions_configured.apikeys_api_restrictions_configured import ( apikeys_api_restrictions_configured, @@ -136,15 +127,12 @@ class Test_apikeys_api_restrictions_configured: apikeys_client.keys = [key] apikeys_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.apikeys.apikeys_api_restrictions_configured.apikeys_api_restrictions_configured.apikeys_client", - new=apikeys_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.apikeys.apikeys_api_restrictions_configured.apikeys_api_restrictions_configured.apikeys_client", + new=apikeys_client, ): from prowler.providers.gcp.services.apikeys.apikeys_api_restrictions_configured.apikeys_api_restrictions_configured import ( apikeys_api_restrictions_configured, diff --git a/tests/providers/gcp/services/apikeys/apikeys_key_exists/apikeys_key_exists_test.py b/tests/providers/gcp/services/apikeys/apikeys_key_exists/apikeys_key_exists_test.py index 3550353b04..211d5822ce 100644 --- a/tests/providers/gcp/services/apikeys/apikeys_key_exists/apikeys_key_exists_test.py +++ b/tests/providers/gcp/services/apikeys/apikeys_key_exists/apikeys_key_exists_test.py @@ -21,15 +21,12 @@ class Test_apikeys_key_exists: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.apikeys.apikeys_key_exists.apikeys_key_exists.apikeys_client", - new=apikeys_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.apikeys.apikeys_key_exists.apikeys_key_exists.apikeys_client", + new=apikeys_client, ): from prowler.providers.gcp.services.apikeys.apikeys_key_exists.apikeys_key_exists import ( apikeys_key_exists, @@ -70,15 +67,12 @@ class Test_apikeys_key_exists: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.apikeys.apikeys_key_exists.apikeys_key_exists.apikeys_client", - new=apikeys_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.apikeys.apikeys_key_exists.apikeys_key_exists.apikeys_client", + new=apikeys_client, ): from prowler.providers.gcp.services.apikeys.apikeys_key_exists.apikeys_key_exists import ( apikeys_key_exists, diff --git a/tests/providers/gcp/services/apikeys/apikeys_key_rotated_in_90_days/apikeys_key_rotated_in_90_days_test.py b/tests/providers/gcp/services/apikeys/apikeys_key_rotated_in_90_days/apikeys_key_rotated_in_90_days_test.py index 9734cf18db..e7547a0e64 100644 --- a/tests/providers/gcp/services/apikeys/apikeys_key_rotated_in_90_days/apikeys_key_rotated_in_90_days_test.py +++ b/tests/providers/gcp/services/apikeys/apikeys_key_rotated_in_90_days/apikeys_key_rotated_in_90_days_test.py @@ -10,15 +10,12 @@ class Test_apikeys_key_rotated_in_90_days: apikeys_client = mock.MagicMock() apikeys_client.keys = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.apikeys.apikeys_key_rotated_in_90_days.apikeys_key_rotated_in_90_days.apikeys_client", - new=apikeys_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.apikeys.apikeys_key_rotated_in_90_days.apikeys_key_rotated_in_90_days.apikeys_client", + new=apikeys_client, ): from prowler.providers.gcp.services.apikeys.apikeys_key_rotated_in_90_days.apikeys_key_rotated_in_90_days import ( apikeys_key_rotated_in_90_days, @@ -46,15 +43,12 @@ class Test_apikeys_key_rotated_in_90_days: apikeys_client.keys = [key] apikeys_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.apikeys.apikeys_key_rotated_in_90_days.apikeys_key_rotated_in_90_days.apikeys_client", - new=apikeys_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.apikeys.apikeys_key_rotated_in_90_days.apikeys_key_rotated_in_90_days.apikeys_client", + new=apikeys_client, ): from prowler.providers.gcp.services.apikeys.apikeys_key_rotated_in_90_days.apikeys_key_rotated_in_90_days import ( apikeys_key_rotated_in_90_days, @@ -89,15 +83,12 @@ class Test_apikeys_key_rotated_in_90_days: apikeys_client.keys = [key] apikeys_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.apikeys.apikeys_key_rotated_in_90_days.apikeys_key_rotated_in_90_days.apikeys_client", - new=apikeys_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.apikeys.apikeys_key_rotated_in_90_days.apikeys_key_rotated_in_90_days.apikeys_client", + new=apikeys_client, ): from prowler.providers.gcp.services.apikeys.apikeys_key_rotated_in_90_days.apikeys_key_rotated_in_90_days import ( apikeys_key_rotated_in_90_days, diff --git a/tests/providers/gcp/services/apikeys/apikeys_service_test.py b/tests/providers/gcp/services/apikeys/apikeys_service_test.py index 1c81610a40..494d997be0 100644 --- a/tests/providers/gcp/services/apikeys/apikeys_service_test.py +++ b/tests/providers/gcp/services/apikeys/apikeys_service_test.py @@ -11,15 +11,12 @@ from tests.providers.gcp.gcp_fixtures import ( class TestAPIKeysService: def test_service(self): - with ( - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", - new=mock_is_api_active, - ), - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", - new=mock_api_client, - ), + with patch( + "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", + new=mock_is_api_active, + ), patch( + "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", + new=mock_api_client, ): api_keys_client = APIKeys( set_mocked_gcp_provider(project_ids=[GCP_PROJECT_ID]) diff --git a/tests/providers/gcp/services/bigquery/bigquery_dataset_cmk_encryption/bigquery_dataset_cmk_encryption_test.py b/tests/providers/gcp/services/bigquery/bigquery_dataset_cmk_encryption/bigquery_dataset_cmk_encryption_test.py index 1097cf0fb3..15e67a5e21 100644 --- a/tests/providers/gcp/services/bigquery/bigquery_dataset_cmk_encryption/bigquery_dataset_cmk_encryption_test.py +++ b/tests/providers/gcp/services/bigquery/bigquery_dataset_cmk_encryption/bigquery_dataset_cmk_encryption_test.py @@ -8,15 +8,12 @@ class Test_bigquery_dataset_cmk_encryption: bigquery_client = mock.MagicMock() bigquery_client.datasets = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.bigquery.bigquery_dataset_cmk_encryption.bigquery_dataset_cmk_encryption.bigquery_client", - new=bigquery_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.bigquery.bigquery_dataset_cmk_encryption.bigquery_dataset_cmk_encryption.bigquery_client", + new=bigquery_client, ): from prowler.providers.gcp.services.bigquery.bigquery_dataset_cmk_encryption.bigquery_dataset_cmk_encryption import ( bigquery_dataset_cmk_encryption, @@ -29,15 +26,12 @@ class Test_bigquery_dataset_cmk_encryption: def test_one_compliant_dataset(self): bigquery_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.bigquery.bigquery_dataset_cmk_encryption.bigquery_dataset_cmk_encryption.bigquery_client", - new=bigquery_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.bigquery.bigquery_dataset_cmk_encryption.bigquery_dataset_cmk_encryption.bigquery_client", + new=bigquery_client, ): from prowler.providers.gcp.services.bigquery.bigquery_dataset_cmk_encryption.bigquery_dataset_cmk_encryption import ( bigquery_dataset_cmk_encryption, @@ -72,15 +66,12 @@ class Test_bigquery_dataset_cmk_encryption: def test_one_non_compliant_dataset(self): bigquery_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.bigquery.bigquery_dataset_cmk_encryption.bigquery_dataset_cmk_encryption.bigquery_client", - new=bigquery_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.bigquery.bigquery_dataset_cmk_encryption.bigquery_dataset_cmk_encryption.bigquery_client", + new=bigquery_client, ): from prowler.providers.gcp.services.bigquery.bigquery_dataset_cmk_encryption.bigquery_dataset_cmk_encryption import ( bigquery_dataset_cmk_encryption, diff --git a/tests/providers/gcp/services/bigquery/bigquery_dataset_public_access/bigquery_dataset_public_access_test.py b/tests/providers/gcp/services/bigquery/bigquery_dataset_public_access/bigquery_dataset_public_access_test.py index b76c88321a..f4c075f49b 100644 --- a/tests/providers/gcp/services/bigquery/bigquery_dataset_public_access/bigquery_dataset_public_access_test.py +++ b/tests/providers/gcp/services/bigquery/bigquery_dataset_public_access/bigquery_dataset_public_access_test.py @@ -8,15 +8,12 @@ class Test_bigquery_dataset_public_access: bigquery_client = mock.MagicMock() bigquery_client.datasets = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.bigquery.bigquery_dataset_public_access.bigquery_dataset_public_access.bigquery_client", - new=bigquery_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.bigquery.bigquery_dataset_public_access.bigquery_dataset_public_access.bigquery_client", + new=bigquery_client, ): from prowler.providers.gcp.services.bigquery.bigquery_dataset_public_access.bigquery_dataset_public_access import ( bigquery_dataset_public_access, @@ -42,15 +39,12 @@ class Test_bigquery_dataset_public_access: bigquery_client.project_ids = [GCP_PROJECT_ID] bigquery_client.datasets = [dataset] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.bigquery.bigquery_dataset_public_access.bigquery_dataset_public_access.bigquery_client", - new=bigquery_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.bigquery.bigquery_dataset_public_access.bigquery_dataset_public_access.bigquery_client", + new=bigquery_client, ): from prowler.providers.gcp.services.bigquery.bigquery_dataset_public_access.bigquery_dataset_public_access import ( bigquery_dataset_public_access, @@ -86,15 +80,12 @@ class Test_bigquery_dataset_public_access: bigquery_client.project_ids = [GCP_PROJECT_ID] bigquery_client.datasets = [dataset] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.bigquery.bigquery_dataset_public_access.bigquery_dataset_public_access.bigquery_client", - new=bigquery_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.bigquery.bigquery_dataset_public_access.bigquery_dataset_public_access.bigquery_client", + new=bigquery_client, ): from prowler.providers.gcp.services.bigquery.bigquery_dataset_public_access.bigquery_dataset_public_access import ( bigquery_dataset_public_access, diff --git a/tests/providers/gcp/services/bigquery/bigquery_service_test.py b/tests/providers/gcp/services/bigquery/bigquery_service_test.py index 02ca32b5df..f4640618c3 100644 --- a/tests/providers/gcp/services/bigquery/bigquery_service_test.py +++ b/tests/providers/gcp/services/bigquery/bigquery_service_test.py @@ -11,15 +11,12 @@ from tests.providers.gcp.gcp_fixtures import ( class TestBigQueryService: def test_service(self): - with ( - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", - new=mock_is_api_active, - ), - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", - new=mock_api_client, - ), + with patch( + "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", + new=mock_is_api_active, + ), patch( + "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", + new=mock_api_client, ): bigquery_client = BigQuery( set_mocked_gcp_provider(project_ids=[GCP_PROJECT_ID]) diff --git a/tests/providers/gcp/services/bigquery/bigquery_table_cmk_encryption/bigquery_table_cmk_encryption_test.py b/tests/providers/gcp/services/bigquery/bigquery_table_cmk_encryption/bigquery_table_cmk_encryption_test.py index d1b8265246..f36304a6bc 100644 --- a/tests/providers/gcp/services/bigquery/bigquery_table_cmk_encryption/bigquery_table_cmk_encryption_test.py +++ b/tests/providers/gcp/services/bigquery/bigquery_table_cmk_encryption/bigquery_table_cmk_encryption_test.py @@ -8,15 +8,12 @@ class Test_bigquery_table_cmk_encryption: bigquery_client = mock.MagicMock() bigquery_client.tables = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.bigquery.bigquery_table_cmk_encryption.bigquery_table_cmk_encryption.bigquery_client", - new=bigquery_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.bigquery.bigquery_table_cmk_encryption.bigquery_table_cmk_encryption.bigquery_client", + new=bigquery_client, ): from prowler.providers.gcp.services.bigquery.bigquery_table_cmk_encryption.bigquery_table_cmk_encryption import ( bigquery_table_cmk_encryption, @@ -29,15 +26,12 @@ class Test_bigquery_table_cmk_encryption: def test_one_compliant_table(self): bigquery_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.bigquery.bigquery_table_cmk_encryption.bigquery_table_cmk_encryption.bigquery_client", - new=bigquery_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.bigquery.bigquery_table_cmk_encryption.bigquery_table_cmk_encryption.bigquery_client", + new=bigquery_client, ): from prowler.providers.gcp.services.bigquery.bigquery_service import Table from prowler.providers.gcp.services.bigquery.bigquery_table_cmk_encryption.bigquery_table_cmk_encryption import ( @@ -72,15 +66,12 @@ class Test_bigquery_table_cmk_encryption: def test_one_non_compliant_table(self): bigquery_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.bigquery.bigquery_table_cmk_encryption.bigquery_table_cmk_encryption.bigquery_client", - new=bigquery_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.bigquery.bigquery_table_cmk_encryption.bigquery_table_cmk_encryption.bigquery_client", + new=bigquery_client, ): from prowler.providers.gcp.services.bigquery.bigquery_service import Table from prowler.providers.gcp.services.bigquery.bigquery_table_cmk_encryption.bigquery_table_cmk_encryption import ( diff --git a/tests/providers/gcp/services/cloudresourcemanager/cloudresourcemanager_service_test.py b/tests/providers/gcp/services/cloudresourcemanager/cloudresourcemanager_service_test.py index 5af6b01461..9cb9cccdb2 100644 --- a/tests/providers/gcp/services/cloudresourcemanager/cloudresourcemanager_service_test.py +++ b/tests/providers/gcp/services/cloudresourcemanager/cloudresourcemanager_service_test.py @@ -13,15 +13,12 @@ from tests.providers.gcp.gcp_fixtures import ( class TestCloudResourceManagerService: def test_service(self): - with ( - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", - new=mock_is_api_active, - ), - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", - new=mock_api_client, - ), + with patch( + "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", + new=mock_is_api_active, + ), patch( + "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", + new=mock_api_client, ): api_keys_client = CloudResourceManager( set_mocked_gcp_provider(project_ids=[GCP_PROJECT_ID]) diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_automated_backups/cloudsql_instance_automated_backups_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_automated_backups/cloudsql_instance_automated_backups_test.py index 5ca2fd954f..6e496bb5eb 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_automated_backups/cloudsql_instance_automated_backups_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_automated_backups/cloudsql_instance_automated_backups_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_automated_backups: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_automated_backups.cloudsql_instance_automated_backups.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_automated_backups.cloudsql_instance_automated_backups.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_automated_backups.cloudsql_instance_automated_backups import ( cloudsql_instance_automated_backups, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_automated_backups: def test_cloudsql_instance_with_automated_backups(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_automated_backups.cloudsql_instance_automated_backups.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_automated_backups.cloudsql_instance_automated_backups.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_automated_backups.cloudsql_instance_automated_backups import ( cloudsql_instance_automated_backups, @@ -83,15 +77,12 @@ class Test_cloudsql_instance_automated_backups: def test_cloudsql_instance_without_automated_backups(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_automated_backups.cloudsql_instance_automated_backups.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_automated_backups.cloudsql_instance_automated_backups.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_automated_backups.cloudsql_instance_automated_backups import ( cloudsql_instance_automated_backups, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_mysql_local_infile_flag/cloudsql_instance_mysql_local_infile_flag_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_mysql_local_infile_flag/cloudsql_instance_mysql_local_infile_flag_test.py index d8da6c9710..5ef3ee1ec4 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_mysql_local_infile_flag/cloudsql_instance_mysql_local_infile_flag_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_mysql_local_infile_flag/cloudsql_instance_mysql_local_infile_flag_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_mysql_local_infile_flag: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_local_infile_flag.cloudsql_instance_mysql_local_infile_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_local_infile_flag.cloudsql_instance_mysql_local_infile_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_local_infile_flag.cloudsql_instance_mysql_local_infile_flag import ( cloudsql_instance_mysql_local_infile_flag, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_mysql_local_infile_flag: def test_cloudsql_postgres_instance(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_local_infile_flag.cloudsql_instance_mysql_local_infile_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_local_infile_flag.cloudsql_instance_mysql_local_infile_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_local_infile_flag.cloudsql_instance_mysql_local_infile_flag import ( cloudsql_instance_mysql_local_infile_flag, @@ -74,15 +68,12 @@ class Test_cloudsql_instance_mysql_local_infile_flag: def test_cloudsql_instance_with_no_flags(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_local_infile_flag.cloudsql_instance_mysql_local_infile_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_local_infile_flag.cloudsql_instance_mysql_local_infile_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_local_infile_flag.cloudsql_instance_mysql_local_infile_flag import ( cloudsql_instance_mysql_local_infile_flag, @@ -123,15 +114,12 @@ class Test_cloudsql_instance_mysql_local_infile_flag: def test_cloudsql_instance_with_local_infile_flag_off(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_local_infile_flag.cloudsql_instance_mysql_local_infile_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_local_infile_flag.cloudsql_instance_mysql_local_infile_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_local_infile_flag.cloudsql_instance_mysql_local_infile_flag import ( cloudsql_instance_mysql_local_infile_flag, @@ -172,15 +160,12 @@ class Test_cloudsql_instance_mysql_local_infile_flag: def test_cloudsql_instance_with_local_infile_flag_on(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_local_infile_flag.cloudsql_instance_mysql_local_infile_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_local_infile_flag.cloudsql_instance_mysql_local_infile_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_local_infile_flag.cloudsql_instance_mysql_local_infile_flag import ( cloudsql_instance_mysql_local_infile_flag, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_mysql_skip_show_database_flag/cloudsql_instance_mysql_skip_show_database_flag_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_mysql_skip_show_database_flag/cloudsql_instance_mysql_skip_show_database_flag_test.py index 91c5c6b595..8e1d541972 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_mysql_skip_show_database_flag/cloudsql_instance_mysql_skip_show_database_flag_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_mysql_skip_show_database_flag/cloudsql_instance_mysql_skip_show_database_flag_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_mysql_skip_show_database_flag: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_instance_mysql_skip_show_database_flag import ( cloudsql_instance_mysql_skip_show_database_flag, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_mysql_skip_show_database_flag: def test_cloudsql_postgres_instance(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_instance_mysql_skip_show_database_flag import ( cloudsql_instance_mysql_skip_show_database_flag, @@ -74,15 +68,12 @@ class Test_cloudsql_instance_mysql_skip_show_database_flag: def test_cloudsql_instance_no_flags(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_instance_mysql_skip_show_database_flag import ( cloudsql_instance_mysql_skip_show_database_flag, @@ -123,15 +114,12 @@ class Test_cloudsql_instance_mysql_skip_show_database_flag: def test_cloudsql_instance_with_skip_show_databases_off(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_instance_mysql_skip_show_database_flag import ( cloudsql_instance_mysql_skip_show_database_flag, @@ -172,15 +160,12 @@ class Test_cloudsql_instance_mysql_skip_show_database_flag: def test_cloudsql_instance_with_skip_show_databases_on(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_mysql_skip_show_database_flag.cloudsql_instance_mysql_skip_show_database_flag import ( cloudsql_instance_mysql_skip_show_database_flag, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_enable_pgaudit_flag/cloudsql_instance_postgres_enable_pgaudit_flag_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_enable_pgaudit_flag/cloudsql_instance_postgres_enable_pgaudit_flag_test.py index 884d21f7ce..2c242be3f0 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_enable_pgaudit_flag/cloudsql_instance_postgres_enable_pgaudit_flag_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_enable_pgaudit_flag/cloudsql_instance_postgres_enable_pgaudit_flag_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_postgres_enable_pgaudit_flag: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_instance_postgres_enable_pgaudit_flag import ( cloudsql_instance_postgres_enable_pgaudit_flag, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_postgres_enable_pgaudit_flag: def test_cloudsql_mysql_instance(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_instance_postgres_enable_pgaudit_flag import ( cloudsql_instance_postgres_enable_pgaudit_flag, @@ -74,15 +68,12 @@ class Test_cloudsql_instance_postgres_enable_pgaudit_flag: def test_cloudsql_instance_no_flags(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_instance_postgres_enable_pgaudit_flag import ( cloudsql_instance_postgres_enable_pgaudit_flag, @@ -123,15 +114,12 @@ class Test_cloudsql_instance_postgres_enable_pgaudit_flag: def test_cloudsql_instance_pgaudit_flag_off(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_instance_postgres_enable_pgaudit_flag import ( cloudsql_instance_postgres_enable_pgaudit_flag, @@ -172,15 +160,12 @@ class Test_cloudsql_instance_postgres_enable_pgaudit_flag: def test_cloudsql_instance_pgaudit_flag_on(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_enable_pgaudit_flag.cloudsql_instance_postgres_enable_pgaudit_flag import ( cloudsql_instance_postgres_enable_pgaudit_flag, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_connections_flag/cloudsql_instance_postgres_log_connections_flag_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_connections_flag/cloudsql_instance_postgres_log_connections_flag_test.py index 994a778792..69d05be8ba 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_connections_flag/cloudsql_instance_postgres_log_connections_flag_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_connections_flag/cloudsql_instance_postgres_log_connections_flag_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_postgres_log_connections_flag: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_connections_flag.cloudsql_instance_postgres_log_connections_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_connections_flag.cloudsql_instance_postgres_log_connections_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_connections_flag.cloudsql_instance_postgres_log_connections_flag import ( cloudsql_instance_postgres_log_connections_flag, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_postgres_log_connections_flag: def test_cloudsql_mysql_instance(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_connections_flag.cloudsql_instance_postgres_log_connections_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_connections_flag.cloudsql_instance_postgres_log_connections_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_connections_flag.cloudsql_instance_postgres_log_connections_flag import ( cloudsql_instance_postgres_log_connections_flag, @@ -74,15 +68,12 @@ class Test_cloudsql_instance_postgres_log_connections_flag: def test_cloudsql_instance_no_flags(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_connections_flag.cloudsql_instance_postgres_log_connections_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_connections_flag.cloudsql_instance_postgres_log_connections_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_connections_flag.cloudsql_instance_postgres_log_connections_flag import ( cloudsql_instance_postgres_log_connections_flag, @@ -123,15 +114,12 @@ class Test_cloudsql_instance_postgres_log_connections_flag: def test_cloudsql_instance_log_connections_flag_off(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_connections_flag.cloudsql_instance_postgres_log_connections_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_connections_flag.cloudsql_instance_postgres_log_connections_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_connections_flag.cloudsql_instance_postgres_log_connections_flag import ( cloudsql_instance_postgres_log_connections_flag, @@ -172,15 +160,12 @@ class Test_cloudsql_instance_postgres_log_connections_flag: def test_cloudsql_instance_log_connections_flag_on(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_connections_flag.cloudsql_instance_postgres_log_connections_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_connections_flag.cloudsql_instance_postgres_log_connections_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_connections_flag.cloudsql_instance_postgres_log_connections_flag import ( cloudsql_instance_postgres_log_connections_flag, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_disconnections_flag/cloudsql_instance_postgres_log_disconnections_flag_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_disconnections_flag/cloudsql_instance_postgres_log_disconnections_flag_test.py index ab9ca6484b..5669661f3b 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_disconnections_flag/cloudsql_instance_postgres_log_disconnections_flag_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_disconnections_flag/cloudsql_instance_postgres_log_disconnections_flag_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_postgres_log_disconnections_flag: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_instance_postgres_log_disconnections_flag import ( cloudsql_instance_postgres_log_disconnections_flag, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_postgres_log_disconnections_flag: def test_cloudsql_mysql_instance(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_instance_postgres_log_disconnections_flag import ( cloudsql_instance_postgres_log_disconnections_flag, @@ -74,15 +68,12 @@ class Test_cloudsql_instance_postgres_log_disconnections_flag: def test_cloudsql_instance_no_flags(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_instance_postgres_log_disconnections_flag import ( cloudsql_instance_postgres_log_disconnections_flag, @@ -123,15 +114,12 @@ class Test_cloudsql_instance_postgres_log_disconnections_flag: def test_cloudsql_instance_log_disconnections_flag_off(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_instance_postgres_log_disconnections_flag import ( cloudsql_instance_postgres_log_disconnections_flag, @@ -172,15 +160,12 @@ class Test_cloudsql_instance_postgres_log_disconnections_flag: def test_cloudsql_instance_log_disconnections_flag_on(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_disconnections_flag.cloudsql_instance_postgres_log_disconnections_flag import ( cloudsql_instance_postgres_log_disconnections_flag, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_error_verbosity_flag/cloudsql_instance_postgres_log_error_verbosity_flag_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_error_verbosity_flag/cloudsql_instance_postgres_log_error_verbosity_flag_test.py index c0ae96fa5f..b5a27d006b 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_error_verbosity_flag/cloudsql_instance_postgres_log_error_verbosity_flag_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_error_verbosity_flag/cloudsql_instance_postgres_log_error_verbosity_flag_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_postgres_log_error_verbosity_flag: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_instance_postgres_log_error_verbosity_flag import ( cloudsql_instance_postgres_log_error_verbosity_flag, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_postgres_log_error_verbosity_flag: def test_cloudsql_mysql_instance(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_instance_postgres_log_error_verbosity_flag import ( cloudsql_instance_postgres_log_error_verbosity_flag, @@ -74,15 +68,12 @@ class Test_cloudsql_instance_postgres_log_error_verbosity_flag: def test_cloudsql_instance_no_flags(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_instance_postgres_log_error_verbosity_flag import ( cloudsql_instance_postgres_log_error_verbosity_flag, @@ -123,15 +114,12 @@ class Test_cloudsql_instance_postgres_log_error_verbosity_flag: def test_cloudsql_instance_log_error_verbosity_flag_off(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_instance_postgres_log_error_verbosity_flag import ( cloudsql_instance_postgres_log_error_verbosity_flag, @@ -172,15 +160,12 @@ class Test_cloudsql_instance_postgres_log_error_verbosity_flag: def test_cloudsql_instance_log_error_verbosity_flag_on(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_error_verbosity_flag.cloudsql_instance_postgres_log_error_verbosity_flag import ( cloudsql_instance_postgres_log_error_verbosity_flag, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_min_duration_statement_flag/cloudsql_instance_postgres_log_min_duration_statement_flag_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_min_duration_statement_flag/cloudsql_instance_postgres_log_min_duration_statement_flag_test.py index 76f6a3c221..80ccb96b0f 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_min_duration_statement_flag/cloudsql_instance_postgres_log_min_duration_statement_flag_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_min_duration_statement_flag/cloudsql_instance_postgres_log_min_duration_statement_flag_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_postgres_log_min_duration_statement_flag: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_instance_postgres_log_min_duration_statement_flag import ( cloudsql_instance_postgres_log_min_duration_statement_flag, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_postgres_log_min_duration_statement_flag: def test_cloudsql_mysql_instance(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_instance_postgres_log_min_duration_statement_flag import ( cloudsql_instance_postgres_log_min_duration_statement_flag, @@ -74,15 +68,12 @@ class Test_cloudsql_instance_postgres_log_min_duration_statement_flag: def test_cloudsql_instance_no_flags(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_instance_postgres_log_min_duration_statement_flag import ( cloudsql_instance_postgres_log_min_duration_statement_flag, @@ -123,15 +114,12 @@ class Test_cloudsql_instance_postgres_log_min_duration_statement_flag: def test_cloudsql_instance_log_min_duration_statement_flag_off(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_instance_postgres_log_min_duration_statement_flag import ( cloudsql_instance_postgres_log_min_duration_statement_flag, @@ -172,15 +160,12 @@ class Test_cloudsql_instance_postgres_log_min_duration_statement_flag: def test_cloudsql_instance_log_min_duration_statement_flag_on(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_duration_statement_flag.cloudsql_instance_postgres_log_min_duration_statement_flag import ( cloudsql_instance_postgres_log_min_duration_statement_flag, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_min_error_statement_flag/cloudsql_instance_postgres_log_min_error_statement_flag_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_min_error_statement_flag/cloudsql_instance_postgres_log_min_error_statement_flag_test.py index 9b9eca528f..007723a912 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_min_error_statement_flag/cloudsql_instance_postgres_log_min_error_statement_flag_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_min_error_statement_flag/cloudsql_instance_postgres_log_min_error_statement_flag_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_postgres_log_min_error_statement_flag: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_instance_postgres_log_min_error_statement_flag import ( cloudsql_instance_postgres_log_min_error_statement_flag, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_postgres_log_min_error_statement_flag: def test_cloudsql_mysql_instance(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_instance_postgres_log_min_error_statement_flag import ( cloudsql_instance_postgres_log_min_error_statement_flag, @@ -74,15 +68,12 @@ class Test_cloudsql_instance_postgres_log_min_error_statement_flag: def test_cloudsql_instance_no_flags(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_instance_postgres_log_min_error_statement_flag import ( cloudsql_instance_postgres_log_min_error_statement_flag, @@ -123,15 +114,12 @@ class Test_cloudsql_instance_postgres_log_min_error_statement_flag: def test_cloudsql_instance_log_min_error_statement_flag_off(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_instance_postgres_log_min_error_statement_flag import ( cloudsql_instance_postgres_log_min_error_statement_flag, @@ -172,15 +160,12 @@ class Test_cloudsql_instance_postgres_log_min_error_statement_flag: def test_cloudsql_instance_log_min_error_statement_flag_on(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_error_statement_flag.cloudsql_instance_postgres_log_min_error_statement_flag import ( cloudsql_instance_postgres_log_min_error_statement_flag, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_min_messages_flag/cloudsql_instance_postgres_log_min_messages_flag_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_min_messages_flag/cloudsql_instance_postgres_log_min_messages_flag_test.py index f26ca5be2f..d40010c3cd 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_min_messages_flag/cloudsql_instance_postgres_log_min_messages_flag_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_min_messages_flag/cloudsql_instance_postgres_log_min_messages_flag_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_postgres_log_min_messages_flag: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_instance_postgres_log_min_messages_flag import ( cloudsql_instance_postgres_log_min_messages_flag, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_postgres_log_min_messages_flag: def test_cloudsql_mysql_instance(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_instance_postgres_log_min_messages_flag import ( cloudsql_instance_postgres_log_min_messages_flag, @@ -74,15 +68,12 @@ class Test_cloudsql_instance_postgres_log_min_messages_flag: def test_cloudsql_instance_no_flags(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_instance_postgres_log_min_messages_flag import ( cloudsql_instance_postgres_log_min_messages_flag, @@ -123,15 +114,12 @@ class Test_cloudsql_instance_postgres_log_min_messages_flag: def test_cloudsql_instance_log_min_messages_flag_off(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_instance_postgres_log_min_messages_flag import ( cloudsql_instance_postgres_log_min_messages_flag, @@ -172,15 +160,12 @@ class Test_cloudsql_instance_postgres_log_min_messages_flag: def test_cloudsql_instance_log_min_messages_flag_on(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_min_messages_flag.cloudsql_instance_postgres_log_min_messages_flag import ( cloudsql_instance_postgres_log_min_messages_flag, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_statement_flag/cloudsql_instance_postgres_log_statement_flag_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_statement_flag/cloudsql_instance_postgres_log_statement_flag_test.py index c28a6795e9..02c82667f8 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_statement_flag/cloudsql_instance_postgres_log_statement_flag_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_postgres_log_statement_flag/cloudsql_instance_postgres_log_statement_flag_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_postgres_log_statement_flag: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_statement_flag.cloudsql_instance_postgres_log_statement_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_statement_flag.cloudsql_instance_postgres_log_statement_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_statement_flag.cloudsql_instance_postgres_log_statement_flag import ( cloudsql_instance_postgres_log_statement_flag, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_postgres_log_statement_flag: def test_cloudsql_mysql_instance(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_statement_flag.cloudsql_instance_postgres_log_statement_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_statement_flag.cloudsql_instance_postgres_log_statement_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_statement_flag.cloudsql_instance_postgres_log_statement_flag import ( cloudsql_instance_postgres_log_statement_flag, @@ -74,15 +68,12 @@ class Test_cloudsql_instance_postgres_log_statement_flag: def test_cloudsql_instance_no_flags(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_statement_flag.cloudsql_instance_postgres_log_statement_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_statement_flag.cloudsql_instance_postgres_log_statement_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_statement_flag.cloudsql_instance_postgres_log_statement_flag import ( cloudsql_instance_postgres_log_statement_flag, @@ -123,15 +114,12 @@ class Test_cloudsql_instance_postgres_log_statement_flag: def test_cloudsql_instance_log_statement_flag_off(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_statement_flag.cloudsql_instance_postgres_log_statement_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_statement_flag.cloudsql_instance_postgres_log_statement_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_statement_flag.cloudsql_instance_postgres_log_statement_flag import ( cloudsql_instance_postgres_log_statement_flag, @@ -172,15 +160,12 @@ class Test_cloudsql_instance_postgres_log_statement_flag: def test_cloudsql_instance_log_statement_flag_on(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_statement_flag.cloudsql_instance_postgres_log_statement_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_statement_flag.cloudsql_instance_postgres_log_statement_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_postgres_log_statement_flag.cloudsql_instance_postgres_log_statement_flag import ( cloudsql_instance_postgres_log_statement_flag, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_public_access/cloudsql_instance_public_access_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_public_access/cloudsql_instance_public_access_test.py index 70d27989e1..b617cfd549 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_public_access/cloudsql_instance_public_access_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_public_access/cloudsql_instance_public_access_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_public_access: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_public_access.cloudsql_instance_public_access.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_public_access.cloudsql_instance_public_access.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_public_access.cloudsql_instance_public_access import ( cloudsql_instance_public_access, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_public_access: def test_cloudsql_instance_no_public_access(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_public_access.cloudsql_instance_public_access.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_public_access.cloudsql_instance_public_access.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_public_access.cloudsql_instance_public_access import ( cloudsql_instance_public_access, @@ -83,15 +77,12 @@ class Test_cloudsql_instance_public_access: def test_cloudsql_instance_public_access(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_public_access.cloudsql_instance_public_access.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_public_access.cloudsql_instance_public_access.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_public_access.cloudsql_instance_public_access import ( cloudsql_instance_public_access, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_public_ip/cloudsql_instance_public_ip_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_public_ip/cloudsql_instance_public_ip_test.py index b38094cb3b..ed234c1185 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_public_ip/cloudsql_instance_public_ip_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_public_ip/cloudsql_instance_public_ip_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_public_ip: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_public_ip.cloudsql_instance_public_ip.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_public_ip.cloudsql_instance_public_ip.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_public_ip.cloudsql_instance_public_ip import ( cloudsql_instance_public_ip, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_public_ip: def test_cloudsql_instance_no_public_ip(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_public_ip.cloudsql_instance_public_ip.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_public_ip.cloudsql_instance_public_ip.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_public_ip.cloudsql_instance_public_ip import ( cloudsql_instance_public_ip, @@ -83,15 +77,12 @@ class Test_cloudsql_instance_public_ip: def test_cloudsql_instance_public_ip(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_public_ip.cloudsql_instance_public_ip.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_public_ip.cloudsql_instance_public_ip.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_public_ip.cloudsql_instance_public_ip import ( cloudsql_instance_public_ip, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_contained_database_authentication_flag/cloudsql_instance_sqlserver_contained_database_authentication_flag_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_contained_database_authentication_flag/cloudsql_instance_sqlserver_contained_database_authentication_flag_test.py index 2a2c21829a..c4e49b3fab 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_contained_database_authentication_flag/cloudsql_instance_sqlserver_contained_database_authentication_flag_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_contained_database_authentication_flag/cloudsql_instance_sqlserver_contained_database_authentication_flag_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_sqlserver_contained_database_authentication_flag: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_instance_sqlserver_contained_database_authentication_flag import ( cloudsql_instance_sqlserver_contained_database_authentication_flag, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_sqlserver_contained_database_authentication_flag: def test_cloudsql_postgres_instance(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_instance_sqlserver_contained_database_authentication_flag import ( cloudsql_instance_sqlserver_contained_database_authentication_flag, @@ -74,15 +68,12 @@ class Test_cloudsql_instance_sqlserver_contained_database_authentication_flag: def test_cloudsql_instance_no_flags(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_instance_sqlserver_contained_database_authentication_flag import ( cloudsql_instance_sqlserver_contained_database_authentication_flag, @@ -123,15 +114,12 @@ class Test_cloudsql_instance_sqlserver_contained_database_authentication_flag: def test_cloudsql_instance_contained_database_authentication_flag_on(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_instance_sqlserver_contained_database_authentication_flag import ( cloudsql_instance_sqlserver_contained_database_authentication_flag, @@ -174,15 +162,12 @@ class Test_cloudsql_instance_sqlserver_contained_database_authentication_flag: def test_cloudsql_instance_contained_database_authentication_flag_off(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_contained_database_authentication_flag.cloudsql_instance_sqlserver_contained_database_authentication_flag import ( cloudsql_instance_sqlserver_contained_database_authentication_flag, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag/cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag/cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag_test.py index dd4494953c..c8ecf9912f 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag/cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag/cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag import ( cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag: def test_cloudsql_postgres_instance(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag import ( cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag, @@ -74,15 +68,12 @@ class Test_cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag: def test_cloudsql_instance_no_flags(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag import ( cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag, @@ -123,15 +114,12 @@ class Test_cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag: def test_cloudsql_instance_cross_db_ownership_flag_on(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag import ( cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag, @@ -172,15 +160,12 @@ class Test_cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag: def test_cloudsql_instance_cross_db_ownership_flag_off(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag.cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag import ( cloudsql_instance_sqlserver_cross_db_ownership_chaining_flag, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_external_scripts_enabled_flag/cloudsql_instance_sqlserver_external_scripts_enabled_flag_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_external_scripts_enabled_flag/cloudsql_instance_sqlserver_external_scripts_enabled_flag_test.py index 2ebe7ac738..8629822de1 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_external_scripts_enabled_flag/cloudsql_instance_sqlserver_external_scripts_enabled_flag_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_external_scripts_enabled_flag/cloudsql_instance_sqlserver_external_scripts_enabled_flag_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_sqlserver_external_scripts_enabled_flag: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_instance_sqlserver_external_scripts_enabled_flag import ( cloudsql_instance_sqlserver_external_scripts_enabled_flag, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_sqlserver_external_scripts_enabled_flag: def test_cloudsql_postgres_instance(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_instance_sqlserver_external_scripts_enabled_flag import ( cloudsql_instance_sqlserver_external_scripts_enabled_flag, @@ -74,15 +68,12 @@ class Test_cloudsql_instance_sqlserver_external_scripts_enabled_flag: def test_cloudsql_sqlserver_instance_no_flags(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_instance_sqlserver_external_scripts_enabled_flag import ( cloudsql_instance_sqlserver_external_scripts_enabled_flag, @@ -123,15 +114,12 @@ class Test_cloudsql_instance_sqlserver_external_scripts_enabled_flag: def test_cloudsql_sqlserver_instance_external_scripts_enabled_flag_on(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_instance_sqlserver_external_scripts_enabled_flag import ( cloudsql_instance_sqlserver_external_scripts_enabled_flag, @@ -172,15 +160,12 @@ class Test_cloudsql_instance_sqlserver_external_scripts_enabled_flag: def test_cloudsql_sqlserver_instance_external_scripts_enabled_flag_off(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_external_scripts_enabled_flag.cloudsql_instance_sqlserver_external_scripts_enabled_flag import ( cloudsql_instance_sqlserver_external_scripts_enabled_flag, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_remote_access_flag/cloudsql_instance_sqlserver_remote_access_flag_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_remote_access_flag/cloudsql_instance_sqlserver_remote_access_flag_test.py index 586a336b3f..18efa9d004 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_remote_access_flag/cloudsql_instance_sqlserver_remote_access_flag_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_remote_access_flag/cloudsql_instance_sqlserver_remote_access_flag_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_sqlserver_remote_access_flag: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_instance_sqlserver_remote_access_flag import ( cloudsql_instance_sqlserver_remote_access_flag, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_sqlserver_remote_access_flag: def test_cloudsql_postgres_instance(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_instance_sqlserver_remote_access_flag import ( cloudsql_instance_sqlserver_remote_access_flag, @@ -74,15 +68,12 @@ class Test_cloudsql_instance_sqlserver_remote_access_flag: def test_cloudsql_instance_no_flags(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_instance_sqlserver_remote_access_flag import ( cloudsql_instance_sqlserver_remote_access_flag, @@ -123,15 +114,12 @@ class Test_cloudsql_instance_sqlserver_remote_access_flag: def test_cloudsql_instance_remote_access_flag_on(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_instance_sqlserver_remote_access_flag import ( cloudsql_instance_sqlserver_remote_access_flag, @@ -172,15 +160,12 @@ class Test_cloudsql_instance_sqlserver_remote_access_flag: def test_cloudsql_instance_remote_access_flag_off(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_remote_access_flag.cloudsql_instance_sqlserver_remote_access_flag import ( cloudsql_instance_sqlserver_remote_access_flag, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_trace_flag/cloudsql_instance_sqlserver_trace_flag_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_trace_flag/cloudsql_instance_sqlserver_trace_flag_test.py index fb916185b0..932e62c333 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_trace_flag/cloudsql_instance_sqlserver_trace_flag_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_trace_flag/cloudsql_instance_sqlserver_trace_flag_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_sqlserver_trace_flag: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_trace_flag.cloudsql_instance_sqlserver_trace_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_trace_flag.cloudsql_instance_sqlserver_trace_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_trace_flag.cloudsql_instance_sqlserver_trace_flag import ( cloudsql_instance_sqlserver_trace_flag, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_sqlserver_trace_flag: def test_cloudsql_postgres_instance(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_trace_flag.cloudsql_instance_sqlserver_trace_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_trace_flag.cloudsql_instance_sqlserver_trace_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_trace_flag.cloudsql_instance_sqlserver_trace_flag import ( cloudsql_instance_sqlserver_trace_flag, @@ -74,15 +68,12 @@ class Test_cloudsql_instance_sqlserver_trace_flag: def test_cloudsql_instance_no_flags(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_trace_flag.cloudsql_instance_sqlserver_trace_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_trace_flag.cloudsql_instance_sqlserver_trace_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_trace_flag.cloudsql_instance_sqlserver_trace_flag import ( cloudsql_instance_sqlserver_trace_flag, @@ -123,15 +114,12 @@ class Test_cloudsql_instance_sqlserver_trace_flag: def test_cloudsql_instance_trace_flag_off(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_trace_flag.cloudsql_instance_sqlserver_trace_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_trace_flag.cloudsql_instance_sqlserver_trace_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_trace_flag.cloudsql_instance_sqlserver_trace_flag import ( cloudsql_instance_sqlserver_trace_flag, @@ -172,15 +160,12 @@ class Test_cloudsql_instance_sqlserver_trace_flag: def test_cloudsql_instance_trace_flag_on(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_trace_flag.cloudsql_instance_sqlserver_trace_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_trace_flag.cloudsql_instance_sqlserver_trace_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_trace_flag.cloudsql_instance_sqlserver_trace_flag import ( cloudsql_instance_sqlserver_trace_flag, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_user_connections_flag/cloudsql_instance_sqlserver_user_connections_flag_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_user_connections_flag/cloudsql_instance_sqlserver_user_connections_flag_test.py index 23b93ee9e5..7032a89d0f 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_user_connections_flag/cloudsql_instance_sqlserver_user_connections_flag_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_user_connections_flag/cloudsql_instance_sqlserver_user_connections_flag_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_sqlserver_user_connections_flag: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_instance_sqlserver_user_connections_flag import ( cloudsql_instance_sqlserver_user_connections_flag, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_sqlserver_user_connections_flag: def test_cloudsql_postgres_instance(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_instance_sqlserver_user_connections_flag import ( cloudsql_instance_sqlserver_user_connections_flag, @@ -74,15 +68,12 @@ class Test_cloudsql_instance_sqlserver_user_connections_flag: def test_cloudsql_sqlserver_instance_no_flags(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_instance_sqlserver_user_connections_flag import ( cloudsql_instance_sqlserver_user_connections_flag, @@ -123,15 +114,12 @@ class Test_cloudsql_instance_sqlserver_user_connections_flag: def test_cloudsql_sqlserver_instance_user_connections_flag_off(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_instance_sqlserver_user_connections_flag import ( cloudsql_instance_sqlserver_user_connections_flag, @@ -172,15 +160,12 @@ class Test_cloudsql_instance_sqlserver_user_connections_flag: def test_cloudsql_sqlserver_instance_user_connections_flag_on(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_connections_flag.cloudsql_instance_sqlserver_user_connections_flag import ( cloudsql_instance_sqlserver_user_connections_flag, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_user_options_flag/cloudsql_instance_sqlserver_user_options_flag_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_user_options_flag/cloudsql_instance_sqlserver_user_options_flag_test.py index de9c9468d5..3535fcb271 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_user_options_flag/cloudsql_instance_sqlserver_user_options_flag_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_sqlserver_user_options_flag/cloudsql_instance_sqlserver_user_options_flag_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_sqlserver_user_options_flag: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_options_flag.cloudsql_instance_sqlserver_user_options_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_options_flag.cloudsql_instance_sqlserver_user_options_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_options_flag.cloudsql_instance_sqlserver_user_options_flag import ( cloudsql_instance_sqlserver_user_options_flag, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_sqlserver_user_options_flag: def test_cloudsql_postgres_instance(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_options_flag.cloudsql_instance_sqlserver_user_options_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_options_flag.cloudsql_instance_sqlserver_user_options_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_options_flag.cloudsql_instance_sqlserver_user_options_flag import ( cloudsql_instance_sqlserver_user_options_flag, @@ -74,15 +68,12 @@ class Test_cloudsql_instance_sqlserver_user_options_flag: def test_cloudsql_sqlserver_instance_no_flags(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_options_flag.cloudsql_instance_sqlserver_user_options_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_options_flag.cloudsql_instance_sqlserver_user_options_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_options_flag.cloudsql_instance_sqlserver_user_options_flag import ( cloudsql_instance_sqlserver_user_options_flag, @@ -123,15 +114,12 @@ class Test_cloudsql_instance_sqlserver_user_options_flag: def test_cloudsql_sqlserver_instance_user_options_flag_empty(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_options_flag.cloudsql_instance_sqlserver_user_options_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_options_flag.cloudsql_instance_sqlserver_user_options_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_options_flag.cloudsql_instance_sqlserver_user_options_flag import ( cloudsql_instance_sqlserver_user_options_flag, @@ -172,15 +160,12 @@ class Test_cloudsql_instance_sqlserver_user_options_flag: def test_cloudsql_sqlserver_instance_user_options_flag_set(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_options_flag.cloudsql_instance_sqlserver_user_options_flag.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_options_flag.cloudsql_instance_sqlserver_user_options_flag.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_sqlserver_user_options_flag.cloudsql_instance_sqlserver_user_options_flag import ( cloudsql_instance_sqlserver_user_options_flag, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_instance_ssl_connections/cloudsql_instance_ssl_connections_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_instance_ssl_connections/cloudsql_instance_ssl_connections_test.py index f22f1b01a3..8b398aee00 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_instance_ssl_connections/cloudsql_instance_ssl_connections_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_instance_ssl_connections/cloudsql_instance_ssl_connections_test.py @@ -11,15 +11,12 @@ class Test_cloudsql_instance_ssl_connections: def test_no_cloudsql_instances(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_ssl_connections.cloudsql_instance_ssl_connections.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_ssl_connections.cloudsql_instance_ssl_connections.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_ssl_connections.cloudsql_instance_ssl_connections import ( cloudsql_instance_ssl_connections, @@ -34,15 +31,12 @@ class Test_cloudsql_instance_ssl_connections: def test_cloudsql_instance_ssl_connections_enabled_and_ssl_mode_encrypted(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_ssl_connections.cloudsql_instance_ssl_connections.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_ssl_connections.cloudsql_instance_ssl_connections.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_ssl_connections.cloudsql_instance_ssl_connections import ( cloudsql_instance_ssl_connections, @@ -83,15 +77,12 @@ class Test_cloudsql_instance_ssl_connections: def test_cloudsql_instance_ssl_connections_disabled_and_ssl_mode_encrypted(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_ssl_connections.cloudsql_instance_ssl_connections.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_ssl_connections.cloudsql_instance_ssl_connections.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_ssl_connections.cloudsql_instance_ssl_connections import ( cloudsql_instance_ssl_connections, @@ -132,15 +123,12 @@ class Test_cloudsql_instance_ssl_connections: def test_cloudsql_instance_ssl_connections_enabled_and_ssl_mode_not_encrypted(self): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_ssl_connections.cloudsql_instance_ssl_connections.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_ssl_connections.cloudsql_instance_ssl_connections.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_ssl_connections.cloudsql_instance_ssl_connections import ( cloudsql_instance_ssl_connections, @@ -183,15 +171,12 @@ class Test_cloudsql_instance_ssl_connections: ): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_ssl_connections.cloudsql_instance_ssl_connections.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_ssl_connections.cloudsql_instance_ssl_connections.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_ssl_connections.cloudsql_instance_ssl_connections import ( cloudsql_instance_ssl_connections, @@ -234,15 +219,12 @@ class Test_cloudsql_instance_ssl_connections: ): cloudsql_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudsql.cloudsql_instance_ssl_connections.cloudsql_instance_ssl_connections.cloudsql_client", - new=cloudsql_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudsql.cloudsql_instance_ssl_connections.cloudsql_instance_ssl_connections.cloudsql_client", + new=cloudsql_client, ): from prowler.providers.gcp.services.cloudsql.cloudsql_instance_ssl_connections.cloudsql_instance_ssl_connections import ( cloudsql_instance_ssl_connections, diff --git a/tests/providers/gcp/services/cloudsql/cloudsql_service_test.py b/tests/providers/gcp/services/cloudsql/cloudsql_service_test.py index 1784f24f2b..d38e1a680d 100644 --- a/tests/providers/gcp/services/cloudsql/cloudsql_service_test.py +++ b/tests/providers/gcp/services/cloudsql/cloudsql_service_test.py @@ -11,15 +11,12 @@ from tests.providers.gcp.gcp_fixtures import ( class TestCloudSQLService: def test_service(self): - with ( - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", - new=mock_is_api_active, - ), - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", - new=mock_api_client, - ), + with patch( + "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", + new=mock_is_api_active, + ), patch( + "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", + new=mock_api_client, ): cloudsql_client = CloudSQL( set_mocked_gcp_provider(project_ids=[GCP_PROJECT_ID]) diff --git a/tests/providers/gcp/services/cloudstorage/cloudstorage_service_test.py b/tests/providers/gcp/services/cloudstorage/cloudstorage_service_test.py index fcd8761db5..d0aaa5986c 100644 --- a/tests/providers/gcp/services/cloudstorage/cloudstorage_service_test.py +++ b/tests/providers/gcp/services/cloudstorage/cloudstorage_service_test.py @@ -13,15 +13,12 @@ from tests.providers.gcp.gcp_fixtures import ( class TestCloudStorageService: def test_service(self): - with ( - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", - new=mock_is_api_active, - ), - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", - new=mock_api_client, - ), + with patch( + "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", + new=mock_is_api_active, + ), patch( + "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", + new=mock_api_client, ): cloudstorage_client = CloudStorage( set_mocked_gcp_provider(project_ids=[GCP_PROJECT_ID]) diff --git a/tests/providers/gcp/services/compute/compute_firewall_rdp_access_from_the_internet_allowed/compute_firewall_rdp_access_from_the_internet_allowed_test.py b/tests/providers/gcp/services/compute/compute_firewall_rdp_access_from_the_internet_allowed/compute_firewall_rdp_access_from_the_internet_allowed_test.py index 930ee14605..019feba685 100644 --- a/tests/providers/gcp/services/compute/compute_firewall_rdp_access_from_the_internet_allowed/compute_firewall_rdp_access_from_the_internet_allowed_test.py +++ b/tests/providers/gcp/services/compute/compute_firewall_rdp_access_from_the_internet_allowed/compute_firewall_rdp_access_from_the_internet_allowed_test.py @@ -9,15 +9,12 @@ class Test_compute_firewall_rdp_access_from_the_internet_allowed: compute_client = mock.MagicMock() compute_client.firewalls = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed import ( compute_firewall_rdp_access_from_the_internet_allowed, @@ -44,15 +41,12 @@ class Test_compute_firewall_rdp_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed import ( compute_firewall_rdp_access_from_the_internet_allowed, @@ -86,15 +80,12 @@ class Test_compute_firewall_rdp_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed import ( compute_firewall_rdp_access_from_the_internet_allowed, @@ -128,15 +119,12 @@ class Test_compute_firewall_rdp_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed import ( compute_firewall_rdp_access_from_the_internet_allowed, @@ -170,15 +158,12 @@ class Test_compute_firewall_rdp_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed import ( compute_firewall_rdp_access_from_the_internet_allowed, @@ -212,15 +197,12 @@ class Test_compute_firewall_rdp_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed import ( compute_firewall_rdp_access_from_the_internet_allowed, @@ -254,15 +236,12 @@ class Test_compute_firewall_rdp_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed import ( compute_firewall_rdp_access_from_the_internet_allowed, @@ -296,15 +275,12 @@ class Test_compute_firewall_rdp_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed import ( compute_firewall_rdp_access_from_the_internet_allowed, @@ -338,15 +314,12 @@ class Test_compute_firewall_rdp_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed import ( compute_firewall_rdp_access_from_the_internet_allowed, @@ -380,15 +353,12 @@ class Test_compute_firewall_rdp_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed import ( compute_firewall_rdp_access_from_the_internet_allowed, @@ -425,15 +395,12 @@ class Test_compute_firewall_rdp_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed import ( compute_firewall_rdp_access_from_the_internet_allowed, @@ -471,15 +438,12 @@ class Test_compute_firewall_rdp_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_rdp_access_from_the_internet_allowed.compute_firewall_rdp_access_from_the_internet_allowed import ( compute_firewall_rdp_access_from_the_internet_allowed, diff --git a/tests/providers/gcp/services/compute/compute_firewall_ssh_access_from_the_internet_allowed/compute_firewall_ssh_access_from_the_internet_allowed_test.py b/tests/providers/gcp/services/compute/compute_firewall_ssh_access_from_the_internet_allowed/compute_firewall_ssh_access_from_the_internet_allowed_test.py index 9939415ab7..eb693d339a 100644 --- a/tests/providers/gcp/services/compute/compute_firewall_ssh_access_from_the_internet_allowed/compute_firewall_ssh_access_from_the_internet_allowed_test.py +++ b/tests/providers/gcp/services/compute/compute_firewall_ssh_access_from_the_internet_allowed/compute_firewall_ssh_access_from_the_internet_allowed_test.py @@ -9,15 +9,12 @@ class Test_compute_firewall_ssh_access_from_the_internet_allowed: compute_client = mock.MagicMock() compute_client.firewalls = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed import ( compute_firewall_ssh_access_from_the_internet_allowed, @@ -44,15 +41,12 @@ class Test_compute_firewall_ssh_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed import ( compute_firewall_ssh_access_from_the_internet_allowed, @@ -86,15 +80,12 @@ class Test_compute_firewall_ssh_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed import ( compute_firewall_ssh_access_from_the_internet_allowed, @@ -128,15 +119,12 @@ class Test_compute_firewall_ssh_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed import ( compute_firewall_ssh_access_from_the_internet_allowed, @@ -170,15 +158,12 @@ class Test_compute_firewall_ssh_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed import ( compute_firewall_ssh_access_from_the_internet_allowed, @@ -212,15 +197,12 @@ class Test_compute_firewall_ssh_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed import ( compute_firewall_ssh_access_from_the_internet_allowed, @@ -254,15 +236,12 @@ class Test_compute_firewall_ssh_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed import ( compute_firewall_ssh_access_from_the_internet_allowed, @@ -296,15 +275,12 @@ class Test_compute_firewall_ssh_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed import ( compute_firewall_ssh_access_from_the_internet_allowed, @@ -338,15 +314,12 @@ class Test_compute_firewall_ssh_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed import ( compute_firewall_ssh_access_from_the_internet_allowed, @@ -380,15 +353,12 @@ class Test_compute_firewall_ssh_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed import ( compute_firewall_ssh_access_from_the_internet_allowed, @@ -425,15 +395,12 @@ class Test_compute_firewall_ssh_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed import ( compute_firewall_ssh_access_from_the_internet_allowed, @@ -471,15 +438,12 @@ class Test_compute_firewall_ssh_access_from_the_internet_allowed: compute_client.firewalls = [firewall] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_firewall_ssh_access_from_the_internet_allowed.compute_firewall_ssh_access_from_the_internet_allowed import ( compute_firewall_ssh_access_from_the_internet_allowed, diff --git a/tests/providers/gcp/services/compute/compute_instance_block_project_wide_ssh_keys_disabled/compute_instance_block_project_wide_ssh_keys_disabled_test.py b/tests/providers/gcp/services/compute/compute_instance_block_project_wide_ssh_keys_disabled/compute_instance_block_project_wide_ssh_keys_disabled_test.py index 657e9a6489..800a3da77a 100644 --- a/tests/providers/gcp/services/compute/compute_instance_block_project_wide_ssh_keys_disabled/compute_instance_block_project_wide_ssh_keys_disabled_test.py +++ b/tests/providers/gcp/services/compute/compute_instance_block_project_wide_ssh_keys_disabled/compute_instance_block_project_wide_ssh_keys_disabled_test.py @@ -10,15 +10,12 @@ class Test_compute_instance_block_project_wide_ssh_keys_disabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_block_project_wide_ssh_keys_disabled.compute_instance_block_project_wide_ssh_keys_disabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_block_project_wide_ssh_keys_disabled.compute_instance_block_project_wide_ssh_keys_disabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_block_project_wide_ssh_keys_disabled.compute_instance_block_project_wide_ssh_keys_disabled import ( compute_instance_block_project_wide_ssh_keys_disabled, @@ -51,15 +48,12 @@ class Test_compute_instance_block_project_wide_ssh_keys_disabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_block_project_wide_ssh_keys_disabled.compute_instance_block_project_wide_ssh_keys_disabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_block_project_wide_ssh_keys_disabled.compute_instance_block_project_wide_ssh_keys_disabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_block_project_wide_ssh_keys_disabled.compute_instance_block_project_wide_ssh_keys_disabled import ( compute_instance_block_project_wide_ssh_keys_disabled, @@ -100,15 +94,12 @@ class Test_compute_instance_block_project_wide_ssh_keys_disabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_block_project_wide_ssh_keys_disabled.compute_instance_block_project_wide_ssh_keys_disabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_block_project_wide_ssh_keys_disabled.compute_instance_block_project_wide_ssh_keys_disabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_block_project_wide_ssh_keys_disabled.compute_instance_block_project_wide_ssh_keys_disabled import ( compute_instance_block_project_wide_ssh_keys_disabled, @@ -149,15 +140,12 @@ class Test_compute_instance_block_project_wide_ssh_keys_disabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_block_project_wide_ssh_keys_disabled.compute_instance_block_project_wide_ssh_keys_disabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_block_project_wide_ssh_keys_disabled.compute_instance_block_project_wide_ssh_keys_disabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_block_project_wide_ssh_keys_disabled.compute_instance_block_project_wide_ssh_keys_disabled import ( compute_instance_block_project_wide_ssh_keys_disabled, diff --git a/tests/providers/gcp/services/compute/compute_instance_confidential_computing_enabled/compute_instance_confidential_computing_enabled_test.py b/tests/providers/gcp/services/compute/compute_instance_confidential_computing_enabled/compute_instance_confidential_computing_enabled_test.py index 2bf8d45963..66ef7c82e2 100644 --- a/tests/providers/gcp/services/compute/compute_instance_confidential_computing_enabled/compute_instance_confidential_computing_enabled_test.py +++ b/tests/providers/gcp/services/compute/compute_instance_confidential_computing_enabled/compute_instance_confidential_computing_enabled_test.py @@ -10,15 +10,12 @@ class Test_compute_instance_confidential_computing_enabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_confidential_computing_enabled.compute_instance_confidential_computing_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_confidential_computing_enabled.compute_instance_confidential_computing_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_confidential_computing_enabled.compute_instance_confidential_computing_enabled import ( compute_instance_confidential_computing_enabled, @@ -51,15 +48,12 @@ class Test_compute_instance_confidential_computing_enabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_confidential_computing_enabled.compute_instance_confidential_computing_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_confidential_computing_enabled.compute_instance_confidential_computing_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_confidential_computing_enabled.compute_instance_confidential_computing_enabled import ( compute_instance_confidential_computing_enabled, @@ -102,15 +96,12 @@ class Test_compute_instance_confidential_computing_enabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_confidential_computing_enabled.compute_instance_confidential_computing_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_confidential_computing_enabled.compute_instance_confidential_computing_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_confidential_computing_enabled.compute_instance_confidential_computing_enabled import ( compute_instance_confidential_computing_enabled, diff --git a/tests/providers/gcp/services/compute/compute_instance_default_service_account_in_use/compute_instance_default_service_account_in_use_test.py b/tests/providers/gcp/services/compute/compute_instance_default_service_account_in_use/compute_instance_default_service_account_in_use_test.py index 46acc87e86..9ad9829f91 100644 --- a/tests/providers/gcp/services/compute/compute_instance_default_service_account_in_use/compute_instance_default_service_account_in_use_test.py +++ b/tests/providers/gcp/services/compute/compute_instance_default_service_account_in_use/compute_instance_default_service_account_in_use_test.py @@ -9,15 +9,12 @@ class Test_compute_instance_default_service_account_in_use: compute_client = mock.MagicMock() compute_client.instances = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use.compute_instance_default_service_account_in_use.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use.compute_instance_default_service_account_in_use.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use.compute_instance_default_service_account_in_use import ( compute_instance_default_service_account_in_use, @@ -50,15 +47,12 @@ class Test_compute_instance_default_service_account_in_use: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use.compute_instance_default_service_account_in_use.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use.compute_instance_default_service_account_in_use.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use.compute_instance_default_service_account_in_use import ( compute_instance_default_service_account_in_use, @@ -101,15 +95,12 @@ class Test_compute_instance_default_service_account_in_use: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use.compute_instance_default_service_account_in_use.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use.compute_instance_default_service_account_in_use.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use.compute_instance_default_service_account_in_use import ( compute_instance_default_service_account_in_use, @@ -152,15 +143,12 @@ class Test_compute_instance_default_service_account_in_use: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use.compute_instance_default_service_account_in_use.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use.compute_instance_default_service_account_in_use.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use.compute_instance_default_service_account_in_use import ( compute_instance_default_service_account_in_use, diff --git a/tests/providers/gcp/services/compute/compute_instance_default_service_account_in_use_with_full_api_access/compute_instance_default_service_account_in_use_with_full_api_access_test.py b/tests/providers/gcp/services/compute/compute_instance_default_service_account_in_use_with_full_api_access/compute_instance_default_service_account_in_use_with_full_api_access_test.py index 37fdb648b6..0f44af37b2 100644 --- a/tests/providers/gcp/services/compute/compute_instance_default_service_account_in_use_with_full_api_access/compute_instance_default_service_account_in_use_with_full_api_access_test.py +++ b/tests/providers/gcp/services/compute/compute_instance_default_service_account_in_use_with_full_api_access/compute_instance_default_service_account_in_use_with_full_api_access_test.py @@ -8,15 +8,12 @@ class Test_compute_instance_default_service_account_in_use_with_full_api_access: compute_client = mock.MagicMock() compute_client.instances = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use_with_full_api_access.compute_instance_default_service_account_in_use_with_full_api_access.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use_with_full_api_access.compute_instance_default_service_account_in_use_with_full_api_access.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use_with_full_api_access.compute_instance_default_service_account_in_use_with_full_api_access import ( compute_instance_default_service_account_in_use_with_full_api_access, @@ -53,15 +50,12 @@ class Test_compute_instance_default_service_account_in_use_with_full_api_access: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use_with_full_api_access.compute_instance_default_service_account_in_use_with_full_api_access.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use_with_full_api_access.compute_instance_default_service_account_in_use_with_full_api_access.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use_with_full_api_access.compute_instance_default_service_account_in_use_with_full_api_access import ( compute_instance_default_service_account_in_use_with_full_api_access, @@ -109,15 +103,12 @@ class Test_compute_instance_default_service_account_in_use_with_full_api_access: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use_with_full_api_access.compute_instance_default_service_account_in_use_with_full_api_access.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use_with_full_api_access.compute_instance_default_service_account_in_use_with_full_api_access.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use_with_full_api_access.compute_instance_default_service_account_in_use_with_full_api_access import ( compute_instance_default_service_account_in_use_with_full_api_access, @@ -165,15 +156,12 @@ class Test_compute_instance_default_service_account_in_use_with_full_api_access: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use_with_full_api_access.compute_instance_default_service_account_in_use_with_full_api_access.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use_with_full_api_access.compute_instance_default_service_account_in_use_with_full_api_access.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_default_service_account_in_use_with_full_api_access.compute_instance_default_service_account_in_use_with_full_api_access import ( compute_instance_default_service_account_in_use_with_full_api_access, diff --git a/tests/providers/gcp/services/compute/compute_instance_encryption_with_csek_enabled/compute_instance_encryption_with_csek_enabled_test.py b/tests/providers/gcp/services/compute/compute_instance_encryption_with_csek_enabled/compute_instance_encryption_with_csek_enabled_test.py index 11c6601ee8..92bc9d3b22 100644 --- a/tests/providers/gcp/services/compute/compute_instance_encryption_with_csek_enabled/compute_instance_encryption_with_csek_enabled_test.py +++ b/tests/providers/gcp/services/compute/compute_instance_encryption_with_csek_enabled/compute_instance_encryption_with_csek_enabled_test.py @@ -10,15 +10,12 @@ class Test_compute_instance_encryption_with_csek_enabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_encryption_with_csek_enabled.compute_instance_encryption_with_csek_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_encryption_with_csek_enabled.compute_instance_encryption_with_csek_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_encryption_with_csek_enabled.compute_instance_encryption_with_csek_enabled import ( compute_instance_encryption_with_csek_enabled, @@ -52,15 +49,12 @@ class Test_compute_instance_encryption_with_csek_enabled: compute_client.instances = [instance] compute_client.region = "us-central1" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_encryption_with_csek_enabled.compute_instance_encryption_with_csek_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_encryption_with_csek_enabled.compute_instance_encryption_with_csek_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_encryption_with_csek_enabled.compute_instance_encryption_with_csek_enabled import ( compute_instance_encryption_with_csek_enabled, @@ -101,15 +95,12 @@ class Test_compute_instance_encryption_with_csek_enabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_encryption_with_csek_enabled.compute_instance_encryption_with_csek_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_encryption_with_csek_enabled.compute_instance_encryption_with_csek_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_encryption_with_csek_enabled.compute_instance_encryption_with_csek_enabled import ( compute_instance_encryption_with_csek_enabled, @@ -150,15 +141,12 @@ class Test_compute_instance_encryption_with_csek_enabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_encryption_with_csek_enabled.compute_instance_encryption_with_csek_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_encryption_with_csek_enabled.compute_instance_encryption_with_csek_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_encryption_with_csek_enabled.compute_instance_encryption_with_csek_enabled import ( compute_instance_encryption_with_csek_enabled, diff --git a/tests/providers/gcp/services/compute/compute_instance_ip_forwarding_is_enabled/compute_instance_ip_forwarding_is_enabled_test.py b/tests/providers/gcp/services/compute/compute_instance_ip_forwarding_is_enabled/compute_instance_ip_forwarding_is_enabled_test.py index c8e15634c7..3b3c4d7907 100644 --- a/tests/providers/gcp/services/compute/compute_instance_ip_forwarding_is_enabled/compute_instance_ip_forwarding_is_enabled_test.py +++ b/tests/providers/gcp/services/compute/compute_instance_ip_forwarding_is_enabled/compute_instance_ip_forwarding_is_enabled_test.py @@ -9,15 +9,12 @@ class Test_compute_instance_ip_forwarding_is_enabled: compute_client = mock.MagicMock() compute_client.instances = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_ip_forwarding_is_enabled.compute_instance_ip_forwarding_is_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_ip_forwarding_is_enabled.compute_instance_ip_forwarding_is_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_ip_forwarding_is_enabled.compute_instance_ip_forwarding_is_enabled import ( compute_instance_ip_forwarding_is_enabled, @@ -50,15 +47,12 @@ class Test_compute_instance_ip_forwarding_is_enabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_ip_forwarding_is_enabled.compute_instance_ip_forwarding_is_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_ip_forwarding_is_enabled.compute_instance_ip_forwarding_is_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_ip_forwarding_is_enabled.compute_instance_ip_forwarding_is_enabled import ( compute_instance_ip_forwarding_is_enabled, @@ -101,15 +95,12 @@ class Test_compute_instance_ip_forwarding_is_enabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_ip_forwarding_is_enabled.compute_instance_ip_forwarding_is_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_ip_forwarding_is_enabled.compute_instance_ip_forwarding_is_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_ip_forwarding_is_enabled.compute_instance_ip_forwarding_is_enabled import ( compute_instance_ip_forwarding_is_enabled, @@ -152,15 +143,12 @@ class Test_compute_instance_ip_forwarding_is_enabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_ip_forwarding_is_enabled.compute_instance_ip_forwarding_is_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_ip_forwarding_is_enabled.compute_instance_ip_forwarding_is_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_ip_forwarding_is_enabled.compute_instance_ip_forwarding_is_enabled import ( compute_instance_ip_forwarding_is_enabled, diff --git a/tests/providers/gcp/services/compute/compute_instance_public_ip/compute_instance_public_ip_test.py b/tests/providers/gcp/services/compute/compute_instance_public_ip/compute_instance_public_ip_test.py index 5dc8cf9bb7..3a86285ea5 100644 --- a/tests/providers/gcp/services/compute/compute_instance_public_ip/compute_instance_public_ip_test.py +++ b/tests/providers/gcp/services/compute/compute_instance_public_ip/compute_instance_public_ip_test.py @@ -8,15 +8,12 @@ class Test_compute_instance_public_ip: compute_client = mock.MagicMock() compute_client.instances = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_public_ip.compute_instance_public_ip.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_public_ip.compute_instance_public_ip.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_public_ip.compute_instance_public_ip import ( compute_instance_public_ip, @@ -29,15 +26,12 @@ class Test_compute_instance_public_ip: def test_no_public_ip_instance(self): compute_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_public_ip.compute_instance_public_ip.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_public_ip.compute_instance_public_ip.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_public_ip.compute_instance_public_ip import ( compute_instance_public_ip, @@ -104,15 +98,12 @@ class Test_compute_instance_public_ip: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_public_ip.compute_instance_public_ip.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_public_ip.compute_instance_public_ip.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_public_ip.compute_instance_public_ip import ( compute_instance_public_ip, diff --git a/tests/providers/gcp/services/compute/compute_loadbalancer_logging_enabled/compute_loadbalancer_logging_enabled_test.py b/tests/providers/gcp/services/compute/compute_loadbalancer_logging_enabled/compute_loadbalancer_logging_enabled_test.py index aaefebcb23..d470a8f39e 100644 --- a/tests/providers/gcp/services/compute/compute_loadbalancer_logging_enabled/compute_loadbalancer_logging_enabled_test.py +++ b/tests/providers/gcp/services/compute/compute_loadbalancer_logging_enabled/compute_loadbalancer_logging_enabled_test.py @@ -10,15 +10,12 @@ class Test_compute_loadbalancer_logging_enabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.load_balancers = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_loadbalancer_logging_enabled.compute_loadbalancer_logging_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_loadbalancer_logging_enabled.compute_loadbalancer_logging_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_loadbalancer_logging_enabled.compute_loadbalancer_logging_enabled import ( compute_loadbalancer_logging_enabled, @@ -44,15 +41,12 @@ class Test_compute_loadbalancer_logging_enabled: compute_client.load_balancers = [load_balancer] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_loadbalancer_logging_enabled.compute_loadbalancer_logging_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_loadbalancer_logging_enabled.compute_loadbalancer_logging_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_loadbalancer_logging_enabled.compute_loadbalancer_logging_enabled import ( compute_loadbalancer_logging_enabled, @@ -88,15 +82,12 @@ class Test_compute_loadbalancer_logging_enabled: compute_client.load_balancers = [load_balancer] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_loadbalancer_logging_enabled.compute_loadbalancer_logging_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_loadbalancer_logging_enabled.compute_loadbalancer_logging_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_loadbalancer_logging_enabled.compute_loadbalancer_logging_enabled import ( compute_loadbalancer_logging_enabled, diff --git a/tests/providers/gcp/services/compute/compute_network_dns_logging_enabled/compute_network_dns_logging_enabled_test.py b/tests/providers/gcp/services/compute/compute_network_dns_logging_enabled/compute_network_dns_logging_enabled_test.py index d525b69cea..1ab7dbc01b 100644 --- a/tests/providers/gcp/services/compute/compute_network_dns_logging_enabled/compute_network_dns_logging_enabled_test.py +++ b/tests/providers/gcp/services/compute/compute_network_dns_logging_enabled/compute_network_dns_logging_enabled_test.py @@ -12,15 +12,12 @@ class Test_compute_network_dns_logging_enabled: compute_client.networks = [] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_network_dns_logging_enabled.compute_network_dns_logging_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_network_dns_logging_enabled.compute_network_dns_logging_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_network_dns_logging_enabled.compute_network_dns_logging_enabled import ( compute_network_dns_logging_enabled, @@ -54,15 +51,12 @@ class Test_compute_network_dns_logging_enabled: dns_client.project_ids = [GCP_PROJECT_ID] dns_client.policies = [policy] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_network_dns_logging_enabled.compute_network_dns_logging_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_network_dns_logging_enabled.compute_network_dns_logging_enabled.compute_client", + new=compute_client, ): with mock.patch( "prowler.providers.gcp.services.compute.compute_network_dns_logging_enabled.compute_network_dns_logging_enabled.dns_client", @@ -110,15 +104,12 @@ class Test_compute_network_dns_logging_enabled: dns_client.project_ids = [GCP_PROJECT_ID] dns_client.policies = [policy] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_network_dns_logging_enabled.compute_network_dns_logging_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_network_dns_logging_enabled.compute_network_dns_logging_enabled.compute_client", + new=compute_client, ): with mock.patch( "prowler.providers.gcp.services.compute.compute_network_dns_logging_enabled.compute_network_dns_logging_enabled.dns_client", diff --git a/tests/providers/gcp/services/compute/compute_network_not_legacy/compute_network_not_legacy_test.py b/tests/providers/gcp/services/compute/compute_network_not_legacy/compute_network_not_legacy_test.py index 16c27e07b3..e50902dee2 100644 --- a/tests/providers/gcp/services/compute/compute_network_not_legacy/compute_network_not_legacy_test.py +++ b/tests/providers/gcp/services/compute/compute_network_not_legacy/compute_network_not_legacy_test.py @@ -11,15 +11,12 @@ class Test_compute_network_not_legacy: compute_client.networks = [] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_network_not_legacy.compute_network_not_legacy.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_network_not_legacy.compute_network_not_legacy.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_network_not_legacy.compute_network_not_legacy import ( compute_network_not_legacy, @@ -44,15 +41,12 @@ class Test_compute_network_not_legacy: compute_client.networks = [network] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_network_not_legacy.compute_network_not_legacy.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_network_not_legacy.compute_network_not_legacy.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_network_not_legacy.compute_network_not_legacy import ( compute_network_not_legacy, @@ -87,15 +81,12 @@ class Test_compute_network_not_legacy: compute_client.networks = [network] compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_network_not_legacy.compute_network_not_legacy.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_network_not_legacy.compute_network_not_legacy.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_network_not_legacy.compute_network_not_legacy import ( compute_network_not_legacy, diff --git a/tests/providers/gcp/services/compute/compute_project_os_login_enabled/compute_project_os_login_enabled_test.py b/tests/providers/gcp/services/compute/compute_project_os_login_enabled/compute_project_os_login_enabled_test.py index b1ace7cc62..c20c75f5df 100644 --- a/tests/providers/gcp/services/compute/compute_project_os_login_enabled/compute_project_os_login_enabled_test.py +++ b/tests/providers/gcp/services/compute/compute_project_os_login_enabled/compute_project_os_login_enabled_test.py @@ -11,15 +11,12 @@ class Test_compute_project_os_login_enabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.projects = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_project_os_login_enabled.compute_project_os_login_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_project_os_login_enabled.compute_project_os_login_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_project_os_login_enabled.compute_project_os_login_enabled import ( compute_project_os_login_enabled, @@ -51,15 +48,12 @@ class Test_compute_project_os_login_enabled: } compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_project_os_login_enabled.compute_project_os_login_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_project_os_login_enabled.compute_project_os_login_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_project_os_login_enabled.compute_project_os_login_enabled import ( compute_project_os_login_enabled, @@ -100,15 +94,12 @@ class Test_compute_project_os_login_enabled: } compute_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_project_os_login_enabled.compute_project_os_login_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_project_os_login_enabled.compute_project_os_login_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_project_os_login_enabled.compute_project_os_login_enabled import ( compute_project_os_login_enabled, diff --git a/tests/providers/gcp/services/compute/compute_public_address_shodan/compute_public_address_shodan_test.py b/tests/providers/gcp/services/compute/compute_public_address_shodan/compute_public_address_shodan_test.py index 7560508d54..50a4337dfe 100644 --- a/tests/providers/gcp/services/compute/compute_public_address_shodan/compute_public_address_shodan_test.py +++ b/tests/providers/gcp/services/compute/compute_public_address_shodan/compute_public_address_shodan_test.py @@ -9,15 +9,12 @@ class Test_compute_public_address_shodan: compute_client = mock.MagicMock() compute_client.addresses = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_public_address_shodan.compute_public_address_shodan.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_public_address_shodan.compute_public_address_shodan.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_public_address_shodan.compute_public_address_shodan import ( compute_public_address_shodan, @@ -52,19 +49,15 @@ class Test_compute_public_address_shodan: ) ] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_public_address_shodan.compute_public_address_shodan.compute_client", - new=compute_client, - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_public_address_shodan.compute_public_address_shodan.shodan.Shodan.host", - return_value=shodan_info, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_public_address_shodan.compute_public_address_shodan.compute_client", + new=compute_client, + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_public_address_shodan.compute_public_address_shodan.shodan.Shodan.host", + return_value=shodan_info, ): from prowler.providers.gcp.services.compute.compute_public_address_shodan.compute_public_address_shodan import ( compute_public_address_shodan, diff --git a/tests/providers/gcp/services/compute/compute_serial_ports_in_use/compute_serial_ports_in_use_test.py b/tests/providers/gcp/services/compute/compute_serial_ports_in_use/compute_serial_ports_in_use_test.py index 626adc47bb..2d9afe5d64 100644 --- a/tests/providers/gcp/services/compute/compute_serial_ports_in_use/compute_serial_ports_in_use_test.py +++ b/tests/providers/gcp/services/compute/compute_serial_ports_in_use/compute_serial_ports_in_use_test.py @@ -10,15 +10,12 @@ class Test_compute_instance_serial_ports_in_use: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_serial_ports_in_use.compute_instance_serial_ports_in_use.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_serial_ports_in_use.compute_instance_serial_ports_in_use.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_serial_ports_in_use.compute_instance_serial_ports_in_use import ( compute_instance_serial_ports_in_use, @@ -51,15 +48,12 @@ class Test_compute_instance_serial_ports_in_use: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_serial_ports_in_use.compute_instance_serial_ports_in_use.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_serial_ports_in_use.compute_instance_serial_ports_in_use.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_serial_ports_in_use.compute_instance_serial_ports_in_use import ( compute_instance_serial_ports_in_use, @@ -100,15 +94,12 @@ class Test_compute_instance_serial_ports_in_use: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_serial_ports_in_use.compute_instance_serial_ports_in_use.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_serial_ports_in_use.compute_instance_serial_ports_in_use.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_serial_ports_in_use.compute_instance_serial_ports_in_use import ( compute_instance_serial_ports_in_use, @@ -149,15 +140,12 @@ class Test_compute_instance_serial_ports_in_use: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_serial_ports_in_use.compute_instance_serial_ports_in_use.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_serial_ports_in_use.compute_instance_serial_ports_in_use.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_serial_ports_in_use.compute_instance_serial_ports_in_use import ( compute_instance_serial_ports_in_use, @@ -198,15 +186,12 @@ class Test_compute_instance_serial_ports_in_use: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_serial_ports_in_use.compute_instance_serial_ports_in_use.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_serial_ports_in_use.compute_instance_serial_ports_in_use.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_serial_ports_in_use.compute_instance_serial_ports_in_use import ( compute_instance_serial_ports_in_use, @@ -247,15 +232,12 @@ class Test_compute_instance_serial_ports_in_use: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_serial_ports_in_use.compute_instance_serial_ports_in_use.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_serial_ports_in_use.compute_instance_serial_ports_in_use.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_serial_ports_in_use.compute_instance_serial_ports_in_use import ( compute_instance_serial_ports_in_use, diff --git a/tests/providers/gcp/services/compute/compute_service_test.py b/tests/providers/gcp/services/compute/compute_service_test.py index 24062d670e..2aa43a7a96 100644 --- a/tests/providers/gcp/services/compute/compute_service_test.py +++ b/tests/providers/gcp/services/compute/compute_service_test.py @@ -11,15 +11,12 @@ from tests.providers.gcp.gcp_fixtures import ( class TestComputeService: def test_service(self): - with ( - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", - new=mock_is_api_active, - ), - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", - new=mock_api_client, - ), + with patch( + "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", + new=mock_is_api_active, + ), patch( + "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", + new=mock_api_client, ): compute_client = Compute(set_mocked_gcp_provider([GCP_PROJECT_ID])) assert compute_client.service == "compute" diff --git a/tests/providers/gcp/services/compute/compute_shielded_vm_enabled/compute_shielded_vm_enabled_test.py b/tests/providers/gcp/services/compute/compute_shielded_vm_enabled/compute_shielded_vm_enabled_test.py index a1d9612b34..2c77bb428f 100644 --- a/tests/providers/gcp/services/compute/compute_shielded_vm_enabled/compute_shielded_vm_enabled_test.py +++ b/tests/providers/gcp/services/compute/compute_shielded_vm_enabled/compute_shielded_vm_enabled_test.py @@ -10,15 +10,12 @@ class Test_compute_instance_shielded_vm_enabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_shielded_vm_enabled.compute_instance_shielded_vm_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_shielded_vm_enabled.compute_instance_shielded_vm_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_shielded_vm_enabled.compute_instance_shielded_vm_enabled import ( compute_instance_shielded_vm_enabled, @@ -51,15 +48,12 @@ class Test_compute_instance_shielded_vm_enabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_shielded_vm_enabled.compute_instance_shielded_vm_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_shielded_vm_enabled.compute_instance_shielded_vm_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_shielded_vm_enabled.compute_instance_shielded_vm_enabled import ( compute_instance_shielded_vm_enabled, @@ -100,15 +94,12 @@ class Test_compute_instance_shielded_vm_enabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_shielded_vm_enabled.compute_instance_shielded_vm_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_shielded_vm_enabled.compute_instance_shielded_vm_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_shielded_vm_enabled.compute_instance_shielded_vm_enabled import ( compute_instance_shielded_vm_enabled, @@ -149,15 +140,12 @@ class Test_compute_instance_shielded_vm_enabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.instances = [instance] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_instance_shielded_vm_enabled.compute_instance_shielded_vm_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_instance_shielded_vm_enabled.compute_instance_shielded_vm_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_instance_shielded_vm_enabled.compute_instance_shielded_vm_enabled import ( compute_instance_shielded_vm_enabled, diff --git a/tests/providers/gcp/services/compute/compute_subnet_flow_logs_enabled/compute_subnet_flow_logs_enabled_test.py b/tests/providers/gcp/services/compute/compute_subnet_flow_logs_enabled/compute_subnet_flow_logs_enabled_test.py index ee2b606e65..402c5c4e44 100644 --- a/tests/providers/gcp/services/compute/compute_subnet_flow_logs_enabled/compute_subnet_flow_logs_enabled_test.py +++ b/tests/providers/gcp/services/compute/compute_subnet_flow_logs_enabled/compute_subnet_flow_logs_enabled_test.py @@ -10,15 +10,12 @@ class Test_compute_subnet_flow_logs_enabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.subnets = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_subnet_flow_logs_enabled.compute_subnet_flow_logs_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_subnet_flow_logs_enabled.compute_subnet_flow_logs_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_subnet_flow_logs_enabled.compute_subnet_flow_logs_enabled import ( compute_subnet_flow_logs_enabled, @@ -44,15 +41,12 @@ class Test_compute_subnet_flow_logs_enabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.subnets = [subnet] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_subnet_flow_logs_enabled.compute_subnet_flow_logs_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_subnet_flow_logs_enabled.compute_subnet_flow_logs_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_subnet_flow_logs_enabled.compute_subnet_flow_logs_enabled import ( compute_subnet_flow_logs_enabled, @@ -88,15 +82,12 @@ class Test_compute_subnet_flow_logs_enabled: compute_client.project_ids = [GCP_PROJECT_ID] compute_client.subnets = [subnet] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.compute.compute_subnet_flow_logs_enabled.compute_subnet_flow_logs_enabled.compute_client", - new=compute_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.compute.compute_subnet_flow_logs_enabled.compute_subnet_flow_logs_enabled.compute_client", + new=compute_client, ): from prowler.providers.gcp.services.compute.compute_subnet_flow_logs_enabled.compute_subnet_flow_logs_enabled import ( compute_subnet_flow_logs_enabled, diff --git a/tests/providers/gcp/services/dataproc/dataproc_encrypted_with_cmks_disabled/dataproc_encrypted_with_cmks_disabled_test.py b/tests/providers/gcp/services/dataproc/dataproc_encrypted_with_cmks_disabled/dataproc_encrypted_with_cmks_disabled_test.py index 6d563419a3..8b54535b69 100644 --- a/tests/providers/gcp/services/dataproc/dataproc_encrypted_with_cmks_disabled/dataproc_encrypted_with_cmks_disabled_test.py +++ b/tests/providers/gcp/services/dataproc/dataproc_encrypted_with_cmks_disabled/dataproc_encrypted_with_cmks_disabled_test.py @@ -9,15 +9,12 @@ class Test_dataproc_encrypted_with_cmks_disabled: dataproc_client = mock.MagicMock() dataproc_client.clusters = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.dataproc.dataproc_encrypted_with_cmks_disabled.dataproc_encrypted_with_cmks_disabled.dataproc_client", - new=dataproc_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.dataproc.dataproc_encrypted_with_cmks_disabled.dataproc_encrypted_with_cmks_disabled.dataproc_client", + new=dataproc_client, ): from prowler.providers.gcp.services.dataproc.dataproc_encrypted_with_cmks_disabled.dataproc_encrypted_with_cmks_disabled import ( dataproc_encrypted_with_cmks_disabled, @@ -31,15 +28,12 @@ class Test_dataproc_encrypted_with_cmks_disabled: dataproc_client = mock.MagicMock() dataproc_client.project_ids = [GCP_PROJECT_ID] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.dataproc.dataproc_encrypted_with_cmks_disabled.dataproc_encrypted_with_cmks_disabled.dataproc_client", - new=dataproc_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.dataproc.dataproc_encrypted_with_cmks_disabled.dataproc_encrypted_with_cmks_disabled.dataproc_client", + new=dataproc_client, ): from prowler.providers.gcp.services.dataproc.dataproc_service import Cluster @@ -72,15 +66,12 @@ class Test_dataproc_encrypted_with_cmks_disabled: dataproc_client = mock.MagicMock() dataproc_client.project_ids = [GCP_PROJECT_ID] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.dataproc.dataproc_encrypted_with_cmks_disabled.dataproc_encrypted_with_cmks_disabled.dataproc_client", - new=dataproc_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.dataproc.dataproc_encrypted_with_cmks_disabled.dataproc_encrypted_with_cmks_disabled.dataproc_client", + new=dataproc_client, ): from prowler.providers.gcp.services.dataproc.dataproc_service import Cluster diff --git a/tests/providers/gcp/services/dataproc/dataproc_service_test.py b/tests/providers/gcp/services/dataproc/dataproc_service_test.py index dc69932bdc..60afee5a96 100644 --- a/tests/providers/gcp/services/dataproc/dataproc_service_test.py +++ b/tests/providers/gcp/services/dataproc/dataproc_service_test.py @@ -10,23 +10,18 @@ from tests.providers.gcp.gcp_fixtures import ( class TestDataprocService: def test_service(self): - with ( - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", - new=mock_is_api_active, - ), - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", - new=mock_api_client, - ), - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.dataproc.dataproc_service.compute_client.regions", - new=["europe-west1-b"], - ), + with patch( + "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", + new=mock_is_api_active, + ), patch( + "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", + new=mock_api_client, + ), patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.dataproc.dataproc_service.compute_client.regions", + new=["europe-west1-b"], ): from prowler.providers.gcp.services.dataproc.dataproc_service import ( Dataproc, diff --git a/tests/providers/gcp/services/dns/dns_dnssec_disabled/dns_dnssec_disabled_test.py b/tests/providers/gcp/services/dns/dns_dnssec_disabled/dns_dnssec_disabled_test.py index 9fc8ec5fcb..383207807e 100644 --- a/tests/providers/gcp/services/dns/dns_dnssec_disabled/dns_dnssec_disabled_test.py +++ b/tests/providers/gcp/services/dns/dns_dnssec_disabled/dns_dnssec_disabled_test.py @@ -14,15 +14,12 @@ class Test_dns_dnssec_disabled: dns_client.managed_zones = [] dns_client.region = GCP_EU1_LOCATION - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.dns.dns_dnssec_disabled.dns_dnssec_disabled.dns_client", - new=dns_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.dns.dns_dnssec_disabled.dns_dnssec_disabled.dns_client", + new=dns_client, ): from prowler.providers.gcp.services.dns.dns_dnssec_disabled.dns_dnssec_disabled import ( dns_dnssec_disabled, @@ -61,15 +58,12 @@ class Test_dns_dnssec_disabled: dns_client.managed_zones = [managed_zone] dns_client.region = GCP_EU1_LOCATION - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.dns.dns_dnssec_disabled.dns_dnssec_disabled.dns_client", - new=dns_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.dns.dns_dnssec_disabled.dns_dnssec_disabled.dns_client", + new=dns_client, ): from prowler.providers.gcp.services.dns.dns_dnssec_disabled.dns_dnssec_disabled import ( dns_dnssec_disabled, @@ -115,15 +109,12 @@ class Test_dns_dnssec_disabled: dns_client.managed_zones = [managed_zone] dns_client.region = GCP_EU1_LOCATION - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.dns.dns_dnssec_disabled.dns_dnssec_disabled.dns_client", - new=dns_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.dns.dns_dnssec_disabled.dns_dnssec_disabled.dns_client", + new=dns_client, ): from prowler.providers.gcp.services.dns.dns_dnssec_disabled.dns_dnssec_disabled import ( dns_dnssec_disabled, diff --git a/tests/providers/gcp/services/dns/dns_rsasha1_in_use_to_key_sign_in_dnssec/dns_rsasha1_in_use_to_key_sign_in_dnssec_test.py b/tests/providers/gcp/services/dns/dns_rsasha1_in_use_to_key_sign_in_dnssec/dns_rsasha1_in_use_to_key_sign_in_dnssec_test.py index 8f4e1609b1..4c99fa4b26 100644 --- a/tests/providers/gcp/services/dns/dns_rsasha1_in_use_to_key_sign_in_dnssec/dns_rsasha1_in_use_to_key_sign_in_dnssec_test.py +++ b/tests/providers/gcp/services/dns/dns_rsasha1_in_use_to_key_sign_in_dnssec/dns_rsasha1_in_use_to_key_sign_in_dnssec_test.py @@ -14,15 +14,12 @@ class Test_dns_rsasha1_in_use_to_key_sign_in_dnssec: dns_client.managed_zones = [] dns_client.region = GCP_EU1_LOCATION - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.dns.dns_rsasha1_in_use_to_key_sign_in_dnssec.dns_rsasha1_in_use_to_key_sign_in_dnssec.dns_client", - new=dns_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.dns.dns_rsasha1_in_use_to_key_sign_in_dnssec.dns_rsasha1_in_use_to_key_sign_in_dnssec.dns_client", + new=dns_client, ): from prowler.providers.gcp.services.dns.dns_rsasha1_in_use_to_key_sign_in_dnssec.dns_rsasha1_in_use_to_key_sign_in_dnssec import ( dns_rsasha1_in_use_to_key_sign_in_dnssec, @@ -61,15 +58,12 @@ class Test_dns_rsasha1_in_use_to_key_sign_in_dnssec: dns_client.managed_zones = [managed_zone] dns_client.region = GCP_EU1_LOCATION - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.dns.dns_rsasha1_in_use_to_key_sign_in_dnssec.dns_rsasha1_in_use_to_key_sign_in_dnssec.dns_client", - new=dns_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.dns.dns_rsasha1_in_use_to_key_sign_in_dnssec.dns_rsasha1_in_use_to_key_sign_in_dnssec.dns_client", + new=dns_client, ): from prowler.providers.gcp.services.dns.dns_rsasha1_in_use_to_key_sign_in_dnssec.dns_rsasha1_in_use_to_key_sign_in_dnssec import ( dns_rsasha1_in_use_to_key_sign_in_dnssec, @@ -115,15 +109,12 @@ class Test_dns_rsasha1_in_use_to_key_sign_in_dnssec: dns_client.managed_zones = [managed_zone] dns_client.region = GCP_EU1_LOCATION - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.dns.dns_rsasha1_in_use_to_key_sign_in_dnssec.dns_rsasha1_in_use_to_key_sign_in_dnssec.dns_client", - new=dns_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.dns.dns_rsasha1_in_use_to_key_sign_in_dnssec.dns_rsasha1_in_use_to_key_sign_in_dnssec.dns_client", + new=dns_client, ): from prowler.providers.gcp.services.dns.dns_rsasha1_in_use_to_key_sign_in_dnssec.dns_rsasha1_in_use_to_key_sign_in_dnssec import ( dns_rsasha1_in_use_to_key_sign_in_dnssec, diff --git a/tests/providers/gcp/services/dns/dns_rsasha1_in_use_to_zone_sign_in_dnssec/dns_rsasha1_in_use_to_zone_sign_in_dnssec_test.py b/tests/providers/gcp/services/dns/dns_rsasha1_in_use_to_zone_sign_in_dnssec/dns_rsasha1_in_use_to_zone_sign_in_dnssec_test.py index f6d8734bb8..f0bb105f9d 100644 --- a/tests/providers/gcp/services/dns/dns_rsasha1_in_use_to_zone_sign_in_dnssec/dns_rsasha1_in_use_to_zone_sign_in_dnssec_test.py +++ b/tests/providers/gcp/services/dns/dns_rsasha1_in_use_to_zone_sign_in_dnssec/dns_rsasha1_in_use_to_zone_sign_in_dnssec_test.py @@ -14,15 +14,12 @@ class Test_dns_rsasha1_in_use_to_zone_sign_in_dnssec: dns_client.managed_zones = [] dns_client.region = GCP_EU1_LOCATION - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.dns.dns_rsasha1_in_use_to_zone_sign_in_dnssec.dns_rsasha1_in_use_to_zone_sign_in_dnssec.dns_client", - new=dns_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.dns.dns_rsasha1_in_use_to_zone_sign_in_dnssec.dns_rsasha1_in_use_to_zone_sign_in_dnssec.dns_client", + new=dns_client, ): from prowler.providers.gcp.services.dns.dns_rsasha1_in_use_to_zone_sign_in_dnssec.dns_rsasha1_in_use_to_zone_sign_in_dnssec import ( dns_rsasha1_in_use_to_zone_sign_in_dnssec, @@ -61,15 +58,12 @@ class Test_dns_rsasha1_in_use_to_zone_sign_in_dnssec: dns_client.managed_zones = [managed_zone] dns_client.region = GCP_EU1_LOCATION - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.dns.dns_rsasha1_in_use_to_zone_sign_in_dnssec.dns_rsasha1_in_use_to_zone_sign_in_dnssec.dns_client", - new=dns_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.dns.dns_rsasha1_in_use_to_zone_sign_in_dnssec.dns_rsasha1_in_use_to_zone_sign_in_dnssec.dns_client", + new=dns_client, ): from prowler.providers.gcp.services.dns.dns_rsasha1_in_use_to_zone_sign_in_dnssec.dns_rsasha1_in_use_to_zone_sign_in_dnssec import ( dns_rsasha1_in_use_to_zone_sign_in_dnssec, @@ -115,15 +109,12 @@ class Test_dns_rsasha1_in_use_to_zone_sign_in_dnssec: dns_client.managed_zones = [managed_zone] dns_client.region = GCP_EU1_LOCATION - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.dns.dns_rsasha1_in_use_to_zone_sign_in_dnssec.dns_rsasha1_in_use_to_zone_sign_in_dnssec.dns_client", - new=dns_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.dns.dns_rsasha1_in_use_to_zone_sign_in_dnssec.dns_rsasha1_in_use_to_zone_sign_in_dnssec.dns_client", + new=dns_client, ): from prowler.providers.gcp.services.dns.dns_rsasha1_in_use_to_zone_sign_in_dnssec.dns_rsasha1_in_use_to_zone_sign_in_dnssec import ( dns_rsasha1_in_use_to_zone_sign_in_dnssec, diff --git a/tests/providers/gcp/services/dns/dns_service_test.py b/tests/providers/gcp/services/dns/dns_service_test.py index 44b2f9043b..49d4fc81ee 100644 --- a/tests/providers/gcp/services/dns/dns_service_test.py +++ b/tests/providers/gcp/services/dns/dns_service_test.py @@ -11,15 +11,12 @@ from tests.providers.gcp.gcp_fixtures import ( class TestDNSService: def test_service(self): - with ( - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", - new=mock_is_api_active, - ), - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", - new=mock_api_client, - ), + with patch( + "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", + new=mock_is_api_active, + ), patch( + "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", + new=mock_api_client, ): api_keys_client = DNS(set_mocked_gcp_provider(project_ids=[GCP_PROJECT_ID])) assert api_keys_client.service == "dns" diff --git a/tests/providers/gcp/services/gke/gke_cluster_no_default_service_account/gke_cluster_no_default_service_account_test.py b/tests/providers/gcp/services/gke/gke_cluster_no_default_service_account/gke_cluster_no_default_service_account_test.py index f5188faee5..ad86f37b0c 100644 --- a/tests/providers/gcp/services/gke/gke_cluster_no_default_service_account/gke_cluster_no_default_service_account_test.py +++ b/tests/providers/gcp/services/gke/gke_cluster_no_default_service_account/gke_cluster_no_default_service_account_test.py @@ -9,15 +9,12 @@ class Test_gke_cluster_no_default_service_account: gke_client = mock.MagicMock() gke_client.clusters = {} - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.gke.gke_cluster_no_default_service_account.gke_cluster_no_default_service_account.gke_client", - new=gke_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.gke.gke_cluster_no_default_service_account.gke_cluster_no_default_service_account.gke_client", + new=gke_client, ): from prowler.providers.gcp.services.gke.gke_cluster_no_default_service_account.gke_cluster_no_default_service_account import ( gke_cluster_no_default_service_account, @@ -45,15 +42,12 @@ class Test_gke_cluster_no_default_service_account: gke_client.project_ids = [GCP_PROJECT_ID] gke_client.clusters = clusters - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.gke.gke_cluster_no_default_service_account.gke_cluster_no_default_service_account.gke_client", - new=gke_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.gke.gke_cluster_no_default_service_account.gke_cluster_no_default_service_account.gke_client", + new=gke_client, ): from prowler.providers.gcp.services.gke.gke_cluster_no_default_service_account.gke_cluster_no_default_service_account import ( gke_cluster_no_default_service_account, @@ -91,15 +85,12 @@ class Test_gke_cluster_no_default_service_account: gke_client.project_ids = [GCP_PROJECT_ID] gke_client.clusters = clusters - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.gke.gke_cluster_no_default_service_account.gke_cluster_no_default_service_account.gke_client", - new=gke_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.gke.gke_cluster_no_default_service_account.gke_cluster_no_default_service_account.gke_client", + new=gke_client, ): from prowler.providers.gcp.services.gke.gke_cluster_no_default_service_account.gke_cluster_no_default_service_account import ( gke_cluster_no_default_service_account, @@ -144,15 +135,12 @@ class Test_gke_cluster_no_default_service_account: gke_client.project_ids = [GCP_PROJECT_ID] gke_client.clusters = clusters - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.gke.gke_cluster_no_default_service_account.gke_cluster_no_default_service_account.gke_client", - new=gke_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.gke.gke_cluster_no_default_service_account.gke_cluster_no_default_service_account.gke_client", + new=gke_client, ): from prowler.providers.gcp.services.gke.gke_cluster_no_default_service_account.gke_cluster_no_default_service_account import ( gke_cluster_no_default_service_account, @@ -197,15 +185,12 @@ class Test_gke_cluster_no_default_service_account: gke_client.project_ids = [GCP_PROJECT_ID] gke_client.clusters = clusters - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.gke.gke_cluster_no_default_service_account.gke_cluster_no_default_service_account.gke_client", - new=gke_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.gke.gke_cluster_no_default_service_account.gke_cluster_no_default_service_account.gke_client", + new=gke_client, ): from prowler.providers.gcp.services.gke.gke_cluster_no_default_service_account.gke_cluster_no_default_service_account import ( gke_cluster_no_default_service_account, diff --git a/tests/providers/gcp/services/gke/gke_service_test.py b/tests/providers/gcp/services/gke/gke_service_test.py index 51dc742583..f708c93434 100644 --- a/tests/providers/gcp/services/gke/gke_service_test.py +++ b/tests/providers/gcp/services/gke/gke_service_test.py @@ -11,15 +11,12 @@ from tests.providers.gcp.gcp_fixtures import ( class TestGKEService: def test_service(self): - with ( - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", - new=mock_is_api_active, - ), - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", - new=mock_api_client, - ), + with patch( + "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", + new=mock_is_api_active, + ), patch( + "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", + new=mock_api_client, ): gke_client = GKE(set_mocked_gcp_provider(project_ids=[GCP_PROJECT_ID])) assert gke_client.service == "container" diff --git a/tests/providers/gcp/services/iam/iam_audit_logs_enabled/iam_audit_logs_enabled_test.py b/tests/providers/gcp/services/iam/iam_audit_logs_enabled/iam_audit_logs_enabled_test.py index de6a2ddc6a..4328d5f7fb 100644 --- a/tests/providers/gcp/services/iam/iam_audit_logs_enabled/iam_audit_logs_enabled_test.py +++ b/tests/providers/gcp/services/iam/iam_audit_logs_enabled/iam_audit_logs_enabled_test.py @@ -12,15 +12,12 @@ class Test_iam_audit_logs_enabled: cloudresourcemanager_client.project_ids = [GCP_PROJECT_ID] cloudresourcemanager_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_audit_logs_enabled.iam_audit_logs_enabled.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_audit_logs_enabled.iam_audit_logs_enabled.cloudresourcemanager_client", + new=cloudresourcemanager_client, ): from prowler.providers.gcp.services.iam.iam_audit_logs_enabled.iam_audit_logs_enabled import ( iam_audit_logs_enabled, @@ -51,15 +48,12 @@ class Test_iam_audit_logs_enabled: } cloudresourcemanager_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_audit_logs_enabled.iam_audit_logs_enabled.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_audit_logs_enabled.iam_audit_logs_enabled.cloudresourcemanager_client", + new=cloudresourcemanager_client, ): from prowler.providers.gcp.services.iam.iam_audit_logs_enabled.iam_audit_logs_enabled import ( iam_audit_logs_enabled, @@ -100,15 +94,12 @@ class Test_iam_audit_logs_enabled: } cloudresourcemanager_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_audit_logs_enabled.iam_audit_logs_enabled.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_audit_logs_enabled.iam_audit_logs_enabled.cloudresourcemanager_client", + new=cloudresourcemanager_client, ): from prowler.providers.gcp.services.iam.iam_audit_logs_enabled.iam_audit_logs_enabled import ( iam_audit_logs_enabled, diff --git a/tests/providers/gcp/services/iam/iam_no_service_roles_at_project_level/iam_no_service_roles_at_project_level_test.py b/tests/providers/gcp/services/iam/iam_no_service_roles_at_project_level/iam_no_service_roles_at_project_level_test.py index ee74496f27..a6c16a230e 100644 --- a/tests/providers/gcp/services/iam/iam_no_service_roles_at_project_level/iam_no_service_roles_at_project_level_test.py +++ b/tests/providers/gcp/services/iam/iam_no_service_roles_at_project_level/iam_no_service_roles_at_project_level_test.py @@ -21,19 +21,15 @@ class Test_iam_no_service_roles_at_project_level: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudresourcemanager.cloudresourcemanager_service.CloudResourceManager", - new=cloudresourcemanager_client, - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_no_service_roles_at_project_level.iam_no_service_roles_at_project_level.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudresourcemanager.cloudresourcemanager_service.CloudResourceManager", + new=cloudresourcemanager_client, + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_no_service_roles_at_project_level.iam_no_service_roles_at_project_level.cloudresourcemanager_client", + new=cloudresourcemanager_client, ): from prowler.providers.gcp.services.iam.iam_no_service_roles_at_project_level.iam_no_service_roles_at_project_level import ( iam_no_service_roles_at_project_level, @@ -84,19 +80,15 @@ class Test_iam_no_service_roles_at_project_level: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudresourcemanager.cloudresourcemanager_service.CloudResourceManager", - new=cloudresourcemanager_client, - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_no_service_roles_at_project_level.iam_no_service_roles_at_project_level.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudresourcemanager.cloudresourcemanager_service.CloudResourceManager", + new=cloudresourcemanager_client, + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_no_service_roles_at_project_level.iam_no_service_roles_at_project_level.cloudresourcemanager_client", + new=cloudresourcemanager_client, ): from prowler.providers.gcp.services.iam.iam_no_service_roles_at_project_level.iam_no_service_roles_at_project_level import ( iam_no_service_roles_at_project_level, @@ -133,19 +125,15 @@ class Test_iam_no_service_roles_at_project_level: cloudresourcemanager_client.bindings = [binding] cloudresourcemanager_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudresourcemanager.cloudresourcemanager_service.CloudResourceManager", - new=cloudresourcemanager_client, - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_no_service_roles_at_project_level.iam_no_service_roles_at_project_level.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudresourcemanager.cloudresourcemanager_service.CloudResourceManager", + new=cloudresourcemanager_client, + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_no_service_roles_at_project_level.iam_no_service_roles_at_project_level.cloudresourcemanager_client", + new=cloudresourcemanager_client, ): from prowler.providers.gcp.services.iam.iam_no_service_roles_at_project_level.iam_no_service_roles_at_project_level import ( iam_no_service_roles_at_project_level, @@ -181,19 +169,15 @@ class Test_iam_no_service_roles_at_project_level: cloudresourcemanager_client.bindings = [binding] cloudresourcemanager_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.cloudresourcemanager.cloudresourcemanager_service.CloudResourceManager", - new=cloudresourcemanager_client, - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_no_service_roles_at_project_level.iam_no_service_roles_at_project_level.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.cloudresourcemanager.cloudresourcemanager_service.CloudResourceManager", + new=cloudresourcemanager_client, + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_no_service_roles_at_project_level.iam_no_service_roles_at_project_level.cloudresourcemanager_client", + new=cloudresourcemanager_client, ): from prowler.providers.gcp.services.iam.iam_no_service_roles_at_project_level.iam_no_service_roles_at_project_level import ( iam_no_service_roles_at_project_level, diff --git a/tests/providers/gcp/services/iam/iam_organization_essential_contacts_configured/iam_organization_essential_contacts_configured_test.py b/tests/providers/gcp/services/iam/iam_organization_essential_contacts_configured/iam_organization_essential_contacts_configured_test.py index cd4bc0951e..cd32be1aa3 100644 --- a/tests/providers/gcp/services/iam/iam_organization_essential_contacts_configured/iam_organization_essential_contacts_configured_test.py +++ b/tests/providers/gcp/services/iam/iam_organization_essential_contacts_configured/iam_organization_essential_contacts_configured_test.py @@ -9,15 +9,12 @@ class Test_iam_organization_essential_contacts_configured: essentialcontacts_client = mock.MagicMock() essentialcontacts_client.organizations = [] essentialcontacts_client.region = "global" - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_organization_essential_contacts_configured.iam_organization_essential_contacts_configured.essentialcontacts_client", - new=essentialcontacts_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_organization_essential_contacts_configured.iam_organization_essential_contacts_configured.essentialcontacts_client", + new=essentialcontacts_client, ): from prowler.providers.gcp.services.iam.iam_organization_essential_contacts_configured.iam_organization_essential_contacts_configured import ( iam_organization_essential_contacts_configured, @@ -31,15 +28,12 @@ class Test_iam_organization_essential_contacts_configured: essentialcontacts_client = mock.MagicMock() essentialcontacts_client.region = "global" essentialcontacts_client.organizations = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_organization_essential_contacts_configured.iam_organization_essential_contacts_configured.essentialcontacts_client", - new=essentialcontacts_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_organization_essential_contacts_configured.iam_organization_essential_contacts_configured.essentialcontacts_client", + new=essentialcontacts_client, ): from prowler.providers.gcp.services.iam.iam_service import Organization @@ -68,15 +62,12 @@ class Test_iam_organization_essential_contacts_configured: essentialcontacts_client = mock.MagicMock() essentialcontacts_client.region = "global" essentialcontacts_client.organizations = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_organization_essential_contacts_configured.iam_organization_essential_contacts_configured.essentialcontacts_client", - new=essentialcontacts_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_organization_essential_contacts_configured.iam_organization_essential_contacts_configured.essentialcontacts_client", + new=essentialcontacts_client, ): from prowler.providers.gcp.services.iam.iam_service import Organization diff --git a/tests/providers/gcp/services/iam/iam_role_kms_enforce_separation_of_duties/iam_role_kms_enforce_separation_of_duties_test.py b/tests/providers/gcp/services/iam/iam_role_kms_enforce_separation_of_duties/iam_role_kms_enforce_separation_of_duties_test.py index 6354340e52..1bc0366e91 100644 --- a/tests/providers/gcp/services/iam/iam_role_kms_enforce_separation_of_duties/iam_role_kms_enforce_separation_of_duties_test.py +++ b/tests/providers/gcp/services/iam/iam_role_kms_enforce_separation_of_duties/iam_role_kms_enforce_separation_of_duties_test.py @@ -21,15 +21,12 @@ class Test_iam_role_kms_enforce_separation_of_duties: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_role_kms_enforce_separation_of_duties.iam_role_kms_enforce_separation_of_duties.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_role_kms_enforce_separation_of_duties.iam_role_kms_enforce_separation_of_duties.cloudresourcemanager_client", + new=cloudresourcemanager_client, ): from prowler.providers.gcp.services.iam.iam_role_kms_enforce_separation_of_duties.iam_role_kms_enforce_separation_of_duties import ( iam_role_kms_enforce_separation_of_duties, @@ -83,15 +80,12 @@ class Test_iam_role_kms_enforce_separation_of_duties: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_role_kms_enforce_separation_of_duties.iam_role_kms_enforce_separation_of_duties.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_role_kms_enforce_separation_of_duties.iam_role_kms_enforce_separation_of_duties.cloudresourcemanager_client", + new=cloudresourcemanager_client, ): from prowler.providers.gcp.services.iam.iam_role_kms_enforce_separation_of_duties.iam_role_kms_enforce_separation_of_duties import ( iam_role_kms_enforce_separation_of_duties, @@ -146,15 +140,12 @@ class Test_iam_role_kms_enforce_separation_of_duties: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_role_kms_enforce_separation_of_duties.iam_role_kms_enforce_separation_of_duties.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_role_kms_enforce_separation_of_duties.iam_role_kms_enforce_separation_of_duties.cloudresourcemanager_client", + new=cloudresourcemanager_client, ): from prowler.providers.gcp.services.iam.iam_role_kms_enforce_separation_of_duties.iam_role_kms_enforce_separation_of_duties import ( iam_role_kms_enforce_separation_of_duties, diff --git a/tests/providers/gcp/services/iam/iam_role_sa_enforce_separation_of_duties/iam_role_sa_enforce_separation_of_duties_test.py b/tests/providers/gcp/services/iam/iam_role_sa_enforce_separation_of_duties/iam_role_sa_enforce_separation_of_duties_test.py index 479f540889..c06a82f065 100644 --- a/tests/providers/gcp/services/iam/iam_role_sa_enforce_separation_of_duties/iam_role_sa_enforce_separation_of_duties_test.py +++ b/tests/providers/gcp/services/iam/iam_role_sa_enforce_separation_of_duties/iam_role_sa_enforce_separation_of_duties_test.py @@ -21,15 +21,12 @@ class Test_iam_role_sa_enforce_separation_of_duties: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_role_sa_enforce_separation_of_duties.iam_role_sa_enforce_separation_of_duties.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_role_sa_enforce_separation_of_duties.iam_role_sa_enforce_separation_of_duties.cloudresourcemanager_client", + new=cloudresourcemanager_client, ): from prowler.providers.gcp.services.iam.iam_role_sa_enforce_separation_of_duties.iam_role_sa_enforce_separation_of_duties import ( iam_role_sa_enforce_separation_of_duties, @@ -83,15 +80,12 @@ class Test_iam_role_sa_enforce_separation_of_duties: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_role_sa_enforce_separation_of_duties.iam_role_sa_enforce_separation_of_duties.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_role_sa_enforce_separation_of_duties.iam_role_sa_enforce_separation_of_duties.cloudresourcemanager_client", + new=cloudresourcemanager_client, ): from prowler.providers.gcp.services.iam.iam_role_sa_enforce_separation_of_duties.iam_role_sa_enforce_separation_of_duties import ( iam_role_sa_enforce_separation_of_duties, @@ -146,15 +140,12 @@ class Test_iam_role_sa_enforce_separation_of_duties: ) } - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_role_sa_enforce_separation_of_duties.iam_role_sa_enforce_separation_of_duties.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_role_sa_enforce_separation_of_duties.iam_role_sa_enforce_separation_of_duties.cloudresourcemanager_client", + new=cloudresourcemanager_client, ): from prowler.providers.gcp.services.iam.iam_role_sa_enforce_separation_of_duties.iam_role_sa_enforce_separation_of_duties import ( iam_role_sa_enforce_separation_of_duties, diff --git a/tests/providers/gcp/services/iam/iam_sa_no_administrative_privileges/iam_sa_no_administrative_privileges_test.py b/tests/providers/gcp/services/iam/iam_sa_no_administrative_privileges/iam_sa_no_administrative_privileges_test.py index f55f8f3f10..5d06d77ea4 100644 --- a/tests/providers/gcp/services/iam/iam_sa_no_administrative_privileges/iam_sa_no_administrative_privileges_test.py +++ b/tests/providers/gcp/services/iam/iam_sa_no_administrative_privileges/iam_sa_no_administrative_privileges_test.py @@ -11,15 +11,12 @@ class Test_iam_sa_no_administrative_privileges: def test_iam_no_sa(self): iam_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.iam_client", + new=iam_client, ): from prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges import ( iam_sa_no_administrative_privileges, @@ -37,19 +34,15 @@ class Test_iam_sa_no_administrative_privileges: cloudresourcemanager_client = mock.MagicMock() iam_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.cloudresourcemanager_client", + new=cloudresourcemanager_client, + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.iam_client", + new=iam_client, ): from prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges import ( iam_sa_no_administrative_privileges, @@ -90,19 +83,15 @@ class Test_iam_sa_no_administrative_privileges: cloudresourcemanager_client = mock.MagicMock() iam_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.cloudresourcemanager_client", + new=cloudresourcemanager_client, + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.iam_client", + new=iam_client, ): from prowler.providers.gcp.services.cloudresourcemanager.cloudresourcemanager_service import ( Binding, @@ -155,19 +144,15 @@ class Test_iam_sa_no_administrative_privileges: cloudresourcemanager_client = mock.MagicMock() iam_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.cloudresourcemanager_client", + new=cloudresourcemanager_client, + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.iam_client", + new=iam_client, ): from prowler.providers.gcp.services.cloudresourcemanager.cloudresourcemanager_service import ( Binding, @@ -220,19 +205,15 @@ class Test_iam_sa_no_administrative_privileges: cloudresourcemanager_client = mock.MagicMock() iam_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.cloudresourcemanager_client", + new=cloudresourcemanager_client, + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.iam_client", + new=iam_client, ): from prowler.providers.gcp.services.cloudresourcemanager.cloudresourcemanager_service import ( Binding, @@ -285,19 +266,15 @@ class Test_iam_sa_no_administrative_privileges: cloudresourcemanager_client = mock.MagicMock() iam_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.cloudresourcemanager_client", + new=cloudresourcemanager_client, + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.iam_client", + new=iam_client, ): from prowler.providers.gcp.services.cloudresourcemanager.cloudresourcemanager_service import ( Binding, @@ -350,19 +327,15 @@ class Test_iam_sa_no_administrative_privileges: cloudresourcemanager_client = mock.MagicMock() iam_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.cloudresourcemanager_client", + new=cloudresourcemanager_client, + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.iam_client", + new=iam_client, ): from prowler.providers.gcp.services.cloudresourcemanager.cloudresourcemanager_service import ( Binding, @@ -415,19 +388,15 @@ class Test_iam_sa_no_administrative_privileges: cloudresourcemanager_client = mock.MagicMock() iam_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.cloudresourcemanager_client", + new=cloudresourcemanager_client, + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.iam_client", + new=iam_client, ): from prowler.providers.gcp.services.cloudresourcemanager.cloudresourcemanager_service import ( Binding, @@ -480,19 +449,15 @@ class Test_iam_sa_no_administrative_privileges: cloudresourcemanager_client = mock.MagicMock() iam_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.cloudresourcemanager_client", - new=cloudresourcemanager_client, - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.cloudresourcemanager_client", + new=cloudresourcemanager_client, + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_administrative_privileges.iam_sa_no_administrative_privileges.iam_client", + new=iam_client, ): from prowler.providers.gcp.services.cloudresourcemanager.cloudresourcemanager_service import ( Binding, diff --git a/tests/providers/gcp/services/iam/iam_sa_no_user_managed_keys/iam_sa_no_user_managed_keys_test.py b/tests/providers/gcp/services/iam/iam_sa_no_user_managed_keys/iam_sa_no_user_managed_keys_test.py index 9b49a21271..f4b1725a71 100644 --- a/tests/providers/gcp/services/iam/iam_sa_no_user_managed_keys/iam_sa_no_user_managed_keys_test.py +++ b/tests/providers/gcp/services/iam/iam_sa_no_user_managed_keys/iam_sa_no_user_managed_keys_test.py @@ -12,15 +12,12 @@ class Test_iam_sa_no_user_managed_keys: def test_iam_no_sa(self): iam_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_user_managed_keys.iam_sa_no_user_managed_keys.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_user_managed_keys.iam_sa_no_user_managed_keys.iam_client", + new=iam_client, ): from prowler.providers.gcp.services.iam.iam_sa_no_user_managed_keys.iam_sa_no_user_managed_keys import ( iam_sa_no_user_managed_keys, @@ -37,15 +34,12 @@ class Test_iam_sa_no_user_managed_keys: def test_iam_sa_no_keys(self): iam_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_user_managed_keys.iam_sa_no_user_managed_keys.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_user_managed_keys.iam_sa_no_user_managed_keys.iam_client", + new=iam_client, ): from prowler.providers.gcp.services.iam.iam_sa_no_user_managed_keys.iam_sa_no_user_managed_keys import ( iam_sa_no_user_managed_keys, @@ -81,15 +75,12 @@ class Test_iam_sa_no_user_managed_keys: def test_iam_sa_no_user_managed_keys(self): iam_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_user_managed_keys.iam_sa_no_user_managed_keys.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_user_managed_keys.iam_sa_no_user_managed_keys.iam_client", + new=iam_client, ): from prowler.providers.gcp.services.iam.iam_sa_no_user_managed_keys.iam_sa_no_user_managed_keys import ( iam_sa_no_user_managed_keys, @@ -136,15 +127,12 @@ class Test_iam_sa_no_user_managed_keys: def test_iam_sa_user_managed_keys(self): iam_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_user_managed_keys.iam_sa_no_user_managed_keys.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_user_managed_keys.iam_sa_no_user_managed_keys.iam_client", + new=iam_client, ): from prowler.providers.gcp.services.iam.iam_sa_no_user_managed_keys.iam_sa_no_user_managed_keys import ( iam_sa_no_user_managed_keys, @@ -191,15 +179,12 @@ class Test_iam_sa_no_user_managed_keys: def test_iam_sa_mixed_keys(self): iam_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_no_user_managed_keys.iam_sa_no_user_managed_keys.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_no_user_managed_keys.iam_sa_no_user_managed_keys.iam_client", + new=iam_client, ): from prowler.providers.gcp.services.iam.iam_sa_no_user_managed_keys.iam_sa_no_user_managed_keys import ( iam_sa_no_user_managed_keys, diff --git a/tests/providers/gcp/services/iam/iam_sa_user_managed_key_rotate_90_days/iam_sa_user_managed_key_rotate_90_days_test.py b/tests/providers/gcp/services/iam/iam_sa_user_managed_key_rotate_90_days/iam_sa_user_managed_key_rotate_90_days_test.py index 934f4f3222..c803111a9d 100644 --- a/tests/providers/gcp/services/iam/iam_sa_user_managed_key_rotate_90_days/iam_sa_user_managed_key_rotate_90_days_test.py +++ b/tests/providers/gcp/services/iam/iam_sa_user_managed_key_rotate_90_days/iam_sa_user_managed_key_rotate_90_days_test.py @@ -12,15 +12,12 @@ class Test_iam_sa_user_managed_key_rotate_90_days: def test_iam_no_sa(self): iam_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_user_managed_key_rotate_90_days.iam_sa_user_managed_key_rotate_90_days.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_user_managed_key_rotate_90_days.iam_sa_user_managed_key_rotate_90_days.iam_client", + new=iam_client, ): from prowler.providers.gcp.services.iam.iam_sa_user_managed_key_rotate_90_days.iam_sa_user_managed_key_rotate_90_days import ( iam_sa_user_managed_key_rotate_90_days, @@ -37,15 +34,12 @@ class Test_iam_sa_user_managed_key_rotate_90_days: def test_iam_sa_no_keys(self): iam_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_user_managed_key_rotate_90_days.iam_sa_user_managed_key_rotate_90_days.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_user_managed_key_rotate_90_days.iam_sa_user_managed_key_rotate_90_days.iam_client", + new=iam_client, ): from prowler.providers.gcp.services.iam.iam_sa_user_managed_key_rotate_90_days.iam_sa_user_managed_key_rotate_90_days import ( iam_sa_user_managed_key_rotate_90_days, @@ -72,15 +66,12 @@ class Test_iam_sa_user_managed_key_rotate_90_days: def test_iam_sa_user_managed_key_rotate_90_days(self): iam_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_user_managed_key_rotate_90_days.iam_sa_user_managed_key_rotate_90_days.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_user_managed_key_rotate_90_days.iam_sa_user_managed_key_rotate_90_days.iam_client", + new=iam_client, ): from prowler.providers.gcp.services.iam.iam_sa_user_managed_key_rotate_90_days.iam_sa_user_managed_key_rotate_90_days import ( iam_sa_user_managed_key_rotate_90_days, @@ -130,15 +121,12 @@ class Test_iam_sa_user_managed_key_rotate_90_days: def test_iam_sa_user_managed_key_no_rotate_90_days(self): iam_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.iam.iam_sa_user_managed_key_rotate_90_days.iam_sa_user_managed_key_rotate_90_days.iam_client", - new=iam_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.iam.iam_sa_user_managed_key_rotate_90_days.iam_sa_user_managed_key_rotate_90_days.iam_client", + new=iam_client, ): from prowler.providers.gcp.services.iam.iam_sa_user_managed_key_rotate_90_days.iam_sa_user_managed_key_rotate_90_days import ( iam_sa_user_managed_key_rotate_90_days, diff --git a/tests/providers/gcp/services/iam/iamgcp_service_test.py b/tests/providers/gcp/services/iam/iamgcp_service_test.py index 3b40b89858..ad0f35547b 100644 --- a/tests/providers/gcp/services/iam/iamgcp_service_test.py +++ b/tests/providers/gcp/services/iam/iamgcp_service_test.py @@ -14,19 +14,15 @@ from tests.providers.gcp.gcp_fixtures import ( class TestIAMService: def test_service(self): - with ( - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", - new=mock_is_api_active, - ), - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", - new=mock_api_client, - ), - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), + with patch( + "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", + new=mock_is_api_active, + ), patch( + "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", + new=mock_api_client, + ), patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), ): from prowler.providers.gcp.services.iam.iam_service import IAM @@ -80,19 +76,15 @@ class TestIAMService: class TestAccessApproval: def test_service(self): - with ( - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", - new=mock_is_api_active, - ), - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", - new=mock_api_client, - ), - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), + with patch( + "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", + new=mock_is_api_active, + ), patch( + "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", + new=mock_api_client, + ), patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), ): from prowler.providers.gcp.services.iam.iam_service import AccessApproval @@ -115,24 +107,19 @@ class TestAccessApproval: class TestEssentialContacts: def test_service(self): - with ( - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", - new=mock_is_api_active, - ), - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", - new=mock_api_client, - ), - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( # Reinstancing the CloudResourceManager client to secure that is not instancied first by a test - "prowler.providers.gcp.services.iam.iam_service.cloudresourcemanager_client", - new=CloudResourceManager( - set_mocked_gcp_provider(), - ), + with patch( + "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", + new=mock_is_api_active, + ), patch( + "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", + new=mock_api_client, + ), patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( # Reinstancing the CloudResourceManager client to secure that is not instancied first by a test + "prowler.providers.gcp.services.iam.iam_service.cloudresourcemanager_client", + new=CloudResourceManager( + set_mocked_gcp_provider(), ), ): from prowler.providers.gcp.services.iam.iam_service import EssentialContacts diff --git a/tests/providers/gcp/services/kms/kms_key_not_publicly_accessible/kms_key_not_publicly_accessible_gcp_test.py b/tests/providers/gcp/services/kms/kms_key_not_publicly_accessible/kms_key_not_publicly_accessible_gcp_test.py index 53e9f5c0df..6ecb07be4b 100644 --- a/tests/providers/gcp/services/kms/kms_key_not_publicly_accessible/kms_key_not_publicly_accessible_gcp_test.py +++ b/tests/providers/gcp/services/kms/kms_key_not_publicly_accessible/kms_key_not_publicly_accessible_gcp_test.py @@ -11,15 +11,12 @@ class Test_kms_key_not_publicly_accessible_gcp: def test_kms_no_key(self): kms_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.kms.kms_key_not_publicly_accessible.kms_key_not_publicly_accessible.kms_client", - new=kms_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.kms.kms_key_not_publicly_accessible.kms_key_not_publicly_accessible.kms_client", + new=kms_client, ): from prowler.providers.gcp.services.kms.kms_key_not_publicly_accessible.kms_key_not_publicly_accessible import ( kms_key_not_publicly_accessible, @@ -36,15 +33,12 @@ class Test_kms_key_not_publicly_accessible_gcp: def test_kms_key_public(self): kms_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.kms.kms_key_not_publicly_accessible.kms_key_not_publicly_accessible.kms_client", - new=kms_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.kms.kms_key_not_publicly_accessible.kms_key_not_publicly_accessible.kms_client", + new=kms_client, ): from prowler.providers.gcp.services.kms.kms_key_not_publicly_accessible.kms_key_not_publicly_accessible import ( kms_key_not_publicly_accessible, @@ -97,15 +91,12 @@ class Test_kms_key_not_publicly_accessible_gcp: def test_kms_key_private(self): kms_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.kms.kms_key_not_publicly_accessible.kms_key_not_publicly_accessible.kms_client", - new=kms_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.kms.kms_key_not_publicly_accessible.kms_key_not_publicly_accessible.kms_client", + new=kms_client, ): from prowler.providers.gcp.services.kms.kms_key_not_publicly_accessible.kms_key_not_publicly_accessible import ( kms_key_not_publicly_accessible, @@ -158,15 +149,12 @@ class Test_kms_key_not_publicly_accessible_gcp: def test_kms_key_no_member(self): kms_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.kms.kms_key_not_publicly_accessible.kms_key_not_publicly_accessible.kms_client", - new=kms_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.kms.kms_key_not_publicly_accessible.kms_key_not_publicly_accessible.kms_client", + new=kms_client, ): from prowler.providers.gcp.services.kms.kms_key_not_publicly_accessible.kms_key_not_publicly_accessible import ( kms_key_not_publicly_accessible, diff --git a/tests/providers/gcp/services/kms/kms_key_rotation_enabled/kms_key_rotation_enabled_test.py b/tests/providers/gcp/services/kms/kms_key_rotation_enabled/kms_key_rotation_enabled_test.py index 5776c858f5..f0761be29f 100644 --- a/tests/providers/gcp/services/kms/kms_key_rotation_enabled/kms_key_rotation_enabled_test.py +++ b/tests/providers/gcp/services/kms/kms_key_rotation_enabled/kms_key_rotation_enabled_test.py @@ -12,15 +12,12 @@ class Test_kms_key_rotation_enabled: def test_kms_no_key(self): kms_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", - new=kms_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", + new=kms_client, ): from prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled import ( kms_key_rotation_enabled, @@ -37,15 +34,12 @@ class Test_kms_key_rotation_enabled: def test_kms_key_no_next_rotation_time_and_no_rotation_period(self): kms_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", - new=kms_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", + new=kms_client, ): from prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled import ( kms_key_rotation_enabled, @@ -96,15 +90,12 @@ class Test_kms_key_rotation_enabled: def test_kms_key_no_next_rotation_time_and_big_rotation_period(self): kms_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", - new=kms_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", + new=kms_client, ): from prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled import ( kms_key_rotation_enabled, @@ -156,15 +147,12 @@ class Test_kms_key_rotation_enabled: def test_kms_key_no_next_rotation_time_and_appropriate_rotation_period(self): kms_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", - new=kms_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", + new=kms_client, ): from prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled import ( kms_key_rotation_enabled, @@ -216,15 +204,12 @@ class Test_kms_key_rotation_enabled: def test_kms_key_no_rotation_period_and_big_next_rotation_time(self): kms_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", - new=kms_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", + new=kms_client, ): from prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled import ( kms_key_rotation_enabled, @@ -279,15 +264,12 @@ class Test_kms_key_rotation_enabled: def test_kms_key_no_rotation_period_and_appropriate_next_rotation_time(self): kms_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", - new=kms_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", + new=kms_client, ): from prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled import ( kms_key_rotation_enabled, @@ -342,15 +324,12 @@ class Test_kms_key_rotation_enabled: def test_kms_key_rotation_period_greater_90_days_and_big_next_rotation_time(self): kms_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", - new=kms_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", + new=kms_client, ): from prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled import ( kms_key_rotation_enabled, @@ -408,15 +387,12 @@ class Test_kms_key_rotation_enabled: ): kms_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", - new=kms_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", + new=kms_client, ): from prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled import ( kms_key_rotation_enabled, @@ -472,15 +448,12 @@ class Test_kms_key_rotation_enabled: def test_kms_key_rotation_period_less_90_days_and_big_next_rotation_time(self): kms_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", - new=kms_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", + new=kms_client, ): from prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled import ( kms_key_rotation_enabled, @@ -538,15 +511,12 @@ class Test_kms_key_rotation_enabled: ): kms_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", - new=kms_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", + new=kms_client, ): from prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled import ( kms_key_rotation_enabled, @@ -602,15 +572,12 @@ class Test_kms_key_rotation_enabled: def test_kms_key_rotation_with_fractional_seconds(self): kms_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - mock.patch( - "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", - new=kms_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), mock.patch( + "prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled.kms_client", + new=kms_client, ): from prowler.providers.gcp.services.kms.kms_key_rotation_enabled.kms_key_rotation_enabled import ( kms_key_rotation_enabled, diff --git a/tests/providers/gcp/services/kms/kmsgcp_service_test.py b/tests/providers/gcp/services/kms/kmsgcp_service_test.py index 9fc537f1d9..1e7bda46b1 100644 --- a/tests/providers/gcp/services/kms/kmsgcp_service_test.py +++ b/tests/providers/gcp/services/kms/kmsgcp_service_test.py @@ -11,15 +11,12 @@ from tests.providers.gcp.gcp_fixtures import ( class TestKMSService: def test_service(self): - with ( - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", - new=mock_is_api_active, - ), - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", - new=mock_api_client, - ), + with patch( + "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", + new=mock_is_api_active, + ), patch( + "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", + new=mock_api_client, ): kms_client = KMS(set_mocked_gcp_provider(project_ids=[GCP_PROJECT_ID])) assert kms_client.service == "cloudkms" diff --git a/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled/logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled_test.py b/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled/logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled_test.py index 7dd0a4ca56..f028c074e8 100644 --- a/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled/logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled_test.py +++ b/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled/logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled_test.py @@ -13,19 +13,15 @@ class Test_logging_log_metric_filter_and_alert_for_audit_configuration_changes_e logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled import ( logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled, @@ -45,19 +41,15 @@ class Test_logging_log_metric_filter_and_alert_for_audit_configuration_changes_e logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled import ( logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled, @@ -97,19 +89,15 @@ class Test_logging_log_metric_filter_and_alert_for_audit_configuration_changes_e logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled import ( logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled, @@ -148,19 +136,15 @@ class Test_logging_log_metric_filter_and_alert_for_audit_configuration_changes_e logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled import ( logging_log_metric_filter_and_alert_for_audit_configuration_changes_enabled, diff --git a/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled/logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled_test.py b/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled/logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled_test.py index 17cf202c76..aae14998c7 100644 --- a/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled/logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled_test.py +++ b/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled/logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled_test.py @@ -13,19 +13,15 @@ class Test_logging_log_metric_filter_and_alert_for_bucket_permission_changes_ena logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled import ( logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled, @@ -45,19 +41,15 @@ class Test_logging_log_metric_filter_and_alert_for_bucket_permission_changes_ena logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled import ( logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled, @@ -97,19 +89,15 @@ class Test_logging_log_metric_filter_and_alert_for_bucket_permission_changes_ena logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled import ( logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled, @@ -148,19 +136,15 @@ class Test_logging_log_metric_filter_and_alert_for_bucket_permission_changes_ena logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled.logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled import ( logging_log_metric_filter_and_alert_for_bucket_permission_changes_enabled, diff --git a/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_custom_role_changes_enabled/logging_log_metric_filter_and_alert_for_custom_role_changes_enabled_test.py b/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_custom_role_changes_enabled/logging_log_metric_filter_and_alert_for_custom_role_changes_enabled_test.py index d08cabcfb3..11b8453bc0 100644 --- a/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_custom_role_changes_enabled/logging_log_metric_filter_and_alert_for_custom_role_changes_enabled_test.py +++ b/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_custom_role_changes_enabled/logging_log_metric_filter_and_alert_for_custom_role_changes_enabled_test.py @@ -13,19 +13,15 @@ class Test_logging_log_metric_filter_and_alert_for_custom_role_changes_enabled: logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled import ( logging_log_metric_filter_and_alert_for_custom_role_changes_enabled, @@ -45,19 +41,15 @@ class Test_logging_log_metric_filter_and_alert_for_custom_role_changes_enabled: logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled import ( logging_log_metric_filter_and_alert_for_custom_role_changes_enabled, @@ -97,19 +89,15 @@ class Test_logging_log_metric_filter_and_alert_for_custom_role_changes_enabled: logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled import ( logging_log_metric_filter_and_alert_for_custom_role_changes_enabled, @@ -148,19 +136,15 @@ class Test_logging_log_metric_filter_and_alert_for_custom_role_changes_enabled: logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled.logging_log_metric_filter_and_alert_for_custom_role_changes_enabled import ( logging_log_metric_filter_and_alert_for_custom_role_changes_enabled, diff --git a/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled/logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled_test.py b/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled/logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled_test.py index 6687c5d3bd..285dfffbfa 100644 --- a/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled/logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled_test.py +++ b/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled/logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled_test.py @@ -13,19 +13,15 @@ class Test_logging_log_metric_filter_and_alert_for_project_ownership_changes_ena logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled import ( logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled, @@ -45,19 +41,15 @@ class Test_logging_log_metric_filter_and_alert_for_project_ownership_changes_ena logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled import ( logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled, @@ -97,19 +89,15 @@ class Test_logging_log_metric_filter_and_alert_for_project_ownership_changes_ena logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled import ( logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled, @@ -148,19 +136,15 @@ class Test_logging_log_metric_filter_and_alert_for_project_ownership_changes_ena logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled.logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled import ( logging_log_metric_filter_and_alert_for_project_ownership_changes_enabled, diff --git a/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled/logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled_test.py b/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled/logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled_test.py index d4bdfaa579..733b0b58f6 100644 --- a/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled/logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled_test.py +++ b/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled/logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled_test.py @@ -13,19 +13,15 @@ class Test_logging_log_metric_filter_and_alert_for_sql_instance_configuration_ch logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled import ( logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled, @@ -45,19 +41,15 @@ class Test_logging_log_metric_filter_and_alert_for_sql_instance_configuration_ch logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled import ( logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled, @@ -97,19 +89,15 @@ class Test_logging_log_metric_filter_and_alert_for_sql_instance_configuration_ch logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled import ( logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled, @@ -148,19 +136,15 @@ class Test_logging_log_metric_filter_and_alert_for_sql_instance_configuration_ch logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled.logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled import ( logging_log_metric_filter_and_alert_for_sql_instance_configuration_changes_enabled, diff --git a/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled/logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled_test.py b/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled/logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled_test.py index cc8dfc6338..1dc34eb7b2 100644 --- a/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled/logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled_test.py +++ b/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled/logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled_test.py @@ -13,19 +13,15 @@ class Test_logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_ena logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled import ( logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled, @@ -45,19 +41,15 @@ class Test_logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_ena logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled import ( logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled, @@ -97,19 +89,15 @@ class Test_logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_ena logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled import ( logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled, @@ -148,19 +136,15 @@ class Test_logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_ena logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled import ( logging_log_metric_filter_and_alert_for_vpc_firewall_rule_changes_enabled, diff --git a/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled/logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled_test.py b/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled/logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled_test.py index d9a166f33e..11d7a56961 100644 --- a/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled/logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled_test.py +++ b/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled/logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled_test.py @@ -13,19 +13,15 @@ class Test_logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled: logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled import ( logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled, @@ -45,19 +41,15 @@ class Test_logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled: logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled import ( logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled, @@ -97,19 +89,15 @@ class Test_logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled: logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled import ( logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled, @@ -148,19 +136,15 @@ class Test_logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled: logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled import ( logging_log_metric_filter_and_alert_for_vpc_network_changes_enabled, diff --git a/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled/logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled_test.py b/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled/logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled_test.py index f96835238c..300235adb9 100644 --- a/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled/logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled_test.py +++ b/tests/providers/gcp/services/logging/logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled/logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled_test.py @@ -13,19 +13,15 @@ class Test_logging_log_metric_filter_and_alert_for_vpc_network_route_changes_ena logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled import ( logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled, @@ -45,19 +41,15 @@ class Test_logging_log_metric_filter_and_alert_for_vpc_network_route_changes_ena logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled import ( logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled, @@ -97,19 +89,15 @@ class Test_logging_log_metric_filter_and_alert_for_vpc_network_route_changes_ena logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled import ( logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled, @@ -148,19 +136,15 @@ class Test_logging_log_metric_filter_and_alert_for_vpc_network_route_changes_ena logging_client = MagicMock() monitoring_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_client", - new=logging_client, - ), - patch( - "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.monitoring_client", - new=monitoring_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_client", + new=logging_client, + ), patch( + "prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.monitoring_client", + new=monitoring_client, ): from prowler.providers.gcp.services.logging.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled.logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled import ( logging_log_metric_filter_and_alert_for_vpc_network_route_changes_enabled, diff --git a/tests/providers/gcp/services/logging/logging_service_test.py b/tests/providers/gcp/services/logging/logging_service_test.py index 0396130c2f..31c9338bc1 100644 --- a/tests/providers/gcp/services/logging/logging_service_test.py +++ b/tests/providers/gcp/services/logging/logging_service_test.py @@ -11,15 +11,12 @@ from tests.providers.gcp.gcp_fixtures import ( class TestLoggingService: def test_service(self): - with ( - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", - new=mock_is_api_active, - ), - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", - new=mock_api_client, - ), + with patch( + "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", + new=mock_is_api_active, + ), patch( + "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", + new=mock_api_client, ): logging_client = Logging( set_mocked_gcp_provider(project_ids=[GCP_PROJECT_ID]) diff --git a/tests/providers/gcp/services/logging/logging_sink_created/logging_sink_created_test.py b/tests/providers/gcp/services/logging/logging_sink_created/logging_sink_created_test.py index 7ad8f9a51f..66b1fd2f1a 100644 --- a/tests/providers/gcp/services/logging/logging_sink_created/logging_sink_created_test.py +++ b/tests/providers/gcp/services/logging/logging_sink_created/logging_sink_created_test.py @@ -12,15 +12,12 @@ class Test_logging_sink_created: def test_no_projects(self): logging_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_sink_created.logging_sink_created.logging_client", - new=logging_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_sink_created.logging_sink_created.logging_client", + new=logging_client, ): from prowler.providers.gcp.services.logging.logging_sink_created.logging_sink_created import ( logging_sink_created, @@ -36,15 +33,12 @@ class Test_logging_sink_created: def test_no_sinks(self): logging_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_sink_created.logging_sink_created.logging_client", - new=logging_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_sink_created.logging_sink_created.logging_client", + new=logging_client, ): from prowler.providers.gcp.services.logging.logging_sink_created.logging_sink_created import ( logging_sink_created, @@ -79,15 +73,12 @@ class Test_logging_sink_created: def test_sink_all(self): logging_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_sink_created.logging_sink_created.logging_client", - new=logging_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_sink_created.logging_sink_created.logging_client", + new=logging_client, ): from prowler.providers.gcp.services.logging.logging_service import Sink from prowler.providers.gcp.services.logging.logging_sink_created.logging_sink_created import ( @@ -121,15 +112,12 @@ class Test_logging_sink_created: def test_sink_not_all(self): logging_client = MagicMock() - with ( - patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_gcp_provider(), - ), - patch( - "prowler.providers.gcp.services.logging.logging_sink_created.logging_sink_created.logging_client", - new=logging_client, - ), + with patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_gcp_provider(), + ), patch( + "prowler.providers.gcp.services.logging.logging_sink_created.logging_sink_created.logging_client", + new=logging_client, ): from prowler.providers.gcp.services.logging.logging_service import Sink from prowler.providers.gcp.services.logging.logging_sink_created.logging_sink_created import ( diff --git a/tests/providers/gcp/services/monitoring/monitoring_service_test.py b/tests/providers/gcp/services/monitoring/monitoring_service_test.py index 0db573a99d..d99f53efd1 100644 --- a/tests/providers/gcp/services/monitoring/monitoring_service_test.py +++ b/tests/providers/gcp/services/monitoring/monitoring_service_test.py @@ -11,15 +11,12 @@ from tests.providers.gcp.gcp_fixtures import ( class TestMonitoringService: def test_service(self): - with ( - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", - new=mock_is_api_active, - ), - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", - new=mock_api_client, - ), + with patch( + "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", + new=mock_is_api_active, + ), patch( + "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", + new=mock_api_client, ): monitoring_client = Monitoring( set_mocked_gcp_provider(project_ids=[GCP_PROJECT_ID]) diff --git a/tests/providers/gcp/services/serviceusage/serviceusage_test.py b/tests/providers/gcp/services/serviceusage/serviceusage_test.py index 4f15800019..223f33c336 100644 --- a/tests/providers/gcp/services/serviceusage/serviceusage_test.py +++ b/tests/providers/gcp/services/serviceusage/serviceusage_test.py @@ -13,15 +13,12 @@ from tests.providers.gcp.gcp_fixtures import ( class TestServiceUsageService: def test_service(self): - with ( - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", - new=mock_is_api_active, - ), - patch( - "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", - new=mock_api_client, - ), + with patch( + "prowler.providers.gcp.lib.service.service.GCPService.__is_api_active__", + new=mock_is_api_active, + ), patch( + "prowler.providers.gcp.lib.service.service.GCPService.__generate_client__", + new=mock_api_client, ): serviceusage_client = ServiceUsage( set_mocked_gcp_provider(project_ids=[GCP_PROJECT_ID]) diff --git a/tests/providers/kubernetes/kubernetes_provider_test.py b/tests/providers/kubernetes/kubernetes_provider_test.py index 681e9b9b39..5238772311 100644 --- a/tests/providers/kubernetes/kubernetes_provider_test.py +++ b/tests/providers/kubernetes/kubernetes_provider_test.py @@ -37,18 +37,15 @@ class TestKubernetesProvider: "cluster": "test-cluster", }, } - with ( - patch( - "prowler.providers.kubernetes.kubernetes_provider.KubernetesProvider.setup_session", - return_value=KubernetesSession( - api_client=client.ApiClient, - context=context, - ), - ), - patch( - "prowler.providers.kubernetes.kubernetes_provider.KubernetesProvider.get_all_namespaces", - return_value=["namespace-1"], + with patch( + "prowler.providers.kubernetes.kubernetes_provider.KubernetesProvider.setup_session", + return_value=KubernetesSession( + api_client=client.ApiClient, + context=context, ), + ), patch( + "prowler.providers.kubernetes.kubernetes_provider.KubernetesProvider.get_all_namespaces", + return_value=["namespace-1"], ): arguments = Namespace() arguments.kubeconfig_file = "dummy_path" diff --git a/tests/providers/microsoft365/services/admincenter/admincenter_groups_not_public_visibility/admincenter_groups_not_public_visibility_test.py b/tests/providers/microsoft365/services/admincenter/admincenter_groups_not_public_visibility/admincenter_groups_not_public_visibility_test.py index 4f7c5c2bc0..8cccfdb027 100644 --- a/tests/providers/microsoft365/services/admincenter/admincenter_groups_not_public_visibility/admincenter_groups_not_public_visibility_test.py +++ b/tests/providers/microsoft365/services/admincenter/admincenter_groups_not_public_visibility/admincenter_groups_not_public_visibility_test.py @@ -13,15 +13,12 @@ class Test_admincenter_groups_not_public_visibility: admincenter_client.audited_tenant = "audited_tenant" admincenter_client.audited_domain = DOMAIN - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_microsoft365_provider(), - ), - mock.patch( - "prowler.providers.microsoft365.services.admincenter.admincenter_groups_not_public_visibility.admincenter_groups_not_public_visibility.admincenter_client", - new=admincenter_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_microsoft365_provider(), + ), mock.patch( + "prowler.providers.microsoft365.services.admincenter.admincenter_groups_not_public_visibility.admincenter_groups_not_public_visibility.admincenter_client", + new=admincenter_client, ): from prowler.providers.microsoft365.services.admincenter.admincenter_groups_not_public_visibility.admincenter_groups_not_public_visibility import ( admincenter_groups_not_public_visibility, @@ -38,15 +35,12 @@ class Test_admincenter_groups_not_public_visibility: admincenter_client.audited_tenant = "audited_tenant" admincenter_client.audited_domain = DOMAIN - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_microsoft365_provider(), - ), - mock.patch( - "prowler.providers.microsoft365.services.admincenter.admincenter_groups_not_public_visibility.admincenter_groups_not_public_visibility.admincenter_client", - new=admincenter_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_microsoft365_provider(), + ), mock.patch( + "prowler.providers.microsoft365.services.admincenter.admincenter_groups_not_public_visibility.admincenter_groups_not_public_visibility.admincenter_client", + new=admincenter_client, ): from prowler.providers.microsoft365.services.admincenter.admincenter_groups_not_public_visibility.admincenter_groups_not_public_visibility import ( admincenter_groups_not_public_visibility, @@ -76,15 +70,12 @@ class Test_admincenter_groups_not_public_visibility: admincenter_client.audited_tenant = "audited_tenant" admincenter_client.audited_domain = DOMAIN - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_microsoft365_provider(), - ), - mock.patch( - "prowler.providers.microsoft365.services.admincenter.admincenter_groups_not_public_visibility.admincenter_groups_not_public_visibility.admincenter_client", - new=admincenter_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_microsoft365_provider(), + ), mock.patch( + "prowler.providers.microsoft365.services.admincenter.admincenter_groups_not_public_visibility.admincenter_groups_not_public_visibility.admincenter_client", + new=admincenter_client, ): from prowler.providers.microsoft365.services.admincenter.admincenter_groups_not_public_visibility.admincenter_groups_not_public_visibility import ( admincenter_groups_not_public_visibility, diff --git a/tests/providers/microsoft365/services/admincenter/admincenter_users_between_two_and_four_global_admins/admincenter_users_between_two_and_four_global_admins_test.py b/tests/providers/microsoft365/services/admincenter/admincenter_users_between_two_and_four_global_admins/admincenter_users_between_two_and_four_global_admins_test.py index b7fafb56a9..ff2290236a 100644 --- a/tests/providers/microsoft365/services/admincenter/admincenter_users_between_two_and_four_global_admins/admincenter_users_between_two_and_four_global_admins_test.py +++ b/tests/providers/microsoft365/services/admincenter/admincenter_users_between_two_and_four_global_admins/admincenter_users_between_two_and_four_global_admins_test.py @@ -13,15 +13,12 @@ class Test_admincenter_users_between_two_and_four_global_admins: admincenter_client.audited_tenant = "audited_tenant" admincenter_client.audited_domain = DOMAIN - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_microsoft365_provider(), - ), - mock.patch( - "prowler.providers.microsoft365.services.admincenter.admincenter_users_between_two_and_four_global_admins.admincenter_users_between_two_and_four_global_admins.admincenter_client", - new=admincenter_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_microsoft365_provider(), + ), mock.patch( + "prowler.providers.microsoft365.services.admincenter.admincenter_users_between_two_and_four_global_admins.admincenter_users_between_two_and_four_global_admins.admincenter_client", + new=admincenter_client, ): from prowler.providers.microsoft365.services.admincenter.admincenter_users_between_two_and_four_global_admins.admincenter_users_between_two_and_four_global_admins import ( admincenter_users_between_two_and_four_global_admins, @@ -38,15 +35,12 @@ class Test_admincenter_users_between_two_and_four_global_admins: admincenter_client.audited_tenant = "audited_tenant" admincenter_client.audited_domain = DOMAIN - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_microsoft365_provider(), - ), - mock.patch( - "prowler.providers.microsoft365.services.admincenter.admincenter_users_between_two_and_four_global_admins.admincenter_users_between_two_and_four_global_admins.admincenter_client", - new=admincenter_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_microsoft365_provider(), + ), mock.patch( + "prowler.providers.microsoft365.services.admincenter.admincenter_users_between_two_and_four_global_admins.admincenter_users_between_two_and_four_global_admins.admincenter_client", + new=admincenter_client, ): from prowler.providers.microsoft365.services.admincenter.admincenter_service import ( DirectoryRole, @@ -89,15 +83,12 @@ class Test_admincenter_users_between_two_and_four_global_admins: admincenter_client.audited_tenant = "audited_tenant" admincenter_client.audited_domain = DOMAIN - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_microsoft365_provider(), - ), - mock.patch( - "prowler.providers.microsoft365.services.admincenter.admincenter_users_between_two_and_four_global_admins.admincenter_users_between_two_and_four_global_admins.admincenter_client", - new=admincenter_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_microsoft365_provider(), + ), mock.patch( + "prowler.providers.microsoft365.services.admincenter.admincenter_users_between_two_and_four_global_admins.admincenter_users_between_two_and_four_global_admins.admincenter_client", + new=admincenter_client, ): from prowler.providers.microsoft365.services.admincenter.admincenter_service import ( DirectoryRole, @@ -151,15 +142,12 @@ class Test_admincenter_users_between_two_and_four_global_admins: admincenter_client.audited_tenant = "audited_tenant" admincenter_client.audited_domain = DOMAIN - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_microsoft365_provider(), - ), - mock.patch( - "prowler.providers.microsoft365.services.admincenter.admincenter_users_between_two_and_four_global_admins.admincenter_users_between_two_and_four_global_admins.admincenter_client", - new=admincenter_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_microsoft365_provider(), + ), mock.patch( + "prowler.providers.microsoft365.services.admincenter.admincenter_users_between_two_and_four_global_admins.admincenter_users_between_two_and_four_global_admins.admincenter_client", + new=admincenter_client, ): from prowler.providers.microsoft365.services.admincenter.admincenter_service import ( DirectoryRole, diff --git a/tests/providers/microsoft365/services/entra/entra_admin_consent_workflow_enabled/entra_admin_consent_workflow_enabled_test.py b/tests/providers/microsoft365/services/entra/entra_admin_consent_workflow_enabled/entra_admin_consent_workflow_enabled_test.py index 9a72389813..b3a719b9e8 100644 --- a/tests/providers/microsoft365/services/entra/entra_admin_consent_workflow_enabled/entra_admin_consent_workflow_enabled_test.py +++ b/tests/providers/microsoft365/services/entra/entra_admin_consent_workflow_enabled/entra_admin_consent_workflow_enabled_test.py @@ -17,15 +17,12 @@ class Test_entra_admin_consent_workflow_enabled: """ entra_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_microsoft365_provider(), - ), - mock.patch( - "prowler.providers.microsoft365.services.entra.entra_admin_consent_workflow_enabled.entra_admin_consent_workflow_enabled.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_microsoft365_provider(), + ), mock.patch( + "prowler.providers.microsoft365.services.entra.entra_admin_consent_workflow_enabled.entra_admin_consent_workflow_enabled.entra_client", + new=entra_client, ): from prowler.providers.microsoft365.services.entra.entra_admin_consent_workflow_enabled.entra_admin_consent_workflow_enabled import ( entra_admin_consent_workflow_enabled, @@ -59,15 +56,12 @@ class Test_entra_admin_consent_workflow_enabled: """ entra_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_microsoft365_provider(), - ), - mock.patch( - "prowler.providers.microsoft365.services.entra.entra_admin_consent_workflow_enabled.entra_admin_consent_workflow_enabled.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_microsoft365_provider(), + ), mock.patch( + "prowler.providers.microsoft365.services.entra.entra_admin_consent_workflow_enabled.entra_admin_consent_workflow_enabled.entra_client", + new=entra_client, ): from prowler.providers.microsoft365.services.entra.entra_admin_consent_workflow_enabled.entra_admin_consent_workflow_enabled import ( entra_admin_consent_workflow_enabled, @@ -101,15 +95,12 @@ class Test_entra_admin_consent_workflow_enabled: """ entra_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_microsoft365_provider(), - ), - mock.patch( - "prowler.providers.microsoft365.services.entra.entra_admin_consent_workflow_enabled.entra_admin_consent_workflow_enabled.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_microsoft365_provider(), + ), mock.patch( + "prowler.providers.microsoft365.services.entra.entra_admin_consent_workflow_enabled.entra_admin_consent_workflow_enabled.entra_client", + new=entra_client, ): from prowler.providers.microsoft365.services.entra.entra_admin_consent_workflow_enabled.entra_admin_consent_workflow_enabled import ( entra_admin_consent_workflow_enabled, @@ -145,15 +136,12 @@ class Test_entra_admin_consent_workflow_enabled: entra_client.admin_consent_policy = None entra_client.tenant_domain = DOMAIN - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_microsoft365_provider(), - ), - mock.patch( - "prowler.providers.microsoft365.services.entra.entra_admin_consent_workflow_enabled.entra_admin_consent_workflow_enabled.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_microsoft365_provider(), + ), mock.patch( + "prowler.providers.microsoft365.services.entra.entra_admin_consent_workflow_enabled.entra_admin_consent_workflow_enabled.entra_client", + new=entra_client, ): from prowler.providers.microsoft365.services.entra.entra_admin_consent_workflow_enabled.entra_admin_consent_workflow_enabled import ( entra_admin_consent_workflow_enabled, diff --git a/tests/providers/microsoft365/services/entra/entra_dynamic_group_for_guests_created/entra_dynamic_group_for_guests_created_test.py b/tests/providers/microsoft365/services/entra/entra_dynamic_group_for_guests_created/entra_dynamic_group_for_guests_created_test.py index 4069310f4a..3bc5fc9b2f 100644 --- a/tests/providers/microsoft365/services/entra/entra_dynamic_group_for_guests_created/entra_dynamic_group_for_guests_created_test.py +++ b/tests/providers/microsoft365/services/entra/entra_dynamic_group_for_guests_created/entra_dynamic_group_for_guests_created_test.py @@ -15,15 +15,12 @@ class Test_entra_dynamic_group_for_guests_created: entra_client = mock.MagicMock() entra_client.groups = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_microsoft365_provider(), - ), - mock.patch( - "prowler.providers.microsoft365.services.entra.entra_dynamic_group_for_guests_created.entra_dynamic_group_for_guests_created.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_microsoft365_provider(), + ), mock.patch( + "prowler.providers.microsoft365.services.entra.entra_dynamic_group_for_guests_created.entra_dynamic_group_for_guests_created.entra_client", + new=entra_client, ): from prowler.providers.microsoft365.services.entra.entra_dynamic_group_for_guests_created.entra_dynamic_group_for_guests_created import ( entra_dynamic_group_for_guests_created, @@ -40,15 +37,12 @@ class Test_entra_dynamic_group_for_guests_created: """ entra_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_microsoft365_provider(), - ), - mock.patch( - "prowler.providers.microsoft365.services.entra.entra_dynamic_group_for_guests_created.entra_dynamic_group_for_guests_created.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_microsoft365_provider(), + ), mock.patch( + "prowler.providers.microsoft365.services.entra.entra_dynamic_group_for_guests_created.entra_dynamic_group_for_guests_created.entra_client", + new=entra_client, ): entra_client.groups = [ Group( @@ -82,15 +76,12 @@ class Test_entra_dynamic_group_for_guests_created: """ entra_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_microsoft365_provider(), - ), - mock.patch( - "prowler.providers.microsoft365.services.entra.entra_dynamic_group_for_guests_created.entra_dynamic_group_for_guests_created.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_microsoft365_provider(), + ), mock.patch( + "prowler.providers.microsoft365.services.entra.entra_dynamic_group_for_guests_created.entra_dynamic_group_for_guests_created.entra_client", + new=entra_client, ): entra_client.groups = [ Group( diff --git a/tests/providers/microsoft365/services/entra/entra_password_hash_sync_enabled/entra_password_hash_sync_enabled_test.py b/tests/providers/microsoft365/services/entra/entra_password_hash_sync_enabled/entra_password_hash_sync_enabled_test.py index bc9e5b3008..b363d4933d 100644 --- a/tests/providers/microsoft365/services/entra/entra_password_hash_sync_enabled/entra_password_hash_sync_enabled_test.py +++ b/tests/providers/microsoft365/services/entra/entra_password_hash_sync_enabled/entra_password_hash_sync_enabled_test.py @@ -10,15 +10,12 @@ class Test_entra_password_hash_sync_enabled: def test_password_hash_sync_enabled(self): entra_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_microsoft365_provider(), - ), - mock.patch( - "prowler.providers.microsoft365.services.entra.entra_password_hash_sync_enabled.entra_password_hash_sync_enabled.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_microsoft365_provider(), + ), mock.patch( + "prowler.providers.microsoft365.services.entra.entra_password_hash_sync_enabled.entra_password_hash_sync_enabled.entra_client", + new=entra_client, ): from prowler.providers.microsoft365.services.entra.entra_password_hash_sync_enabled.entra_password_hash_sync_enabled import ( entra_password_hash_sync_enabled, @@ -48,15 +45,12 @@ class Test_entra_password_hash_sync_enabled: def test_password_hash_sync_disabled(self): entra_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_microsoft365_provider(), - ), - mock.patch( - "prowler.providers.microsoft365.services.entra.entra_password_hash_sync_enabled.entra_password_hash_sync_enabled.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_microsoft365_provider(), + ), mock.patch( + "prowler.providers.microsoft365.services.entra.entra_password_hash_sync_enabled.entra_password_hash_sync_enabled.entra_client", + new=entra_client, ): from prowler.providers.microsoft365.services.entra.entra_password_hash_sync_enabled.entra_password_hash_sync_enabled import ( entra_password_hash_sync_enabled, @@ -100,15 +94,12 @@ class Test_entra_password_hash_sync_enabled: def test_password_hash_sync_disabled_two_org(self): entra_client = mock.MagicMock() - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_microsoft365_provider(), - ), - mock.patch( - "prowler.providers.microsoft365.services.entra.entra_password_hash_sync_enabled.entra_password_hash_sync_enabled.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_microsoft365_provider(), + ), mock.patch( + "prowler.providers.microsoft365.services.entra.entra_password_hash_sync_enabled.entra_password_hash_sync_enabled.entra_client", + new=entra_client, ): from prowler.providers.microsoft365.services.entra.entra_password_hash_sync_enabled.entra_password_hash_sync_enabled import ( entra_password_hash_sync_enabled, @@ -139,15 +130,12 @@ class Test_entra_password_hash_sync_enabled: entra_client = mock.MagicMock() entra_client.organization = [] - with ( - mock.patch( - "prowler.providers.common.provider.Provider.get_global_provider", - return_value=set_mocked_microsoft365_provider(), - ), - mock.patch( - "prowler.providers.microsoft365.services.entra.entra_password_hash_sync_enabled.entra_password_hash_sync_enabled.entra_client", - new=entra_client, - ), + with mock.patch( + "prowler.providers.common.provider.Provider.get_global_provider", + return_value=set_mocked_microsoft365_provider(), + ), mock.patch( + "prowler.providers.microsoft365.services.entra.entra_password_hash_sync_enabled.entra_password_hash_sync_enabled.entra_client", + new=entra_client, ): from prowler.providers.microsoft365.services.entra.entra_password_hash_sync_enabled.entra_password_hash_sync_enabled import ( entra_password_hash_sync_enabled,