Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix Tever.reload to work with RegStateRecord and add unit test #668

Merged
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
44 changes: 19 additions & 25 deletions src/keri/vdr/eventing.py
Original file line number Diff line number Diff line change
Expand Up @@ -682,7 +682,7 @@ class Tever:
"""
NoBackers = False

def __init__(self, cues=None, stt=None, serder=None, seqner=None, saider=None,
def __init__(self, cues=None, rsr=None, serder=None, seqner=None, saider=None,
bigers=None, db=None, reger=None, noBackers=None, estOnly=None,
regk=None, local=False):
""" Create incepting tever and state from registry inception serder
Expand All @@ -691,7 +691,7 @@ def __init__(self, cues=None, stt=None, serder=None, seqner=None, saider=None,

Parameters:
serder (Serder): instance of registry inception event
stt (Serder): transaction state notice state message Serder
rsr (RegStateRecord): transaction state notice state message Serder
seqner (Seqner): issuing event sequence number from controlling KEL.
saider (Saider): issuing event said from controlling KEL.
bigers (list): list of Siger instances of indexed backer signatures of
Expand All @@ -711,7 +711,7 @@ def __init__(self, cues=None, stt=None, serder=None, seqner=None, saider=None,

"""

if not (stt or serder):
if not (rsr or serder):
raise ValueError("Missing required arguments. Need state or serder")

self.reger = reger if reger is not None else viring.Reger()
Expand All @@ -720,8 +720,8 @@ def __init__(self, cues=None, stt=None, serder=None, seqner=None, saider=None,
self.db = db if db is not None else basing.Baser(reopen=True)
self.local = True if local else False

if stt: # preload from state
self.reload(stt)
if rsr: # preload from state
self.reload(rsr)
return

self.version = serder.version
Expand Down Expand Up @@ -758,40 +758,34 @@ def __init__(self, cues=None, stt=None, serder=None, seqner=None, saider=None,

self.regk = self.prefixer.qb64

def reload(self, ksn):
def reload(self, rsr):
""" Reload Tever attributes (aka its state) from state serder

Reload Tever attributes (aka its state) from state serder

Parameters:
ksn (Serder): instance of key stat notice 'ksn' message body
rsr (RegStateRecord): instance of key stat notice 'ksn' message body

"""

for k in TSN_LABELS:
if k not in ksn.ked:
raise ValidationError("Missing element = {} from {} event."
" evt = {}.".format(k, Ilks.ksn,
ksn.pretty()))
ked = asdict(rsr)

self.version = ksn.version
self.pre = ksn.ked["ii"]
self.regk = ksn.ked["i"]
self.version = rsr.vn
self.pre = ked["ii"]
self.regk = ked["i"]
self.prefixer = Prefixer(qb64=self.regk)
self.sn = ksn.sn
self.ilk = ksn.ked["et"]
self.toad = int(ksn.ked["bt"], 16)
self.baks = ksn.ked["b"]
self.cuts = ksn.ked["br"]
self.adds = ksn.ked["ba"]
self.sn = int(ked['s'], 16)
self.ilk = ked["et"]
self.toad = int(ked["bt"], 16)
self.baks = ked["b"]

self.noBackers = True if TraitDex.NoBackers in ksn.ked["c"] else False
self.estOnly = True if TraitDex.EstOnly in ksn.ked["c"] else False
self.noBackers = True if TraitDex.NoBackers in ked["c"] else False
self.estOnly = True if TraitDex.EstOnly in ked["c"] else False

if (raw := self.reger.getTvt(key=dgKey(pre=self.prefixer.qb64,
dig=ksn.ked['d']))) is None:
dig=ked['d']))) is None:
raise kering.MissingEntryError("Corresponding event for state={} not found."
"".format(ksn.pretty()))
"".format(ked))
self.serder = serdering.SerderKERI(raw=bytes(raw))

def state(self): #state(self, kind=Serials.json)
Expand Down
2 changes: 1 addition & 1 deletion src/keri/vdr/viring.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ def __getitem__(self, k):
if (rsr := self.reger.states.get(keys=k)) is None:
raise ex # reraise KeyError
try:
tever = eventing.Tever(stt=rsr, db=self.db, reger=self.reger)
tever = eventing.Tever(rsr=rsr, db=self.db, reger=self.reger)
except kering.MissingEntryError: # no kel event for keystate
raise ex # reraise KeyError
super(rbdict, self).__setitem__(k, tever)
Expand Down
26 changes: 26 additions & 0 deletions tests/vdr/test_txn_state.py
Original file line number Diff line number Diff line change
Expand Up @@ -525,3 +525,29 @@ def test_credential_tsn_message(mockHelpingNowUTC, mockCoringRandomNonce, mockHe
keys = (creder.said, bobHab.pre)
saider = bamReger.txnsb.saiderdb.get(keys=keys)
assert saider.qb64b == b'EIxhyBA8h6BMmtWEJzqNkoAquIkMucpXbdY3kQX25GQu'


def test_tever_reload(mockHelpingNowUTC, mockCoringRandomNonce, mockHelpingNowIso8601):

with habbing.openHby(name="bob", base="test") as hby:

bobHab = hby.makeHab(name="bob", isith='1', icount=1,)
assert bobHab.pre == 'EA_SbBUZYwqLVlAAn14d6QUBQCSReJlZ755JqTgmRhXH'

regery = credentialing.Regery(hby=hby, name="test", temp=True)
issuer = regery.makeRegistry(prefix=bobHab.pre, name=bobHab.name)
rseal = SealEvent(issuer.regk, "0", issuer.regd)._asdict()
bobHab.interact(data=[rseal])
seqner = coring.Seqner(sn=bobHab.kever.sn)
issuer.anchorMsg(pre=issuer.regk,
regd=issuer.regd,
seqner=seqner,
saider=coring.Saider(qb64=bobHab.kever.serder.said))
regery.processEscrows()

assert issuer.regk == 'ECbNKwkTjZqsfwNLxTnraPImegy1YeQ2-pCrTBQmu3i6'

rsr = regery.reger.states.get(keys=issuer.regk)
tever = eventing.Tever(rsr=rsr, reger=regery.reger)
assert tever.regk == issuer.regk
assert tever.pre == bobHab.pre
Loading