test: complete test for editing role positions

This commit is contained in:
izzy
2025-06-08 11:32:42 +01:00
parent 3e8a401077
commit d30ceea373
5 changed files with 83 additions and 60 deletions

View File

@@ -61,22 +61,6 @@
"d": 0 "d": 0
}, },
"rank": 0 "rank": 0
},
"__ID:7__": {
"name": "Lower Rank 1",
"permissions": {
"a": 0,
"d": 0
},
"rank": 2
},
"__ID:8__": {
"name": "Lower Rank 2",
"permissions": {
"a": 0,
"d": 0
},
"rank": 2
} }
}, },
"default_permissions": 4000322560 "default_permissions": 4000322560

View File

@@ -1,10 +1,15 @@
use std::{collections::{HashMap, HashSet}, ops::BitXor, time::Duration}; use std::{
collections::{HashMap, HashSet},
ops::BitXor,
time::Duration,
};
use crate::{ use crate::{
mongodb::{ mongodb::{
bson::{doc, from_bson, from_document, to_document, Bson, DateTime, Document}, bson::{doc, from_bson, from_document, to_document, Bson, DateTime, Document},
options::FindOptions, options::FindOptions,
}, AbstractChannels, AbstractServers, Channel, Invite, MongoDb, Server, User, DISCRIMINATOR_SEARCH_SPACE },
AbstractChannels, AbstractServers, Channel, Invite, MongoDb, User, DISCRIMINATOR_SEARCH_SPACE,
}; };
use bson::{oid::ObjectId, to_bson}; use bson::{oid::ObjectId, to_bson};
use futures::StreamExt; use futures::StreamExt;
@@ -1165,7 +1170,9 @@ pub async fn run_migrations(db: &MongoDb, revision: i32) -> i32 {
} }
if revision <= 41 { if revision <= 41 {
info!("Running migration [revision 41 / 05-06-2025]: convert role ranks to uniform numbers."); info!(
"Running migration [revision 41 / 05-06-2025]: convert role ranks to uniform numbers."
);
#[derive(Serialize, Deserialize, Clone)] #[derive(Serialize, Deserialize, Clone)]
struct Role { struct Role {
@@ -1180,7 +1187,8 @@ pub async fn run_migrations(db: &MongoDb, revision: i32) -> i32 {
pub roles: HashMap<String, Role>, pub roles: HashMap<String, Role>,
} }
let mut servers = db.db() let mut servers = db
.db()
.collection::<Server>("servers") .collection::<Server>("servers")
.find(doc! { .find(doc! {
"roles": { "roles": {
@@ -1196,19 +1204,23 @@ pub async fn run_migrations(db: &MongoDb, revision: i32) -> i32 {
while let Some(server) = servers.next().await { while let Some(server) = servers.next().await {
let mut ordered_roles = server.roles.clone().into_iter().collect::<Vec<_>>(); let mut ordered_roles = server.roles.clone().into_iter().collect::<Vec<_>>();
ordered_roles.sort_by(|(_, role_a), (_, role_b)| role_a.rank.cmp(&role_b.rank)); ordered_roles.sort_by(|(_, role_a), (_, role_b)| role_a.rank.cmp(&role_b.rank));
let ordered_roles = ordered_roles.into_iter().map(|(id, _)| id).collect::<Vec<_>>(); let ordered_roles = ordered_roles
.into_iter()
.map(|(id, _)| id)
.collect::<Vec<_>>();
let mut doc = doc! {}; let mut doc = doc! {};
for id in server.roles.keys() { for id in server.roles.keys() {
doc.insert(format!("roles.{id}.rank"), ordered_roles.iter().position(|x| id == x).unwrap() as i64); doc.insert(
}; format!("roles.{id}.rank"),
ordered_roles.iter().position(|x| id == x).unwrap() as i64,
);
}
db.db().collection::<Server>("servers") db.db()
.update_one( .collection::<Server>("servers")
doc! { "_id": &server.id }, .update_one(doc! { "_id": &server.id }, doc! { "$set": doc })
doc! { "$set": doc }
)
.await .await
.unwrap(); .unwrap();
} }

View File

@@ -1,28 +0,0 @@
[
{
"_object_type": "User",
"_id": "__ID:0__",
"username": "Owner",
"discriminator": "0001"
},
{
"_object_type": "User",
"_id": "__ID:1__",
"username": "Member",
"discriminator": "0001"
},
{
"_object_type": "User",
"_id": "__ID:2__",
"username": "Member",
"discriminator": "0002"
},
{
"_object_type": "Channel",
"_id": "__ID:3__",
"channel_type": "Group",
"name": "My Group",
"owner": "__ID:0__",
"recipients": ["__ID:0__", "__ID:1__"]
}
]

View File

@@ -3,18 +3,21 @@
"_object_type": "User", "_object_type": "User",
"_id": "__ID:0__", "_id": "__ID:0__",
"username": "Owner", "username": "Owner",
"last_acknowledged_policy_change": "2025-06-07T04:04:48+0000",
"discriminator": "0001" "discriminator": "0001"
}, },
{ {
"_object_type": "User", "_object_type": "User",
"_id": "__ID:1__", "_id": "__ID:1__",
"username": "Moderator", "username": "Moderator",
"last_acknowledged_policy_change": "2025-06-07T04:04:48+0000",
"discriminator": "0001" "discriminator": "0001"
}, },
{ {
"_object_type": "User", "_object_type": "User",
"_id": "__ID:2__", "_id": "__ID:2__",
"username": "User", "username": "User",
"last_acknowledged_policy_change": "2025-06-07T04:04:48+0000",
"discriminator": "0001" "discriminator": "0001"
}, },
{ {
@@ -46,7 +49,7 @@
"__ID:5__": { "__ID:5__": {
"name": "Moderator", "name": "Moderator",
"permissions": { "permissions": {
"a": 545270208, "a": 545270216,
"d": 0 "d": 0
}, },
"rank": 1 "rank": 1

View File

@@ -85,10 +85,9 @@ mod test {
async fn edit_role_rankings() { async fn edit_role_rankings() {
let harness = TestHarness::new().await; let harness = TestHarness::new().await;
fixture!(harness.db, "server_with_roles", fixture!(harness.db, "server_with_many_roles",
owner user 0 owner user 0
moderator user 1 moderator user 1
user user 2
server server 4); server server 4);
// Moderator can re-order the roles below them // Moderator can re-order the roles below them
@@ -104,7 +103,7 @@ mod test {
let response = harness let response = harness
.client .client
.post(format!("/servers/{}/roles/ranks", server.id)) .patch(format!("/servers/{}/roles/ranks", server.id))
.header(ContentType::JSON) .header(ContentType::JSON)
.body( .body(
json!(v0::DataEditRoleRanks { json!(v0::DataEditRoleRanks {
@@ -121,5 +120,58 @@ mod test {
assert_eq!(response.status(), Status::Ok); assert_eq!(response.status(), Status::Ok);
drop(response); drop(response);
// ... but not above them
let mut target_order: Vec<String> = server
.ordered_roles()
.into_iter()
.map(|(id, _)| id)
.collect();
// Swap the two lower ranked roles
target_order.swap(0, 1);
let response = harness
.client
.patch(format!("/servers/{}/roles/ranks", server.id))
.header(ContentType::JSON)
.body(
json!(v0::DataEditRoleRanks {
ranks: target_order.clone()
})
.to_string(),
)
.header(Header::new(
"x-session-token",
moderator_session.token.to_string(),
))
.dispatch()
.await;
assert_eq!(response.status(), Status::Forbidden);
drop(response);
// The owner can set any order they want
let (_, owner_session) = harness.account_from_user(owner.id).await;
let response = harness
.client
.patch(format!("/servers/{}/roles/ranks", server.id))
.header(ContentType::JSON)
.body(
json!(v0::DataEditRoleRanks {
ranks: target_order.clone()
})
.to_string(),
)
.header(Header::new(
"x-session-token",
owner_session.token.to_string(),
))
.dispatch()
.await;
assert_eq!(response.status(), Status::Ok);
drop(response);
} }
} }