131 lines
4.1 KiB
Rust
131 lines
4.1 KiB
Rust
use std::sync::{Arc, RwLock};
|
|
|
|
use chorus::gateway::Gateway;
|
|
use chorus::{
|
|
instance::{Instance, UserMeta},
|
|
types::{
|
|
Channel, ChannelCreateSchema, Guild, GuildCreateSchema, RegisterSchema,
|
|
RoleCreateModifySchema, RoleObject,
|
|
},
|
|
UrlBundle,
|
|
};
|
|
|
|
#[allow(dead_code)]
|
|
#[derive(Debug)]
|
|
pub(crate) struct TestBundle {
|
|
pub urls: UrlBundle,
|
|
pub user: UserMeta,
|
|
pub instance: Instance,
|
|
pub guild: Arc<RwLock<Guild>>,
|
|
pub role: Arc<RwLock<RoleObject>>,
|
|
pub channel: Arc<RwLock<Channel>>,
|
|
}
|
|
|
|
#[allow(unused)]
|
|
impl TestBundle {
|
|
pub(crate) async fn create_user(&mut self, username: &str) -> UserMeta {
|
|
let register_schema = RegisterSchema {
|
|
username: username.to_string(),
|
|
consent: true,
|
|
date_of_birth: Some("2000-01-01".to_string()),
|
|
..Default::default()
|
|
};
|
|
self.instance
|
|
.register_account(®ister_schema)
|
|
.await
|
|
.unwrap()
|
|
}
|
|
pub(crate) async fn clone_user_without_gateway(&self) -> UserMeta {
|
|
UserMeta {
|
|
belongs_to: self.user.belongs_to.clone(),
|
|
token: self.user.token.clone(),
|
|
limits: self.user.limits.clone(),
|
|
settings: self.user.settings.clone(),
|
|
object: self.user.object.clone(),
|
|
gateway: Gateway::new(self.instance.urls.wss.clone()).await.unwrap(),
|
|
}
|
|
}
|
|
}
|
|
|
|
// Set up a test by creating an Instance and a User. Reduces Test boilerplate.
|
|
pub(crate) async fn setup() -> TestBundle {
|
|
let urls = UrlBundle::new(
|
|
"http://localhost:3001/api".to_string(),
|
|
"ws://localhost:3001".to_string(),
|
|
"http://localhost:3001".to_string(),
|
|
);
|
|
let mut instance = Instance::new(urls.clone(), true).await.unwrap();
|
|
// Requires the existance of the below user.
|
|
let reg = RegisterSchema {
|
|
username: "integrationtestuser".into(),
|
|
consent: true,
|
|
date_of_birth: Some("2000-01-01".to_string()),
|
|
..Default::default()
|
|
};
|
|
let guild_create_schema = GuildCreateSchema {
|
|
name: Some("Test-Guild!".to_string()),
|
|
region: None,
|
|
icon: None,
|
|
channels: None,
|
|
guild_template_code: None,
|
|
system_channel_id: None,
|
|
rules_channel_id: None,
|
|
};
|
|
let channel_create_schema = ChannelCreateSchema {
|
|
name: "testchannel".to_string(),
|
|
channel_type: Some(0),
|
|
topic: None,
|
|
icon: None,
|
|
bitrate: None,
|
|
user_limit: None,
|
|
rate_limit_per_user: None,
|
|
position: None,
|
|
permission_overwrites: None,
|
|
parent_id: None,
|
|
id: None,
|
|
nsfw: Some(false),
|
|
rtc_region: None,
|
|
default_auto_archive_duration: None,
|
|
default_reaction_emoji: None,
|
|
flags: Some(0),
|
|
default_thread_rate_limit_per_user: Some(0),
|
|
video_quality_mode: None,
|
|
};
|
|
let mut user = instance.register_account(®).await.unwrap();
|
|
let guild = Guild::create(&mut user, guild_create_schema).await.unwrap();
|
|
let channel = Channel::create(&mut user, guild.id, channel_create_schema)
|
|
.await
|
|
.unwrap();
|
|
|
|
let role_create_schema: chorus::types::RoleCreateModifySchema = RoleCreateModifySchema {
|
|
name: Some("Bundle role".to_string()),
|
|
permissions: Some("8".to_string()), // Administrator permissions
|
|
hoist: Some(true),
|
|
icon: None,
|
|
unicode_emoji: Some("".to_string()),
|
|
mentionable: Some(true),
|
|
position: None,
|
|
color: None,
|
|
};
|
|
let role = chorus::types::RoleObject::create(&mut user, guild.id, role_create_schema)
|
|
.await
|
|
.unwrap();
|
|
|
|
TestBundle {
|
|
urls,
|
|
user,
|
|
instance,
|
|
guild: Arc::new(RwLock::new(guild)),
|
|
role: Arc::new(RwLock::new(role)),
|
|
channel: Arc::new(RwLock::new(channel)),
|
|
}
|
|
}
|
|
|
|
// Teardown method to clean up after a test.
|
|
#[allow(dead_code)]
|
|
pub(crate) async fn teardown(mut bundle: TestBundle) {
|
|
let id = bundle.guild.read().unwrap().id;
|
|
Guild::delete(&mut bundle.user, id).await.unwrap();
|
|
bundle.user.delete().await.unwrap()
|
|
}
|