runbooks 0.2.3__py3-none-any.whl → 0.6.1__py3-none-any.whl

runbooks/inventory/tests/common_test_functions.py

@@ -0,0 +1,204 @@
+import pytest
+from botocore import client, session
+
+ERASE_LINE = "\x1b[2K"
+
+
+def AWSAccount_from_AWSKeyID(AWSKeyID):
+    """
+    Determines the AWS Account number from the AWS Key ID.
+    The original idea here: https://hackingthe.cloud/aws/enumeration/get-account-id-from-keys/
+    @param AWSKeyID: This is a *fake* AWS Key in the format of "xxxx" (four characters)
+     and then should be the account number. We'll toss out the rest of the data as irrelevant
+    @return: Returns the account number as a string
+    """
+    trimmed_AWSKeyID = AWSKeyID[4:16]  # remove KeyID prefix
+    return str(trimmed_AWSKeyID)
+
+
+def AWSKeyID_from_AWSAccount(AWSAccountNumber):
+    """
+    Makes up an AWS Key ID from the AWS Account number.
+    @param AWSAccountNumber: This is the AWS Account number as a string
+    @return: Returns the AWS Key ID as a string
+    """
+    # An AWS Key looks like this: "AIDAJ74HIVAJJXOVUHYO6" (21 characters)
+    AWSKey = f"xxxx{str(AWSAccountNumber)}xxxxx"
+    return AWSKey
+
+
+def _amend_create_boto3_session(test_data, mocker):
+    orig = session.Session.create_client
+
+    def amend_create_client(
+        self,
+        service_name,
+        region_name=None,
+        api_version=None,
+        use_ssl=True,
+        verify=None,
+        endpoint_url=None,
+        aws_access_key_id=None,
+        aws_secret_access_key=None,
+        aws_session_token=None,
+        config=None,
+    ):
+        # Intercept boto3 Session, in hopes of sending back a client that includes the Account Number
+        # if aws_access_key_id == '*****AccessKeyHere*****':
+        print(test_data["FunctionName"])
+        if aws_access_key_id == "MeantToFail":
+            print(f"Failed Access Key: {aws_access_key_id}")
+            return ()
+        else:
+            print(f"Not Failed Access Key: {aws_access_key_id}")
+            return_response = orig(
+                self,
+                service_name,
+                region_name,
+                api_version,
+                use_ssl,
+                verify,
+                endpoint_url,
+                aws_access_key_id,
+                aws_secret_access_key,
+                aws_session_token,
+                config,
+            )
+            return return_response
+
+    mocker.patch("botocore.session.Session.create_client", new=amend_create_client)
+    print()
+
+
+def _amend_make_api_call_orig(test_key, test_value, mocker):
+    orig = client.BaseClient._make_api_call
+
+    def amend_make_api_call(self, operation_name, kwargs):
+        # Intercept boto3 operations for <secretsmanager.get_secret_value>. Optionally, you can also
+        # check on the argument <SecretId> and control how you want the response would be. This is
+        # a very flexible solution as you have full control over the whole process of fetching a
+        # secret.
+        if operation_name == "ListAccounts":
+            if isinstance(test_value, Exception):
+                raise test_value
+            # Implied break and exit of the function here...
+            print(
+                f"Operation Name mocked: {operation_name}\n"
+                f"Key Name: {test_key}\n"
+                f"kwargs: {kwargs}\n"
+                f"mocked return_response: {test_value}"
+            )
+            return test_value
+
+        return_response = orig(self, operation_name, kwargs)
+        print(
+            f"Operation Name passed through: {operation_name}\n"
+            f"Key name: {test_key}\n"
+            f"kwargs: {kwargs}\n"
+            f"Actual return response: {return_response}"
+        )
+        return return_response
+
+    mocker.patch("botocore.client.BaseClient._make_api_call", new=amend_make_api_call)
+
+
+def _amend_make_api_call(meta_key_dict, test_dict, mocker):
+    orig = client.BaseClient._make_api_call
+
+    def amend_make_api_call(self, operation_name, kwargs):
+        # Intercept boto3 operations for <secretsmanager.get_secret_value>. Optionally, you can also
+        # check on the argument <SecretId> and control how you want the response would be. This is
+        # a very flexible solution as you have full control over the whole process of fetching a
+        # secret.
+        for op_name in test_dict:
+            test_value = op_name["test_result"]
+            region = self.meta.region_name
+            if operation_name == op_name["operation_name"]:
+                if isinstance(test_value, Exception):
+                    # Implied break and exit of the function here...
+                    raise test_value
+                print(
+                    f"Operation Name mocked: {operation_name}\n"
+                    f"Function Name: {meta_key_dict['FunctionName']}\n"
+                    f"kwargs: {kwargs}\n"
+                    f"mocked return_response: {op_name['test_result']}"
+                )
+                return op_name["test_result"]
+        try:
+            print(
+                f"Trying: Operation Name passed through: {operation_name}\n"
+                f"Key Name: {meta_key_dict['FunctionName']}\n"
+                f"kwargs: {kwargs}\n"
+            )
+            return_response = orig(self, operation_name, kwargs)
+            print(f"Actual return_response: {return_response}")
+        except Exception as my_Error:
+            raise ConnectionError("Operation Failed")
+        return return_response
+
+    mocker.patch("botocore.client.BaseClient._make_api_call", new=amend_make_api_call)
+
+
+def _amend_make_api_call_specific(meta_key_dict, test_dict, mocker):
+    orig = client.BaseClient._make_api_call
+
+    def amend_make_api_call(self, operation_name, kwargs):
+        # Intercept boto3 operations for <secretsmanager.get_secret_value>. Optionally, you can also
+        # check on the argument <SecretId> and control how you want the response would be. This is
+        # a very flexible solution as you have full control over the whole process of fetching a
+        # secret.
+        # This goes through the operations in the dictionary and checks for each operation, whether it matches the operation we're patching right now.
+        for op_name in test_dict:
+            if operation_name == op_name["operation_name"]:
+                # We are able to capture the region from the API call, and use that to differentiate data to return
+                region = self.meta.region_name
+                # For the various sets of return data that we have...
+                test_value = None
+                for set_of_result_data in op_name["test_result"]:
+                    # Check to see which set of return data matches the region we're calling for now...
+                    if set_of_result_data["Region"] == region:
+                        test_value = set_of_result_data["mocked_response"]
+                        break
+                if test_value is not None and isinstance(test_value, Exception):
+                    # Implied break and exit of the function here...
+                    print("Expected Error...")
+                    raise test_value
+                elif test_value is None:
+                    print(f"No test data offered for this credentials in region {region}")
+                    continue
+                print(
+                    f"Operation Name mocked: {operation_name}\n"
+                    f"Function Name: {meta_key_dict['FunctionName']}\n"
+                    f"kwargs: {kwargs}\n"
+                    f"mocked return_response: {test_value}"
+                )
+                return test_value
+
+        print(
+            f"Operation Name passed through: {operation_name}\n"
+            f"Function Name: {meta_key_dict['FunctionName']}\n"
+            f"kwargs: {kwargs}\n"
+        )
+        return_response = orig(self, operation_name, kwargs)
+        print(f"Actual return_response: {return_response}")
+        return return_response
+
+    mocker.patch("botocore.client.BaseClient._make_api_call", new=amend_make_api_call)
+
+
+# mocker.patch('botocore.session', new=amend_make_api_call)
+
+
+def mock_find_all_instances2(creds: dict, region: str):
+    """
+    This is a mock function that will return a list of all the instances in the region that we're looking for.
+    :param creds: Credentials object, where 'AccountNumber' is the account number of the account that we're looking for.
+    :param region: string for region we're checking
+    :return: The output from the EC2 API for "list_instances"
+    """
+    from Tests.common_test_data import All_Instances_Response_Data
+
+    for mock_data_set in All_Instances_Response_Data:
+        if mock_data_set["Region"] == region and mock_data_set["AccountNumber"] == creds["AccountNumber"]:
+            return mock_data_set["instance_data"]
+    raise KeyError(f"No data for {creds['AccountNumber']} found for region {region}")

runbooks/inventory/tests/setup.py

@@ -0,0 +1,24 @@
+from setuptools import setup
+
+setup(
+    name="cli_skeleton",  # Update Name
+    packages=["cli_skeleton"],  # Name of CLI
+    version="0.1.0",  # Version
+    description="A cli for X",  # Description
+    author="NAME",  # Author
+    url="https://github.com/REPO",  # To be updated
+    author_email="NAME@EMAIL.com",  # Email
+    download_url="https://github.com/REPO",  # To be updated
+    keywords=["aws", "python"],  # Key words for cli
+    classifiers=[],
+    install_requires=[
+        "PACKAGE"  # Add all packages
+    ],
+    setup_requires=[],
+    tests_require=[],
+    entry_points={
+        "console_scripts": [
+            "cli_skeleton = cli_skeleton.__main__:main",
+        ],
+    },
+)

runbooks/inventory/tests/src.py

@@ -0,0 +1,18 @@
+import boto3
+from account_class import aws_acct_access
+
+sendgrid_api_key_arn = "some_name"
+region_name = "us-east-1"
+credentials = {"profile_name": "LZ1"}
+
+
+def do_stuff():
+    # aws_acct = aws_acct_access()
+    # session = boto3.session.Session(**credentials)
+    # client = session.client(service_name="organizations", region_name=region_name)
+    # response1 = client.list_accounts()
+    # response2 = client.list_roots()
+    # responsedict = {'response1': response1, 'response2': response2, 'response3': aws_acct}
+    # # responsedict = {'response1': response1, 'response2': response2}
+    # return responsedict
+    print("Hello World")

runbooks/inventory/tests/test_cfn_describe_stacks.py

@@ -0,0 +1,208 @@
+import sys
+import unittest
+from io import StringIO
+from pprint import pprint
+from unittest.mock import MagicMock, patch
+
+import pytest
+
+# Import the functions you want to test
+from cfn_describe_stacks import (
+    collect_cfnstacks,
+    display_stacks,
+    modify_stacks,
+    parse_args,
+    setup_auth_accounts_and_regions,
+)
+
+
+class TestScriptFunctions(unittest.TestCase):
+    def setUp(self):
+        # This is the parameters provided.
+        self.mock_profile = "mock_profile"
+        self.mock_regions = ["us-east-1", "us-west-1"]
+        self.fragments = ["Stack"]
+        self.accounts = ["123456789012", "234567890123"]
+        self.exact = False
+        self.timing = True
+        self.SkipAccounts = None
+        self.SkipProfiles = None
+        self.loglevel = 50
+        self.status = None
+        self.stackid = False
+        self.rootonly = False
+        self.deletionrun = False
+        # This is the parameters that will be passed to the script
+        self.mock_args = ["-p", self.mock_profile, "-rs", self.mock_regions, "--time", "-f", self.fragments]
+        # This is the parameters that have been instantiated within the script, including default values
+        self.expected_args = {
+            # 'AccessRoles' : None,
+            "Accounts": self.accounts,
+            "DeletionRun": self.deletionrun,
+            "Exact": self.exact,
+            "Fragments": self.fragments,
+            "Profile": self.mock_profile,
+            "Regions": self.mock_regions,
+            "RootOnly": self.rootonly,
+            "SkipAccounts": self.SkipAccounts,
+            "SkipProfiles": self.SkipProfiles,
+            "Time": self.timing,
+            "loglevel": self.loglevel,
+            "stackid": self.stackid,
+            "status": self.status,
+            # Add other expected arguments as needed
+        }
+
+    def test_parse_args(self):
+        with patch("sys.argv", self.mock_args):
+            args = parse_args(sys.argv)
+            for arg, value in self.expected_args.items():
+                self.assertEqual(getattr(args, arg), value)
+
+    @patch("all_my_cfnstacks.aws_acct_access")
+    @patch("all_my_cfnstacks.Inventory_Modules.get_regions3")
+    # @patch('all_my_cfnstacks.Inventory_Modules.RemoveCoreAccounts')
+    # def test_setup_auth_accounts_and_regions(self, mock_remove_core_accounts, mock_get_regions3, mock_aws_acct_access):
+    def test_setup_auth_accounts_and_regions(self, mock_get_regions3, mock_aws_acct):
+        # Set up mock objects and data
+        mock_aws_acct_access = MagicMock()
+        mock_aws_acct_access.ChildAccounts = [{"AccountId": "123456789012"}, {"AccountId": "234567890123"}]
+        mock_aws_acct.return_value = mock_aws_acct_access
+        # mock_aws_acct.ChildAccounts = [{'AccountId': '123456789012'}, {'AccountId': '234567890123'}]
+        mock_get_regions3.return_value = ["us-east-1", "us-west-1"]
+        # mock_remove_core_accounts.return_value =
+        # mock_remove_core_accounts.side_effect = lambda accounts, skip_accounts: accounts
+
+        # Call the function
+        aws_acct, account_list, region_list = setup_auth_accounts_and_regions(
+            self.mock_profile,
+            mock_get_regions3,
+            self.accounts,
+            self.SkipAccounts,
+            self.fragments,
+            self.exact,
+            self.deletionrun,
+        )
+
+        # Assert the results
+        self.assertEqual(aws_acct, mock_aws_acct.return_value)
+        self.assertEqual(account_list, ["123456789012", "234567890123"])
+        self.assertEqual(region_list, mock_get_regions3.return_value)
+
+    @patch("all_my_cfnstacks.Inventory_Modules.find_stacks2")
+    def test_collect_cfnstacks(self, mock_find_stacks2):
+        # Set up mock data
+        mock_credential_list = [
+            {"AccountId": "123456789012", "Region": "us-east-1", "Success": True},
+            {"AccountId": "234567890123", "Region": "us-west-1", "Success": False},
+        ]
+        mock_stacks = [
+            {
+                "StackName": "my-stack-1",
+                "StackStatus": "CREATE_COMPLETE",
+                "StackId": "arn:aws:cloudformation:...",
+                "CreationTime": "2023-04-01T00:00:00Z",
+            },
+            {
+                "StackName": "my-stack-2",
+                "StackStatus": "UPDATE_IN_PROGRESS",
+                "StackId": "arn:aws:cloudformation:...",
+                "CreationTime": "2023-04-02T00:00:00Z",
+            },
+        ]
+        mock_find_stacks2.return_value = mock_stacks
+
+        # Call the function
+        all_stacks = collect_cfnstacks(mock_credential_list)
+
+        # Assert the results
+        expected_stacks = [
+            {
+                "Account": "123456789012",
+                "Region": "us-east-1",
+                "AccessKeyId": None,
+                "SecretAccessKey": None,
+                "SessionToken": None,
+                "AccountNumber": None,
+                "StackName": "my-stack-1",
+                "StackCreate": None,
+            }
+        ]
+
+    # Need some asserts here
+
+    @patch("script.display_results")
+    @patch("builtins.print")
+    def test_display_stacks(self, mock_print, mock_display_results):
+        global AccountList, RegionList
+        AccountList = ["111111111111", "222222222222"]
+        RegionList = ["us-east-1", "us-west-2"]
+        all_stacks = [
+            {
+                "Account": "111111111111",
+                "Region": "us-east-1",
+                "StackStatus": "CREATE_COMPLETE",
+                "StackCreate": "2023-04-01",
+                "StackName": "stack1",
+                "StackArn": "arn:aws:cloudformation:us-east-1:111111111111:stack/stack1/abcd1234",
+            },
+            {
+                "Account": "222222222222",
+                "Region": "us-west-2",
+                "StackStatus": "UPDATE_COMPLETE",
+                "StackCreate": "2023-03-15",
+                "StackName": "stack2",
+                "StackArn": "arn:aws:cloudformation:us-west-2:222222222222:stack/stack2/efgh5678",
+            },
+        ]
+
+        display_stacks(all_stacks)
+
+        mock_display_results.assert_called_once()
+        expected_print_calls = [
+            mock.call("\x1b[2K"),
+            mock.call("\x1b[31mFound 2 stacks across 2 accounts across 2 regions\x1b[0m"),
+            mock.call(),
+            mock.call("The list of accounts and regions:"),
+            mock.ANY,  # Skipping the pprint call
+        ]
+        mock_print.assert_has_calls(expected_print_calls)
+
+    @patch("builtins.input", return_value="y")
+    @patch("all_my_cfnstacks.Inventory_Modules.delete_stack2")
+    def test_modify_stacks(self, mock_delete_stack2, mock_input):
+        global pStackfrag, DeletionRun
+        pStackfrag = "my-stack"
+        DeletionRun = True
+
+        stack_found1 = {
+            "Account": "111111111111",
+            "Region": "us-east-1",
+            "StackName": "stack1",
+            "StackStatus": "CREATE_COMPLETE",
+        }
+        stack_found2 = {
+            "Account": "222222222222",
+            "Region": "us-west-2",
+            "StackName": "stack2",
+            "StackStatus": "DELETE_FAILED",
+        }
+        stacks_found = [stack_found1, stack_found2]
+
+        mock_delete_stack2.side_effect = ["delete_response1", "delete_response2"]
+
+        modify_result = modify_stacks(stacks_found)
+
+        self.assertEqual(modify_result, ["delete_response1", "delete_response2"])
+        mock_input.assert_called_once_with("Deletion of stacks has been requested, are you still sure? (y/n): ")
+        mock_delete_stack2.assert_any_call(stack_found1, "us-east-1", "stack1")
+        mock_delete_stack2.assert_any_call(
+            stack_found2,
+            "us-west-2",
+            "stack2",
+            RetainResources=True,
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()

runbooks/inventory/tests/test_ec2_describe_instances.py

@@ -0,0 +1,162 @@
+"""
+python
+"""
+
+import sys
+import unittest
+from unittest.mock import patch
+
+import pytest
+from common_test_data import (
+    All_Instances_Response_Data,
+    CredentialResponseData,
+    mock_instances_1,
+    mock_profile_list_1,
+    mock_profile_list_2,
+    mock_profile_list_3,
+    mock_profile_list_4,
+    mock_region_list_1,
+    mock_region_list_2,
+    mock_region_list_3,
+    mock_region_list_4,
+)
+from common_test_functions import mock_find_all_instances2
+from ec2_describe_instances import find_all_instances, parse_args
+
+
+class TestScriptFunctions(unittest.TestCase):
+    def setUp(self):
+        # This is the parameters provided. Note that this
+        self.account_mapping = [
+            {"AccountNumber": "111122223333", "Region": "us-east-1", "mock_instances_returned": "mock_instances_1"},
+            {"AccountNumber": "444455556666", "Region": "us-east-2", "mock_instances_returned": "mock_instances_2"},
+            {"AccountNumber": "555566667777", "Region": "us-west-2", "mock_instances_returned": "mock_instances_3"},
+            {"AccountNumber": "555566667777", "Region": "eu-west-1", "mock_instances_returned": "mock_instances_4"},
+            {"AccountNumber": "666677775555", "Region": "eu-central-1", "mock_instances_returned": "mock_instances_5"},
+            {"AccountNumber": "777755556666", "Region": "eu-north-1", "mock_instances_returned": "mock_instances_6"},
+            {"AccountNumber": "777755556666", "Region": "eu-west-2", "mock_instances_returned": "mock_instances_7"},
+            {"AccountNumber": "666677778888", "Region": "ap-south-1", "mock_instances_returned": "mock_instances_8"},
+            {"AccountNumber": "777788886666", "Region": "il-central-1", "mock_instances_returned": "mock_instances_9"},
+            {"AccountNumber": "888866667777", "Region": "af-south-1", "mock_instances_returned": "mock_instances_10"},
+        ]
+        self.expected_args = {
+            "AccessRoles": None,
+            "Accounts": None,
+            "Profiles": ["mock_profile"],
+            "Regions": ["us-east-1"],
+            "RootOnly": False,
+            "SkipAccounts": None,
+            "SkipProfiles": None,
+            "Time": True,
+            "loglevel": 20,
+            "pStatus": "running",
+            # Add other expected arguments as needed
+        }
+        self.mock_args = ["-p", "mock_profile", "-rs", "us-east-1", "-s", "running", "--time", "-vvv"]
+        self.mock_profile_list = ["mock_profile_1", "mock_profile_2"]
+        self.mock_region_list = ["us-east-1", "us-east-2"]
+
+    # This is the parameters that have been instantiated within the script, including default values
+
+    def test_parse_args(self):
+        with patch("sys.argv", self.mock_args):
+            args = parse_args(sys.argv)
+            for arg, value in self.expected_args.items():
+                self.assertEqual(getattr(args, arg), value)
+
+    # @patch('all_my_instances.Inventory_Modules.get_regions3')
+    # @patch('all_my_instances.Inventory_Modules.get_profiles')
+    # @patch('all_my_instances.get_credentials_for_accounts_in_org')
+
+    # # @pytest.mark.parametrize(
+    # 	"mock_org_credentials, mock_profile_list, mock_region_list",
+    # 	[
+    # 		(CredentialResponseData, mock_profile_list_1, mock_region_list_1),
+    # 		(CredentialResponseData, mock_profile_list_2, mock_region_list_2),
+    # 		(CredentialResponseData, mock_profile_list_3, mock_region_list_3),
+    # 		(CredentialResponseData, mock_profile_list_4, mock_region_list_4),
+    # 		],
+    # 	)
+    # def test_get_credentials(self, mock_get_credentials_for_accounts_in_org, mock_get_profiles, mock_get_regions3, mock_org_credentials, mock_profile_list, mock_region_list):
+
+    """
+	def test_get_credentials(self, mock_get_credentials_for_accounts_in_org, mock_get_profiles, mock_get_regions3):
+		mock_get_profiles.return_value = self.mock_profile_list
+		mock_get_regions3.return_value = self.mock_region_list
+		mock_get_credentials_for_accounts_in_org.return_value = CredentialResponseData
+
+		# The credentials supplied here absolutely do not matter, since the Credential Response is also hard-coded above.
+		# def get_credentials(fProfile_list: list, fRegion_list: list, fSkipProfiles: list = None, fSkipAccounts: list = None, fRootOnly: bool = False, fAccounts: list = None, fAccessRoles: list = None, fTiming=False) -> list:
+		credentials = get_credentials(mock_get_profiles, mock_get_regions3)
+		self.assertEqual(len(credentials), len(self.mock_profile_list) * len(CredentialResponseData))
+		self.assertEqual(credentials[0]['MgmtAccount'], '111122223333')
+		self.assertEqual(credentials[0]['AccountId'], '111122223333')
+		self.assertEqual(credentials[0]['Region'], 'us-east-1')
+		self.assertEqual(credentials[0]['Profile'], 'mock_profile')
+		self.assertEqual(credentials[0]['AccountStatus'], 'ACTIVE')
+		self.assertEqual(credentials[0]['Role'], 'Use Profile')
+		self.assertEqual(credentials[1]['MgmtAccount'], '111122223333')
+		self.assertEqual(credentials[1]['AccountId'], '444455556666')
+		self.assertEqual(credentials[1]['Region'], 'us-east-2')
+		self.assertEqual(credentials[1]['Profile'], None)
+		self.assertEqual(credentials[1]['AccountStatus'], 'ACTIVE')
+		self.assertEqual(credentials[1]['Role'], 'AWSCloudFormationStackSetExecutionRole')
+	"""
+
+    @patch("all_my_instances.Inventory_Modules.find_account_instances2", wraps=mock_find_all_instances2)
+    def test_find_all_instances(self, mock_find_account_instances2):
+        test_creds = CredentialResponseData[1:3]
+        instances = find_all_instances(test_creds, "running")
+        # self.assertEqual(len(instances), (len(mock_instances_1) * len(CredentialResponseData)))
+        for mock_profile in All_Instances_Response_Data:
+            for instance in instances:
+                mock_data_set = (
+                    mock_profile["instance_data"]["Reservations"]
+                    if mock_profile["mock_profile"] == instance["ParentProfile"]
+                    else None
+                )
+
+        self.assertEqual(instances[0]["InstanceType"], "t2.micro")
+        self.assertEqual(instances[0]["InstanceId"], "i-1234567890abcdef")
+        self.assertEqual(instances[0]["PublicDNSName"], "ec2-1-2-3-4.us-east-1.compute.amazonaws.com")
+        self.assertEqual(instances[0]["State"], "running")
+        self.assertEqual(instances[0]["Name"], "Instance1")
+        self.assertEqual(instances[0]["AccountId"], "111122223333")
+        self.assertEqual(instances[0]["Region"], "us-east-1")
+        self.assertEqual(instances[0]["MgmtAccount"], "111122223333")
+        self.assertEqual(instances[0]["ParentProfile"], "mock_profile")
+
+    # def test_main(self):
+    # 	# Capture the output of the script
+    # 	captured_output = StringIO()
+    # 	sys.stdout = captured_output
+    #
+    # 	# Call the main function with mock arguments
+    # 	with patch('sys.argv', self.mock_args):
+    # 		with patch('all_my_instances.get_credentials', return_value=CredentialResponseData):
+    # 			with patch('all_my_instances.find_all_instances',
+    # 			           return_value=[{'InstanceType': 't2.micro',
+    # 			                          'InstanceId'  : 'i-1234567890abcdef',
+    # 			                          'State'       : 'running'}]):
+    # 				all_my_instances.init()
+    #
+    # 	# Restore the original stdout
+    # 	sys.stdout = sys.__stdout__
+    #
+    # 	# Check the captured output
+    # 	output = captured_output.getvalue()
+    # 	self.assertIn('Found 1 instances across 1 accounts across 1 regions', output)
+
+
+if __name__ == "__main__":
+    unittest.main()
+
+    """
+	In the test_find_all_instances method, I added assertions to check if the instance details are correctly populated in the returned list.
+	
+	Additionally, I added a test_main method to test the main function of the script. This test captures the output of the script using StringIO and checks if the expected output is present in the captured output.
+	
+	Note that I've used the patch decorator from the unittest.mock module to mock the get_credentials and find_all_instances functions in the test_main method. You might need to adjust the mocked return values based on your specific use case.
+	
+	With these additions, the unit test script should be complete and ready to run using python test_script.py (assuming the test script file is named test_script.py).
+	"""

runbooks/inventory/tests/test_inventory_modules.py

@@ -0,0 +1,55 @@
+import pytest
+from common_test_data import (
+    AssumeRoleResponseData,
+    DescribeOrganizationsResponseData,
+    DescribeRegionsResponseData,
+    GetCallerIdentity,
+    ListAccountsResponseData,
+    cli_provided_parameters1,
+)
+from common_test_functions import _amend_make_api_call
+from Inventory_Modules import get_all_credentials
+
+get_all_credentials_test_result_dict = [
+    {"operation_name": "GetCallerIdentity", "test_result": GetCallerIdentity},
+    {"operation_name": "DescribeOrganization", "test_result": DescribeOrganizationsResponseData},
+    {"operation_name": "AssumeRole", "test_result": AssumeRoleResponseData},
+    {"operation_name": "ListAccounts", "test_result": ListAccountsResponseData},
+    {"operation_name": "DescribeRegions", "test_result": DescribeRegionsResponseData},
+]
+
+
+# Skipped for now, since I know the get_credential testing needs more work
+@pytest.mark.skip
+@pytest.mark.parametrize(
+    "parameters, test_value_dict",
+    [
+        (cli_provided_parameters1, get_all_credentials_test_result_dict),
+    ],
+)
+def test_get_all_credentials(parameters, test_value_dict, mocker):
+    pProfiles = parameters["pProfiles"]
+    pRegionList = parameters["pRegionList"]
+    pSkipProfiles = parameters["pSkipProfiles"]
+    pSkipAccounts = parameters["pSkipAccounts"]
+    pAccountList = parameters["pAccountList"]
+    pTiming = parameters["pTiming"]
+    pRootOnly = parameters["pRootOnly"]
+    pRoleList = parameters["pRoleList"]
+    test_data = {"FunctionName": "get_all_credentials", "AccountSpecific": True, "RegionSpecific": True}
+    _amend_make_api_call(test_data, test_value_dict, mocker)
+
+    # if isinstance(test_value, Exception):
+    # 	print("Expected Error...")
+    # 	with pytest.raises(type(test_value)) as error:
+    # 		get_all_credentials(pProfiles, pTiming, pSkipProfiles, pSkipAccounts, pRootOnly, pAccountList, pRegionList, pRoleList)
+    # 	result = error
+    # else:
+    result = get_all_credentials(
+        pProfiles, pTiming, pSkipProfiles, pSkipAccounts, pRootOnly, pAccountList, pRegionList, pRoleList
+    )
+    for cred in result:
+        assert cred["Success"]
+    print("Result:", result)
+
+    return result