def client(): with moto.mock_secretsmanager(): yield SecretsManagerClient( cloudfront_key_pair_name=cloudfront_key_pair_name, post_verification_api_creds_name=post_verification_api_creds_name, google_client_ids_name=google_client_ids_name, )
def test_aggregate_metrics_no_docker(mock_process, mock_unpgn, mock_verify, mock_pgn, mock_sort, capfd, monkeypatch, aws_credentials): monkeypatch.setenv('docker_bool', 'n') monkeypatch.setenv('user_agent_header', 'test-user-agent-header') repo_name = 'test-repo-name' mock_process.side_effect = lambda name: ({ 'test-' + name: { 'test-key': 'test-val' } }, { 'test-name-2': 'test-param-2' }) unpgn_metric = {'test-github-unpgn': 0} mock_unpgn.return_value = unpgn_metric mock_verify.side_effect = [({ 'test-github-unpgn': 0 }, {}), ({}, { 'test-docker': 'hello' })] mock_pgn.return_value = {'test-github-pgn': 12} mock_sort.side_effect = lambda metric, text, param_to_name: print( 'Requested metric data:', metric, '\nRequested text data:', text) with mock_secretsmanager(): boto3.setup_default_session() boto3.client('secretsmanager').create_secret(Name='github_auth_token', SecretString='1234') github_docker.aggregate_metrics('test-owner', repo_name) out, err = capfd.readouterr() assert "Requested metric data: {'test-github-unpgn': 0, 'test-github-pgn': 12}" in out assert "Requested text data: {'test-docker': 'hello'}" not in out
def mock_secrets_manager(): """ Starts moto at the beginning of tests, and closes it at the end of the session """ mock = mock_secretsmanager() mock.start() yield mock.stop()
def secrets_manager(): test_secret = { "twitter-consumer-key": "consumer-key", "twitter-consumer-secret": "consumer-secret", "twitter-access-token": "access-token", "twitter-access-secret": "access-secret" } with mock_secretsmanager(): client = boto3.client('secretsmanager') client.create_secret( Name='test-secret', SecretString=json.dumps(test_secret), ) yield client
def setUp(self): self.dynamo_mock = mock_dynamodb2() self.dynamo_mock.start() self.s3_mock = mock_s3() self.s3_mock.start() self.secrets_mock = mock_secretsmanager() self.secrets_mock.start() self.sqs_mock = mock_sqs() self.sqs_mock.start() self.sts_mock = mock_sts() self.sts_mock.start() self.matrix_infra_config = MatrixInfraConfig() self.redshift_config = MatrixRedshiftConfig() self.sqs = boto3.resource('sqs') self.sqs.create_queue(QueueName=f"test_query_job_q_name") self.sqs.create_queue(QueueName=f"test_deadletter_query_job_q_name") self.sqs.create_queue(QueueName=f"test_notification_q_url")
def setUp(self): with warnings.catch_warnings(): warnings.filterwarnings("ignore", category=DeprecationWarning) self.mock = mock_secretsmanager() self.mock.start() current_account_id = "111122223333" region = "us-east-1" self.client = get_boto3_client(profile=None, service="secretsmanager", region=region) response = self.client.create_secret(Name=MY_RESOURCE, SecretString="foosecret") self.example = SecretsManagerSecret( name=MY_RESOURCE, region=region, client=self.client, current_account_id=current_account_id) self.secrets = SecretsManagerSecrets( client=self.client, current_account_id=current_account_id, region=region)
def secretsmanager(aws_credentials): with mock_secretsmanager(): yield boto3.client("secretsmanager", region_name="us-east-1")
def secrets_manager(region): with moto.mock_secretsmanager(): yield boto3.client("secretsmanager", region_name=region)
def create_secret(): with mock_secretsmanager(): boto3.setup_default_session() boto3.client('secretsmanager').create_secret(Name='github_auth_token', SecretString='1234') yield
def secretsmanager(aws_credentials): with mock_secretsmanager(): yield boto3.client('secretsmanager', region_name='us-east-1')
def create_secret(name: str = SECRET_NAME, payload: Dict = SECRET_PAYLOAD): with moto.mock_secretsmanager(): client = secret_client() client.create_secret(Name=name, Description='My cool test secret', SecretString=json.dumps(payload))
def secrets_manager(aws_credentials): with mock_secretsmanager(): secrets_manager = SecretsManagerHandler() yield secrets_manager
def secret_client(): with moto.mock_secretsmanager(): session = boto3.session.Session(region_name=REGION) return session.client( service_name='secretsmanager', endpoint_url=f'https://secretsmanager.{REGION}.amazonaws.com' )
def sm(): with mock_secretsmanager(): client = boto3.client("secretsmanager") yield client
def create_mocked_secret_manager_connection(): with mock_secretsmanager(): yield boto3.session.Session().client(service_name="secretsmanager", region_name="us-east-1")
def fake_secrets(): with mock_secretsmanager(): sm = boto3.client("secretsmanager", region_name=FAKE_REGION) for name, value in FAKE_SECRET_VALUES.items(): sm.create_secret(Name=name, SecretString=json.dumps(value)) yield
def mock_secretsmanager_resource(): with mock_secretsmanager(): yield boto3.client("secretsmanager")