Changeset 6457 for branches/HeuristicLab.Hive-3.4/sources/HeuristicLab.Services.Hive.Tests

Timestamp:

06/20/11 17:04:35 (13 years ago)

Author:

cneumuel

Message:

#1233

• added methods for granting and revoking hive experiment permissions
• added unit tests for hive experiment permissions
• added a status webpage to see how some statistics and current status

Location:

branches/HeuristicLab.Hive-3.4/sources/HeuristicLab.Services.Hive.Tests

Files:

• Mocks/MockAuthorizationManager.cs (modified) (2 diffs)
• Mocks/MockServiceLocator.cs (modified) (3 diffs)
• ServiceTests.cs (modified) (6 diffs)

branches/HeuristicLab.Hive-3.4/sources/HeuristicLab.Services.Hive.Tests/Mocks/MockAuthorizationManager.cs

@@ -24 +24 @@
 namespace HeuristicLab.Services.Hive.Tests.Mocks {
   public class MockAuthorizationManager : IAuthorizationManager {
-    private Guid userId = new Guid("6C7911A1-46EA-4E51-97DF-5582653AAFCE");
+    public static Guid MockUserId1 = new Guid("6C7911A1-46EA-4E51-97DF-5582653AAFCE");
+    public static Guid MockUserId2 = new Guid("897660EB-C90F-4054-988C-D39D530A0A02");
+    public static Guid MockUserId3 = new Guid("9F7B4EA8-A38D-4BC2-802F-E148AC7A6A87");
+
+    private Guid userId = MockUserId1;
 
     public Guid UserId {
@@ -37 +41 @@
       // o.k.
     }
-
+    
+    internal void SetUserId(Guid id) {
+      this.userId = id;
+    }
   }
 }

branches/HeuristicLab.Hive-3.4/sources/HeuristicLab.Services.Hive.Tests/Mocks/MockServiceLocator.cs

@@ -22 +22 @@
 using HeuristicLab.Services.Hive.DataAccess;
 using HeuristicLab.Services.Hive.Tests.Mocks;
+using System;
 
 namespace HeuristicLab.Services.Hive.Tests {
@@ -35 +36 @@
     }
 
+    private MockAuthorizationManager mockAuthorizationManager = new MockAuthorizationManager();
     public IAuthorizationManager AuthorizationManager {
-      get { return new MockAuthorizationManager(); }
+      get { return mockAuthorizationManager; }
     }
 
@@ -53 +55 @@
       get { return defaultServiceLocator.HeartbeatManager; }
     }
+
+    #region Special mocking methods
+    public void SetCurrentUser(Guid id) {
+      mockAuthorizationManager.SetUserId(id);
+    }
+
+    #endregion
   }
 }

branches/HeuristicLab.Hive-3.4/sources/HeuristicLab.Services.Hive.Tests/ServiceTests.cs

@@ -26 +26 @@
 using HeuristicLab.Services.Hive.Common.DataTransfer;
 using HeuristicLab.Services.Hive.Common.ServiceContracts;
+using HeuristicLab.Services.Hive.Tests.Mocks;
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 using DT = HeuristicLab.Services.Hive.Common.DataTransfer;
@@ -32 +33 @@
   [TestClass]
   public class ServiceTests {
+    // use the mock service locator to modify service properties (such as current user)
+    private static MockServiceLocator mockServiceLocator;
 
     [ClassInitialize]
     public static void MyClassInitialize(TestContext testContext) {
-      ServiceLocator.Instance = new MockServiceLocator(ServiceLocator.Instance);
+      mockServiceLocator = new MockServiceLocator(ServiceLocator.Instance);
+      ServiceLocator.Instance = mockServiceLocator;
     }
 
@@ -47 +51 @@
 
       // create hive experiment
-      DT.HiveExperiment experiment = new DT.HiveExperiment() {
-        Name = "TestExperiment",
-        Description = ""
-      };
+      DT.HiveExperiment experiment = new DT.HiveExperiment() { Name = "TestExperiment", Description = "" };
 
       // create job
-      DT.Job job = new DT.Job() {
-        CoresNeeded = 1,
-        MemoryNeeded = 0,
-        Priority = 0
-      };
+      DT.Job job = new DT.Job() { CoresNeeded = 1, MemoryNeeded = 0, Priority = 0 };
       job.State = JobState.Offline;
       job.StateLog.Add(new StateLog { State = JobState.Offline, DateTime = DateTime.Now });
@@ -78 +75 @@
       plugin1.DateCreated = DateTime.Now;
       plugin1.Hash = hash;
-      
+
       DT.PluginData pluginData1 = new DT.PluginData();
       pluginData1.FileName = "Tests.MyPlugin-1.0.dll";
@@ -206 +203 @@
 
       // create hive experiment
-      DT.HiveExperiment experiment = new DT.HiveExperiment() {
-        Name = "TestExperiment",
-        Description = ""
-      };
+      DT.HiveExperiment experiment = new DT.HiveExperiment() { Name = "TestExperiment", Description = "" };
 
       // create parent job
@@ -304 +298 @@
       service.DeleteSlave(slave.Id);
     }
+
+
+    [TestMethod]
+    public void TestHiveExperimentPermissions() {
+      var service = GetLocalService();
+      mockServiceLocator.SetCurrentUser(MockAuthorizationManager.MockUserId1);
+
+      // create hive experiment
+      DT.HiveExperiment e1 = new DT.HiveExperiment() { Name = "TestExperiment", Description = "" };
+      e1.Id = service.AddHiveExperiment(e1);
+      
+      var e1loaded = service.GetHiveExperiment(e1.Id);
+      Assert.AreEqual(Permission.Full, e1loaded.Permission);
+
+      // change to user2
+      mockServiceLocator.SetCurrentUser(MockAuthorizationManager.MockUserId2);
+      e1loaded = service.GetHiveExperiment(e1.Id);
+      Assert.AreEqual(null, e1loaded); // no access
+
+      // user2 should not be able to grant permissions
+      try {
+        service.GrantPermission(e1.Id, MockAuthorizationManager.MockUserId2, Permission.ReadOnly);
+        Assert.Fail("Should not be possible to grant permission due to missing permission for User2");
+      }
+      catch { /* ok, cool */ }
+
+      // switch back to user1 (owner) and grant user2 permissions
+      mockServiceLocator.SetCurrentUser(MockAuthorizationManager.MockUserId1);
+      service.GrantPermission(e1.Id, MockAuthorizationManager.MockUserId2, Permission.ReadOnly);
+
+      // back to user2
+      mockServiceLocator.SetCurrentUser(MockAuthorizationManager.MockUserId2);
+      e1loaded = service.GetHiveExperiment(e1.Id);
+      Assert.AreEqual(Permission.ReadOnly, e1loaded.Permission);
+
+      // user2 should still not be able to grant permissions
+      try {
+        service.GrantPermission(e1.Id, MockAuthorizationManager.MockUserId2, Permission.ReadOnly);
+        Assert.Fail("Should not be possible to grant permission due to missing permission for User2");
+      }
+      catch { /* ok, cool */ }
+
+      // back to user1
+      mockServiceLocator.SetCurrentUser(MockAuthorizationManager.MockUserId1);
+      service.GrantPermission(e1.Id, MockAuthorizationManager.MockUserId2, Permission.Full);
+
+      // back to user2
+      mockServiceLocator.SetCurrentUser(MockAuthorizationManager.MockUserId2);
+      e1loaded = service.GetHiveExperiment(e1.Id);
+      Assert.AreEqual(Permission.Full, e1loaded.Permission);
+
+      // grant rights to user3, now this should be possible due to full permissions
+      service.GrantPermission(e1.Id, MockAuthorizationManager.MockUserId3, Permission.ReadOnly);
+
+      // back to user1 and revoke rights for user2
+      mockServiceLocator.SetCurrentUser(MockAuthorizationManager.MockUserId1);
+      service.RevokePermission(e1.Id, MockAuthorizationManager.MockUserId2);
+
+      // back to user2
+      mockServiceLocator.SetCurrentUser(MockAuthorizationManager.MockUserId2);
+      e1loaded = service.GetHiveExperiment(e1.Id);
+      Assert.AreEqual(null, e1loaded); // no access
+
+      service.DeleteHiveExperiment(e1.Id);
+    }
   }
 }