source: trunk/sources/HeuristicLab.Hive.Server.Core/DbTestApp.cs @ 1377

#region License Information
/* HeuristicLab
 * Copyright (C) 2002-2008 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
 *
 * This file is part of HeuristicLab.
 *
 * HeuristicLab is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * HeuristicLab is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with HeuristicLab. If not, see <http://www.gnu.org/licenses/>.
 */
#endregion

using System;
using System.Collections.Generic;
using System.Text;
using HeuristicLab.PluginInfrastructure;
using System.Net;
using HeuristicLab.Hive.Contracts;
using HeuristicLab.Hive.Contracts.Interfaces;
using HeuristicLab.Hive.Server.DataAccess;
using HeuristicLab.Hive.Contracts.BusinessObjects;
using System.Diagnostics;
using HeuristicLab.DataAccess.Interfaces;

namespace HeuristicLab.Hive.Server {
  [ClassInfo(Name = "Hive DB Test App",
      Description = "Test Application for the Hive DataAccess Layer",
      AutoRestart = true)]
  class HiveDbTestApplication : ApplicationBase {
    private void TestClientAdapter() {
      IClientAdapter clientAdapter =
        ServiceLocator.GetClientAdapter();

      ClientInfo client = new ClientInfo();
      client.Login = DateTime.Now;
      client.ClientId = Guid.NewGuid();
      clientAdapter.Update(client);

      ClientInfo clientRead =
        clientAdapter.GetById(client.ClientId);
      Debug.Assert(
        clientRead != null &&
        client.ClientId == clientRead.ClientId);

      client.CpuSpeedPerCore = 2000;
      clientAdapter.Update(client);
      clientRead =
        clientAdapter.GetById(client.ClientId);
      Debug.Assert(
       clientRead != null &&
       client.ClientId == clientRead.ClientId &&
       clientRead.CpuSpeedPerCore == 2000);

      ICollection<ClientInfo> clients = 
        clientAdapter.GetAll();
      int count = clients.Count;

      clientAdapter.Delete(client);

      clients = clientAdapter.GetAll();
      Debug.Assert(count - 1 == clients.Count);
    }

    private void TestClientGroupAdapter() {
      IClientGroupAdapter clientGroupAdapter =
       ServiceLocator.GetClientGroupAdapter();

      ClientInfo client =
        new ClientInfo();
      client.Name = "Stefan";
      client.ClientId = Guid.NewGuid();
      client.Login = DateTime.Now;

      ClientInfo client2 =
        new ClientInfo();
      client2.Name = "Martin";
      client2.ClientId = Guid.NewGuid();
      client2.Login = DateTime.Now;

      ClientInfo client3 =
        new ClientInfo();
      client3.Name = "Heinz";
      client3.ClientId = Guid.NewGuid();
      client3.Login = DateTime.Now;

      ClientGroup group =
        new ClientGroup();

      ClientGroup subGroup =
        new ClientGroup();
      subGroup.Resources.Add(client);

      group.Resources.Add(client3);
      group.Resources.Add(client2);
      group.Resources.Add(subGroup);

      clientGroupAdapter.Update(group);

      ClientGroup read =
        clientGroupAdapter.GetById(group.Id);

      ICollection<ClientGroup> clientGroups =
        clientGroupAdapter.GetAll();

      IClientAdapter clientAdapter =
        ServiceLocator.GetClientAdapter();

      clientAdapter.Delete(client3);

      read =
         clientGroupAdapter.GetById(group.Id);

      clientGroupAdapter.Delete(subGroup);

      read =
         clientGroupAdapter.GetById(group.Id);

      clientGroups =
        clientGroupAdapter.GetAll();

      clientGroupAdapter.Delete(group);

      clientGroups =
        clientGroupAdapter.GetAll();

      clientAdapter.Delete(client);
      clientAdapter.Delete(client2);
    }

    private void TestJobAdapter() {
      IJobAdapter jobAdapter = 
        ServiceLocator.GetJobAdapter();
      IClientAdapter clientAdapter =
        ServiceLocator.GetClientAdapter();

      Job job = new Job();

      ClientInfo client = new ClientInfo();
      client.ClientId = Guid.NewGuid();
      client.Login = DateTime.Now;

      job.Client = client;
      jobAdapter.Update(job);

      ICollection<Job> jobs = jobAdapter.GetAll();

      jobAdapter.Delete(job);
      clientAdapter.Delete(client);

      jobs = jobAdapter.GetAll();
    }

    private void TestJobResultsAdapter() {
      Job job = new Job();

      ClientInfo client = new ClientInfo();
      client.ClientId = Guid.NewGuid();
      client.Login = DateTime.Now;

      job.Client = client;

      IJobResultsAdapter resultsAdapter = 
        ServiceLocator.GetJobResultsAdapter();

      byte[] resultByte = {0x0f, 0x1f, 0x2f, 0x3f, 0x4f};

      JobResult result = new JobResult();
      result.Client = client;
      result.Job = job;
      result.Result = resultByte;

      resultsAdapter.Update(result);

      JobResult read =
        resultsAdapter.GetById(result.Id);
      Debug.Assert(
        read.Id == result.Id &&
        result.Client.Id == read.Client.Id &&
        result.Job.Id == read.Job.Id &&
        result.Result == result.Result);

      int count =
        resultsAdapter.GetAll().Count;

      resultsAdapter.Delete(result);

      ICollection<JobResult> allResults =
        resultsAdapter.GetAll();

      Debug.Assert(allResults.Count == count - 1);

      IJobAdapter jboAdapter =
        ServiceLocator.GetJobAdapter();
      jboAdapter.Delete(job);
      IClientAdapter clientAdapter =
        ServiceLocator.GetClientAdapter();
      clientAdapter.Delete(client);
    }

    public override void Run() {
      IDBSynchronizer transactionManager =
        ServiceLocator.GetDBSynchronizer();
      
      TestClientAdapter();
      transactionManager.UpdateDB();

      TestClientGroupAdapter();
      transactionManager.UpdateDB();   

      TestJobAdapter();
      transactionManager.UpdateDB();  

      TestJobResultsAdapter();
      transactionManager.UpdateDB();     
    }
  }
}