src/identity/application/services/owner_manage_store_employee_service.rs

@@ -246,6 +246,44 @@ mod tests {
 
             std::sync::Arc::new(m)
         }
+
+        pub fn mock_service_remove_employee_from_role(
+            times: Option<usize>,
+            cmd: RemoveEmployeeFromRoleCommand,
+        ) -> OwnerManageStoreEmployeesServiceObj {
+            let res = EmployeeRemovedFromRoleEvent::get_event(&cmd);
+            let mut m = MockOwnerManageStoreEmployeesUseCase::default();
+
+            if let Some(times) = times {
+                m.expect_remove_employee_from_role()
+                    .times(times)
+                    .returning(move |_| Ok(res.clone()));
+            } else {
+                m.expect_remove_employee_from_role()
+                    .returning(move |_| Ok(res.clone()));
+            }
+
+            std::sync::Arc::new(m)
+        }
+
+        pub fn mock_service_set_role_to_employee(
+            times: Option<usize>,
+            cmd: SetRoleToEmployeeCommand,
+        ) -> OwnerManageStoreEmployeesServiceObj {
+            let res = RoleSetToEmployeeEvent::get_event(&cmd);
+            let mut m = MockOwnerManageStoreEmployeesUseCase::default();
+
+            if let Some(times) = times {
+                m.expect_set_role_to_employee()
+                    .times(times)
+                    .returning(move |_| Ok(res.clone()));
+            } else {
+                m.expect_set_role_to_employee()
+                    .returning(move |_| Ok(res.clone()));
+            }
+
+            std::sync::Arc::new(m)
+        }
     }
 
     #[actix_rt::test]