def test_db_version(mocker, version, is_ok):
mocker.patch.object(app.puppetdb, "current_version", return_value=version)
if is_ok:
utils.check_db_version(app.puppetdb)
else:
with pytest.raises(SystemExit) as e:
utils.check_db_version(app.puppetdb)
assert e.code == 1
def test_db_empty_response(mocker, mock_err_log):
err = "Empty Response"
def connection_error():
raise EmptyResponseError(err)
mocker.patch.object(app.puppetdb, 'current_version',
side_effect=connection_error)
with pytest.raises(SystemExit) as e:
utils.check_db_version(app.puppetdb)
assert e.code == 2
def test_db_connection_error(mocker, mock_err_log):
err = "ConnectionError"
def connection_error():
x = Response()
x.status_code = 500
x.reason = err
raise ConnectionError(err, response=x)
mocker.patch.object(app.puppetdb, 'current_version',
side_effect=connection_error)
with pytest.raises(SystemExit) as e:
utils.check_db_version(app.puppetdb)
assert e.code == 2
def test_db_http_error(mocker, mock_err_log):
err = "NotFound"
def raise_http_error():
x = Response()
x.status_code = 404
x.reason = err
raise HTTPError(err, response=x)
mocker.patch.object(app.puppetdb, 'current_version',
side_effect=raise_http_error)
with pytest.raises(SystemExit) as e:
utils.check_db_version(app.puppetdb)
assert e.code == 2
def before_first_request():
check_db_version(puppetdb)
def test_db_invalid_version(mocker, mock_err_log):
mocker.patch.object(app.puppetdb, 'current_version', return_value='4')
with pytest.raises(SystemExit) as e:
utils.check_db_version(app.puppetdb)
assert e.code == 2