Skip to content

Commit 92d064e

Browse files
committed
refactor(tests): Pass decryption_settings in to send_and_receive_encrypted_to_device_test_helper
To allow passing in different values in future tests.
1 parent e638677 commit 92d064e

File tree

2 files changed

+26
-3
lines changed

2 files changed

+26
-3
lines changed

crates/matrix-sdk-crypto/src/machine/test_helpers.rs

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -188,6 +188,7 @@ pub async fn send_and_receive_encrypted_to_device_test_helper(
188188
recipient: &OlmMachine,
189189
event_type: &str,
190190
content: &Value,
191+
decryption_settings: &DecryptionSettings,
191192
) -> ProcessedToDeviceEvent {
192193
let device =
193194
sender.get_device(recipient.user_id(), recipient.device_id(), None).await.unwrap().unwrap();
@@ -218,11 +219,9 @@ pub async fn send_and_receive_encrypted_to_device_test_helper(
218219
next_batch_token: None,
219220
};
220221

221-
let decryption_settings =
222-
DecryptionSettings { sender_device_trust_requirement: TrustRequirement::Untrusted };
223-
224222
let (decrypted, _) =
225223
recipient.receive_sync_changes(sync_changes, &decryption_settings).await.unwrap();
224+
226225
assert_eq!(1, decrypted.len());
227226
decrypted[0].clone()
228227
}

crates/matrix-sdk-crypto/src/machine/tests/send_encrypted_to_device.rs

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,11 +51,15 @@ async fn test_send_encrypted_to_device() {
5151
"rooms": ["!726s6s6q:example.com"]
5252
});
5353

54+
let decryption_settings =
55+
DecryptionSettings { sender_device_trust_requirement: TrustRequirement::Untrusted };
56+
5457
let processed_event = send_and_receive_encrypted_to_device_test_helper(
5558
&alice,
5659
&bob,
5760
custom_event_type,
5861
&custom_content,
62+
&decryption_settings,
5963
)
6064
.await;
6165

@@ -124,11 +128,15 @@ async fn test_receive_custom_encrypted_to_device_fails_if_device_unknown() {
124128
"rooms": ["!726s6s6q:example.com"]
125129
});
126130

131+
let decryption_settings =
132+
DecryptionSettings { sender_device_trust_requirement: TrustRequirement::Untrusted };
133+
127134
let processed_event = send_and_receive_encrypted_to_device_test_helper(
128135
&alice,
129136
&bob,
130137
custom_event_type,
131138
&custom_content,
139+
&decryption_settings,
132140
)
133141
.await;
134142

@@ -153,11 +161,15 @@ async fn test_send_olm_encryption_info_unverified_identity() {
153161
"rooms": ["!726s6s6q:example.com"]
154162
});
155163

164+
let decryption_settings =
165+
DecryptionSettings { sender_device_trust_requirement: TrustRequirement::Untrusted };
166+
156167
let processed_event = send_and_receive_encrypted_to_device_test_helper(
157168
&alice,
158169
&bob,
159170
custom_event_type,
160171
&custom_content,
172+
&decryption_settings,
161173
)
162174
.await;
163175

@@ -193,11 +205,15 @@ async fn test_send_olm_encryption_info_verified_identity() {
193205
"rooms": ["!726s6s6q:example.com"]
194206
});
195207

208+
let decryption_settings =
209+
DecryptionSettings { sender_device_trust_requirement: TrustRequirement::Untrusted };
210+
196211
let processed_event = send_and_receive_encrypted_to_device_test_helper(
197212
&alice,
198213
&bob,
199214
custom_event_type,
200215
&custom_content,
216+
&decryption_settings,
201217
)
202218
.await;
203219

@@ -229,11 +245,15 @@ async fn test_send_olm_encryption_info_verified_locally() {
229245
.await
230246
.unwrap();
231247

248+
let decryption_settings =
249+
DecryptionSettings { sender_device_trust_requirement: TrustRequirement::Untrusted };
250+
232251
let processed_event = send_and_receive_encrypted_to_device_test_helper(
233252
&alice,
234253
&bob,
235254
custom_event_type,
236255
&custom_content,
256+
&decryption_settings,
237257
)
238258
.await;
239259

@@ -271,11 +291,15 @@ async fn test_send_olm_encryption_info_verification_violation() {
271291
"rooms": ["!726s6s6q:example.com"]
272292
});
273293

294+
let decryption_settings =
295+
DecryptionSettings { sender_device_trust_requirement: TrustRequirement::Untrusted };
296+
274297
let processed_event = send_and_receive_encrypted_to_device_test_helper(
275298
&alice,
276299
&bob,
277300
custom_event_type,
278301
&custom_content,
302+
&decryption_settings,
279303
)
280304
.await;
281305

0 commit comments

Comments
 (0)