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
35
36
37
|
import datetime
from datetime import timezone
from unittest import mock
import pytest
import aiobotocore.credentials
import aiobotocore.session
import aiobotocore.signers
@pytest.mark.moto
@pytest.mark.asyncio
async def test_signers_generate_db_auth_token(rds_client):
hostname = 'prod-instance.us-east-1.rds.amazonaws.com'
port = 3306
username = 'someusername'
clock = datetime.datetime(2016, 11, 7, 17, 39, 33, tzinfo=timezone.utc)
with mock.patch('datetime.datetime') as dt:
dt.utcnow.return_value = clock
result = await aiobotocore.signers.generate_db_auth_token(
rds_client, hostname, port, username
)
result2 = await rds_client.generate_db_auth_token(
hostname, port, username
)
# A scheme needs to be appended to the beginning or urlsplit may fail
# on certain systems.
assert result.startswith(
'prod-instance.us-east-1.rds.amazonaws.com:3306/?AWSAccessKeyId=xxx&'
)
assert result2.startswith(
'prod-instance.us-east-1.rds.amazonaws.com:3306/?AWSAccessKeyId=xxx&'
)
|
