PixiEditor/PixiEditorTests/ModelsTests/ControllersTests/ClipboardControllerTests.cs

using System.IO;
using System.Linq;
using System.Windows;
using System.Windows.Media;
using System.Windows.Media.Imaging;
using PixiEditor.Models.Controllers;
using PixiEditor.Models.Layers;
using PixiEditor.Models.Position;
using Xunit;

namespace PixiEditorTests.ModelsTests.ControllersTests
{
    public class ClipboardControllerTests
    {
        private const string Text = "Text data";
        private readonly Color testColor = Colors.Coral;

        [StaFact]
        public void TestThatClipboardControllerIgnoresNonImageDataInClipboard()
        {
            Clipboard.Clear();
            Clipboard.SetText(Text);
            WriteableBitmap img = ClipboardController.GetImageFromClipboard();
            Assert.Null(img);
        }

        [StaFact]
        public void TestThatIsImageInClipboardWorksForDib()
        {
            Clipboard.Clear();
            Clipboard.SetImage(BitmapFactory.New(10, 10));
            Assert.True(ClipboardController.IsImageInClipboard());
        }

        [StaFact]
        public void TestThatClipboardControllerSavesImageToClipboard()
        {
            Layer testLayer = new Layer("test layer", 10, 10);
            ClipboardController.CopyToClipboard(new[] { testLayer }, CoordinatesCalculator.RectangleToCoordinates(0, 0, 9, 9).ToArray(), 10, 10);
            Assert.True(ClipboardController.IsImageInClipboard());
        }

        [StaFact]
        public void TestThatCopyToClipboardWithSelectionSavesCorrectBitmap()
        {
            Clipboard.Clear();

            Layer testLayer = new Layer("test layer", 10, 10);
            Layer testLayer2 = new Layer("test layer", 10, 10);
            testLayer.SetPixel(new Coordinates(4, 4), testColor);
            testLayer2.SetPixel(new Coordinates(5, 5), testColor);

            ClipboardController.CopyToClipboard(
                new[] { testLayer, testLayer2 },
                new[] { new Coordinates(4, 4), new Coordinates(5, 5) },
                10,
                10);

            BitmapSource img = Clipboard.GetImage(); // Using default Clipboard get image to avoid false positives from faulty ClipboardController GetImage

            Assert.True(ClipboardController.IsImageInClipboard());
            Assert.NotNull(img);
            Assert.Equal(2, img.PixelWidth);
            Assert.Equal(2, img.PixelHeight);

            WriteableBitmap bmp = new WriteableBitmap(img);
            Assert.Equal(testColor, bmp.GetPixel(0, 0));
            Assert.Equal(testColor, bmp.GetPixel(1, 1));
        }

        [StaFact]
        public void TestThatClipboardControllerGetsCorrectImageInDibFormatFromClipboard()
        {
            Clipboard.Clear();
            WriteableBitmap bmp = BitmapFactory.New(10, 10);
            bmp.SetPixel(4, 4, testColor);
            Clipboard.SetImage(bmp);

            WriteableBitmap img = ClipboardController.GetImageFromClipboard();
            Assert.NotNull(img);
            Assert.Equal(10, img.PixelWidth);
            Assert.Equal(10, img.PixelHeight);
            Assert.Equal(testColor, bmp.GetPixel(4, 4));
        }

        [StaFact]
        public void TestThatClipboardControllerGetsCorrectImageInPngFormatFromClipboard()
        {
            Clipboard.Clear();
            WriteableBitmap bmp = BitmapFactory.New(10, 10);
            bmp.SetPixel(4, 4, testColor);
            using (MemoryStream pngStream = new MemoryStream())
            {
                DataObject data = new DataObject();

                PngBitmapEncoder encoder = new PngBitmapEncoder();
                encoder.Frames.Add(BitmapFrame.Create(bmp));
                encoder.Save(pngStream);
                data.SetData("PNG", pngStream, false); // PNG, supports transparency
                Clipboard.SetDataObject(data, true);
            }

            WriteableBitmap img = ClipboardController.GetImageFromClipboard();
            Assert.NotNull(img);
            Assert.Equal(10, img.PixelWidth);
            Assert.Equal(10, img.PixelHeight);
            Assert.Equal(testColor, bmp.GetPixel(4, 4));
        }

        [StaFact]
        public void TestThatClipboardControllerGetsCorrectImageInBitmapFormatFromClipboard()
        {
            Clipboard.Clear();
            WriteableBitmap bmp = BitmapFactory.New(10, 10);
            bmp.SetPixel(4, 4, testColor);

            DataObject data = new DataObject();
            data.SetData(DataFormats.Bitmap, bmp, false); // PNG, supports transparency
            Clipboard.SetDataObject(data, true);

            WriteableBitmap img = ClipboardController.GetImageFromClipboard();
            Assert.NotNull(img);
            Assert.Equal(10, img.PixelWidth);
            Assert.Equal(10, img.PixelHeight);
            Assert.Equal(testColor, bmp.GetPixel(4, 4));
        }
    }
}