blob: 86533bc6db999eb0109e198a83cada51ab88dc44 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
from src.secrets_manager import sm_client, create_secret, list_secret
import boto3
from moto import mock_aws
import json
import pytest
import os
pytest.fixture(scope='class')
def mock_aws_credentials():
"""Mocked AWS Credentials for moto."""
os.environ["AWS_ACCESS_KEY_ID"] = "testing"
os.environ["AWS_SECRET_ACCESS_KEY"] = "testing"
os.environ["AWS_SECURITY_TOKEN"] = "testing"
os.environ["AWS_SESSION_TOKEN"] = "testing"
os.environ["AWS_DEFAULT_REGION"] = "eu-west-2"
@pytest.fixture(scope='class')
def mock_sm_client(mock_aws_credentials):
with mock_aws():
yield boto3.client('secretsmanager')
def test_create_secret_stores_secrets(mock_sm_client):
cohort_id = "test_cohort_id"
user = "test_user_id"
password = "test_password"
host = "test_host"
database = "test_database"
port = "test_port"
secret_name = "test_secret"
response = create_secret(mock_sm_client, secret_name, cohort_id, user, password, host, database, port)
assert response['Name'] == secret_name
|
