feat(Audit_Info): Unifying import set audit info for different providers (#1538)

tests/providers/aws/aws_provider_test.py

@@ -1,46 +1,14 @@
-import json
-
 import boto3
 import sure  # noqa
-from moto import mock_iam, mock_organizations, mock_sts
+from moto import mock_iam, mock_sts
 
-from prowler.providers.aws.aws_provider import (
-    assume_role,
-    get_organizations_metadata,
-    get_region_global_service,
-    validate_credentials,
-)
+from prowler.providers.aws.aws_provider import assume_role, get_region_global_service
 from prowler.providers.aws.lib.audit_info.models import AWS_Assume_Role, AWS_Audit_Info
 
 ACCOUNT_ID = 123456789012
 
 
 class Test_AWS_Provider:
-    @mock_sts
-    @mock_iam
-    def test_validate_credentials(self):
-        # Create a mock IAM user
-        iam_client = boto3.client("iam", region_name="us-east-1")
-        iam_user = iam_client.create_user(UserName="test-user")["User"]
-        # Create a mock IAM access keys
-        access_key = iam_client.create_access_key(UserName=iam_user["UserName"])[
-            "AccessKey"
-        ]
-        access_key_id = access_key["AccessKeyId"]
-        secret_access_key = access_key["SecretAccessKey"]
-        # Create AWS session to validate
-        session = boto3.session.Session(
-            aws_access_key_id=access_key_id,
-            aws_secret_access_key=secret_access_key,
-            region_name="us-east-1",
-        )
-        # Validate AWS session
-        get_caller_identity = validate_credentials(session)
-
-        get_caller_identity["Arn"].should.equal(iam_user["Arn"])
-        get_caller_identity["UserId"].should.equal(iam_user["UserId"])
-        # assert get_caller_identity["UserId"] == str(ACCOUNT_ID)
-
     @mock_iam
     @mock_sts
     def test_assume_role(self):
@@ -114,60 +82,6 @@ class Test_AWS_Provider:
             21 + 1 + len(sessionName)
         )
 
-    @mock_organizations
-    @mock_sts
-    @mock_iam
-    def test_organizations(self):
-        client = boto3.client("organizations", region_name="us-east-1")
-        iam_client = boto3.client("iam", region_name="us-east-1")
-        sts_client = boto3.client("sts", region_name="us-east-1")
-
-        mockname = "mock-account"
-        mockdomain = "moto-example.org"
-        mockemail = "@".join([mockname, mockdomain])
-
-        org_id = client.create_organization(FeatureSet="ALL")["Organization"]["Id"]
-        account_id = client.create_account(AccountName=mockname, Email=mockemail)[
-            "CreateAccountStatus"
-        ]["AccountId"]
-
-        client.tag_resource(
-            ResourceId=account_id, Tags=[{"Key": "key", "Value": "value"}]
-        )
-
-        trust_policy_document = {
-            "Version": "2012-10-17",
-            "Statement": {
-                "Effect": "Allow",
-                "Principal": {
-                    "AWS": "arn:aws:iam::{account_id}:root".format(
-                        account_id=ACCOUNT_ID
-                    )
-                },
-                "Action": "sts:AssumeRole",
-            },
-        }
-        iam_role_arn = iam_client.role_arn = iam_client.create_role(
-            RoleName="test-role",
-            AssumeRolePolicyDocument=json.dumps(trust_policy_document),
-        )["Role"]["Arn"]
-        session_name = "new-session"
-        assumed_role = sts_client.assume_role(
-            RoleArn=iam_role_arn, RoleSessionName=session_name
-        )
-
-        org = get_organizations_metadata(account_id, assumed_role)
-
-        org.account_details_email.should.equal(mockemail)
-        org.account_details_name.should.equal(mockname)
-        org.account_details_arn.should.equal(
-            "arn:aws:organizations::{0}:account/{1}/{2}".format(
-                ACCOUNT_ID, org_id, account_id
-            )
-        )
-        org.account_details_org.should.equal(org_id)
-        org.account_details_tags.should.equal("key:value,")
-
     def test_get_region_global_service(self):
         # Create mock audit_info
         input_audit_info = AWS_Audit_Info(