Add MasWriter test for e-mail and unsupported threepids

Author: reivilibre

crates/syn2mas/src/mas_writer/mod.rs

@@ -854,7 +854,9 @@ mod test {
     use uuid::Uuid;
 
     use crate::{
-        mas_writer::{MasNewUser, MasNewUserPassword},
+        mas_writer::{
+            MasNewEmailThreepid, MasNewUnsupportedThreepid, MasNewUser, MasNewUserPassword,
+        },
         LockedMasDatabase, MasWriter,
     };
 
@@ -1018,4 +1020,68 @@ mod test {
 
         assert_db_snapshot!(&mut conn);
     }
+
+    /// Tests writing a single user, with an e-mail address associated.
+    #[sqlx::test(migrator = "mas_storage_pg::MIGRATOR")]
+    async fn test_write_user_with_email(pool: PgPool) {
+        let mut conn = pool.acquire().await.unwrap();
+        let mut writer = make_mas_writer(&pool, &mut conn).await;
+
+        writer
+            .write_users(vec![MasNewUser {
+                user_id: Uuid::from_u128(1u128),
+                username: "alice".to_owned(),
+                created_at: DateTime::default(),
+                locked_at: None,
+                can_request_admin: false,
+            }])
+            .await
+            .expect("failed to write user");
+
+        writer
+            .write_email_threepids(vec![MasNewEmailThreepid {
+                user_email_id: Uuid::from_u128(2u128),
+                user_id: Uuid::from_u128(1u128),
+                email: "[email protected]".to_owned(),
+                created_at: DateTime::default(),
+            }])
+            .await
+            .expect("failed to write e-mail");
+
+        writer.finish().await.expect("failed to finish MasWriter");
+
+        assert_db_snapshot!(&mut conn);
+    }
+
+    /// Tests writing a single user, with a unsupported third-party ID associated.
+    #[sqlx::test(migrator = "mas_storage_pg::MIGRATOR")]
+    async fn test_write_user_with_unsupported_threepid(pool: PgPool) {
+        let mut conn = pool.acquire().await.unwrap();
+        let mut writer = make_mas_writer(&pool, &mut conn).await;
+
+        writer
+            .write_users(vec![MasNewUser {
+                user_id: Uuid::from_u128(1u128),
+                username: "alice".to_owned(),
+                created_at: DateTime::default(),
+                locked_at: None,
+                can_request_admin: false,
+            }])
+            .await
+            .expect("failed to write user");
+
+        writer
+            .write_unsupported_threepids(vec![MasNewUnsupportedThreepid {
+                user_id: Uuid::from_u128(1u128),
+                medium: "msisdn".to_owned(),
+                address: "441189998819991197253".to_owned(),
+                created_at: DateTime::default(),
+            }])
+            .await
+            .expect("failed to write phone number (unsupported threepid)");
+
+        writer.finish().await.expect("failed to finish MasWriter");
+
+        assert_db_snapshot!(&mut conn);
+    }
 }

crates/syn2mas/src/mas_writer/snapshots/syn2mas__mas_writer__test__write_user_with_email.snap

@@ -0,0 +1,17 @@
+---
+source: crates/syn2mas/src/mas_writer/mod.rs
+expression: db_snapshot
+---
+user_emails:
+  - confirmed_at: "1970-01-01 00:00:00+00"
+    created_at: "1970-01-01 00:00:00+00"
+    email: alice@example.org
+    user_email_id: 00000000-0000-0000-0000-000000000002
+    user_id: 00000000-0000-0000-0000-000000000001
+users:
+  - can_request_admin: "false"
+    created_at: "1970-01-01 00:00:00+00"
+    locked_at: ~
+    primary_user_email_id: ~
+    user_id: 00000000-0000-0000-0000-000000000001
+    username: alice

crates/syn2mas/src/mas_writer/snapshots/syn2mas__mas_writer__test__write_user_with_unsupported_threepid.snap

@@ -0,0 +1,16 @@
+---
+source: crates/syn2mas/src/mas_writer/mod.rs
+expression: db_snapshot
+---
+user_unsupported_third_party_ids:
+  - address: "441189998819991197253"
+    created_at: "1970-01-01 00:00:00+00"
+    medium: msisdn
+    user_id: 00000000-0000-0000-0000-000000000001
+users:
+  - can_request_admin: "false"
+    created_at: "1970-01-01 00:00:00+00"
+    locked_at: ~
+    primary_user_email_id: ~
+    user_id: 00000000-0000-0000-0000-000000000001
+    username: alice