PowerToys/src/modules/fancyzones/tests/UnitTests/Zone.Spec.cpp

#include "pch.h"
#include "lib\Zone.h"

using namespace Microsoft::VisualStudio::CppUnitTestFramework;

namespace FancyZonesUnitTests
{
    TEST_CLASS(ZoneUnitTests){
        public:
            TEST_METHOD(TestCreateZone){
                RECT zoneRect{ 10, 10, 200, 200 };
    winrt::com_ptr<IZone> zone = MakeZone(zoneRect);
    Assert::IsNotNull(&zone);
    CustomAssert::AreEqual(zoneRect, zone->GetZoneRect());

    constexpr size_t id = 10;
    zone->SetId(id);
    Assert::AreEqual(zone->Id(), id);
}

TEST_METHOD(ContainsWindow)
{
    RECT zoneRect{ 10, 10, 200, 200 };
    winrt::com_ptr<IZone> zone = MakeZone(zoneRect);
    HWND newWindow = Mocks::Window();
    Assert::IsFalse(zone->ContainsWindow(newWindow));
}

TEST_METHOD(TestAddRemoveWindow)
{
    RECT zoneRect{ 10, 10, 200, 200 };
    winrt::com_ptr<IZone> zone = MakeZone(zoneRect);
    HWND newWindow = Mocks::Window();

    Assert::IsFalse(zone->ContainsWindow(newWindow));
    zone->AddWindowToZone(newWindow, Mocks::Window(), true);
    Assert::IsTrue(zone->ContainsWindow(newWindow));

    zone->RemoveWindowFromZone(newWindow, false);
    Assert::IsFalse(zone->ContainsWindow(newWindow));
}

TEST_METHOD(TestRemoveInvalidWindow)
{
    RECT zoneRect{ 10, 10, 200, 200 };
    winrt::com_ptr<IZone> zone = MakeZone(zoneRect);
    HWND newWindow = Mocks::Window();
    zone->RemoveWindowFromZone(newWindow, false);
}
}
;
}