def test_hframe_rw_db():
"""
Create a pb buffer
write to db
read from db
:return:
"""
global engine_g
""" Create table """
hyperframe.HyperFrameRecord.create_table(engine_g)
""" Create some PB records """
hf1 = _make_hframe_record('inner_record')
hf2 = _make_hframe_record('outer_record', hframes=[
hf1,
])
""" Write out PBs as rows """
hf_hash = hashlib.md5(hf2.pb.SerializeToString()).hexdigest()
w_pb_db(hf2, engine_g)
""" Read in PBs as rows"""
hf_results = r_pb_db(hyperframe.HyperFrameRecord, engine_g)
hf_hash2 = None
for x in hf_results:
hf_hash2 = hashlib.md5(x.pb.SerializeToString()).hexdigest()
assert (hf_hash == hf_hash2)
def test_linkauth_rw_db():
"""
Create a pb buffer
write to db
read from db
:return:
"""
global engine_g
""" Create table """
hyperframe.LinkAuthBase.create_table(engine_g)
""" Create some PB records """
slar = _make_linkauth_records()
""" Write out PBs as rows """
slar_hash = hashlib.md5(slar.pb.SerializeToString()).hexdigest()
w_pb_db(slar, engine_g)
""" Read in PBs as rows"""
link_auth_results = r_pb_db(hyperframe.LinkAuthBase, engine_g)
slar_hash2 = None
for x in link_auth_results:
if x.pb.WhichOneof('auth') == 's3_auth':
slar_hash2 = hashlib.md5(x.pb.SerializeToString()).hexdigest()
assert (slar_hash == slar_hash2)
def test_link_rw_db():
"""
Create a pb buffer
write to db
read from db
:return:
"""
global engine_g
""" Create table """
hyperframe.LinkBase.create_table(engine_g)
""" Create some PB records """
local_link, s3_link, db_link = _make_link_records()
""" Write out PBs as rows """
local_hash = w_pb_db(local_link, engine_g)
s3_hash = w_pb_db(s3_link, engine_g)
db_hash = w_pb_db(db_link, engine_g)
""" Read in PBs as rows"""
link_results = r_pb_db(hyperframe.LinkBase, engine_g)
local_hash2 = None
s3_hash2 = None
db_hash2 = None
for x in link_results:
if x.pb.WhichOneof('link') == 'local':
local_hash2 = hashlib.md5(local_link.pb.SerializeToString()).hexdigest()
if x.pb.WhichOneof('link') == 's3':
s3_hash2 = hashlib.md5(s3_link.pb.SerializeToString()).hexdigest()
if x.pb.WhichOneof('link') == 'database':
db_hash2 = hashlib.md5(db_link.pb.SerializeToString()).hexdigest()
assert (local_hash == local_hash2)
assert (s3_hash == s3_hash2)
assert (db_hash == db_hash2)
