def error_handler(errors):
# filter out the rather common MSG_ONE_OF errors
log_errors = []
for error in errors:
if error.get('description', '') != MSG_ONE_OF:
log_errors.append(error)
if log_errors:
get_client('ichnaea').debug('error_handler' + repr(log_errors))
return _JSONError(errors, errors.status)
def setUpClass(cls):
super(DBTestCase, cls).setup_engine()
# Clobber the stream with a debug version
from heka.holder import get_client
heka_client = get_client('ichnaea')
use_hekatest(heka_client)
def test_get_client_with_cfg(self):
cfg = {'logger': 'amo.dev',
'stream': {'class': 'heka.streams.UdpStream',
'host': ['logstash1', 'logstash2'],
'port': '5566'},
}
heka = get_client('amo.dev', cfg)
assert heka != None
def test_db_config(self):
self.db_master = _make_db()
self.db_slave = _make_db(create=False)
self.setup_session()
app = _make_app(_db_master=self.db_master,
_db_slave=self.db_slave,
_heka_client=get_client('ichnaea'))
app.get('/stats_location.json', status=200)
def test_get_client_with_cfg(self):
cfg = {
'logger': 'amo.dev',
'stream': {
'class': 'heka.streams.UdpStream',
'host': ['logstash1', 'logstash2'],
'port': '5566'
},
}
heka = get_client('amo.dev', cfg)
assert heka != None
def setUpClass(cls):
super(AppTestCase, cls).setup_engine()
# Clobber the stream with a debug version
from heka.holder import get_client
heka_client = get_client('ichnaea')
use_hekatest(heka_client)
cls.app = _make_app(_db_master=cls.db_master,
_db_slave=cls.db_slave,
_heka_client=heka_client)
def test_db_hooks(self):
settings = {
'db_master': SQLURI,
'db_master_socket': SQLSOCKET,
'db_slave': SQLURI,
'db_slave_socket': SQLSOCKET,
'_heka_client': get_client('ichnaea'),
}
app = _make_app(**settings)
self.db_master = app.app.registry.db_master
self.db_slave = app.app.registry.db_slave
self.setup_session()
app.get('/stats_location.json', status=200)
def configure_heka(registry_settings={}):
# If a test client is defined just use that instead of whatever is
# defined in the configuration
if '_heka_client' in registry_settings:
return registry_settings['_heka_client']
# deal with konfig's include/extends syntax and construct a merged
# file-like object from all the files
merged_stream = StringIO()
konfig = config()
konfig.write(merged_stream)
merged_stream.seek(0)
client = get_client('ichnaea')
client = client_from_text_config(merged_stream.read(), 'heka', client)
return client
def setUp(self):
CeleryTestCase.setUp(self)
self.heka_client = get_client('ichnaea')
self.heka_client.stream.msgs.clear()
def test_get_client(self):
heka = get_client('new_client')
assert heka != None
def get_heka_client():
return get_client('ichnaea')
def test_get_client(self):
heka = get_client('new_client')
assert heka != None
def get_heka_client():
return get_client('ichnaea')
def __call__(self, *args, **kw):
with get_client('ichnaea').timer("task." + self.shortname):
result = super(DatabaseTask, self).__call__(*args, **kw)
return result
def get_heka_client():
return get_client("ichnaea")