test: add utilities for common setup

Signed-off-by: phazeschift <51278042+phazeschift@users.noreply.github.com>
This commit is contained in:
phazeschift
2025-03-15 18:47:04 -07:00
committed by Zomatree
parent 6f0e80eef9
commit 6737100b4b
2 changed files with 92 additions and 46 deletions

View File

@@ -75,10 +75,9 @@ mod test {
use crate::{rocket, util::test::TestHarness};
use revolt_database::{Channel, Server};
use revolt_models::v0::{
DataCreateGroup, DataCreateServer, DataCreateServerChannel, Invite, InviteResponse,
LegacyServerChannelType,
DataCreateGroup, DataCreateServerChannel, Invite, InviteResponse, LegacyServerChannelType,
};
use rocket::http::{Header, Status};
use rocket::http::Status;
#[rocket::async_test]
async fn success_fetch_group_invite() {
@@ -94,12 +93,13 @@ mod test {
)
.await
.expect("`Channel`");
let create_response = harness
.client
.post(format!("/channels/{}/invites", group.id()))
.header(Header::new("x-session-token", session.token.to_string()))
.dispatch()
.await;
let create_response = TestHarness::with_session(
session,
harness
.client
.post(format!("/channels/{}/invites", group.id())),
)
.await;
assert_eq!(create_response.status(), Status::Ok);
let invite_from_create: Invite = create_response.into_json().await.expect("`Invite`");
let invite_code = match invite_from_create {
@@ -143,25 +143,15 @@ mod test {
async fn success_fetch_text_channel_invite() {
let harness = TestHarness::new().await;
let (_, session, user) = harness.new_user().await;
let (_, channels) = Server::create(
&harness.db,
DataCreateServer {
name: "Test Server".to_string(),
..Default::default()
},
&user,
true,
)
.await
.expect("Failed to create test server");
let (_, channels) = harness.new_server(&user).await;
let channel = channels.first().expect("Server Channel");
let create_response = harness
.client
.post(format!("/channels/{}/invites", channel.id()))
.header(Header::new("x-session-token", session.token.to_string()))
.dispatch()
.await;
let create_response = TestHarness::with_session(
session,
harness
.client
.post(format!("/channels/{}/invites", channel.id())),
)
.await;
assert_eq!(create_response.status(), Status::Ok);
let invite_from_create: Invite = create_response.into_json().await.expect("`Invite`");
let invite_code = match invite_from_create {
@@ -194,17 +184,7 @@ mod test {
async fn success_fetch_voice_channel_invite() {
let harness = TestHarness::new().await;
let (_, session, user) = harness.new_user().await;
let (server, _) = Server::create(
&harness.db,
DataCreateServer {
name: "Test Server".to_string(),
..Default::default()
},
&user,
true,
)
.await
.expect("Failed to create test server");
let (server, _) = harness.new_server(&user).await;
let server_mut: &mut Server = &mut server.clone();
let channel = Channel::create_server_channel(
@@ -220,12 +200,13 @@ mod test {
)
.await
.expect("Failed to make new channel");
let create_response = harness
.client
.post(format!("/channels/{}/invites", channel.id()))
.header(Header::new("x-session-token", session.token.to_string()))
.dispatch()
.await;
let create_response = TestHarness::with_session(
session,
harness
.client
.post(format!("/channels/{}/invites", channel.id())),
)
.await;
assert_eq!(create_response.status(), Status::Ok);
let invite_from_create: Invite = create_response.into_json().await.expect("`Invite`");
let invite_code = match invite_from_create {