diff --git a/src/lib.rs b/src/lib.rs index 3ab313b..5ba1afd 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -108,15 +108,15 @@ impl Sway { let n = self.stream.read(&mut buffer)?; match serde_json::from_slice::(&buffer[14..n]) { - Ok(result) => return Ok(result), - Err(e) => return Err(e.into()), + Ok(result) => Ok(result), + Err(e) => Err(e.into()), } } fn parse_events(&mut self, events: Vec<&str>) -> Result { match serde_json::to_string(&events) { Ok(json) => Ok(json), - Err(e) => return Err(e.into()), + Err(e) => Err(e.into()), } } } @@ -136,50 +136,31 @@ mod tests { #[test] fn run_command() { - match sway_ipc().run_command("reload") { - Ok(response) => assert_eq!(response.into_iter().nth(0).unwrap().success, true), - Err(e) => panic!("{}", e.to_string()), - } + assert_eq!(sway_ipc().run_command("reload").unwrap()[0].success, true); } #[test] fn get_workspaces() { - match sway_ipc().get_workspaces() { - Ok(response) => { - for workspace in response { - assert_eq!(workspace._type, "workspace"); - } - }, - Err(e) => panic!("{}", e.to_string()), + for workspace in sway_ipc().get_workspaces().unwrap() { + assert_eq!(workspace._type, "workspace"); } } #[test] fn subscribe() { - match sway_ipc().subscribe(vec!(EVENT_WORKSPACE, EVENT_MODE)) { - Ok(response) => assert_eq!(response.success, true), - Err(e) => panic!("{}", e.to_string()), - } + assert_eq!(sway_ipc().subscribe(vec!(EVENT_WORKSPACE, EVENT_MODE)).unwrap().success, true); } #[test] fn get_outputs() { - match sway_ipc().get_outputs() { - Ok(response) => { - for output in response { - assert_eq!(output._type, "output"); - } - }, - Err(e) => panic!("{}", e.to_string()), + for output in sway_ipc().get_outputs().unwrap() { + assert_eq!(output._type, "output"); } } #[test] fn get_tree() { - match sway_ipc().get_tree() { - Ok(response) => assert_eq!(response.id, 1), - Err(e) => panic!("{}", e.to_string()), - } + assert_eq!(sway_ipc().get_tree().unwrap().id, 1); } #[test] @@ -187,106 +168,68 @@ mod tests { let mut sway = sway_ipc(); sway.run_command("unmark").unwrap(); sway.run_command("mark test").unwrap(); - match sway.get_marks() { - Ok(response) => assert_eq!(response.into_iter().nth(0).unwrap(), "test"), - Err(e) => panic!("{}", e.to_string()), - } + assert_eq!(sway.get_marks().unwrap()[0], "test"); sway.run_command("unmark").unwrap(); } #[test] fn get_bar_config() { - match sway_ipc().get_bar_config("bar-0") { - Ok(response) => assert_eq!(response.id, "bar-0"), - Err(e) => panic!("{}", e.to_string()) - } + assert_eq!(sway_ipc().get_bar_config("bar-0").unwrap().id, "bar-0"); } #[test] fn get_version() { - match sway_ipc().get_version() { - Ok(response) => assert_eq!(response.major, 1), - Err(e) => panic!("{}", e.to_string()), - } + assert_eq!(sway_ipc().get_version().unwrap().major, 1); } #[test] fn get_binding_modes() { - match sway_ipc().get_binding_modes() { - Ok(response) => { - assert_eq!(response.into_iter().nth(0).unwrap(), "default") - }, - Err(e) => panic!("{}", e.to_string()), - } + assert_eq!(sway_ipc().get_binding_modes().unwrap()[0], "default"); } #[test] fn get_config() { - match sway_ipc().get_config() { - Ok(_) => {}, - Err(e) => panic!("{}", e.to_string()) - } + sway_ipc().get_config().unwrap().config; } #[test] fn send_tick() { - match sway_ipc().send_tick("") { - Ok(response) => assert_eq!(response.success, true), - Err(e) => panic!("{}", e.to_string()), - } + assert_eq!(sway_ipc().send_tick("").unwrap().success, true); } #[test] fn sync() { - match sway_ipc().sync() { - Ok(response) => assert_eq!(response.success, false), - Err(e) => panic!("{}", e.to_string()), - } + assert_eq!(sway_ipc().sync().unwrap().success, false); } #[test] fn get_binding_state() { - match sway_ipc().get_binding_state() { - Ok(response) => assert_eq!(response.name, "default"), - Err(e) => panic!("{}", e.to_string()), - } + assert_eq!(sway_ipc().get_binding_state().unwrap().name, "default"); } #[test] fn get_inputs() { - match sway_ipc().get_inputs() { - Ok(response) => { - for input in response { - if !(input._type.eq("keyboard") | input._type.eq("pointer")) { - panic!("invalid input: {}", input._type); - } - } - }, - Err(e) => panic!("{}", e.to_string()), + for input in sway_ipc().get_inputs().unwrap() { + if !(input._type.eq("keyboard") | input._type.eq("pointer")) { + panic!("invalid input: {}", input._type); + } } } #[test] fn get_seats() { - match sway_ipc().get_seats() { - Ok(response) => assert_eq!(response.into_iter().nth(0).unwrap().name, "seat0"), - Err(e) => panic!("{}", e.to_string()) - } + assert_eq!(sway_ipc().get_seats().unwrap()[0].name, "seat0"); } #[test] fn construct_packet() { - match I3msg::new("reload", IPC_RUN_COMMAND).construct_packet() { - Ok(buffer) => { - assert_eq!(buffer, vec![ - 0x69, 0x33, 0x2D, 0x69, - 0x70, 0x63, 0x06, 0x00, - 0x00, 0x00, 0x00, 0x00, - 0x00, 0x00, 0x72, 0x65, - 0x6C, 0x6F, 0x61, 0x64, - ]); - }, - Err(e) => panic!("{}", e.to_string()), - } + let packet = I3msg::new("reload", IPC_RUN_COMMAND).construct_packet().unwrap(); + assert_eq!(packet, vec![ + 0x69, 0x33, 0x2D, 0x69, + 0x70, 0x63, 0x06, 0x00, + 0x00, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x72, 0x65, + 0x6C, 0x6F, 0x61, 0x64, + ]); } } \ No newline at end of file