From db2822b2cb1c0f06919d009067dfe9c191af7aaa Mon Sep 17 00:00:00 2001 From: epi Date: Sun, 19 Dec 2021 13:03:44 -0600 Subject: [PATCH] revised a test --- src/scan_manager/tests.rs | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/src/scan_manager/tests.rs b/src/scan_manager/tests.rs index 993fe93..085b327 100644 --- a/src/scan_manager/tests.rs +++ b/src/scan_manager/tests.rs @@ -621,17 +621,20 @@ fn menu_get_command_input_from_user_returns_cancel() { fn menu_get_command_input_from_user_returns_add() { let menu = Menu::new(); - for cmd in &["add", "Addd", "a", "A"] { + for cmd in ["add", "Addd", "a", "A", "None"] { let test_url = "http://happyfuntimes.commmm"; let full_cmd = format!("{} {}\n", cmd, test_url); - let result = menu.get_command_input_from_user(&full_cmd).unwrap(); + if cmd != "None" { + let result = menu.get_command_input_from_user(&full_cmd).unwrap(); + assert!(matches!(result, MenuCmd::Add(_))); - assert!(matches!(result, MenuCmd::Add(_))); - - if let MenuCmd::Add(url) = result { - assert_eq!(url, test_url); - } + if let MenuCmd::Add(url) = result { + assert_eq!(url, test_url); + } + } else { + assert!(menu.get_command_input_from_user(&full_cmd).is_none()); + }; } }