1010import deltachat as dc
1111from deltachat import account_hookimpl , Message
1212from deltachat .tracker import ImexTracker
13+ from deltachat .testplugin import E2EE_INFO_MSGS
1314
1415
1516def test_basic_imap_api (acfactory , tmp_path ):
@@ -408,6 +409,10 @@ def test_forward_messages(acfactory, lp):
408409 msg_out = chat .send_text ("message2" )
409410
410411 lp .sec ("ac2: wait for receive" )
412+ ev = ac2 ._evtracker .get_matching ("DC_EVENT_INCOMING_MSG|DC_EVENT_MSGS_CHANGED" )
413+ msg_in = ac2 .get_message_by_id (ev .data2 )
414+ assert msg_in .text == "Messages are end-to-end encrypted."
415+
411416 ev = ac2 ._evtracker .get_matching ("DC_EVENT_INCOMING_MSG|DC_EVENT_MSGS_CHANGED" )
412417 assert ev .data2 == msg_out .id
413418 msg_in = ac2 .get_message_by_id (msg_out .id )
@@ -622,6 +627,11 @@ def test_moved_markseen(acfactory):
622627
623628 with ac2 .direct_imap .idle () as idle2 :
624629 ac2 .start_io ()
630+
631+ ev = ac2 ._evtracker .get_matching ("DC_EVENT_INCOMING_MSG|DC_EVENT_MSGS_CHANGED" )
632+ msg = ac2 .get_message_by_id (ev .data2 )
633+ assert msg .text == "Messages are end-to-end encrypted."
634+
625635 ev = ac2 ._evtracker .get_matching ("DC_EVENT_INCOMING_MSG|DC_EVENT_MSGS_CHANGED" )
626636 msg = ac2 .get_message_by_id (ev .data2 )
627637
@@ -738,15 +748,15 @@ def test_mdn_asymmetric(acfactory, lp):
738748 lp .sec ("sending text message from ac1 to ac2" )
739749 msg_out = chat .send_text ("message1" )
740750
741- assert len (chat .get_messages ()) == 1
751+ assert len (chat .get_messages ()) == 1 + E2EE_INFO_MSGS
742752
743753 lp .sec ("disable ac1 MDNs" )
744754 ac1 .set_config ("mdns_enabled" , "0" )
745755
746756 lp .sec ("wait for ac2 to receive message" )
747757 msg = ac2 ._evtracker .wait_next_incoming_message ()
748758
749- assert len (msg .chat .get_messages ()) == 1
759+ assert len (msg .chat .get_messages ()) == 1 + E2EE_INFO_MSGS
750760
751761 lp .sec ("ac2: mark incoming message as seen" )
752762 ac2 .mark_seen_messages ([msg ])
@@ -755,7 +765,7 @@ def test_mdn_asymmetric(acfactory, lp):
755765 # MDN should be moved even though MDNs are already disabled
756766 ac1 ._evtracker .get_matching ("DC_EVENT_IMAP_MESSAGE_MOVED" )
757767
758- assert len (chat .get_messages ()) == 1
768+ assert len (chat .get_messages ()) == 1 + E2EE_INFO_MSGS
759769
760770 # Wait for the message to be marked as seen on IMAP.
761771 ac1 ._evtracker .get_info_contains ("Marked messages [0-9]+ in folder DeltaChat as seen." )
@@ -1123,6 +1133,11 @@ def ac_outgoing_message(self, message):
11231133 assert m == msg_out
11241134
11251135 lp .sec ("wait for ac2 to receive message" )
1136+
1137+ ev = ac2 ._evtracker .get_matching ("DC_EVENT_MSGS_CHANGED|DC_EVENT_INCOMING_MSG" )
1138+ msg_in = ac2 .get_message_by_id (ev .data2 )
1139+ assert msg_in .text == "Messages are end-to-end encrypted."
1140+
11261141 ev = ac2 ._evtracker .get_matching ("DC_EVENT_MSGS_CHANGED|DC_EVENT_INCOMING_MSG" )
11271142 assert ev .data2 == msg_out .id
11281143 msg_in = ac2 .get_message_by_id (msg_out .id )
@@ -1158,10 +1173,10 @@ def assert_account_is_proper(ac):
11581173 assert contact2 .addr == some1_addr
11591174 chat2 = contact2 .create_chat ()
11601175 messages = chat2 .get_messages ()
1161- assert len (messages ) == 3
1162- assert messages [0 ].text == "msg1"
1163- assert messages [1 ].filemime == "image/png"
1164- assert os .stat (messages [1 ].filename ).st_size == os .stat (original_image_path ).st_size
1176+ assert len (messages ) == 3 + E2EE_INFO_MSGS
1177+ assert messages [0 + E2EE_INFO_MSGS ].text == "msg1"
1178+ assert messages [1 + E2EE_INFO_MSGS ].filemime == "image/png"
1179+ assert os .stat (messages [1 + E2EE_INFO_MSGS ].filename ).st_size == os .stat (original_image_path ).st_size
11651180 ac .set_config ("displayname" , "new displayname" )
11661181 assert ac .get_config ("displayname" ) == "new displayname"
11671182
@@ -1414,8 +1429,8 @@ def test_connectivity(acfactory, lp):
14141429 ac1 .maybe_network ()
14151430 ac1 ._evtracker .wait_for_connectivity (dc .const .DC_CONNECTIVITY_CONNECTED )
14161431 msgs = ac1 .create_chat (ac2 ).get_messages ()
1417- assert len (msgs ) == 1
1418- assert msgs [0 ].text == "Hi"
1432+ assert len (msgs ) == 1 + E2EE_INFO_MSGS
1433+ assert msgs [0 + E2EE_INFO_MSGS ].text == "Hi"
14191434
14201435 lp .sec ("Test that the connectivity changes to WORKING while new messages are fetched" )
14211436
@@ -1425,8 +1440,8 @@ def test_connectivity(acfactory, lp):
14251440 ac1 ._evtracker .wait_for_connectivity_change (dc .const .DC_CONNECTIVITY_WORKING , dc .const .DC_CONNECTIVITY_CONNECTED )
14261441
14271442 msgs = ac1 .create_chat (ac2 ).get_messages ()
1428- assert len (msgs ) == 2
1429- assert msgs [1 ].text == "Hi 2"
1443+ assert len (msgs ) == 2 + E2EE_INFO_MSGS
1444+ assert msgs [1 + E2EE_INFO_MSGS ].text == "Hi 2"
14301445
14311446
14321447def test_fetch_deleted_msg (acfactory , lp ):
0 commit comments