def test_get_aws_keys_from_profile(self): logger = self._logger aws_access_key_id = "test_access_key" aws_secret_access_key = "test_secret_key" aws_session = get_aws_session( logger, dict(aws_access_key_id=aws_access_key_id, aws_secret_access_key=aws_secret_access_key)) access_key, secret_key = get_aws_keys_from_profile(logger, aws_session) self.assertEqual(access_key, aws_access_key_id) self.assertEqual(secret_key, aws_secret_access_key)
def test_get_aws_session_invalid_profile(self): logger = self._logger aws_session = get_aws_session( logger, dict(profile_name="invalid_profile", region_name="eu-west-1")) self.assertIsNone(aws_session)
def test_get_aws_session_invalid_params(self): logger = self._logger aws_session = get_aws_session( logger, dict(region_name="eu-west-1", invalid_param="invalid_param")) self.assertIsNone(aws_session)
def test_get_aws_session(self): logger = self._logger aws_session = get_aws_session(logger, dict(region_name="eu-west-1")) self.assertEqual(type(aws_session).__name__, "Session")
def test_get_s3_resource(self): logger = self._logger aws_session = get_aws_session(logger, dict(region_name="eu-west-1")) s3_resource = get_s3_resource(logger, aws_session) self.assertEqual(type(s3_resource).__name__, "s3.ServiceResource")
def test_get_s3_client(self): logger = self._logger aws_session = get_aws_session(logger, dict(region_name="eu-west-1")) s3_client = get_s3_client(logger, aws_session) self.assertEqual(type(s3_client).__name__, "S3")