blah/blahrs
1
0
Fork 0
mirror of https://github.com/Blah-IM/blahrs.git synced 2025-07-01 03:55:33 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add tests for room join/leave and fix response status code

Browse source
This commit is contained in:
oxalica 2024-09-09 04:13:10 -04:00
parent 5cd45232f4
commit e40ec6a324
3 changed files with 115 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

101
blahd/tests/basic.rs
View file

@ -6,8 +6,8 @@ use std::sync::{Arc, LazyLock};
use anyhow::Result;
use blah::types::{
get_timestamp, AuthPayload, CreateRoomPayload, Id, MemberPermission, RoomAttrs, RoomMember,
RoomMemberList, ServerPermission, UserKey, WithSig,
get_timestamp, AuthPayload, CreateRoomPayload, Id, MemberPermission, RoomAdminOp,
RoomAdminPayload, RoomAttrs, RoomMember, RoomMemberList, ServerPermission, UserKey, WithSig,
};
use blahd::{ApiError, AppState, Database, RoomList, RoomMetadata};
use ed25519_dalek::SigningKey;
@ -31,6 +31,9 @@ fn mock_rng() -> impl RngCore {
rand::rngs::mock::StepRng::new(9, 1)
}
#[derive(Debug, Deserialize)]
enum NoContent {}
trait ResultExt {
fn expect_api_err(self, status: StatusCode, code: &str);
}
@ -258,3 +261,97 @@ async fn room_create_get(server: Server, #[case] public: bool) {
.unwrap();
assert_eq!(got_joined, expect_list(false));
}
#[rstest]
#[tokio::test]
async fn room_join_leave(server: Server) {
let rng = &mut mock_rng();
let rid_pub = create_room(
&server,
&ALICE_PRIV,
rng,
RoomAttrs::PUBLIC_JOINABLE,
"public room",
)
.await
.unwrap();
let rid_priv = create_room(
&server,
&ALICE_PRIV,
rng,
RoomAttrs::empty(),
"private room",
)
.await
.unwrap();
let mut join = |rid: Id, key: &SigningKey| {
let req = sign(
key,
rng,
RoomAdminPayload {
room: rid,
op: RoomAdminOp::AddMember {
permission: MemberPermission::MAX_SELF_ADD,
user: UserKey(key.verifying_key().to_bytes()),
},
},
);
server.request::<_, NoContent>(Method::POST, format!("/room/{rid}/admin"), None, Some(req))
};
// Ok.
join(rid_pub, &BOB_PRIV).await.unwrap();
// Already joined.
join(rid_pub, &BOB_PRIV)
.await
.expect_api_err(StatusCode::CONFLICT, "exists");
// Not permitted.
join(rid_priv, &BOB_PRIV)
.await
.expect_api_err(StatusCode::NOT_FOUND, "not_found");
// Not exists.
join(Id::INVALID, &BOB_PRIV)
.await
.expect_api_err(StatusCode::NOT_FOUND, "not_found");
// Bob is joined now.
assert_eq!(
server
.get::<RoomList>("/room?filter=joined", Some(&auth(&BOB_PRIV, rng)))
.await
.unwrap()
.rooms
.len(),
1,
);
let mut leave = |rid: Id, key: &SigningKey| {
let req = sign(
key,
rng,
RoomAdminPayload {
room: rid,
op: RoomAdminOp::RemoveMember {
user: UserKey(key.verifying_key().to_bytes()),
},
},
);
server.request::<_, NoContent>(Method::POST, format!("/room/{rid}/admin"), None, Some(req))
};
// Ok.
leave(rid_pub, &BOB_PRIV).await.unwrap();
// Already left.
leave(rid_pub, &BOB_PRIV)
.await
.expect_api_err(StatusCode::NOT_FOUND, "not_found");
// Unpermitted and not inside.
leave(rid_priv, &BOB_PRIV)
.await
.expect_api_err(StatusCode::NOT_FOUND, "not_found");
// Unpermitted and not inside.
leave(Id::INVALID, &BOB_PRIV)
.await
.expect_api_err(StatusCode::NOT_FOUND, "not_found");
}