using BarberShop;
using NUnit.Framework;
using System.Collections.Generic;

public class Test_22
{
    private BarberShop.BarberShop barberShop;
    private Barber b1 = new Barber("a", 1, 1);
    private Barber b2 = new Barber("b", 2, 3);
    private Barber b3 = new Barber("c", 2, 2);

    private Client c1 = new Client("a", 1, Gender.MALE);
    private Client c2 = new Client("b", 1, Gender.FEMALE);
    private Client c3 = new Client("c", 2, Gender.FEMALE);
    private Client c4 = new Client("d", 6, Gender.FEMALE);
    private Client c5 = new Client("e", 5, Gender.FEMALE);

    [SetUp]
    public void Setup()
    {
        this.barberShop = new BarberShop.BarberShop();
    }

    [Test]
    public void TestGetAllBarbersSortedWithClientsCountDesc1()
    {
        this.barberShop.AddBarber(b1);
        this.barberShop.AddBarber(b2);
        this.barberShop.AddBarber(b3);
        this.barberShop.AddClient(c1);
        this.barberShop.AddClient(c2);
        this.barberShop.AddClient(c3);
        this.barberShop.AddClient(c4);
        this.barberShop.AddClient(c5);
        this.barberShop.AssignClient(b2, c1);
        this.barberShop.AssignClient(b2, c2);
        this.barberShop.AssignClient(b1, c3);
        this.barberShop.AssignClient(b2, c4);
        this.barberShop.AssignClient(b2, c5);

        var res = this.barberShop.GetAllBarbersSortedWithClientsCountDesc();
        var expected = new List<Barber>() { b2, b1, b3 };
        CollectionAssert.AreEqual(res, expected);
    }
}