Fake Context

FakeHQContext.cs

using System;
using System.Data.Entity;
using System.Linq;
using System.Reflection;
using Entities.HQ;
using Repositories.Contexts;

namespace TestUtils
{
  public class FakeHQContext : IHQContext
  {
    public FakeHQContext()
    {
      HQCodes = new FakeHQCodesSet();
    }

    public IDbSet<HQCodes> HQCodes { get; set; }

    public Database Database
    {
        get { return null; }
    } 

    public IDbSet<T> Set<T>() where T : class
    {
      foreach (PropertyInfo property in typeof (FakeHQContext).GetProperties())
      {
        if (property.PropertyType == typeof (IDbSet<T>))
        {
          return property.GetValue(this, null) as IDbSet<T>;
        }
      } 
        
      throw new Exception("Type collection not found");
    }

    public void SaveChanges()
    {
      // do nothing
    }
  }

  public class FakeHQCodesSet : FakeDbSet<HQCodes>
  {
    public override HQCodes Find(params object[] keyValues)
    {
      return this.SingleOrDefault(e => e.CodeId == (long) keyValues.Single());
    }
  }
}