namespace WebApp.Mocks
{
    public class MockModel1Set : MockDbSet<Model1>
    {
        public override Model1 Find(params object[] keyValues)
        {
            return this.SingleOrDefault(s => s.Id == (int)keyValues.Single());
        }
    }

    public class MockModel2Set : MockDbSet<Model2> {}
}