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")
