Changeset 1765 for trunk/sources/HeuristicLab.Security.Core/3.2/SecurityCoreTest.cs

Timestamp:

05/08/09 14:08:46 (15 years ago)

Author:

mbecirov

Message:

#586 Extended test data insert

File:

• trunk/sources/HeuristicLab.Security.Core/3.2/SecurityCoreTest.cs (modified) (5 diffs)

trunk/sources/HeuristicLab.Security.Core/3.2/SecurityCoreTest.cs

@@ -5 +5 @@
 using HeuristicLab.PluginInfrastructure;
 using HeuristicLab.Security.Contracts.BusinessObjects;
+using HeuristicLab.Hive.Server.Core;
 
 namespace HeuristicLab.Security.Core {
@@ -11 +12 @@
         AutoRestart = true)]
   class SecurityCoreTest : ApplicationBase {
+
+      SecurityManager manager = new SecurityManager();
+      PermissionManager permManager =  new PermissionManager();
+
+
       private void testGroups() {
-        SecurityManager manager =
-          new SecurityManager();
 
         User user = new User();
@@ -56 +60 @@
 
       private void testPermissions() {
-        SecurityManager manager =
-           new SecurityManager();
-
-        PermissionManager permManager =
-          new PermissionManager();
 
         Permission permission = new Permission();
@@ -99 +98 @@
 
       private void testPermissionsGroup() {
-        SecurityManager manager =
-          new SecurityManager();
-
-        PermissionManager permManager =
-          new PermissionManager();
 
         User user = new User();
@@ -170 +164 @@
         manager.RemoveUserGroup(group2.Id);
 
-        manager.RemovePermission(permission.Id);                        
-      }
-      
+        manager.RemovePermission(permission.Id);
+      }
+      #region TEST DATA INSERT SECTION for MB
+
+      private void InsertTestDataForPermissionCheck() {
+        int numberOfUsers = 10;
+
+        List<User> users = new List<User>();
+        for (int i = 0; i < numberOfUsers; i++) {
+          users.Add(MB_InsertTestUser());
+        }
+        MB_AssignGroups(users);
+        
+      }
+
+      private User MB_InsertTestUser() {
+        Random rand = new Random(DateTime.Now.Millisecond);
+        int usr = rand.Next(1000);
+        User user = new User();
+        user.Login = "test" + usr.ToString();
+        user.SetHashedPassword("test");
+        user.Name = "test" + usr.ToString();
+        return manager.AddNewUser(user);
+      }
+
+      /// <summary>
+      /// Splits the given users into two groups and assigns them. Some will be not assigned.
+      /// </summary>
+      /// <param name="users"></param>
+      private void MB_AssignGroups(List<User> users) {
+        UserGroup group01 = new UserGroup();
+        group01.Name = "Test Group 01";
+
+        UserGroup group02 = new UserGroup();
+        group02.Name = "Test Group 02";
+
+        //three-way split users into group1, group2 and none
+        int idx = users.Count / 3;
+        for (int i = 0; i < idx; i++) {
+          group01.Members.Add(users[i]);
+        }
+        for (int i = idx; i < users.Count / 2; i++) {
+          group02.Members.Add(users[i]);
+        }
+        manager.AddNewUserGroup(group01);
+        manager.AddNewUserGroup(group02);
+        
+      }
+
+      private void MB_AddPermission(List<User> users) {
+        if (users.Count < 2) return;
+        Permission permission = new Permission();
+        //permission.Id = PermissiveSecurityAction.Add_Job;
+        permission.Name = "ADD_JOB";
+        permission.Description = "Add new jobs";
+        permission.Plugin = "HeuristicLab.Hive.Server";
+        
+        manager.AddPermission(permission);
+        //grant permission to random users
+        Random rand = new Random(DateTime.Now.Millisecond);
+        for (int i = 0; i < users.Count/2; i++) {
+          int idx = rand.Next(users.Count);
+          manager.GrantPermission(users[i].Id, permission.Id, Guid.Empty);
+        }
+      }
+
+      #endregion
       public override void Run() {
-        testPermissionsGroup();
+        //testPermissionsGroup();
+        InsertTestDataForPermissionCheck();
       }
     }