def test_retrieve(self, mock_get_method):
""" tests that the retrieve method works expected. """
_setup_mock_response_for_retriever(mock_get_method,
self._public_key_pem)
retriever = HTTPSMultiRepositoryPublicKeyRetriever(self.keystore_urls)
self.assertEqual(retriever.retrieve('example/eg'),
self._public_key_pem)
def test_retrieve_with_500_error(self, mock_get_method):
""" tests that the retrieve method works as expected
when the first key repository returns a server error response.
"""
retriever = HTTPSMultiRepositoryPublicKeyRetriever(self.keystore_urls)
_setup_mock_response_for_retriever(mock_get_method,
self._public_key_pem)
valid_response = mock_get_method.return_value
del mock_get_method.return_value
server_exception = requests.exceptions.HTTPError(response=mock.Mock(
status_code=500))
mock_get_method.side_effect = [server_exception, valid_response]
self.assertEqual(retriever.retrieve('example/eg'),
self._public_key_pem)