def __init__(self, queue_resource=None):
LOG.debug('Creating OrdersController')
self.order_repo = repo.get_order_repository()
self.queue = queue_resource or async_client.TaskClient()
self.type_order_validator = validators.TypeOrderValidator()
self.quota_enforcer = quota.QuotaEnforcer('orders', self.order_repo)
def __init__(self):
self.consumer_repo = repo.get_container_consumer_repository()
self.container_repo = repo.get_container_repository()
self.secret_repo = repo.get_secret_repository()
self.validator = validators.ContainerValidator()
self.quota_enforcer = quota.QuotaEnforcer('containers',
self.container_repo)
def __init__(self, container_id):
self.container_id = container_id
self.consumer_repo = repo.get_container_consumer_repository()
self.container_repo = repo.get_container_repository()
self.project_repo = repo.get_project_repository()
self.validator = validators.ContainerConsumerValidator()
self.quota_enforcer = quota.QuotaEnforcer('consumers',
self.consumer_repo)
def __init__(self, secret):
super().__init__()
self.secret = secret
self.secret_id = secret.id
self.consumer_repo = repo.get_secret_consumer_repository()
self.secret_repo = repo.get_secret_repository()
self.project_repo = repo.get_project_repository()
self.validator = validators.SecretConsumerValidator()
self.quota_enforcer = quota.QuotaEnforcer('consumers',
self.consumer_repo)
def __init__(self):
LOG.debug('Creating CertificateAuthoritiesController')
self.ca_repo = repo.get_ca_repository()
self.project_ca_repo = repo.get_project_ca_repository()
self.preferred_ca_repo = repo.get_preferred_ca_repository()
self.project_repo = repo.get_project_repository()
self.validator = validators.NewCAValidator()
self.quota_enforcer = quota.QuotaEnforcer('cas', self.ca_repo)
# Populate the CA table at start up
cert_resources.refresh_certificate_resources()
def test_should_pass_below_limit(self):
test_repo = DummyRepoForTestingQuotaEnforcement(4)
quota_enforcer = quota.QuotaEnforcer('secrets', test_repo)
five_project_quotas = {
'consumers': 5,
'containers': 5,
'orders': 5,
'secrets': 5,
'cas': 5
}
self.quota_driver.set_project_quotas(self.project.external_id,
five_project_quotas)
quota_enforcer.enforce(self.project)
def test_should_raise_above_limit(self):
test_repo = DummyRepoForTestingQuotaEnforcement(6)
quota_enforcer = quota.QuotaEnforcer('secrets', test_repo)
five_project_quotas = {
'consumers': 5,
'containers': 5,
'orders': 5,
'secrets': 5,
'cas': 5
}
self.quota_driver.set_project_quotas(self.project.external_id,
five_project_quotas)
exception = self.assertRaises(excep.QuotaReached,
quota_enforcer.enforce, self.project)
self.assertIn('Quota reached for project', six.text_type(exception))
self.assertIn('my_keystone_id', six.text_type(exception))
self.assertIn('secrets', six.text_type(exception))
self.assertIn(str(5), six.text_type(exception))
def test_should_raise_disabled_value(self):
test_repo = DummyRepoForTestingQuotaEnforcement(0)
quota_enforcer = quota.QuotaEnforcer('secrets', test_repo)
disabled_project_quotas = {
'consumers': 0,
'containers': 0,
'orders': 0,
'secrets': 0,
'cas': 0
}
self.quota_driver.set_project_quotas(self.project.external_id,
disabled_project_quotas)
exception = self.assertRaises(excep.QuotaReached,
quota_enforcer.enforce, self.project)
self.assertIn('Quota reached for project', exception.message)
self.assertIn('my_keystone_id', exception.message)
self.assertIn('secrets', exception.message)
self.assertIn(str(0), exception.message)
def __init__(self):
LOG.debug('=== Creating MutualAttestationController ===')
self.secret_repo = repo.get_secret_repository()
self.quota_enforcer = quota.QuotaEnforcer('secrets', self.secret_repo)
def test_should_pass_default_unlimited(self):
test_repo = DummyRepoForTestingQuotaEnforcement(0)
quota_enforcer = quota.QuotaEnforcer('secrets', test_repo)
quota_enforcer.enforce(self.project)
def __init__(self):
LOG.debug('Creating SecretsController')
self.validator = validators.NewSecretValidator()
self.secret_repo = repo.get_secret_repository()
self.quota_enforcer = quota.QuotaEnforcer('secrets', self.secret_repo)
