def _save_packets(qeez_token, res_dc, sync=False):
'''Saves data packets (to all possible DBs)
'''
save_packets_to_stat(qeez_token, res_dc, redis_conn=get_stat_redis())
if sync:
return direct_stat_save(qeez_token, res_dc, atime=gmtime())
job = enqueue_stat_save(qeez_token,
res_dc,
atime=gmtime(),
redis_conn=get_save_redis())
return bool(job)
def test_save_packets_to_stat():
res_dcs = [
{'': '', 'a': 'b'},
{b'': b'', b'a': b'b'},
{b'': b'', b'1:2:3:4:5:6:7': b'1:2:3'},
{b'7:6:5:4:3:2:1': b'2,3,1:6.5:4', b'1:2:3:4:5:6:7': b'1:2:3'},
]
redis_conn = get_redis(CFG['STAT_REDIS'])
for res_dc in res_dcs:
try:
assert utils.save_packets_to_stat(qeez_token, res_dc) is True
except Exception as exc:
assert exc is None
try:
assert utils.save_packets_to_stat(
qeez_token, res_dc, redis_conn) is True
except Exception as exc:
assert exc is None
def test_save_packets_to_stat():
res_dcs = [
{'': '', 'a': 'b'},
{b'': b'', b'a': b'b'},
{b'': b'', b'1:2:3:4:5:6:7': b'1:2:3'},
{b'7:6:5:4:3:2:1': b'2,3,1:6.5:4', b'1:2:3:4:5:6:7': b'1:2:3'},
]
redis_conn = get_redis(CFG['STAT_REDIS'])
for res_dc in res_dcs:
try:
assert utils.save_packets_to_stat(qeez_token, res_dc) is True
except Exception as exc:
assert exc is None
try:
assert utils.save_packets_to_stat(
qeez_token, res_dc, redis_conn) is True
except Exception as exc:
assert exc is None
def test_pull_stat_res_ok():
from qeez_stats.utils import save_packets_to_stat
stat_id = CFG['STAT_CALC_FN']
qeez_token = get_token()
res_dc = {
b'7:6:5:4:3:2:1': b'2,3,1:6.5:4',
b'1:2:3:4:5:6:7': b'1:2:3',
}
save_packets_to_stat(qeez_token, res_dc, redis_conn=None)
job = queues.enqueue_stat_calc(stat_id, qeez_token, redis_conn=None)
assert isinstance(job, Job)
assert job.id
queue = Queue(name=job.origin, connection=job.connection)
worker = SimpleWorker([queue], connection=queue.connection)
worker.work(burst=True)
res = queues.pull_stat_res(stat_id, qeez_token, redis_conn=None)
assert isinstance(res, float)
assert res == 123.1
def _save_packets(qeez_token, res_dc):
'''Saves data packets (to all possible DBs)
'''
save_packets_to_stat(qeez_token, res_dc, redis_conn=get_stat_redis())
enqueue_stat_save(
qeez_token, res_dc, atime=gmtime(), redis_conn=get_save_redis())
