dotnet/tests/CommunityToolkit.Common.UnitTests/Collections/ObservableGroupedCollectionTests.cs

// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

using System.Collections.Generic;
using System.Linq;
using CommunityToolkit.Common.Collections;
using Microsoft.VisualStudio.TestTools.UnitTesting;

namespace CommunityToolkit.Common.UnitTests.Collections;

[TestClass]
public class ObservableGroupedCollectionTests
{
    [TestMethod]
    public void Ctor_ShouldHaveExpectedValues()
    {
        ObservableGroupedCollection<string, int>? groupCollection = new();

        Assert.AreEqual(groupCollection.Count, 0);
    }

    [TestMethod]
    public void Ctor_WithGroups_ShouldHaveExpectedValues()
    {
        List<IGrouping<string, int>>? groups = new()
        {
            new IntGroup("A", new[] { 1, 3, 5 }),
            new IntGroup("B", new[] { 2, 4, 6 }),
        };
        ObservableGroupedCollection<string, int>? groupCollection = new(groups);

        Assert.AreEqual(groupCollection.Count, 2);

        Assert.AreEqual(groupCollection.ElementAt(0).Key, "A");
        CollectionAssert.AreEqual(groupCollection.ElementAt(0), new[] { 1, 3, 5 });

        Assert.AreEqual(groupCollection.ElementAt(1).Key, "B");
        CollectionAssert.AreEqual(groupCollection.ElementAt(1), new[] { 2, 4, 6 });
    }
}