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