def test_returns_public_key_in_range(self):
cloudtrail_client = Mock()
cloudtrail_client.list_public_keys.return_value = {
'PublicKeyList': [
{
'Fingerprint': 'a',
'OtherData': 'a',
'Value': 'a'
},
{
'Fingerprint': 'b',
'OtherData': 'b',
'Value': 'b'
},
{
'Fingerprint': 'c',
'OtherData': 'c',
'Value': 'c'
},
]
}
provider = PublicKeyProvider(cloudtrail_client)
start_date = START_DATE
end_date = start_date + timedelta(days=2)
keys = provider.get_public_keys(start_date, end_date)
self.assertEquals(
{
'a': {
'Fingerprint': 'a',
'OtherData': 'a',
'Value': 'a'
},
'b': {
'Fingerprint': 'b',
'OtherData': 'b',
'Value': 'b'
},
'c': {
'Fingerprint': 'c',
'OtherData': 'c',
'Value': 'c'
},
}, keys)
cloudtrail_client.list_public_keys.assert_has_calls(
[call(EndTime=end_date, StartTime=start_date)])
def test_returns_public_key_in_range(self):
cloudtrail_client = Mock()
cloudtrail_client.list_public_keys.return_value = {'PublicKeyList': [
{'Fingerprint': 'a', 'OtherData': 'a', 'Value': 'a'},
{'Fingerprint': 'b', 'OtherData': 'b', 'Value': 'b'},
{'Fingerprint': 'c', 'OtherData': 'c', 'Value': 'c'},
]}
provider = PublicKeyProvider(cloudtrail_client)
start_date = START_DATE
end_date = start_date + timedelta(days=2)
keys = provider.get_public_keys(start_date, end_date)
self.assertEquals({
'a': {'Fingerprint': 'a', 'OtherData': 'a', 'Value': 'a'},
'b': {'Fingerprint': 'b', 'OtherData': 'b', 'Value': 'b'},
'c': {'Fingerprint': 'c', 'OtherData': 'c', 'Value': 'c'},
}, keys)
cloudtrail_client.list_public_keys.assert_has_calls(
call(EndTime=end_date, StartTime=start_date))
def test_returns_public_key_in_range(self):
cloudtrail_client = Mock()
cloudtrail_client.list_public_keys.return_value = {
"PublicKeyList": [
{"Fingerprint": "a", "OtherData": "a", "Value": "a"},
{"Fingerprint": "b", "OtherData": "b", "Value": "b"},
{"Fingerprint": "c", "OtherData": "c", "Value": "c"},
]
}
provider = PublicKeyProvider(cloudtrail_client)
start_date = START_DATE
end_date = start_date + timedelta(days=2)
keys = provider.get_public_keys(start_date, end_date)
self.assertEquals(
{
"a": {"Fingerprint": "a", "OtherData": "a", "Value": "a"},
"b": {"Fingerprint": "b", "OtherData": "b", "Value": "b"},
"c": {"Fingerprint": "c", "OtherData": "c", "Value": "c"},
},
keys,
)
cloudtrail_client.list_public_keys.assert_has_calls([call(EndTime=end_date, StartTime=start_date)])