def test_404_endpoint(self, mocker):
mocker.patch.object(update,
'urlopen',
return_value=StringIO('404 Page Not Found'))
assert update.latest_version() is None
def test_http_error(self, mocker):
mocker.patch.object(update, 'urlopen', side_effect=IOError())
assert update.latest_version() is None
def test_healthy_endpoint(self, mocker):
version = '1.2.3'
mocker.patch.object(update, 'urlopen', return_value=StringIO(version))
assert update.latest_version() == version