#![cfg(feature = "unstable-msc3381")] use std::{collections::BTreeMap, ops::Range}; use assert_matches2::assert_matches; use js_int::{uint, UInt}; use ruma_common::{ events::{ message::TextContentBlock, poll::{ compile_poll_results, end::PollEndEventContent, response::{OriginalSyncPollResponseEvent, PollResponseEventContent}, start::{ OriginalSyncPollStartEvent, PollAnswer, PollAnswers, PollAnswersError, PollContentBlock, PollKind, PollStartEventContent, }, }, relation::Reference, AnyMessageLikeEvent, MessageLikeEvent, }, owned_event_id, MilliSecondsSinceUnixEpoch, }; use serde_json::{from_value as from_json_value, json, to_value as to_json_value}; #[test] fn poll_answers_deserialization_valid() { let json_data = json!([ { "org.matrix.msc3381.v2.id": "aaa", "org.matrix.msc1767.text": [{ "body": "First answer" }] }, { "org.matrix.msc3381.v2.id": "bbb", "org.matrix.msc1767.text": [{ "body": "Second answer" }] }, ]); let answers = from_json_value::(json_data).unwrap(); assert_eq!(answers.len(), 2); } #[test] fn poll_answers_deserialization_truncate() { let json_data = json!([ { "org.matrix.msc3381.v2.id": "aaa", "org.matrix.msc1767.text": [{ "body": "1st answer" }] }, { "org.matrix.msc3381.v2.id": "bbb", "org.matrix.msc1767.text": [{ "body": "2nd answer" }] }, { "org.matrix.msc3381.v2.id": "ccc", "org.matrix.msc1767.text": [{ "body": "3rd answer" }] }, { "org.matrix.msc3381.v2.id": "ddd", "org.matrix.msc1767.text": [{ "body": "4th answer" }] }, { "org.matrix.msc3381.v2.id": "eee", "org.matrix.msc1767.text": [{ "body": "5th answer" }] }, { "org.matrix.msc3381.v2.id": "fff", "org.matrix.msc1767.text": [{ "body": "6th answer" }] }, { "org.matrix.msc3381.v2.id": "ggg", "org.matrix.msc1767.text": [{ "body": "7th answer" }] }, { "org.matrix.msc3381.v2.id": "hhh", "org.matrix.msc1767.text": [{ "body": "8th answer" }] }, { "org.matrix.msc3381.v2.id": "iii", "org.matrix.msc1767.text": [{ "body": "9th answer" }] }, { "org.matrix.msc3381.v2.id": "jjj", "org.matrix.msc1767.text": [{ "body": "10th answer" }] }, { "org.matrix.msc3381.v2.id": "kkk", "org.matrix.msc1767.text": [{ "body": "11th answer" }] }, { "org.matrix.msc3381.v2.id": "lll", "org.matrix.msc1767.text": [{ "body": "12th answer" }] }, { "org.matrix.msc3381.v2.id": "mmm", "org.matrix.msc1767.text": [{ "body": "13th answer" }] }, { "org.matrix.msc3381.v2.id": "nnn", "org.matrix.msc1767.text": [{ "body": "14th answer" }] }, { "org.matrix.msc3381.v2.id": "ooo", "org.matrix.msc1767.text": [{ "body": "15th answer" }] }, { "org.matrix.msc3381.v2.id": "ppp", "org.matrix.msc1767.text": [{ "body": "16th answer" }] }, { "org.matrix.msc3381.v2.id": "qqq", "org.matrix.msc1767.text": [{ "body": "17th answer" }] }, { "org.matrix.msc3381.v2.id": "rrr", "org.matrix.msc1767.text": [{ "body": "18th answer" }] }, { "org.matrix.msc3381.v2.id": "sss", "org.matrix.msc1767.text": [{ "body": "19th answer" }] }, { "org.matrix.msc3381.v2.id": "ttt", "org.matrix.msc1767.text": [{ "body": "20th answer" }] }, { "org.matrix.msc3381.v2.id": "uuu", "org.matrix.msc1767.text": [{ "body": "21th answer" }] }, { "org.matrix.msc3381.v2.id": "vvv", "org.matrix.msc1767.text": [{ "body": "22th answer" }] }, ]); let answers = from_json_value::(json_data).unwrap(); assert_eq!(answers.len(), 20); } #[test] fn poll_answers_deserialization_not_enough() { let json_data = json!([]); let err = from_json_value::(json_data).unwrap_err(); assert!(err.is_data()); assert_eq!(err.to_string(), PollAnswersError::NotEnoughValues.to_string()); } #[test] fn start_content_serialization() { let event_content = PollStartEventContent::with_plain_text( "How's the weather?\n1. Not bad…\n2. Fine.\n3. Amazing!", PollContentBlock::new( TextContentBlock::plain("How's the weather?"), vec![ PollAnswer::new("not-bad".to_owned(), TextContentBlock::plain("Not bad…")), PollAnswer::new("fine".to_owned(), TextContentBlock::plain("Fine.")), PollAnswer::new("amazing".to_owned(), TextContentBlock::plain("Amazing!")), ] .try_into() .unwrap(), ), ); assert_eq!( to_json_value(&event_content).unwrap(), json!({ "org.matrix.msc1767.text": [ { "body": "How's the weather?\n1. Not bad…\n2. Fine.\n3. Amazing!" } ], "org.matrix.msc3381.v2.poll": { "question": { "org.matrix.msc1767.text": [{ "body": "How's the weather?" }] }, "answers": [ { "org.matrix.msc3381.v2.id": "not-bad", "org.matrix.msc1767.text": [{ "body": "Not bad…" }] }, { "org.matrix.msc3381.v2.id": "fine", "org.matrix.msc1767.text": [{ "body": "Fine." }] }, { "org.matrix.msc3381.v2.id": "amazing", "org.matrix.msc1767.text": [{ "body": "Amazing!" }] }, ], }, }) ); } #[test] fn start_event_serialization() { let mut poll = PollContentBlock::new( TextContentBlock::plain("How's the weather?"), vec![ PollAnswer::new("not-bad".to_owned(), TextContentBlock::plain("Not bad…")), PollAnswer::new("fine".to_owned(), TextContentBlock::plain("Fine.")), PollAnswer::new("amazing".to_owned(), TextContentBlock::plain("Amazing!")), ] .try_into() .unwrap(), ); poll.kind = PollKind::Disclosed; poll.max_selections = uint!(2); let content = PollStartEventContent::with_plain_text( "How's the weather?\n1. Not bad…\n2. Fine.\n3. Amazing!", poll, ); assert_eq!( to_json_value(&content).unwrap(), json!({ "org.matrix.msc1767.text": [ { "body": "How's the weather?\n1. Not bad…\n2. Fine.\n3. Amazing!" } ], "org.matrix.msc3381.v2.poll": { "question": { "org.matrix.msc1767.text": [{ "body": "How's the weather?" }] }, "kind": "org.matrix.msc3381.v2.disclosed", "max_selections": 2, "answers": [ { "org.matrix.msc3381.v2.id": "not-bad", "org.matrix.msc1767.text": [{ "body": "Not bad…" }] }, { "org.matrix.msc3381.v2.id": "fine", "org.matrix.msc1767.text": [{ "body": "Fine." }] }, { "org.matrix.msc3381.v2.id": "amazing", "org.matrix.msc1767.text": [{ "body": "Amazing!" }] }, ] }, }) ); } #[test] fn start_event_deserialization() { let json_data = json!({ "content": { "org.matrix.msc1767.text": [ { "body": "How's the weather?\n1. Not bad…\n2. Fine.\n3. Amazing!" } ], "org.matrix.msc3381.v2.poll": { "question": { "org.matrix.msc1767.text": [{ "body": "How's the weather?" }] }, "max_selections": 2, "answers": [ { "org.matrix.msc3381.v2.id": "not-bad", "org.matrix.msc1767.text": [{ "body": "Not bad…" }] }, { "org.matrix.msc3381.v2.id": "fine", "org.matrix.msc1767.text": [{ "body": "Fine." }] }, { "org.matrix.msc3381.v2.id": "amazing", "org.matrix.msc1767.text": [{ "body": "Amazing!" }] }, ] }, }, "event_id": "$event:notareal.hs", "origin_server_ts": 134_829_848, "room_id": "!roomid:notareal.hs", "sender": "@user:notareal.hs", "type": "org.matrix.msc3381.v2.poll.start", }); let event = from_json_value::(json_data).unwrap(); assert_matches!( event, AnyMessageLikeEvent::PollStart(MessageLikeEvent::Original(message_event)) ); assert_eq!( message_event.content.text[0].body, "How's the weather?\n1. Not bad…\n2. Fine.\n3. Amazing!" ); let poll = message_event.content.poll; assert_eq!(poll.question.text[0].body, "How's the weather?"); assert_eq!(poll.kind, PollKind::Undisclosed); assert_eq!(poll.max_selections, uint!(2)); let answers = poll.answers; assert_eq!(answers.len(), 3); assert_eq!(answers[0].id, "not-bad"); assert_eq!(answers[0].text[0].body, "Not bad…"); assert_eq!(answers[1].id, "fine"); assert_eq!(answers[1].text[0].body, "Fine."); assert_eq!(answers[2].id, "amazing"); assert_eq!(answers[2].text[0].body, "Amazing!"); } #[test] fn response_content_serialization() { let event_content = PollResponseEventContent::new( vec!["my-answer".to_owned()].into(), owned_event_id!("$related_event:notareal.hs"), ); assert_eq!( to_json_value(&event_content).unwrap(), json!({ "org.matrix.msc3381.v2.selections": ["my-answer"], "m.relates_to": { "rel_type": "m.reference", "event_id": "$related_event:notareal.hs", } }) ); } #[test] fn response_event_serialization() { let content = PollResponseEventContent::new( vec!["first-answer".to_owned(), "second-answer".to_owned()].into(), owned_event_id!("$related_event:notareal.hs"), ); assert_eq!( to_json_value(&content).unwrap(), json!({ "org.matrix.msc3381.v2.selections": ["first-answer", "second-answer"], "m.relates_to": { "rel_type": "m.reference", "event_id": "$related_event:notareal.hs", }, }) ); } #[test] fn response_event_deserialization() { let json_data = json!({ "content": { "org.matrix.msc3381.v2.selections": ["my-answer"], "m.relates_to": { "rel_type": "m.reference", "event_id": "$related_event:notareal.hs", } }, "event_id": "$event:notareal.hs", "origin_server_ts": 134_829_848, "room_id": "!roomid:notareal.hs", "sender": "@user:notareal.hs", "type": "org.matrix.msc3381.v2.poll.response", }); let event = from_json_value::(json_data).unwrap(); assert_matches!( event, AnyMessageLikeEvent::PollResponse(MessageLikeEvent::Original(message_event)) ); let selections = message_event.content.selections; assert_eq!(selections.len(), 1); assert_eq!(selections[0], "my-answer"); assert_matches!(message_event.content.relates_to, Reference { event_id, .. }); assert_eq!(event_id, "$related_event:notareal.hs"); } #[test] fn end_content_serialization() { let event_content = PollEndEventContent::with_plain_text( "The poll has closed. Top answer: Amazing!", owned_event_id!("$related_event:notareal.hs"), ); assert_eq!( to_json_value(&event_content).unwrap(), json!({ "org.matrix.msc1767.text": [ { "body": "The poll has closed. Top answer: Amazing!" } ], "m.relates_to": { "rel_type": "m.reference", "event_id": "$related_event:notareal.hs", } }) ); } #[test] fn end_event_serialization() { let mut content = PollEndEventContent::with_plain_text( "The poll has closed. Top answer: Amazing!", owned_event_id!("$related_event:notareal.hs"), ); content.poll_results = Some( BTreeMap::from([ ("not-bad".to_owned(), uint!(1)), ("fine".to_owned(), uint!(5)), ("amazing".to_owned(), uint!(14)), ]) .into(), ); assert_eq!( to_json_value(&content).unwrap(), json!({ "org.matrix.msc1767.text": [ { "body": "The poll has closed. Top answer: Amazing!" }, ], "org.matrix.msc3381.v2.poll.results": { "not-bad": 1, "fine": 5, "amazing": 14, }, "m.relates_to": { "rel_type": "m.reference", "event_id": "$related_event:notareal.hs", }, }) ); } #[test] fn end_event_deserialization() { let json_data = json!({ "content": { "org.matrix.msc1767.text": [ { "body": "The poll has closed. Top answer: Amazing!" }, ], "m.relates_to": { "rel_type": "m.reference", "event_id": "$related_event:notareal.hs", } }, "event_id": "$event:notareal.hs", "origin_server_ts": 134_829_848, "room_id": "!roomid:notareal.hs", "sender": "@user:notareal.hs", "type": "org.matrix.msc3381.v2.poll.end", }); let event = from_json_value::(json_data).unwrap(); assert_matches!(event, AnyMessageLikeEvent::PollEnd(MessageLikeEvent::Original(message_event))); assert_eq!(message_event.content.text[0].body, "The poll has closed. Top answer: Amazing!"); assert_matches!(message_event.content.relates_to, Reference { event_id, .. }); assert_eq!(event_id, "$related_event:notareal.hs"); } fn new_poll_response( event_id: &str, user_id: &str, ts: UInt, selections: &[&str], ) -> OriginalSyncPollResponseEvent { from_json_value(json!({ "type": "org.matrix.msc3381.v2.poll.response", "sender": user_id, "origin_server_ts": ts, "event_id": event_id, "content": { "m.relates_to": { "rel_type": "m.reference", "event_id": "$poll_start_event_id" }, "org.matrix.msc3381.v2.selections": selections, } })) .unwrap() } fn generate_poll_responses( range: Range, selections: &[&str], ) -> Vec { let mut responses = Vec::with_capacity(range.len()); for i in range { let event_id = format!("$valid_event_{i}"); let user_id = format!("@valid_user_{i}:localhost"); let ts = 1000 + i as u16; responses.push(new_poll_response(&event_id, &user_id, ts.into(), selections)); } responses } #[test] fn compute_results() { let poll: OriginalSyncPollStartEvent = from_json_value(json!({ "type": "org.matrix.msc3381.v2.poll.start", "sender": "@alice:localhost", "event_id": "$poll_start_event_id", "origin_server_ts": 1, "content": { "org.matrix.msc1767.text": [ { "mimetype": "text/plain", "body": "What should we order for the party?\n1. Pizza 🍕\n2. Poutine 🍟\n3. Italian 🍝\n4. Wings 🔥" } ], "org.matrix.msc3381.v2.poll": { "kind": "m.disclosed", "max_selections": 2, "question": { "org.matrix.msc1767.text": [{"body": "What should we order for the party?"}] }, "answers": [ {"org.matrix.msc3381.v2.id": "pizza", "org.matrix.msc1767.text": [{"body": "Pizza 🍕"}]}, {"org.matrix.msc3381.v2.id": "poutine", "org.matrix.msc1767.text": [{"body": "Poutine 🍟"}]}, {"org.matrix.msc3381.v2.id": "italian", "org.matrix.msc1767.text": [{"body": "Italian 🍝"}]}, {"org.matrix.msc3381.v2.id": "wings", "org.matrix.msc1767.text": [{"body": "Wings 🔥"}]}, ] } } })).unwrap(); // Populate responses. let mut responses = generate_poll_responses(0..5, &["pizza"]); responses.extend(generate_poll_responses(5..10, &["poutine"])); responses.extend(generate_poll_responses(10..15, &["italian"])); responses.extend(generate_poll_responses(15..20, &["wings"])); let counted = compile_poll_results(&poll.content.poll, &responses, None); assert_eq!(counted.get("pizza").unwrap().len(), 5); assert_eq!(counted.get("poutine").unwrap().len(), 5); assert_eq!(counted.get("italian").unwrap().len(), 5); assert_eq!(counted.get("wings").unwrap().len(), 5); let mut iter = counted.keys(); assert_eq!(iter.next(), Some(&"pizza")); assert_eq!(iter.next(), Some(&"poutine")); assert_eq!(iter.next(), Some(&"italian")); assert_eq!(iter.next(), Some(&"wings")); assert_eq!(iter.next(), None); let poll_end = poll.compile_results(&responses); let results = poll_end.poll_results.unwrap(); assert_eq!(*results.get("pizza").unwrap(), uint!(5)); assert_eq!(*results.get("poutine").unwrap(), uint!(5)); assert_eq!(*results.get("italian").unwrap(), uint!(5)); assert_eq!(*results.get("wings").unwrap(), uint!(5)); assert_eq!( results.sorted().as_slice(), &[("italian", uint!(5)), ("pizza", uint!(5)), ("poutine", uint!(5)), ("wings", uint!(5))] ); assert_eq!( poll_end.text.find_plain(), Some("The poll has closed. Top answers: Pizza 🍕, Poutine 🍟, Italian 🍝, Wings 🔥") ); responses.extend(vec![ new_poll_response( "$multi_event_1", "@multi_user_1:localhost", uint!(2000), &["poutine", "wings"], ), new_poll_response( "$multi_event_2", "@multi_user_2:localhost", uint!(2200), &["poutine", "italian"], ), ]); let counted = compile_poll_results(&poll.content.poll, &responses, None); assert_eq!(counted.get("pizza").unwrap().len(), 5); assert_eq!(counted.get("poutine").unwrap().len(), 7); assert_eq!(counted.get("italian").unwrap().len(), 6); assert_eq!(counted.get("wings").unwrap().len(), 6); let mut iter = counted.keys(); assert_eq!(iter.next(), Some(&"poutine")); assert_eq!(iter.next(), Some(&"italian")); assert_eq!(iter.next(), Some(&"wings")); assert_eq!(iter.next(), Some(&"pizza")); assert_eq!(iter.next(), None); let poll_end = poll.compile_results(&responses); let results = poll_end.poll_results.unwrap(); assert_eq!(*results.get("pizza").unwrap(), uint!(5)); assert_eq!(*results.get("poutine").unwrap(), uint!(7)); assert_eq!(*results.get("italian").unwrap(), uint!(6)); assert_eq!(*results.get("wings").unwrap(), uint!(6)); assert_eq!( results.sorted().as_slice(), &[("poutine", uint!(7)), ("italian", uint!(6)), ("wings", uint!(6)), ("pizza", uint!(5))] ); assert_eq!(poll_end.text.find_plain(), Some("The poll has closed. Top answer: Poutine 🍟")); responses.extend(vec![ new_poll_response( "$multi_same_event_1", "@multi_same_user_1:localhost", uint!(3000), &["poutine", "poutine"], ), new_poll_response( "$multi_same_event_2", "@multi_same_user_2:localhost", uint!(3300), &["pizza", "pizza"], ), ]); let counted = compile_poll_results(&poll.content.poll, &responses, None); assert_eq!(counted.get("pizza").unwrap().len(), 6); assert_eq!(counted.get("poutine").unwrap().len(), 8); assert_eq!(counted.get("italian").unwrap().len(), 6); assert_eq!(counted.get("wings").unwrap().len(), 6); let poll_end = poll.compile_results(&responses); let results = poll_end.poll_results.unwrap(); assert_eq!(*results.get("pizza").unwrap(), uint!(6)); assert_eq!(*results.get("poutine").unwrap(), uint!(8)); assert_eq!(*results.get("italian").unwrap(), uint!(6)); assert_eq!(*results.get("wings").unwrap(), uint!(6)); let changing_user_1 = "@changing_user_1:localhost"; let changing_user_2 = "@changing_user_2:localhost"; let changing_user_3 = "@changing_user_3:localhost"; responses.extend(vec![ new_poll_response("$valid_for_now_event_1", changing_user_1, uint!(4000), &["wings"]), new_poll_response("$valid_for_now_event_2", changing_user_2, uint!(4100), &["wings"]), new_poll_response("$valid_for_now_event_3", changing_user_3, uint!(4200), &["wings"]), ]); let counted = compile_poll_results(&poll.content.poll, &responses, None); assert_eq!(counted.get("pizza").unwrap().len(), 6); assert_eq!(counted.get("poutine").unwrap().len(), 8); assert_eq!(counted.get("italian").unwrap().len(), 6); assert_eq!(counted.get("wings").unwrap().len(), 9); let mut iter = counted.keys(); assert_eq!(iter.next(), Some(&"wings")); assert_eq!(iter.next(), Some(&"poutine")); assert_eq!(iter.next(), Some(&"pizza")); assert_eq!(iter.next(), Some(&"italian")); assert_eq!(iter.next(), None); let poll_end = poll.compile_results(&responses); let results = poll_end.poll_results.unwrap(); assert_eq!(*results.get("pizza").unwrap(), uint!(6)); assert_eq!(*results.get("poutine").unwrap(), uint!(8)); assert_eq!(*results.get("italian").unwrap(), uint!(6)); assert_eq!(*results.get("wings").unwrap(), uint!(9)); assert_eq!( results.sorted().as_slice(), &[("wings", uint!(9)), ("poutine", uint!(8)), ("italian", uint!(6)), ("pizza", uint!(6))] ); assert_eq!(poll_end.text.find_plain(), Some("The poll has closed. Top answer: Wings 🔥")); // Change with new selection. responses.push(new_poll_response( "$change_vote_event", changing_user_1, uint!(4400), &["italian"], )); let counted = compile_poll_results(&poll.content.poll, &responses, None); assert_eq!(counted.get("pizza").unwrap().len(), 6); assert_eq!(counted.get("poutine").unwrap().len(), 8); assert_eq!(counted.get("italian").unwrap().len(), 7); assert_eq!(counted.get("wings").unwrap().len(), 8); let poll_end = poll.compile_results(&responses); let results = poll_end.poll_results.unwrap(); assert_eq!(*results.get("pizza").unwrap(), uint!(6)); assert_eq!(*results.get("poutine").unwrap(), uint!(8)); assert_eq!(*results.get("italian").unwrap(), uint!(7)); assert_eq!(*results.get("wings").unwrap(), uint!(8)); // Change with no selection. responses.push(new_poll_response( "$no_selection_vote_event", changing_user_1, uint!(4500), &[], )); let counted = compile_poll_results(&poll.content.poll, &responses, None); assert_eq!(counted.get("pizza").unwrap().len(), 6); assert_eq!(counted.get("poutine").unwrap().len(), 8); assert_eq!(counted.get("italian").unwrap().len(), 6); assert_eq!(counted.get("wings").unwrap().len(), 8); let poll_end = poll.compile_results(&responses); let results = poll_end.poll_results.unwrap(); assert_eq!(*results.get("pizza").unwrap(), uint!(6)); assert_eq!(*results.get("poutine").unwrap(), uint!(8)); assert_eq!(*results.get("italian").unwrap(), uint!(6)); assert_eq!(*results.get("wings").unwrap(), uint!(8)); // Change with invalid selection. responses.push(new_poll_response( "$invalid_vote_event", changing_user_2, uint!(4500), &["indian"], )); let counted = compile_poll_results(&poll.content.poll, &responses, None); assert_eq!(counted.get("pizza").unwrap().len(), 6); assert_eq!(counted.get("poutine").unwrap().len(), 8); assert_eq!(counted.get("italian").unwrap().len(), 6); assert_eq!(counted.get("wings").unwrap().len(), 7); let poll_end = poll.compile_results(&responses); let results = poll_end.poll_results.unwrap(); assert_eq!(*results.get("pizza").unwrap(), uint!(6)); assert_eq!(*results.get("poutine").unwrap(), uint!(8)); assert_eq!(*results.get("italian").unwrap(), uint!(6)); assert_eq!(*results.get("wings").unwrap(), uint!(7)); // Response older than most recent one is ignored. responses.push(new_poll_response("$past_event", changing_user_3, uint!(1), &["pizza"])); let counted = compile_poll_results(&poll.content.poll, &responses, None); assert_eq!(counted.get("pizza").unwrap().len(), 6); assert_eq!(counted.get("poutine").unwrap().len(), 8); assert_eq!(counted.get("italian").unwrap().len(), 6); assert_eq!(counted.get("wings").unwrap().len(), 7); let poll_end = poll.compile_results(&responses); let results = poll_end.poll_results.unwrap(); assert_eq!(*results.get("pizza").unwrap(), uint!(6)); assert_eq!(*results.get("poutine").unwrap(), uint!(8)); assert_eq!(*results.get("italian").unwrap(), uint!(6)); assert_eq!(*results.get("wings").unwrap(), uint!(7)); // Response in the future is ignored. let future_ts = MilliSecondsSinceUnixEpoch::now().0 + uint!(100_000); responses.push(new_poll_response("$future_event", changing_user_3, future_ts, &["pizza"])); let counted = compile_poll_results(&poll.content.poll, &responses, None); assert_eq!(counted.get("pizza").unwrap().len(), 6); assert_eq!(counted.get("poutine").unwrap().len(), 8); assert_eq!(counted.get("italian").unwrap().len(), 6); assert_eq!(counted.get("wings").unwrap().len(), 7); let poll_end = poll.compile_results(&responses); let results = poll_end.poll_results.unwrap(); assert_eq!(*results.get("pizza").unwrap(), uint!(6)); assert_eq!(*results.get("poutine").unwrap(), uint!(8)); assert_eq!(*results.get("italian").unwrap(), uint!(6)); assert_eq!(*results.get("wings").unwrap(), uint!(7)); }