@@ -289,12 +289,16 @@ mod tests {
289
289
use assert_matches2:: assert_let;
290
290
use matrix_sdk_test:: async_test;
291
291
use ruma:: { device_id, user_id} ;
292
- use serde_json:: { self , Value } ;
292
+ use serde_json;
293
293
use vodozemac:: olm:: { OlmMessage , SessionConfig } ;
294
294
295
295
use crate :: {
296
- identities:: ReadOnlyDevice , olm:: Account ,
297
- types:: events:: room:: encrypted:: ToDeviceEncryptedEventContent ,
296
+ identities:: ReadOnlyDevice ,
297
+ olm:: Account ,
298
+ types:: events:: {
299
+ dummy:: DummyEventContent , olm_v1:: DecryptedOlmV1Event ,
300
+ room:: encrypted:: ToDeviceEncryptedEventContent ,
301
+ } ,
298
302
} ;
299
303
300
304
#[ async_test]
@@ -349,10 +353,8 @@ mod tests {
349
353
. unwrap ( ) ;
350
354
351
355
// Also ensure that the encrypted payload has the device keys.
352
- let plaintext: Value = serde_json:: from_str ( & bob_session_result. plaintext ) . unwrap ( ) ;
353
- assert_eq ! (
354
- plaintext[ "org.matrix.msc4147.device_keys" ] [ "user_id" ] . as_str( ) ,
355
- Some ( "@alice:localhost" )
356
- ) ;
356
+ let plaintext: DecryptedOlmV1Event < DummyEventContent > =
357
+ serde_json:: from_str ( & bob_session_result. plaintext ) . unwrap ( ) ;
358
+ assert_eq ! ( plaintext. device_keys. unwrap( ) , alice. device_keys( ) ) ;
357
359
}
358
360
}
0 commit comments