From 8a8f590788271d6d721d7db169f2a28734715745 Mon Sep 17 00:00:00 2001 From: Devin R Date: Thu, 23 Apr 2020 09:55:59 -0400 Subject: [PATCH] room: tests for calculate_room --- src/models/room.rs | 87 +++++++++- src/test_builder.rs | 27 ++++ tests/data/events/aliases.json | 15 ++ tests/data/events/name.json | 2 +- tests/data/sync_with_summary.json | 256 ++++++++++++++++++++++++++++++ 5 files changed, 382 insertions(+), 5 deletions(-) create mode 100644 tests/data/events/aliases.json create mode 100644 tests/data/sync_with_summary.json diff --git a/src/models/room.rs b/src/models/room.rs index efe8c6c8..d2e19ef6 100644 --- a/src/models/room.rs +++ b/src/models/room.rs @@ -163,7 +163,12 @@ impl RoomName { let mut names = members .values() .take(3) - .map(|mem| mem.user_id.localpart().to_string()) + .map(|mem| { + mem.display_name + .clone() + .unwrap_or(mem.user_id.localpart().to_string()) + .to_string() + }) .collect::>(); // stabilize ordering names.sort(); @@ -172,7 +177,11 @@ impl RoomName { let mut names = members .values() .take(3) - .map(|mem| mem.user_id.localpart().to_string()) + .map(|mem| { + mem.display_name + .clone() + .unwrap_or(mem.user_id.localpart().to_string()) + }) .collect::>(); names.sort(); // TODO what is the length the spec wants us to use here and in the `else` @@ -246,7 +255,7 @@ impl Room { true } - fn set_name_room(&mut self, name: &str) -> bool { + fn set_room_name(&mut self, name: &str) -> bool { self.room_name.set_name(name); true } @@ -338,7 +347,7 @@ impl Room { /// Returns true if the room name changed, false otherwise. pub fn handle_room_name(&mut self, event: &NameEvent) -> bool { match event.content.name() { - Some(name) => self.set_name_room(name), + Some(name) => self.set_room_name(name), _ => false, } } @@ -527,4 +536,74 @@ mod test { .unwrap(); assert_eq!(admin.power_level.unwrap(), js_int::Int::new(100).unwrap()); } + + #[test] + fn calculate_aliases() { + let rid = RoomId::try_from("!roomid:room.com").unwrap(); + let uid = UserId::try_from("@example:localhost").unwrap(); + + let mut bld = EventBuilder::default() + .add_state_event_from_file("./tests/data/events/aliases.json", StateEvent::RoomAliases) + .build_room_runner(&rid, &uid); + + let room = bld.to_room(); + + assert_eq!("tutorial", room.calculate_name()); + } + + #[test] + fn calculate_alias() { + let rid = RoomId::try_from("!roomid:room.com").unwrap(); + let uid = UserId::try_from("@example:localhost").unwrap(); + + let mut bld = EventBuilder::default() + .add_state_event_from_file( + "./tests/data/events/alias.json", + StateEvent::RoomCanonicalAlias, + ) + .build_room_runner(&rid, &uid); + + let room = bld.to_room(); + + assert_eq!("tutorial", room.calculate_name()); + } + + #[test] + fn calculate_name() { + let rid = RoomId::try_from("!roomid:room.com").unwrap(); + let uid = UserId::try_from("@example:localhost").unwrap(); + + let mut bld = EventBuilder::default() + .add_state_event_from_file("./tests/data/events/name.json", StateEvent::RoomName) + .build_room_runner(&rid, &uid); + + let room = bld.to_room(); + + assert_eq!("room name", room.calculate_name()); + } + + #[tokio::test] + async fn calculate_room_names_from_summary() { + let homeserver = Url::from_str(&mockito::server_url()).unwrap(); + + let mut bld = EventBuilder::default().build_with_response( + // this sync has no room.name or room.alias events so only relies on summary + "tests/data/sync_with_summary.json", + "GET", + Matcher::Regex(r"^/_matrix/client/r0/sync\?.*$".to_string()), + ); + + let session = Session { + access_token: "1234".to_owned(), + user_id: UserId::try_from("@example:localhost").unwrap(), + device_id: "DEVICEID".to_owned(), + }; + let client = AsyncClient::new(homeserver, Some(session)).unwrap(); + let client = bld.set_client(client).to_client().await.unwrap(); + + let sync_settings = SyncSettings::new().timeout(Duration::from_millis(3000)); + let _response = client.sync(sync_settings).await.unwrap(); + + assert_eq!(vec!["example, example2"], client.get_room_names().await); + } } diff --git a/src/test_builder.rs b/src/test_builder.rs index ff637b1c..b28b0139 100644 --- a/src/test_builder.rs +++ b/src/test_builder.rs @@ -165,6 +165,33 @@ impl EventBuilder { self } + /// Consumes `ResponseBuilder and returns a `TestRunner`. + /// + /// The `TestRunner` responds to requests made by the `AsyncClient`. + pub fn build_with_response(mut self, path: P, method: &str, matcher: M) -> MockTestRunner + where + M: Into, + P: AsRef, + { + let body = fs::read_to_string(path.as_ref()) + .expect(&format!("file not found {:?}", path.as_ref())); + let mock = Some( + mock(method, matcher) + .with_status(200) + .with_body(body) + .create(), + ); + MockTestRunner { + client: None, + ephemeral: Vec::new(), + account_data: Vec::new(), + room_events: Vec::new(), + presence_events: Vec::new(), + state_events: Vec::new(), + mock, + } + } + /// Consumes `ResponseBuilder and returns a `TestRunner`. /// /// The `TestRunner` streams the events to the client and holds methods to make assertions diff --git a/tests/data/events/aliases.json b/tests/data/events/aliases.json new file mode 100644 index 00000000..5b81aa2d --- /dev/null +++ b/tests/data/events/aliases.json @@ -0,0 +1,15 @@ +{ + "content": { + "aliases": [ + "#tutorial:localhost" + ] + }, + "event_id": "$15139375516NUgtD:localhost", + "origin_server_ts": 1513937551720, + "sender": "@example:localhost", + "state_key": "localhost", + "type": "m.room.aliases", + "unsigned": { + "age": 7034220174 + } +} diff --git a/tests/data/events/name.json b/tests/data/events/name.json index 79946006..fa70a945 100644 --- a/tests/data/events/name.json +++ b/tests/data/events/name.json @@ -1,6 +1,6 @@ { "content": { - "name": "#tutorial:localhost" + "name": "room name" }, "event_id": "$15139375513VdeRF:localhost", "origin_server_ts": 1513937551461, diff --git a/tests/data/sync_with_summary.json b/tests/data/sync_with_summary.json new file mode 100644 index 00000000..f655b000 --- /dev/null +++ b/tests/data/sync_with_summary.json @@ -0,0 +1,256 @@ +{ + "device_one_time_keys_count": {}, + "next_batch": "s526_47314_0_7_1_1_1_11444_1", + "device_lists": { + "changed": [ + "@example:example.org" + ], + "left": [] + }, + "rooms": { + "invite": {}, + "join": { + "!SVkFJHzfwvuaIEawgC:localhost": { + "summary": { + "m.heroes": [ + "@alice:example.com", + "@bob:example.com" + ], + "m.joined_member_count": 2, + "m.invited_member_count": 0 + }, + "account_data": { + "events": [ + { + "content": { + "event_id": "$someplace:example.org" + }, + "room_id": "!roomid:room.com", + "type": "m.fully_read" + }, + { + "content": { + "ignored_users": { + "@someone:example.org": {} + } + }, + "type": "m.ignored_user_list" + } + ] + }, + "ephemeral": { + "events": [ + { + "content": { + "$151680659217152dPKjd:localhost": { + "m.read": { + "@example:localhost": { + "ts": 1516809890615 + } + } + } + }, + "type": "m.receipt" + } + ] + }, + "state": { + "events": [ + { + "content": { + "join_rule": "public" + }, + "event_id": "$15139375514WsgmR:localhost", + "origin_server_ts": 1513937551539, + "sender": "@example:localhost", + "state_key": "", + "type": "m.room.join_rules", + "unsigned": { + "age": 7034220355 + } + }, + { + "content": { + "avatar_url": null, + "displayname": "example", + "membership": "join" + }, + "event_id": "$151800140517rfvjc:localhost", + "membership": "join", + "origin_server_ts": 1518001405556, + "sender": "@example:localhost", + "state_key": "@example:localhost", + "type": "m.room.member", + "unsigned": { + "age": 2970366338, + "replaces_state": "$151800111315tsynI:localhost" + } + }, + { + "content": { + "history_visibility": "shared" + }, + "event_id": "$15139375515VaJEY:localhost", + "origin_server_ts": 1513937551613, + "sender": "@example:localhost", + "state_key": "", + "type": "m.room.history_visibility", + "unsigned": { + "age": 7034220281 + } + }, + { + "content": { + "creator": "@example:localhost" + }, + "event_id": "$15139375510KUZHi:localhost", + "origin_server_ts": 1513937551203, + "sender": "@example:localhost", + "state_key": "", + "type": "m.room.create", + "unsigned": { + "age": 7034220691 + } + }, + { + "content": { + "topic": "\ud83d\ude00" + }, + "event_id": "$151957878228ssqrJ:localhost", + "origin_server_ts": 1519578782185, + "sender": "@example:localhost", + "state_key": "", + "type": "m.room.topic", + "unsigned": { + "age": 1392989709, + "prev_content": { + "topic": "test" + }, + "prev_sender": "@example:localhost", + "replaces_state": "$151957069225EVYKm:localhost" + } + }, + { + "content": { + "ban": 50, + "events": { + "m.room.avatar": 50, + "m.room.canonical_alias": 50, + "m.room.history_visibility": 100, + "m.room.name": 50, + "m.room.power_levels": 100 + }, + "events_default": 0, + "invite": 0, + "kick": 50, + "redact": 50, + "state_default": 50, + "users": { + "@example:localhost": 100 + }, + "users_default": 0 + }, + "event_id": "$15139375512JaHAW:localhost", + "origin_server_ts": 1513937551359, + "sender": "@example:localhost", + "state_key": "", + "type": "m.room.power_levels", + "unsigned": { + "age": 7034220535 + } + }, + { + "content": { + "avatar_url": null, + "displayname": "example2", + "membership": "join" + }, + "event_id": "$152034824468gOeNB:localhost", + "membership": "join", + "origin_server_ts": 1520348244605, + "sender": "@example2:localhost", + "state_key": "@example2:localhost", + "type": "m.room.member", + "unsigned": { + "age": 623527289, + "prev_content": { + "membership": "leave" + }, + "prev_sender": "@example:localhost", + "replaces_state": "$152034819067QWJxM:localhost" + } + }, + { + "content": { + "membership": "leave", + "reason": "offline", + "avatar_url": "avatar.com", + "displayname": "example" + }, + "event_id": "$1585345508297748AIUBh:matrix.org", + "origin_server_ts": 1585345508223, + "sender": "@example:localhost", + "state_key": "@example:localhost", + "type": "m.room.member", + "unsigned": { + "replaces_state": "$1585345354296486IGZfp:localhost", + "prev_content": { + "avatar_url": "avatar.com", + "displayname": "example", + "membership": "join" + }, + "prev_sender": "@example2:localhost", + "age": 6992 + }, + "room_id": "!roomid:room.com" + } + ] + }, + "timeline": { + "events": [ + { + "content": { + "body": "baba", + "format": "org.matrix.custom.html", + "formatted_body": "baba", + "msgtype": "m.text" + }, + "event_id": "$152037280074GZeOm:localhost", + "origin_server_ts": 1520372800469, + "sender": "@example:localhost", + "type": "m.room.message", + "unsigned": { + "age": 598971425 + } + } + ], + "limited": true, + "prev_batch": "t392-516_47314_0_7_1_1_1_11444_1" + }, + "unread_notifications": { + "highlight_count": 0, + "notification_count": 11 + } + } + }, + "leave": {} + }, + "to_device": { + "events": [] + }, + "presence": { + "events": [ + { + "content": { + "avatar_url": "mxc://localhost:wefuiwegh8742w", + "currently_active": false, + "last_active_ago": 1, + "presence": "online", + "status_msg": "Making cupcakes" + }, + "sender": "@example:localhost", + "type": "m.presence" + } + ] + } +}