Skip to content

Commit

Permalink
remove the regions
Browse files Browse the repository at this point in the history
  • Loading branch information
WillGibson committed Nov 13, 2024
1 parent 9d1d5c6 commit 77ca9e4
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 6 deletions.
4 changes: 2 additions & 2 deletions postgres/manage_users.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,8 +73,8 @@ def handler(event, context):
username = event['Username']
user_permissions = event['Permissions']

secrets_manager = boto3.client("secretsmanager", region_name="eu-west-2")
ssm = boto3.client("ssm", region_name="eu-west-2")
secrets_manager = boto3.client("secretsmanager")
ssm = boto3.client("ssm")

master_user = json.loads(secrets_manager.get_secret_value(SecretId=db_master_user_secret_arn)["SecretString"])

Expand Down
8 changes: 4 additions & 4 deletions postgres/tests/test_manage_users.py
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,7 @@ def test_create_or_update_db_user_when_user_exists(self):

@mock_aws
def test_create_or_update_user_secret(self):
ssm = boto3.client("ssm", region_name="eu-west-2")
ssm = boto3.client("ssm")
user_secret_name = "/test/secret"
user_secret_string = {"username": "test_user", "password": "test_password"}

Expand All @@ -97,7 +97,7 @@ def test_create_or_update_user_secret(self):

@mock_aws
def test_create_or_update_user_secret_overwrites(self):
ssm = boto3.client("ssm", region_name="eu-west-2")
ssm = boto3.client("ssm")
user_secret_name = "/test/secret"
user_secret_string = {"username": "test_user", "password": "test_password"}
ssm.put_parameter(Name=user_secret_name, Value="blah", Type="String")
Expand All @@ -114,7 +114,7 @@ def test_create_or_update_user_secret_overwrites(self):
@patch("postgres.manage_users.psycopg2.connect")
@mock_aws
def test_handler(self, mock_connect, mock_create_or_update_db_user):
secretsmanager = boto3.client("secretsmanager", region_name="eu-west-2")
secretsmanager = boto3.client("secretsmanager")
secret_id = secretsmanager.create_secret(
Name=self.secret_name, SecretString=self.secret_string
)["ARN"]
Expand All @@ -126,7 +126,7 @@ def test_handler(self, mock_connect, mock_create_or_update_db_user):

handler(self.event, self.context)

user_password = json.loads(boto3.client("ssm", region_name="eu-west-2").get_parameter(Name=self.secret_name, WithDecryption=True)["Parameter"]["Value"])[
user_password = json.loads(boto3.client("ssm").get_parameter(Name=self.secret_name, WithDecryption=True)["Parameter"]["Value"])[
"password"
]

Expand Down

0 comments on commit 77ca9e4

Please sign in to comment.