Check that Ursula sent back the proper number of CFrags.

pull/142/head
jMyles 2017-12-07 20:36:09 -08:00
parent 2b1ca0d0d9
commit ad0a2a59cf
1 changed files with 6 additions and 1 deletions

View File

@ -306,8 +306,13 @@ class Bob(Character):
return generated_work_orders
def get_reencrypted_c_frag(self, networky_stuff, work_order):
def get_reencrypted_c_frags(self, networky_stuff, work_order):
cfrags = networky_stuff.reencrypt(work_order)
if not len(work_order) == len(cfrags):
raise ValueError("Ursula gave back the wrong number of cfrags. She's up to something.")
for counter, pfrag in enumerate(work_order.pfrags):
# TODO: Ursula is actually supposed to sign this. See #141.
self._saved_work_orders[work_order.ursula_id].append(work_order)
return cfrags
def get_ursula(self, ursula_id):