From 3a6058e4b532ee9a1cfd113c956ca01ba8e4e514 Mon Sep 17 00:00:00 2001 From: heck Date: Thu, 25 Mar 2021 01:52:15 +0100 Subject: [PATCH] PYADPT-110: Update sync testing (sync_handshake.py) to work with new sync impl --- test/sync_handshake.py | 73 ++++++++++++++++++++---------------------- 1 file changed, 35 insertions(+), 38 deletions(-) diff --git a/test/sync_handshake.py b/test/sync_handshake.py index c027183..b75ef49 100755 --- a/test/sync_handshake.py +++ b/test/sync_handshake.py @@ -112,37 +112,36 @@ def add_debug_info(msg): return msg -class UserInterface(pEp.UserInterface): - def notifyHandshake(self, me, partner, signal): - print(colored(str(signal), "yellow"), end=" ") - output("on " + device_name + "" if not me.fpr else - "for identities " + str(me.fpr) + " " + str(partner.fpr)) - if me.fpr and partner.fpr: - assert me.fpr != partner.fpr - - if signal in ( - pEp.sync_handshake_signal.SYNC_NOTIFY_INIT_ADD_OTHER_DEVICE, - pEp.sync_handshake_signal.SYNC_NOTIFY_INIT_ADD_OUR_DEVICE, - pEp.sync_handshake_signal.SYNC_NOTIFY_INIT_FORM_GROUP - ): - if isinstance(end_on, list): - end_on.extend([ - pEp.sync_handshake_signal.SYNC_NOTIFY_SOLE, - pEp.sync_handshake_signal.SYNC_NOTIFY_IN_GROUP, - ]) - sleep(.5) # user is reading message - try: - if not options.noanswer: - if options.reject: - self.deliverHandshakeResult(SYNC_HANDSHAKE_REJECTED) - else: - self.deliverHandshakeResult(SYNC_HANDSHAKE_ACCEPTED) - - except NameError: - self.deliverHandshakeResult(SYNC_HANDSHAKE_ACCEPTED) - if signal in end_on: - global the_end - the_end = True + +def this_notifyHandshake(me, partner, signal): + print(colored(str(signal), "yellow"), end=" ") + output("on " + device_name + "" if not me.fpr else + "for identities " + str(me.fpr) + " " + str(partner.fpr)) + if me.fpr and partner.fpr: + assert me.fpr != partner.fpr + + if signal in ( + pEp.sync_handshake_signal.SYNC_NOTIFY_INIT_ADD_OTHER_DEVICE, + pEp.sync_handshake_signal.SYNC_NOTIFY_INIT_ADD_OUR_DEVICE, + pEp.sync_handshake_signal.SYNC_NOTIFY_INIT_FORM_GROUP + ): + if isinstance(end_on, list): + end_on.extend([ + pEp.sync_handshake_signal.SYNC_NOTIFY_SOLE, + pEp.sync_handshake_signal.SYNC_NOTIFY_IN_GROUP, + ]) + sleep(.5) # user is reading message + try: + if not options.noanswer: + if options.reject: + pEp.deliver_handshake_result(SYNC_HANDSHAKE_REJECTED) + else: + pEp.deliver_handshake_result(SYNC_HANDSHAKE_ACCEPTED) + except NameError: + pEp.deliver_handshake_result(SYNC_HANDSHAKE_ACCEPTED) + if signal in end_on: + global the_end + the_end = True def shutdown_sync(): @@ -153,6 +152,7 @@ def run(name, color=None, imap=False, own_ident=1, leave=False): global device_name device_name = name + pEp.notify_handshake = this_notifyHandshake if color: global output @@ -170,7 +170,7 @@ def run(name, color=None, imap=False, own_ident=1, leave=False): me = pEp.Identity(imap_settings.IMAP_EMAIL, name + " of " + imap_settings.IMAP_USER, name) pEp.myself(me) - pEp.messageToSend = messageImapToSend + pEp.message_to_send = messageImapToSend else: me = pEp.Identity("alice@peptest.ch", name + " of Alice Neuman", name) pEp.myself(me) @@ -183,22 +183,19 @@ def run(name, color=None, imap=False, own_ident=1, leave=False): me3 = pEp.Identity("alice@pep.foundation", name + " of Alice Neuman", name) pEp.myself(me3) - pEp.messageToSend = messageToSend + pEp.message_to_send = messageToSend if multithreaded: from threading import Thread def sync_thread(): print(colored("********* ", "yellow") + colored("sync_thread entered", color)) - ui = UserInterface() print(colored("********* ", "yellow") + colored("UserInterface object created", color)) - pEp.do_sync_protocol() + pEp.Sync.run() print(colored("********* ", "yellow") + colored("leaving sync_thread", color)) sync = Thread(target=sync_thread) sync.start() else: - pEp.script_is_implementing_sync() - sync = None - ui = UserInterface() + pEp.start_sync(); try: if leave: