21 lines
863 B
Python
21 lines
863 B
Python
import pytest
|
|
|
|
from util.config.validators import ConfigValidationException
|
|
from util.config.validators.validate_database import DatabaseValidator
|
|
|
|
@pytest.mark.parametrize('unvalidated_config,user,user_password,expected', [
|
|
(None, None, None, TypeError),
|
|
({}, None, None, KeyError),
|
|
({'DB_URI': 'sqlite:///:memory:'}, None, None, None),
|
|
({'DB_URI': 'invalid:///:memory:'}, None, None, KeyError),
|
|
({'DB_NOTURI': 'sqlite:///:memory:'}, None, None, KeyError),
|
|
({'DB_URI': 'mysql:///someinvalid'}, None, None, ConfigValidationException),
|
|
])
|
|
def test_validate_database(unvalidated_config, user, user_password, expected):
|
|
validator = DatabaseValidator()
|
|
|
|
if expected is not None:
|
|
with pytest.raises(expected):
|
|
validator.validate(unvalidated_config, user, user_password)
|
|
else:
|
|
validator.validate(unvalidated_config, user, user_password)
|