From c85ac2d663f0b09c2b06d3042456a5bc3dd46064 Mon Sep 17 00:00:00 2001
From: bitfl0wer <florian@pro-weber.com>
Date: Sat, 27 May 2023 22:12:07 +0200
Subject: [PATCH] Move common code into common::<>

---
 tests/common/mod.rs | 101 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 101 insertions(+)
 create mode 100644 tests/common/mod.rs

diff --git a/tests/common/mod.rs b/tests/common/mod.rs
new file mode 100644
index 0000000..fd58140
--- /dev/null
+++ b/tests/common/mod.rs
@@ -0,0 +1,101 @@
+
+use chorus::{
+    instance::{Instance, UserMeta},
+    types::{Channel, ChannelCreateSchema, Guild, GuildCreateSchema, RegisterSchema},
+    URLBundle,
+};
+
+#[derive(Debug)]
+pub struct TestBundle {
+    pub urls: URLBundle,
+    pub user: UserMeta,
+    pub instance: Instance,
+    pub guild_id: String,
+    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 = RegisterSchema::new(
+        "integrationtestuser".to_string(),
+        None,
+        true,
+        None,
+        None,
+        None,
+        Some("2000-01-01".to_string()),
+        None,
+        None,
+        None,
+    )
+    .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_id = Guild::create(&mut user, urls.get_api(), guild_create_schema)
+        .await
+        .unwrap();
+    let channel = Channel::create(
+        &user.token,
+        urls.get_api(),
+        guild_id.as_str(),
+        channel_create_schema,
+        &mut user.limits,
+        &mut instance.limits,
+    )
+    .await
+    .unwrap();
+
+    TestBundle {
+        urls,
+        user,
+        instance,
+        guild_id,
+        channel,
+    }
+}
+
+// Teardown method to clean up after a test.
+pub async fn teardown(mut bundle: TestBundle) {
+    Guild::delete(
+        &mut bundle.user,
+        bundle.instance.urls.get_api(),
+        bundle.guild_id,
+    )
+    .await;
+    bundle.user.delete().await;
+}