def test_quotas(mocker):
from iris.sender.quota import ApplicationQuota
from iris.metrics import stats
from gevent import sleep
mocker.patch('iris.sender.quota.ApplicationQuota.get_new_rules',
return_value=[(u'testapp', 5, 2, 120, 120, u'testuser',
u'user', u'iris-plan', 10)])
mocker.patch('iris.sender.quota.ApplicationQuota.notify_incident')
mocker.patch('iris.sender.quota.ApplicationQuota.notify_target')
quotas = ApplicationQuota(None, None, None)
sleep(1)
assert quotas.allow_send({'application': 'testapp'})
assert quotas.allow_send({'application': 'testapp'})
assert quotas.allow_send({'application': 'testapp'}) # Breach soft quota
assert quotas.allow_send({'application': 'testapp'})
assert quotas.allow_send({'application': 'testapp'})
assert not quotas.allow_send({'application': 'testapp'
}) # Breach hard quota
assert not quotas.allow_send({'application': 'testapp'})
assert stats['quota_soft_exceed_cnt'] == 3
assert stats['quota_hard_exceed_cnt'] == 2
assert stats['app_testapp_quota_hard_usage_pct'] == 100
assert stats['app_testapp_quota_soft_usage_pct'] == 200
for _ in xrange(10):
assert quotas.allow_send({'application': 'app_without_quota'})
def init_sender(config):
db.init(config)
cache.init(config)
metrics.init(config, 'iris-sender', default_sender_metrics)
api_cache.cache_priorities()
api_cache.cache_applications()
api_cache.cache_modes()
global should_mock_gwatch_renewer, send_message
if config['sender'].get('debug'):
logger.info('DEBUG MODE')
should_mock_gwatch_renewer = True
should_skip_send = True
else:
should_skip_send = False
should_mock_gwatch_renewer = should_mock_gwatch_renewer or config.get(
'skipgmailwatch', False)
should_skip_send = should_skip_send or config.get('skipsend', False)
if should_skip_send:
config['vendors'] = [{
'type': 'iris_dummy',
'name': 'iris dummy vendor'
}]
global quota
quota = ApplicationQuota(db, cache.targets_for_role,
config['sender'].get('sender_app'))
def init_sender(config):
gevent.signal(signal.SIGINT, sender_shutdown)
gevent.signal(signal.SIGTERM, sender_shutdown)
gevent.signal(signal.SIGQUIT, sender_shutdown)
api_host = config['sender'].get('api_host', 'http://localhost:16649')
db.init(config)
cache.init(api_host, config)
metrics.init(config, 'iris-sender', default_sender_metrics)
api_cache.cache_priorities()
api_cache.cache_applications()
api_cache.cache_modes()
global should_mock_gwatch_renewer, send_message
if config['sender'].get('debug'):
logger.info('DEBUG MODE')
should_mock_gwatch_renewer = True
should_skip_send = True
else:
should_skip_send = False
should_mock_gwatch_renewer = should_mock_gwatch_renewer or config.get(
'skipgmailwatch', False)
should_skip_send = should_skip_send or config.get('skipsend', False)
if should_skip_send:
config['vendors'] = [{
'type': 'iris_dummy',
'name': 'iris dummy vendor'
}]
global quota
quota = ApplicationQuota(db, cache.targets_for_role,
config['sender'].get('sender_app'))
global coordinator
zk_hosts = config['sender'].get('zookeeper_cluster', False)
if zk_hosts:
logger.info('Initializing coordinator with ZK: %s', zk_hosts)
coordinator = Coordinator(zk_hosts=zk_hosts,
hostname=socket.gethostname(),
port=config['sender'].get('port', 2321),
join_cluster=True)
else:
logger.info(
'ZK cluster info not specified. Using master status from config')
coordinator = NonClusterCoordinator(
is_master=config['sender'].get('is_master', True),
slaves=config['sender'].get('slaves', []))