chorus/tests/common/mod.rs

use chorus::{
    instance::{Instance, UserMeta},
    types::{
        Channel, ChannelCreateSchema, Guild, GuildCreateSchema, RegisterSchema,
        RegisterSchemaOptions, RoleCreateModifySchema, RoleObject,
    },
    URLBundle,
};

#[derive(Debug)]
pub struct TestBundle {
    pub urls: URLBundle,
    pub user: UserMeta,
    pub instance: Instance,
    pub guild: Guild,
    pub role: RoleObject,
    pub channel: Channel,
}

// Set up a test by creating an Instance and a User. Reduces Test boilerplate.
pub 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()).await.unwrap();
    // Requires the existance of the below user.
    let reg = RegisterSchemaOptions {
        date_of_birth: Some("2000-01-01".to_string()),
        ..RegisterSchema::builder("integrationtestuser", true)
    }
    .build()
    .unwrap();
    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(&reg).await.unwrap();
    let guild = Guild::create(&mut user, guild_create_schema).await.unwrap();
    let channel = Channel::create(&mut user, &guild.id.to_string(), 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 guild_id = guild.id.clone().to_string();
    let role = chorus::types::RoleObject::create(&mut user, &guild_id, role_create_schema)
        .await
        .unwrap();

    TestBundle {
        urls,
        user,
        instance,
        guild,
        role,
        channel,
    }
}

// Teardown method to clean up after a test.
pub async fn teardown(mut bundle: TestBundle) {
    Guild::delete(&mut bundle.user, &bundle.guild.id.to_string()).await;
    bundle.user.delete().await;
}