Some UserManager functions and BDD steps for them

Author: dmikhalin

src/user/user_manager.rs

@@ -66,14 +66,43 @@ impl UserManager {
         Err(ConnectionError::ClusterAllNodesFailed(error_buffer.join("\n")))?
     }
 
-    // pub fn all() -> Vec<User> {
-    //
-    // }
-    // async fn run_failsafe<F, P, R>(&self, name: String, task: F) -> Result<R>
-    //     where
-    //         F: Fn(ServerDatabase, ServerConnection, bool) -> P,
-    //         P: Future<Output = Result<R>>,
-    // {
-    //     Database::get(name, self.connection.clone()).await?.run_failsafe(&task).await
-    // }
+    pub async fn create(&self, username: String, password: String) -> Result<()> {
+        let mut error_buffer = Vec::with_capacity(self.connection.server_count());
+        for server_connection in self.connection.connections() {
+            match server_connection.create_user(username.clone(), password.clone()).await {
+                Ok(()) => {
+                    return Ok(());
+                }
+                Err(err) => error_buffer.push(format!("- {}: {}", server_connection.address(), err)),
+            }
+        }
+        Err(ConnectionError::ClusterAllNodesFailed(error_buffer.join("\n")))?
+    }
+
+    pub async fn delete(&self, username: String) -> Result<()> {
+        let mut error_buffer = Vec::with_capacity(self.connection.server_count());
+        for server_connection in self.connection.connections() {
+            match server_connection.delete_user(username.clone()).await {
+                Ok(()) => {
+                    return Ok(());
+                }
+                Err(err) => error_buffer.push(format!("- {}: {}", server_connection.address(), err)),
+            }
+        }
+        Err(ConnectionError::ClusterAllNodesFailed(error_buffer.join("\n")))?
+    }
+
+    pub async fn set_password(&self, username: String, password: String) -> Result<()> {
+        let mut error_buffer = Vec::with_capacity(self.connection.server_count());
+        for server_connection in self.connection.connections() {
+            match server_connection.set_user_password(username.clone(), password.clone()).await {
+                Ok(()) => {
+                    return Ok(());
+                }
+                Err(err) => error_buffer.push(format!("- {}: {}", server_connection.address(), err)),
+            }
+        }
+        Err(ConnectionError::ClusterAllNodesFailed(error_buffer.join("\n")))?
+    }
+
 }

tests/behaviour/connection/user/steps.rs

@@ -59,11 +59,29 @@ generic_step_impl! {
         assert!(res.unwrap());
     }
 
+    #[step(expr = "users not contains: {word}")]
+    async fn users_not_contains(context: &mut Context, username: String) {
+        let res = context.users.contains(username).await;
+        assert!(res.is_ok(), "{:?}", res.err());
+        assert!(!res.unwrap());
+    }
+
+    #[step(expr = "users create: {word}, {word}")]
+    async fn users_create(context: &mut Context, username: String, password: String) {
+        let res = context.users.create(username, password).await;
+        assert!(res.is_ok(), "{:?}", res.err());
+    }
+
+    #[step(expr = "users password set: {word}, {word}")]
+    async fn users_password_set(context: &mut Context, username: String, password: String) {
+        let res = context.users.set_password(username, password).await;
+        assert!(res.is_ok(), "{:?}", res.err());
+    }
+
     #[step(expr = "users delete: {word}")]
-    async fn user_delete(context: &mut Context, user: String) {
-        // for session_tracker in &mut context.session_trackers {
-        //     session_tracker.open_transaction(type_.transaction_type).await.unwrap();
-        // }
+    async fn user_delete(context: &mut Context, username: String) {
+        let res = context.users.delete(username).await;
+        assert!(res.is_ok(), "{:?}", res.err());
     }
 
 }