pEpPythonAdapter/test/mp_sync_test.py

248 lines
10 KiB
Python

"""
= Tests for keysync =
See multipEp.py for command line switches.
* Example to launch with latest built pEp Python Adapter (no install):
PYTHONPATH=`pwd`/build/lib.macosx-10.12-x86_64-3.4 \
python3.4 test/mp_sync_test.py
* Example to debug at a particular step in "group_of_3_members"
python3.4 test/mp_sync_test.py only_group_of_3_members debug_GroupA3_3
* Example test with both GPG and NetPGP :
In that example only GroupA1 is using NetPGP, others use GPG
- In pEpEngine :
hg clean --all
make install DEBUG=YES OPENPGP=GPG PREFIX=$HOME/pep_gpg
hg clean --all
make install DEBUG=YES OPENPGP=NETPGP PREFIX=$HOME/pep_netpgp
- In pEpPythonAdapter :
DYLD_LIBRARY_PATH=$HOME/pep_gpg/lib \
PYTHONPATH=`pwd`/build/lib.macosx-10.12-x86_64-3.4 \
python3.4 test/mp_sync_test.py libs_GroupA1=$HOME/pep_netpgp/lib
"""
from multipEp import *
#("instance name", [instance_action_func, [args], {kwargs}], result_func),
#(manager_action_func, [args], {kwargs}, result_func),
def message_to_self():
yield ("GroupA1", [create_account, ["first@group.a", "GroupA First"]])
self_msg = yield ("GroupA1", [encrypted_message, ["first@group.a",
"first@group.a",
"GroupA First to GroupA First -- encrypted",
"GroupA First to GroupA First -- long encrypted"]])
yield ("GroupA1", [decrypt_message, [self_msg]], expect(pEp.PEP_rating.PEP_rating_trusted_and_anonymized))
def pre_existing_peers_with_encrypted_mail():
for action in [
("GroupA1", [create_account, ["first@group.a", "GroupA First"]]),
("SoloA", [create_account, ["first@solo.a", "First SoloA"]]),
(flush_all_mails,),
# key exchange
("SoloA", [send_message, ["first@solo.a",
"first@group.a",
"SoloA First to GroupA First",
"SoloA First to GroupA First -- long"]]),
("GroupA1", [send_message, ["first@group.a",
"first@solo.a",
"GroupA First to SoloA First",
"GroupA First to SoloA First -- long"]])
] : yield action
enc_msg = yield ("SoloA", [encrypted_message, ["first@solo.a",
"first@group.a",
"SoloA First to GroupA First -- encrypted",
"SoloA First to GroupA First -- long encrypted"]])
for action in [
(flush_all_mails,),
("GroupA1", [decrypt_message, [enc_msg]], expect(pEp.PEP_rating.PEP_rating_reliable)),
] : yield action
return enc_msg
def group_on_keygen():
enc_msg = yield from pre_existing_peers_with_encrypted_mail()
for action in [
("GroupA2", [create_account, ["first@group.a", "GroupA First"]]),
(cycle_until_no_change, ["GroupA1", "GroupA2"], expect(3)),
("GroupA2", [decrypt_message, [enc_msg]], expect(pEp.PEP_rating.PEP_rating_reliable))
] : yield action
return enc_msg
def message_to_self_after_sync():
yield from group_on_keygen()
yield (flush_all_mails,)
self_msg = yield ("GroupA2", [encrypted_message, ["first@group.a",
"first@group.a",
"GroupA First to GroupA First -- encrypted",
"GroupA First to GroupA First -- long encrypted"]])
for action in [
("GroupA1", [decrypt_message, [self_msg]], expect(pEp.PEP_rating.PEP_rating_trusted_and_anonymized)),
("GroupA2", [decrypt_message, [self_msg]], expect(pEp.PEP_rating.PEP_rating_trusted_and_anonymized))
] : yield action
self_msg = yield ("GroupA1", [encrypted_message, ["first@group.a",
"first@group.a",
"GroupA First to GroupA First -- encrypted",
"GroupA First to GroupA First -- long encrypted"]])
for action in [
("GroupA1", [decrypt_message, [self_msg]], expect(pEp.PEP_rating.PEP_rating_trusted_and_anonymized)),
("GroupA2", [decrypt_message, [self_msg]], expect(pEp.PEP_rating.PEP_rating_trusted_and_anonymized))
] : yield action
def group_on_cannotdecrypt():
enc_msg = yield from pre_existing_peers_with_encrypted_mail()
for action in [
("GroupA2", [create_account, ["first@group.a", "GroupA First"]]),
(flush_all_mails,),
("GroupA2", [decrypt_message, [enc_msg]], expect(pEp.PEP_rating.PEP_rating_have_no_key)),
(cycle_until_no_change, ["GroupA1", "GroupA2"], expect(3)),
("GroupA2", [decrypt_message, [enc_msg]], expect(pEp.PEP_rating.PEP_rating_reliable)),
] : yield action
def group_gets_losing_key():
yield from group_on_keygen()
# device losing key election is the youngest one, last created, GroupA2
self_msg = yield ("GroupA2", [encrypted_message, ["first@group.a",
"first@group.a",
"GroupA First to GroupA First -- encrypted",
"GroupA First to GroupA First -- long encrypted"]])
for action in [
(flush_all_mails,),
("GroupA1", [decrypt_message, [self_msg]], expect(pEp.PEP_rating.PEP_rating_trusted_and_anonymized)),
] : yield action
def group_of_3_members():
enc_msg = yield from group_on_keygen()
for action in [
("GroupA3", [create_account, ["first@group.a", "GroupA First"]]),
(cycle_until_no_change, ["GroupA1", "GroupA2", "GroupA3"], expect(3)),
("GroupA3", [decrypt_message, [enc_msg]], expect(pEp.PEP_rating.PEP_rating_reliable))
] : yield action
return enc_msg
def new_address_peer_and_mail(identity_flag = None):
for action in [
("SoloB", [create_account, ["first@solo.b", "First SoloB"]]),
("GroupA3", [create_account, ["second@group.a", "GroupA Second", identity_flag]]),
# key exchange
("SoloB", [send_message, ["first@solo.b",
"second@group.a",
"SoloB First to GroupA second",
"SoloB First to GroupA second -- long"]]),
("GroupA3", [send_message, ["second@group.a",
"first@solo.b",
"GroupA second to SoloB First",
"GroupA second to SoloB First"]]),
] : yield action
enc_msg = yield ("SoloB", [encrypted_message, ["first@solo.b",
"second@group.a",
"SoloB First to GroupA Second -- encrypted",
"SoloB First to GroupA Second -- long encrypted"]])
for action in [
("GroupA3", [decrypt_message, [enc_msg]], expect(pEp.PEP_rating.PEP_rating_reliable)),
] : yield action
return enc_msg
def keygen_in_a_group_of_3_members(pre_actions=[]):
yield from group_of_3_members()
enc_msg = yield from new_address_peer_and_mail()
for action in pre_actions + [
(cycle_until_no_change, ["GroupA1", "GroupA2", "GroupA3"], expect(1)),
(flush_all_mails,),
("GroupA1", [create_account, ["second@group.a", "GroupA Second"]]),
(flush_all_mails, expect(0)),
("GroupA2", [create_account, ["second@group.a", "GroupA Second"]]),
(flush_all_mails, expect(0)),
("GroupA2", [decrypt_message, [enc_msg]], expect(pEp.PEP_rating.PEP_rating_reliable)),
("GroupA1", [decrypt_message, [enc_msg]], expect(pEp.PEP_rating.PEP_rating_reliable)),
] : yield action
def group_survives_restart():
yield from keygen_in_a_group_of_3_members([
(restart_instance, ["GroupA1"]),
(restart_instance, ["GroupA2"]),
(restart_instance, ["GroupA3"])
])
def nokey_in_a_group_of_3_members():
yield from group_of_3_members()
enc_msg = yield from new_address_peer_and_mail()
for action in [
(flush_all_mails,),
("GroupA1", [create_account, ["second@group.a", "GroupA Second"]]),
(flush_all_mails,),
("GroupA1", [decrypt_message, [enc_msg]], expect(pEp.PEP_rating.PEP_rating_have_no_key)),
(cycle_until_no_change, ["GroupA1", "GroupA2", "GroupA3"], expect(3)),
("GroupA1", [decrypt_message, [enc_msg]], expect(pEp.PEP_rating.PEP_rating_reliable)),
("GroupA2", [create_account, ["second@group.a", "GroupA Second"]]),
("GroupA2", [decrypt_message, [enc_msg]], expect(pEp.PEP_rating.PEP_rating_reliable)),
] : yield action
def not_for_sync_in_a_group_of_3_members(pre_actions=[]):
yield from group_of_3_members()
enc_msg = yield from new_address_peer_and_mail(pEp.identity_flags.PEP_idf_not_for_sync)
for action in pre_actions + [
(cycle_until_no_change, ["GroupA1", "GroupA2", "GroupA3"], expect(1)),
(flush_all_mails,),
("GroupA1", [create_account, ["second@group.a", "GroupA Second", pEp.identity_flags.PEP_idf_not_for_sync]]),
(flush_all_mails,),
("GroupA2", [create_account, ["second@group.a", "GroupA Second", pEp.identity_flags.PEP_idf_not_for_sync]]),
(flush_all_mails,),
("GroupA2", [decrypt_message, [enc_msg]], expect(pEp.PEP_rating.PEP_rating_have_no_key)),
("GroupA1", [decrypt_message, [enc_msg]], expect(pEp.PEP_rating.PEP_rating_have_no_key)),
] : yield action
def timeout_while_group_on_keygen():
for action in [
(disable_auto_handshake,),
] : yield action
enc_msg = yield from pre_existing_peers_with_encrypted_mail()
for action in [
("GroupA2", [create_account, ["first@group.a", "GroupA First"]]),
(cycle_until_no_change, ["GroupA1", "GroupA2"], expect(3)),
("GroupA1", [simulate_timeout, []]),
("GroupA2", [simulate_timeout, []]),
(flush_all_mails,),
(enable_auto_handshake,),
("GroupA2", [decrypt_message, [enc_msg]], expect(pEp.PEP_rating.PEP_rating_have_no_key)),
(cycle_until_no_change, ["GroupA1", "GroupA2"], expect(3)),
("GroupA2", [decrypt_message, [enc_msg]], expect(pEp.PEP_rating.PEP_rating_reliable))
] : yield action
return enc_msg
if __name__ == "__main__":
run_scenario(message_to_self)
run_scenario(group_on_keygen)
run_scenario(group_on_cannotdecrypt)
run_scenario(group_gets_losing_key)
# run_scenario(group_of_3_members)
# run_scenario(keygen_in_a_group_of_3_members)
# run_scenario(group_survives_restart)
# run_scenario(nokey_in_a_group_of_3_members)
# run_scenario(not_for_sync_in_a_group_of_3_members)
run_scenario(message_to_self_after_sync)
run_scenario(timeout_while_group_on_keygen)