Skip to content

Commit ddc2f55

Browse files
committed
test: encrypt 15 more Rust tests
- chat::chat_tests::test_forward_group - chat::chat_tests::test_resend_foreign_message_fails - chat::chat_tests::test_resend_info_message_fails - ephemeral::ephemeral_tests::test_ephemeral_timer_non_member - receive_imf::receive_imf_tests::test_delayed_removal_is_ignored - receive_imf::receive_imf_tests::test_dont_readd_with_normal_msg - receive_imf::receive_imf_tests::test_dont_recreate_contacts_on_add_remove - receive_imf::receive_imf_tests::test_member_left_does_not_create_chat - receive_imf::receive_imf_tests::test_outgoing_private_reply_multidevice - receive_imf::receive_imf_tests::test_recreate_member_list_on_missing_add_of_self - receive_imf::receive_imf_tests::test_references - receive_imf::receive_imf_tests::test_send_as_bot - receive_imf::receive_imf_tests::test_unsigned_chat_group_hdr - securejoin::securejoin_tests::test_unknown_sender - webxdc::webxdc_tests::test_webxdc_reject_updates_from_non_groupmembers
1 parent 8f3fc10 commit ddc2f55

File tree

6 files changed

+128
-128
lines changed

6 files changed

+128
-128
lines changed

src/chat/chat_tests.rs

Lines changed: 21 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -2093,19 +2093,20 @@ async fn test_forward_group() -> Result<()> {
20932093
let mut tcm = TestContextManager::new();
20942094
let alice = tcm.alice().await;
20952095
let bob = tcm.bob().await;
2096+
let charlie = tcm.charlie().await;
20962097

20972098
let alice_chat = alice.create_chat(&bob).await;
20982099
let bob_chat = bob.create_chat(&alice).await;
20992100

21002101
// Alice creates a group with Bob.
21012102
let alice_group_chat_id =
21022103
create_group_chat(&alice, ProtectionStatus::Unprotected, "Group").await?;
2103-
let bob_id = Contact::create(&alice, "Bob", "bob@example.net").await?;
2104-
let claire_id = Contact::create(&alice, "Claire", "claire@example.net").await?;
2104+
let bob_id = alice.add_or_lookup_contact_id(&bob).await;
2105+
let charlie_id = alice.add_or_lookup_contact_id(&charlie).await;
21052106
add_contact_to_chat(&alice, alice_group_chat_id, bob_id).await?;
2106-
add_contact_to_chat(&alice, alice_group_chat_id, claire_id).await?;
2107+
add_contact_to_chat(&alice, alice_group_chat_id, charlie_id).await?;
21072108
let sent_group_msg = alice
2108-
.send_text(alice_group_chat_id, "Hi Bob and Claire")
2109+
.send_text(alice_group_chat_id, "Hi Bob and Charlie")
21092110
.await;
21102111
let bob_group_chat_id = bob.recv_msg(&sent_group_msg).await.chat_id;
21112112

@@ -2388,19 +2389,15 @@ async fn test_resend_own_message() -> Result<()> {
23882389

23892390
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
23902391
async fn test_resend_foreign_message_fails() -> Result<()> {
2391-
let alice = TestContext::new_alice().await;
2392-
let alice_grp = create_group_chat(&alice, ProtectionStatus::Unprotected, "grp").await?;
2393-
add_contact_to_chat(
2394-
&alice,
2395-
alice_grp,
2396-
Contact::create(&alice, "", "bob@example.net").await?,
2397-
)
2398-
.await?;
2392+
let mut tcm = TestContextManager::new();
2393+
let alice = &tcm.alice().await;
2394+
let bob = &tcm.bob().await;
2395+
let alice_grp = create_group_chat(alice, ProtectionStatus::Unprotected, "grp").await?;
2396+
add_contact_to_chat(alice, alice_grp, alice.add_or_lookup_contact_id(bob).await).await?;
23992397
let sent1 = alice.send_text(alice_grp, "alice->bob").await;
24002398

2401-
let bob = TestContext::new_bob().await;
24022399
let msg = bob.recv_msg(&sent1).await;
2403-
assert!(resend_msgs(&bob, &[msg.id]).await.is_err());
2400+
assert!(resend_msgs(bob, &[msg.id]).await.is_err());
24042401

24052402
Ok(())
24062403
}
@@ -2444,24 +2441,23 @@ async fn test_resend_opportunistically_encryption() -> Result<()> {
24442441

24452442
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
24462443
async fn test_resend_info_message_fails() -> Result<()> {
2447-
let alice = TestContext::new_alice().await;
2448-
let alice_grp = create_group_chat(&alice, ProtectionStatus::Unprotected, "grp").await?;
2449-
add_contact_to_chat(
2450-
&alice,
2451-
alice_grp,
2452-
Contact::create(&alice, "", "bob@example.net").await?,
2453-
)
2454-
.await?;
2444+
let mut tcm = TestContextManager::new();
2445+
let alice = &tcm.alice().await;
2446+
let bob = &tcm.bob().await;
2447+
let charlie = &tcm.charlie().await;
2448+
2449+
let alice_grp = create_group_chat(alice, ProtectionStatus::Unprotected, "grp").await?;
2450+
add_contact_to_chat(alice, alice_grp, alice.add_or_lookup_contact_id(bob).await).await?;
24552451
alice.send_text(alice_grp, "alice->bob").await;
24562452

24572453
add_contact_to_chat(
2458-
&alice,
2454+
alice,
24592455
alice_grp,
2460-
Contact::create(&alice, "", "claire@example.org").await?,
2456+
alice.add_or_lookup_contact_id(charlie).await,
24612457
)
24622458
.await?;
24632459
let sent2 = alice.pop_sent_msg().await;
2464-
assert!(resend_msgs(&alice, &[sent2.sender_msg_id]).await.is_err());
2460+
assert!(resend_msgs(alice, &[sent2.sender_msg_id]).await.is_err());
24652461

24662462
Ok(())
24672463
}

src/ephemeral/ephemeral_tests.rs

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@ use crate::chat::{
55
};
66
use crate::config::Config;
77
use crate::constants::DC_CHAT_ID_ARCHIVED_LINK;
8-
use crate::contact::Contact;
98
use crate::download::DownloadState;
109
use crate::location;
1110
use crate::message::markseen_msgs;
@@ -791,7 +790,7 @@ async fn test_ephemeral_timer_non_member() -> Result<()> {
791790
let alice = &tcm.alice().await;
792791
let bob = &tcm.bob().await;
793792

794-
let alice_bob_contact_id = Contact::create(alice, "Bob", "bob@example.net").await?;
793+
let alice_bob_contact_id = alice.add_or_lookup_contact_id(bob).await;
795794
let alice_chat_id =
796795
create_group_chat(alice, ProtectionStatus::Unprotected, "Group name").await?;
797796
add_contact_to_chat(alice, alice_chat_id, alice_bob_contact_id).await?;

0 commit comments

Comments
 (0)