49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
|
import pytest
|
||
|
|
|
||
|
|
from mock import patch
|
||
|
|
|
||
|
|
from util.config.validators import ConfigValidationException
|
||
|
|
from util.config.validators.validate_license import LicenseValidator
|
||
|
|
from util.morecollections import AttrDict
|
||
|
|
from util.license import License, QUAY_DEPLOYMENTS_ENTITLEMENT, QUAY_ENTITLEMENT
|
||
|
|
|
||
|
|
from test.fixtures import *
|
||
|
|
|
||
|
|
@pytest.mark.parametrize('deployments, allowed_deployments', [
|
||
|
|
(1, 1),
|
||
|
|
(3, 3),
|
||
|
|
(3, 2),
|
||
|
|
(3, 1),
|
||
|
|
])
|
||
|
|
def test_too_many_storage_engines(deployments, allowed_deployments, app):
|
||
|
|
def get_license():
|
||
|
|
decoded = {
|
||
|
|
'expirationDate': '2157-12-1',
|
||
|
|
'subscriptions': {
|
||
|
|
'someSubscription': {
|
||
|
|
'serviceEnd': '2157-12-1',
|
||
|
|
'durationPeriod': 'yearly',
|
||
|
|
'entitlements': {
|
||
|
|
QUAY_ENTITLEMENT: 1,
|
||
|
|
QUAY_DEPLOYMENTS_ENTITLEMENT: allowed_deployments,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
return License(decoded)
|
||
|
|
|
||
|
|
storage_configs = [(str(i), {}) for i in range(0, deployments)]
|
||
|
|
|
||
|
|
with patch('app.config_provider.get_license', get_license):
|
||
|
|
validator = LicenseValidator()
|
||
|
|
|
||
|
|
if allowed_deployments < deployments:
|
||
|
|
with pytest.raises(ConfigValidationException):
|
||
|
|
validator.validate({
|
||
|
|
'DISTRIBUTED_STORAGE_CONFIG': storage_configs,
|
||
|
|
}, None, None)
|
||
|
|
else:
|
||
|
|
validator.validate({
|
||
|
|
'DISTRIBUTED_STORAGE_CONFIG': storage_configs,
|
||
|
|
}, None, None)
|