diff --git a/examples/login.rs b/examples/login.rs index e89d8d2..2449d6c 100644 --- a/examples/login.rs +++ b/examples/login.rs @@ -24,5 +24,5 @@ async fn main() { .await .expect("An error occurred during the login process"); dbg!(user.belongs_to); - dbg!(&user.object.read().unwrap().username); + dbg!(&user.object.unwrap().as_ref().read().unwrap().username); } diff --git a/tests/auth.rs b/tests/auth.rs index 705328a..c7da756 100644 --- a/tests/auth.rs +++ b/tests/auth.rs @@ -85,8 +85,10 @@ async fn test_login_with_token() { .await .unwrap(); assert_eq!( - bundle.user.object.read().unwrap().id, - other_user.object.read().unwrap().id + bundle.user.object.as_ref().unwrap() + .read().unwrap() + .id, + other_user.object.unwrap().read().unwrap().id ); assert_eq!(bundle.user.token, other_user.token); diff --git a/tests/channels.rs b/tests/channels.rs index e00744a..5127fd6 100644 --- a/tests/channels.rs +++ b/tests/channels.rs @@ -2,7 +2,11 @@ // License, v. 2.0. If a copy of the MPL was not distributed with this // file, You can obtain one at http://mozilla.org/MPL/2.0/. -use chorus::types::{self, Channel, GetChannelMessagesSchema, MessageSendSchema, PermissionFlags, PermissionOverwrite, PermissionOverwriteType, PrivateChannelCreateSchema, RelationshipType, Snowflake}; +use chorus::types::{ + self, Channel, GetChannelMessagesSchema, MessageSendSchema, PermissionFlags, + PermissionOverwrite, PermissionOverwriteType, PrivateChannelCreateSchema, RelationshipType, + Snowflake, +}; mod common; @@ -67,7 +71,7 @@ async fn modify_channel() { assert_eq!(modified_channel.name, Some(CHANNEL_NAME.to_string())); let permission_override = PermissionFlags::MANAGE_CHANNELS | PermissionFlags::MANAGE_MESSAGES; - let user_id: types::Snowflake = bundle.user.object.read().unwrap().id; + let user_id: types::Snowflake = bundle.user.object.as_ref().unwrap().read().unwrap().id; let permission_override = PermissionOverwrite { id: user_id, overwrite_type: PermissionOverwriteType::Member, @@ -155,7 +159,13 @@ async fn create_dm() { let other_user = bundle.create_user("integrationtestuser2").await; let user = &mut bundle.user; let private_channel_create_schema = PrivateChannelCreateSchema { - recipients: Some(Vec::from([other_user.object.read().unwrap().id])), + recipients: Some(Vec::from([other_user + .object + .as_ref() + .unwrap() + .read() + .unwrap() + .id])), access_tokens: None, nicks: None, }; @@ -175,7 +185,7 @@ async fn create_dm() { .unwrap() .id .clone(), - other_user.object.read().unwrap().id + other_user.object.unwrap().read().unwrap().id ); assert_eq!( dm_channel @@ -188,7 +198,7 @@ async fn create_dm() { .unwrap() .id .clone(), - user.object.read().unwrap().id.clone() + user.object.as_ref().unwrap().read().unwrap().id.clone() ); common::teardown(bundle).await; } @@ -200,9 +210,9 @@ async fn remove_add_person_from_to_dm() { let mut bundle = common::setup().await; let mut other_user = bundle.create_user("integrationtestuser2").await; let mut third_user = bundle.create_user("integrationtestuser3").await; - let third_user_id = third_user.object.read().unwrap().id; - let other_user_id = other_user.object.read().unwrap().id; - let user_id = bundle.user.object.read().unwrap().id; + let third_user_id = third_user.object.as_ref().unwrap().read().unwrap().id; + let other_user_id = other_user.object.as_ref().unwrap().read().unwrap().id; + let user_id = bundle.user.object.as_ref().unwrap().read().unwrap().id; let user = &mut bundle.user; let private_channel_create_schema = PrivateChannelCreateSchema { recipients: Some(Vec::from([other_user_id, third_user_id])), diff --git a/tests/common/mod.rs b/tests/common/mod.rs index 315db38..d0af796 100644 --- a/tests/common/mod.rs +++ b/tests/common/mod.rs @@ -47,6 +47,7 @@ impl TestBundle { ChorusUser { belongs_to: self.user.belongs_to.clone(), token: self.user.token.clone(), + mfa_token: None, limits: self.user.limits.clone(), settings: self.user.settings.clone(), object: self.user.object.clone(), diff --git a/tests/guilds.rs b/tests/guilds.rs index e3a73f5..85dd3ce 100644 --- a/tests/guilds.rs +++ b/tests/guilds.rs @@ -60,7 +60,8 @@ async fn guild_create_ban() { .await .unwrap(); other_user.accept_invite(&invite.code, None).await.unwrap(); - let other_user_id = other_user.object.read().unwrap().id; + let other_user_id = other_user.object.as_ref().unwrap() + .read().unwrap().id; Guild::create_ban( guild.id, other_user_id, @@ -112,7 +113,9 @@ async fn guild_remove_member() { .await .unwrap(); other_user.accept_invite(&invite.code, None).await.unwrap(); - let other_user_id = other_user.object.read().unwrap().id; + let other_user_id = other_user.object + .as_ref().unwrap() + .read().unwrap().id; Guild::remove_member(guild.id, other_user_id, None, &mut bundle.user) .await .unwrap(); diff --git a/tests/members.rs b/tests/members.rs index a66d25a..0fc45d4 100644 --- a/tests/members.rs +++ b/tests/members.rs @@ -16,7 +16,8 @@ async fn add_remove_role() -> ChorusResult<()> { let mut bundle = common::setup().await; let guild = bundle.guild.read().unwrap().id; let role = bundle.role.read().unwrap().id; - let member_id = bundle.user.object.read().unwrap().id; + let member_id = bundle.user.object.as_ref().unwrap() + .read().unwrap().id; GuildMember::add_role(&mut bundle.user, guild, member_id, role).await?; let member = GuildMember::get(&mut bundle.user, guild, member_id) .await diff --git a/tests/messages.rs b/tests/messages.rs index 3ca6e16..b1de770 100644 --- a/tests/messages.rs +++ b/tests/messages.rs @@ -106,7 +106,7 @@ async fn search_messages() { let _arg = Some(&vec_attach); let message = bundle.user.send_message(message, channel.id).await.unwrap(); let query = MessageSearchQuery { - author_id: Some(Vec::from([bundle.user.object.read().unwrap().id])), + author_id: Some(Vec::from([bundle.user.object.as_ref().unwrap().read().unwrap().id])), ..Default::default() }; let guild_id = bundle.guild.read().unwrap().id; diff --git a/tests/relationships.rs b/tests/relationships.rs index 2eea5b3..4cd1492 100644 --- a/tests/relationships.rs +++ b/tests/relationships.rs @@ -16,9 +16,10 @@ async fn test_get_mutual_relationships() { let mut bundle = common::setup().await; let mut other_user = bundle.create_user("integrationtestuser2").await; let user = &mut bundle.user; - let username = user.object.read().unwrap().username.clone(); - let discriminator = user.object.read().unwrap().discriminator.clone(); - let other_user_id: types::Snowflake = other_user.object.read().unwrap().id; + + let username = user.object.as_ref().unwrap().read().unwrap().username.clone(); + let discriminator = user.object.as_ref().unwrap().read().unwrap().discriminator.clone(); + let other_user_id: types::Snowflake = other_user.object.as_ref().unwrap().read().unwrap().id; let friend_request_schema = types::FriendRequestSendSchema { username, discriminator: Some(discriminator), @@ -38,8 +39,8 @@ async fn test_get_relationships() { let mut bundle = common::setup().await; let mut other_user = bundle.create_user("integrationtestuser2").await; let user = &mut bundle.user; - let username = user.object.read().unwrap().username.clone(); - let discriminator = user.object.read().unwrap().discriminator.clone(); + let username = user.object.as_ref().unwrap().read().unwrap().username.clone(); + let discriminator = user.object.as_ref().unwrap().read().unwrap().discriminator.clone(); let friend_request_schema = types::FriendRequestSendSchema { username, discriminator: Some(discriminator), @@ -51,7 +52,7 @@ async fn test_get_relationships() { let relationships = user.get_relationships().await.unwrap(); assert_eq!( relationships.get(0).unwrap().id, - other_user.object.read().unwrap().id + other_user.object.unwrap().read().unwrap().id ); common::teardown(bundle).await } @@ -62,8 +63,8 @@ async fn test_modify_relationship_friends() { let mut bundle = common::setup().await; let mut other_user = bundle.create_user("integrationtestuser2").await; let user = &mut bundle.user; - let user_id: types::Snowflake = user.object.read().unwrap().id; - let other_user_id: types::Snowflake = other_user.object.read().unwrap().id; + let user_id: types::Snowflake = user.object.as_ref().unwrap().read().unwrap().id; + let other_user_id: types::Snowflake = other_user.object.as_ref().unwrap().read().unwrap().id; other_user .modify_user_relationship(user_id, types::RelationshipType::Friends) @@ -72,7 +73,7 @@ async fn test_modify_relationship_friends() { let relationships = user.get_relationships().await.unwrap(); assert_eq!( relationships.get(0).unwrap().id, - other_user.object.read().unwrap().id + other_user.object.as_ref().unwrap().read().unwrap().id ); assert_eq!( relationships.get(0).unwrap().relationship_type, @@ -81,7 +82,7 @@ async fn test_modify_relationship_friends() { let relationships = other_user.get_relationships().await.unwrap(); assert_eq!( relationships.get(0).unwrap().id, - user.object.read().unwrap().id + user.object.as_ref().unwrap().read().unwrap().id ); assert_eq!( relationships.get(0).unwrap().relationship_type, @@ -114,7 +115,7 @@ async fn test_modify_relationship_block() { let mut bundle = common::setup().await; let mut other_user = bundle.create_user("integrationtestuser2").await; let user = &mut bundle.user; - let user_id: types::Snowflake = user.object.read().unwrap().id; + let user_id: types::Snowflake = user.object.as_ref().unwrap().read().unwrap().id; other_user .modify_user_relationship(user_id, types::RelationshipType::Blocked) @@ -125,7 +126,7 @@ async fn test_modify_relationship_block() { let relationships = other_user.get_relationships().await.unwrap(); assert_eq!( relationships.get(0).unwrap().id, - user.object.read().unwrap().id + user.object.as_ref().unwrap().read().unwrap().id ); assert_eq!( relationships.get(0).unwrap().relationship_type,