PowerToys/src/tests/win-app-driver/FancyZonesTests/EditorGridZoneResizeTests.cs

using Microsoft.VisualStudio.TestTools.UnitTesting;
using OpenQA.Selenium.Appium;
using OpenQA.Selenium.Appium.Windows;
using OpenQA.Selenium.Interactions;
using System;
using System.Collections.ObjectModel;
using System.Windows.Forms;

namespace PowerToysTests
{ 
    [TestClass]
    public class FancyZonesEditorGridZoneResizeTests : FancyZonesEditor
    {
        private const int moveStep = 5;

        private void Move(AppiumWebElement thumb, int border, bool moveAscending, bool moveHorizontally, int clickShift = 0)
        {
            Actions action = new Actions(session);
            action.MoveToElement(thumb).MoveByOffset(0, clickShift).ClickAndHold();

            int thumbCenter = 0;
            if (moveHorizontally)
            {
                thumbCenter = thumb.Rect.X + thumb.Rect.Width / 2;
            }
            else
            {
                thumbCenter = thumb.Rect.Y + thumb.Rect.Height / 2;
            }

            int moves = Math.Abs(thumbCenter - border) / moveStep;
            for (int j = 0; j < moves; j++)
            {
                int step = moveAscending ? moveStep : -moveStep;
                if (moveHorizontally)
                {
                    action.MoveByOffset(step, 0);
                }
                else
                {
                    action.MoveByOffset(0, step);
                }
            }

            action.Release().Perform();
        }

        [TestMethod]
        public void MoveVerticalSplitter()
        {
            OpenCreatorWindow("Columns", "Custom table layout creator", "EditTemplateButton");
            WindowsElement gridEditor = session.FindElementByClassName("GridEditor");
            Assert.IsNotNull(gridEditor);

            Assert.AreEqual(3, session.FindElementsByClassName("GridZone").Count);
            ReadOnlyCollection<AppiumWebElement> thumbs = gridEditor.FindElementsByClassName("Thumb");
            Assert.AreEqual(2, thumbs.Count);

            //move left
            for (int i = 0; i < thumbs.Count; i++)
            {
                AppiumWebElement thumb = thumbs[i];
                int border = i == 0 ? 0 : thumbs[i - 1].Rect.Right;
                Move(thumb, border, false, true);

                Assert.IsTrue(thumb.Rect.Left - border <= moveStep);
                Assert.IsTrue(thumb.Rect.Right > border);
            }

            //move right
            for (int i = thumbs.Count - 1; i >= 0; i--)
            {
                AppiumWebElement thumb = thumbs[i];
                int border = i == thumbs.Count - 1 ? Screen.PrimaryScreen.WorkingArea.Right : thumbs[i + 1].Rect.Left;
                Move(thumb, border, true, true);

                Assert.IsTrue(border - thumb.Rect.Right <= moveStep);
                Assert.IsTrue(thumb.Rect.Left < border);
            }

            //move up
            foreach (AppiumWebElement thumb in thumbs)
            {
                int expected = thumb.Rect.X;
                
                Move(thumb, 0, false, false);
                int actual = thumb.Rect.X;

                Assert.AreEqual(expected, actual);
            }

            //move down
            foreach (AppiumWebElement thumb in thumbs)
            {
                int expected = thumb.Rect.X;

                Move(thumb, Screen.PrimaryScreen.WorkingArea.Right, false, false);
                int actual = thumb.Rect.X;

                Assert.AreEqual(expected, actual);
            }
        }

        [TestMethod]
        public void MoveHorizontalSplitter()
        {
            OpenCreatorWindow("Rows", "Custom table layout creator", "EditTemplateButton");
            WindowsElement gridEditor = session.FindElementByClassName("GridEditor");
            Assert.IsNotNull(gridEditor);

            Assert.AreEqual(3, session.FindElementsByClassName("GridZone").Count);
            ReadOnlyCollection<AppiumWebElement> thumbs = gridEditor.FindElementsByClassName("Thumb");
            Assert.AreEqual(2, thumbs.Count);

            //move up
            for (int i = 0; i < thumbs.Count; i++)
            {
                AppiumWebElement thumb = thumbs[i];
                int border = i == 0 ? 0 : thumbs[i - 1].Rect.Bottom;
                Move(thumb, border, false, false);

                Assert.IsTrue(thumb.Rect.Top - border <= moveStep);
                Assert.IsTrue(thumb.Rect.Right > border);
            }

            //move down
            for (int i = thumbs.Count - 1; i >= 0; i--)
            {
                AppiumWebElement thumb = thumbs[i];
                int border = i == thumbs.Count - 1 ? Screen.PrimaryScreen.WorkingArea.Bottom : thumbs[i + 1].Rect.Top;
                Move(thumb, border, true, false);

                Assert.IsTrue(border - thumb.Rect.Bottom <= moveStep);
                Assert.IsTrue(thumb.Rect.Top < border);
            }

            //move left
            foreach (AppiumWebElement thumb in thumbs)
            {
                int expected = thumb.Rect.Y;

                Move(thumb, 0, false, true);
                int actual = thumb.Rect.Y;

                Assert.AreEqual(expected, actual);
            }

            //move right
            foreach (AppiumWebElement thumb in thumbs)
            {
                int expected = thumb.Rect.Y;

                Move(thumb, Screen.PrimaryScreen.WorkingArea.Right, true, true);
                int actual = thumb.Rect.Y;

                Assert.AreEqual(expected, actual);
            }
        }

        [TestMethod]
        public void CreateSplitter()
        {
            OpenCreatorWindow("Columns", "Custom table layout creator", "EditTemplateButton");
            WindowsElement gridEditor = session.FindElementByClassName("GridEditor");
            Assert.IsNotNull(gridEditor);

            ReadOnlyCollection<AppiumWebElement> thumbs = gridEditor.FindElementsByClassName("Thumb");
            Assert.AreEqual(3, session.FindElementsByClassName("GridZone").Count);
            Assert.AreEqual(2, thumbs.Count);

            new Actions(session).MoveToElement(thumbs[0]).MoveByOffset(-30, 0).Click().Perform();
            Assert.AreEqual(3, gridEditor.FindElementsByClassName("Thumb").Count);

            ReadOnlyCollection<WindowsElement> zones = session.FindElementsByClassName("GridZone");
            Assert.AreEqual(4, zones.Count);

            //check that zone was splitted horizontally 
            Assert.AreNotEqual(zones[0].Rect.Height, zones[1].Rect.Height);
            Assert.AreNotEqual(zones[3].Rect.Height, zones[1].Rect.Height);
            Assert.AreEqual(zones[1].Rect.Height, zones[2].Rect.Height);
        }

        [TestMethod]
        public void CreateSplitterWithShiftPressed()
        {
            OpenCreatorWindow("Columns", "Custom table layout creator", "EditTemplateButton");
            WindowsElement gridEditor = session.FindElementByClassName("GridEditor");
            Assert.IsNotNull(gridEditor);

            ReadOnlyCollection<AppiumWebElement> thumbs = gridEditor.FindElementsByClassName("Thumb");
            Assert.AreEqual(3, session.FindElementsByClassName("GridZone").Count);
            Assert.AreEqual(2, thumbs.Count);

            new Actions(session).MoveToElement(thumbs[0]).MoveByOffset(-100, 0)
                .KeyDown(OpenQA.Selenium.Keys.Shift).Click().KeyUp(OpenQA.Selenium.Keys.Shift)
                .Perform();
            Assert.AreEqual(3, gridEditor.FindElementsByClassName("Thumb").Count);

            ReadOnlyCollection<WindowsElement> zones = session.FindElementsByClassName("GridZone");
            Assert.AreEqual(4, zones.Count);
            
            //check that zone was splitted vertically
            Assert.AreEqual(zones[0].Rect.Height, zones[1].Rect.Height);
            Assert.AreEqual(zones[1].Rect.Height, zones[2].Rect.Height);
            Assert.AreEqual(zones[2].Rect.Height, zones[3].Rect.Height);
        }

        [TestMethod]
        public void CreateSplitterWithShiftPressedFocusOnGridEditor()
        {
            OpenCreatorWindow("Columns", "Custom table layout creator", "EditTemplateButton");
            WindowsElement gridEditor = session.FindElementByClassName("GridEditor");
            Assert.IsNotNull(gridEditor);

            ReadOnlyCollection<AppiumWebElement> thumbs = gridEditor.FindElementsByClassName("Thumb");
            Assert.AreEqual(3, session.FindElementsByClassName("GridZone").Count);
            Assert.AreEqual(2, thumbs.Count);

            new Actions(session).MoveToElement(thumbs[0]).Click().MoveByOffset(-100, 0)
                .KeyDown(OpenQA.Selenium.Keys.Shift).Click().KeyUp(OpenQA.Selenium.Keys.Shift)
                .Perform();
            Assert.AreEqual(3, gridEditor.FindElementsByClassName("Thumb").Count);

            ReadOnlyCollection<WindowsElement> zones = session.FindElementsByClassName("GridZone");
            Assert.AreEqual(4, zones.Count);

            //check that zone was splitted vertically
            Assert.AreEqual(zones[0].Rect.Height, zones[1].Rect.Height);
            Assert.AreEqual(zones[1].Rect.Height, zones[2].Rect.Height);
            Assert.AreEqual(zones[2].Rect.Height, zones[3].Rect.Height);
        }

        [TestMethod]
        public void MoveHorizontallyWithLimiter()
        {
            OpenCreatorWindow("Columns", "Custom table layout creator", "EditTemplateButton");
            WindowsElement gridEditor = session.FindElementByClassName("GridEditor");
            Assert.IsNotNull(gridEditor);

            Assert.AreEqual(3, session.FindElementsByClassName("GridZone").Count);
            ReadOnlyCollection<AppiumWebElement> thumbs = gridEditor.FindElementsByClassName("Thumb");
            Assert.AreEqual(2, thumbs.Count);

            //create new zones
            new Actions(session).MoveToElement(thumbs[0]).Click().MoveByOffset(-30, 0)
                .KeyDown(OpenQA.Selenium.Keys.Shift).Click().KeyUp(OpenQA.Selenium.Keys.Shift)
                .Perform();
            thumbs = gridEditor.FindElementsByClassName("Thumb");
            Assert.AreEqual(4, session.FindElementsByClassName("GridZone").Count);
            Assert.AreEqual(3, thumbs.Count);

            //move thumbs
            AppiumWebElement limiter = gridEditor.FindElementsByClassName("Thumb")[0];
            AppiumWebElement movable = gridEditor.FindElementsByClassName("Thumb")[1];

            Move(movable, 0, false, true);
            Assert.IsTrue(movable.Rect.X > limiter.Rect.X);
            Assert.IsTrue(movable.Rect.X - limiter.Rect.X < movable.Rect.Width);

            Move(limiter, limiter.Rect.X - (limiter.Rect.X / 2), false, true);

            Move(movable, 0, false, true);
            Assert.IsTrue(movable.Rect.X > limiter.Rect.X);
            Assert.IsTrue(movable.Rect.X - limiter.Rect.X < movable.Rect.Width);
        }

        [TestMethod]
        public void MoveVerticallyWithLimiter()
        {
            OpenCreatorWindow("Rows", "Custom table layout creator", "EditTemplateButton"); 
            WindowsElement gridEditor = session.FindElementByClassName("GridEditor");
            Assert.IsNotNull(gridEditor);

            Assert.AreEqual(3, session.FindElementsByClassName("GridZone").Count);
            ReadOnlyCollection<AppiumWebElement> thumbs = gridEditor.FindElementsByClassName("Thumb");
            Assert.AreEqual(2, thumbs.Count);

            //create new zones
            new Actions(session).MoveToElement(thumbs[0]).Click().MoveByOffset(0, -(thumbs[0].Rect.Y / 2))
                .KeyDown(OpenQA.Selenium.Keys.Shift).Click().KeyUp(OpenQA.Selenium.Keys.Shift)
                .Perform();
            thumbs = gridEditor.FindElementsByClassName("Thumb");
            Assert.AreEqual(4, session.FindElementsByClassName("GridZone").Count);
            Assert.AreEqual(3, thumbs.Count);

            //move thumbs
            AppiumWebElement limiter = gridEditor.FindElementsByClassName("Thumb")[0];
            AppiumWebElement movable = gridEditor.FindElementsByClassName("Thumb")[1];

            Move(movable, 0, false, false);
            Assert.IsTrue(movable.Rect.Y > limiter.Rect.Y);
            Assert.IsTrue(movable.Rect.Y - limiter.Rect.Y < movable.Rect.Height);

            Move(limiter, limiter.Rect.Y - (limiter.Rect.Y / 2), false, false, -5);

            Move(movable, 0, false, false);
            Assert.IsTrue(movable.Rect.Y > limiter.Rect.Y);
            Assert.IsTrue(movable.Rect.Y - limiter.Rect.Y < movable.Rect.Height);
        }

        [TestMethod]
        public void MergeZones()
        {
            OpenCreatorWindow("Columns", "Custom table layout creator", "EditTemplateButton");
            WindowsElement gridEditor = session.FindElementByClassName("GridEditor");
            Assert.IsNotNull(gridEditor);

            ReadOnlyCollection<WindowsElement> zones = session.FindElementsByClassName("GridZone");
            ReadOnlyCollection<AppiumWebElement> thumbs = gridEditor.FindElementsByClassName("Thumb");
            Assert.AreEqual(3, zones.Count);
            Assert.AreEqual(2, thumbs.Count);

            Move(zones[0], thumbs[0].Rect.X + thumbs[0].Rect.Width + 10, true, true, -(zones[0].Rect.Height / 2) + 10);

            WindowsElement mergeButton = session.FindElementByName("Merge zones");
            Assert.IsNotNull(mergeButton, "Cannot merge: no merge button");
            new Actions(session).Click(mergeButton).Perform();

            Assert.AreEqual(2, session.FindElementsByClassName("GridZone").Count);
            Assert.AreEqual(1, gridEditor.FindElementsByClassName("Thumb").Count);
        }

        [TestMethod]
        public void MoveAfterMerge()
        {
            OpenCreatorWindow("Columns", "Custom table layout creator", "EditTemplateButton");
            WindowsElement gridEditor = session.FindElementByClassName("GridEditor");
            Assert.IsNotNull(gridEditor);

            ReadOnlyCollection<AppiumWebElement> thumbs = gridEditor.FindElementsByClassName("Thumb");

            //create new zones
            new Actions(session).MoveToElement(thumbs[0]).Click().MoveByOffset(-(thumbs[0].Rect.X / 2), 0)
                .KeyDown(OpenQA.Selenium.Keys.Shift).Click().KeyUp(OpenQA.Selenium.Keys.Shift)
                .Perform();
            thumbs = gridEditor.FindElementsByClassName("Thumb");

            //merge zones
            ReadOnlyCollection<WindowsElement> zones = session.FindElementsByClassName("GridZone");
            Move(zones[0], thumbs[0].Rect.X + thumbs[0].Rect.Width + 10, true, true, -(zones[0].Rect.Height / 2) + 10);
            WindowsElement mergeButton = session.FindElementByName("Merge zones");
            Assert.IsNotNull(mergeButton, "Cannot merge: no merge button");
            new Actions(session).Click(mergeButton).Perform();

            //move thumb
            AppiumWebElement thumb = thumbs[1]; //thumb from merged zone is still present 
            Move(thumb, 0, false, true); 
            Assert.IsTrue(thumb.Rect.Left <= moveStep);
            Assert.IsTrue(thumb.Rect.Right > 0);
        }

        [ClassInitialize]
        public static void ClassInitialize(TestContext context)
        {
            Setup(context, false);
            ResetSettings();

            if (!isPowerToysLaunched)
            {
                LaunchPowerToys();
            }
            OpenEditor();
            OpenTemplates();
        }

        [ClassCleanup]
        public static void ClassCleanup()
        {
            CloseEditor();
            TearDown();
        }

        [TestInitialize]
        public void TestInitialize()
        {

        }

        [TestCleanup]
        public void TestCleanup()
        {
            WindowsElement cancelButton = session.FindElementByXPath("//Window[@Name=\"FancyZones Editor\"]/Window/Button[@Name=\"Cancel\"]");
            Assert.IsNotNull(cancelButton);
            new Actions(session).MoveToElement(cancelButton).Click().Perform();
        }
    }
}