tripleo-ci/scripts/emit_releases_file/test_get_dlrn_hash.py
John Trowbridge 6f06040bad Wire in get_dlrn_hash
This wires in the get_dlrn_hash function to the release dictionary,
so that we end up with an actual dlrn hash instead of a named hash.

Release dictionary tests updated to mock this function, including
verification of the expected calls to get_dlrn_hash.

This also adds more comprehensive logging and unittests to the
get_dlrn_hash function.

Change-Id: I41087bf0c247b933a641ea0da582b5f2b368840e
2018-05-21 08:25:09 -04:00

165 lines
6.9 KiB
Python

from emit_releases_file import get_dlrn_hash
import mock
import pytest
@mock.patch('logging.getLogger')
@mock.patch('requests.get')
def test_get_dlrn_hash_ok(mock_get, mock_logging):
mock_logger = mock.MagicMock()
mock_logging.return_value = mock_logger
mock_log_exception = mock.MagicMock()
mock_log_warning = mock.MagicMock()
mock_log_info = mock.MagicMock()
mock_logger.exception = mock_log_exception
mock_logger.warning = mock_log_warning
mock_logger.info = mock_log_info
mock_response = mock.Mock()
mock_response.ok = True
mock_response.text = (
'[delorean]\nname=delorean-openstack-nova-81c23c04'
'7e8e0fc03b54164921f49fdb4103202c\nbaseurl=https:/'
'/trunk.rdoproject.org/centos7/81/c2/81c23c047e8e0'
'fc03b54164921f49fdb4103202c_b333f915\nenabled=1\n'
'gpgcheck=0\npriority=1')
mock_get.return_value = mock_response
release = 'master'
hash_name = 'current-tripleo'
dlrn_hash = '81c23c047e8e0fc03b54164921f49fdb4103202c_b333f915'
repo_url = ('https://trunk.rdoproject.org/centos7-%s/%s/delorean.repo' %
(release, hash_name))
assert get_dlrn_hash(release, hash_name) == dlrn_hash
mock_get.assert_called_once_with(repo_url, timeout=4)
mock_log_info.assert_called_once_with("Got DLRN hash: {} for the named "
"hash: {} on the {} "
"release".format(dlrn_hash,
hash_name,
release))
mock_log_warning.assert_not_called()
mock_log_exception.assert_not_called()
@mock.patch('logging.getLogger')
@mock.patch('requests.get')
def test_null_response_raises_runtimeerror(mock_get, mock_logging):
mock_logger = mock.MagicMock()
mock_logging.return_value = mock_logger
mock_log_exception = mock.MagicMock()
mock_log_warning = mock.MagicMock()
mock_log_info = mock.MagicMock()
mock_logger.exception = mock_log_exception
mock_logger.warning = mock_log_warning
mock_logger.info = mock_log_info
release = 'master'
hash_name = 'current-tripleo'
repo_url = ('https://trunk.rdoproject.org/centos7-%s/%s/delorean.repo' %
(release, hash_name))
mock_get.return_value = None
with pytest.raises(RuntimeError):
get_dlrn_hash(release, hash_name)
mock_get.assert_called_with(repo_url, timeout=4)
assert mock_get.call_count == 10
mock_log_info.assert_not_called()
mock_log_warning.assert_called_with("Attempt 10 of 10 to get DLRN hash "
"failed to get a response.")
assert mock_log_warning.call_count == 10
mock_log_exception.assert_not_called()
@mock.patch('logging.getLogger')
@mock.patch('requests.get')
def test_get_dlrn_hash_500_then_200(mock_get, mock_logging):
mock_logger = mock.MagicMock()
mock_logging.return_value = mock_logger
mock_log_exception = mock.MagicMock()
mock_log_warning = mock.MagicMock()
mock_log_info = mock.MagicMock()
mock_logger.exception = mock_log_exception
mock_logger.warning = mock_log_warning
mock_logger.info = mock_log_info
mock_response_ok = mock.Mock()
mock_response_ok.ok = True
mock_response_ok.text = (
'[delorean]\nname=delorean-openstack-nova-81c23c04'
'7e8e0fc03b54164921f49fdb4103202c\nbaseurl=https:/'
'/trunk.rdoproject.org/centos7/81/c2/81c23c047e8e0'
'fc03b54164921f49fdb4103202c_b333f915\nenabled=1\n'
'gpgcheck=0\npriority=1')
mock_response_bad = mock.Mock()
mock_response_bad.ok = False
mock_response_bad.status_code = 500
release = 'master'
hash_name = 'current-tripleo'
dlrn_hash = '81c23c047e8e0fc03b54164921f49fdb4103202c_b333f915'
repo_url = ('https://trunk.rdoproject.org/centos7-%s/%s/delorean.repo' %
(release, hash_name))
mock_get.side_effect = [mock_response_bad, mock_response_ok]
assert get_dlrn_hash(release, hash_name, retries=10) == dlrn_hash
mock_get.assert_called_with(repo_url, timeout=4)
mock_log_info.assert_called_once_with("Got DLRN hash: {} for the named "
"hash: {} on the {} "
"release".format(dlrn_hash,
hash_name,
release))
mock_log_warning.assert_called_once_with("Attempt 1 of 10 to get DLRN "
"hash returned status code 500.")
mock_log_exception.assert_not_called()
@mock.patch('logging.getLogger')
@mock.patch('requests.get')
def test_get_dlrn_hash_timeout(mock_get, mock_logging):
mock_logger = mock.MagicMock()
mock_logging.return_value = mock_logger
mock_log_exception = mock.MagicMock()
mock_log_warning = mock.MagicMock()
mock_log_info = mock.MagicMock()
mock_logger.exception = mock_log_exception
mock_logger.warning = mock_log_warning
mock_logger.info = mock_log_info
release = 'master'
hash_name = 'current-tripleo'
repo_url = ('https://trunk.rdoproject.org/centos7-%s/%s/delorean.repo' %
(release, hash_name))
mock_get_exception = Exception("We need more power!")
mock_get.side_effect = mock_get_exception
with pytest.raises(RuntimeError):
get_dlrn_hash(release, hash_name, retries=10)
mock_get.assert_called_with(repo_url, timeout=4)
mock_log_info.assert_not_called()
mock_log_warning.assert_called_with("Attempt 10 of 10 to get DLRN hash "
"threw an exception.")
assert mock_log_warning.call_count == 10
mock_log_exception.assert_called_with(mock_get_exception)
assert mock_log_exception.call_count == 10
@mock.patch('logging.getLogger')
@mock.patch('requests.get')
def test_get_dlrn_hash_500_10_times(mock_get, mock_logging):
mock_logger = mock.MagicMock()
mock_logging.return_value = mock_logger
mock_log_exception = mock.MagicMock()
mock_log_warning = mock.MagicMock()
mock_log_info = mock.MagicMock()
mock_logger.exception = mock_log_exception
mock_logger.warning = mock_log_warning
mock_logger.info = mock_log_info
mock_response = mock.Mock()
mock_response.ok = False
mock_response.status_code = 500
release = 'master'
hash_name = 'current-tripleo'
repo_url = ('https://trunk.rdoproject.org/centos7-%s/%s/delorean.repo' %
(release, hash_name))
mock_get.return_value = mock_response
with pytest.raises(RuntimeError):
get_dlrn_hash(release, hash_name, retries=10)
mock_get.assert_called_with(repo_url, timeout=4)
mock_log_info.assert_not_called()
mock_log_warning.assert_called_with("Attempt 10 of 10 to get DLRN hash "
"returned status code 500.")
assert mock_log_warning.call_count == 10
mock_log_exception.assert_not_called()