// Copyright (c) Microsoft Open Technologies, Inc. All rights reserved. See License.txt in the project root for license information.

namespace PlanCompilerTests
{
    using System.Data.Entity;
    using System.Linq;
    using AdvancedPatternsModel;
    using Xunit;

    /// <summary>
    /// Tests for GroupBy statements in Linq queries.
    /// </summary>
    public class LinqGroupByTests : FunctionalTestBase
    {
        #region Infrastructure/setup

        #endregion

        #region Tests for GroupBy that trigger an aggregate pushdown

        [Fact]
        private void GroupBy_aggregate_pushdown_single_key()
        {
            using (var context = new AdvancedPatternsMasterContext())
            {
                var groupByQuery = from workOrder in context.WorkOrders
                                   group new
                                       {
                                           workOrder.WorkOrderId,
                                           workOrder.Details
                                       } by workOrder.EmployeeId
                                   into ordersByEmployeeGroup
                                   select new
                                       {
                                           EmployeeId = ordersByEmployeeGroup.Key,
                                           OrderCount = ordersByEmployeeGroup.Count(),
                                           MaxOrderId = ordersByEmployeeGroup.Max(o => o.WorkOrderId)
                                       };
                var sql = groupByQuery.ToString();
                Assert.True(sql != null && sql.ToUpper().Contains("GROUP BY"));
            }
        }

        // Dev11 448362   
        [Fact]
        private void GroupBy_aggregate_pushdown_translates_NewRecordOp()
        {
            using (var context = new AdvancedPatternsMasterContext())
            {
                var groupByNewQuery = from workOrder in context.WorkOrders
                                      group new
                                          {
                                              workOrder.WorkOrderId,
                                              workOrder.Details
                                          } by new
                                              {
                                                  workOrder.EmployeeId
                                              }
                                      into ordersByEmployeeGroup
                                      select new
                                          {
                                              ordersByEmployeeGroup.Key.EmployeeId,
                                              OrderCount = ordersByEmployeeGroup.Count(),
                                              MaxOrderId = ordersByEmployeeGroup.Max(o => o.WorkOrderId)
                                          };
                var sql = groupByNewQuery.ToString();
                Assert.True(sql != null && sql.ToUpper().Contains("GROUP BY"));
            }
        }

        #endregion
    }
}
