added update beginnings; init of tests

1 files changed, 6 insertions, 6 deletions

diff --git a/src/input.rs b/src/input.rs
index 061f1fd..0c13e67 100644
--- a/src/input.rs
+++ b/src/input.rs
@@ -1,5 +1,5 @@
 use std::io::{stdin, Error, ErrorKind};
-use crate::{add, config::Cfg, list};
+use crate::{config::Cfg, list, modification, update};
 
 pub struct Input {
     pub command: String,
@@ -41,15 +41,15 @@ pub async fn get_input(config: Cfg) -> Result<(), Box<dyn std::error::Error>> {
     let input = Input::from(user_input.trim().to_string())?;
 
     match input.command.as_str() {
-        "add" => { 
-            if input.args == None { return Err(Box::new(Error::new(ErrorKind::InvalidInput, "TOO_FEW_ARGUMENTS"))) };
-            if input.args.as_ref().unwrap().len() != 1 { return Err(Box::new(Error::new(ErrorKind::InvalidInput, "TOO_MANY_ARGUMENTS"))) };
-            add(config, input.args.unwrap()[0].to_string()).await?;
-            Ok(()) 
+        "mod" => { 
+            modification(config, input.args).await
         },
         "list" => {
             list(config, input.args)
         },
+        "update" => {
+            update(config).await
+        },
         _ => Err(Box::new(Error::new(ErrorKind::InvalidInput, "UNKNOWN_COMMAND"))),
     }
 }