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

Timestamp:

05/17/11 18:29:41 (13 years ago)

Author:

cneumuel

Message:

#1233

• added basic statistics recording for (once per minute)
  • executiontime per user
  • usedcores, usedmemory per slave

File:

• branches/HeuristicLab.Hive-3.4/sources/HeuristicLab.Services.Hive.Tests/DaoTests.cs (modified) (3 diffs)

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

@@ -7 +7 @@
 
 namespace HeuristicLab.Services.Hive.Tests {
+  using System.Collections.Generic;
   using System.Threading;
   using DT = HeuristicLab.Services.Hive.Common.DataTransfer;
@@ -83 +84 @@
         Assert.IsTrue(Math.Abs((job1.StateLog[i].DateTime - job1loaded.StateLog[i].DateTime).TotalSeconds) < 1);
       }
-      
+
       job1 = job1loaded;
 
@@ -236 +237 @@
 
     }
+
+
+    [TestMethod]
+    public void TestStatistics() {
+      IHiveDao dao = ServiceLocator.Instance.HiveDao;
+
+      var stats = new DT.Statistics();
+      stats.TimeStamp = DateTime.Now;
+      var slaveStats = new List<DT.SlaveStatistics>();
+      slaveStats.Add(new DT.SlaveStatistics() { Cores = 8, FreeCores = 3, Memory = 8000, FreeMemory = 3000, CpuUtilization = 0.8, SlaveId = Guid.NewGuid() });
+      slaveStats.Add(new DT.SlaveStatistics() { Cores = 2, FreeCores = 0, Memory = 1024, FreeMemory = 100, CpuUtilization = 0.99, SlaveId = Guid.NewGuid() });
+      slaveStats.Add(new DT.SlaveStatistics() { Cores = 4, FreeCores = 4, Memory = 3024, FreeMemory = 2300, CpuUtilization = 0.01, SlaveId = Guid.NewGuid() });
+      stats.SlaveStatistics = slaveStats;
+
+      var userStats = new List<DT.UserStatistics>();
+      userStats.Add(new DT.UserStatistics() { UsedCores = 15, ExecutionTime = TimeSpan.FromHours(123), UserId = Guid.NewGuid() });
+      userStats.Add(new DT.UserStatistics() { UsedCores = 42, ExecutionTime = TimeSpan.FromHours(945), UserId = Guid.NewGuid() });
+      stats.UserStatistics = userStats;
+
+      stats.Id = dao.AddStatistics(stats);
+
+      var statsLoaded = dao.GetStatistic(stats.Id);
+      Assert.AreEqual(stats.Id, statsLoaded.Id);
+      Assert.IsTrue(Math.Abs((stats.TimeStamp - statsLoaded.TimeStamp).TotalSeconds) < 1);
+
+      for (int i = 0; i  < slaveStats.Count; i ++) {
+        Assert.AreEqual(slaveStats[i].Cores, statsLoaded.SlaveStatistics.Single(x => x.SlaveId == slaveStats[i].SlaveId).Cores);
+        Assert.AreEqual(slaveStats[i].FreeCores, statsLoaded.SlaveStatistics.Single(x => x.SlaveId == slaveStats[i].SlaveId).FreeCores);
+        Assert.AreEqual(slaveStats[i].Memory, statsLoaded.SlaveStatistics.Single(x => x.SlaveId == slaveStats[i].SlaveId).Memory);
+        Assert.AreEqual(slaveStats[i].FreeMemory, statsLoaded.SlaveStatistics.Single(x => x.SlaveId == slaveStats[i].SlaveId).FreeMemory);
+        Assert.AreEqual(slaveStats[i].CpuUtilization, statsLoaded.SlaveStatistics.Single(x => x.SlaveId == slaveStats[i].SlaveId).CpuUtilization);
+      }
+
+      for (int i = 0; i < userStats.Count; i++) {
+        Assert.AreEqual(userStats[i].ExecutionTime, statsLoaded.UserStatistics.Single(x => x.UserId == userStats[i].UserId).ExecutionTime);
+        Assert.AreEqual(userStats[i].UsedCores, statsLoaded.UserStatistics.Single(x => x.UserId == userStats[i].UserId).UsedCores);        
+      }
+
+      dao.DeleteStatistics(stats.Id);
+      Assert.AreEqual(null, dao.GetStatistic(stats.Id));
+    }
   }
 }