Avalonia/tests/Avalonia.Visuals.UnitTests/TestVisual.cs

// Copyright (c) The Avalonia Project. All rights reserved.
// Licensed under the MIT license. See licence.md file in the project root for full license information.

using System;
using System.Collections.Generic;
using System.Linq;
using Avalonia.Rendering;
using Avalonia.VisualTree;

namespace Avalonia.Visuals.UnitTests
{
    public class ParamEventArgs<T> : EventArgs
    {
        public ParamEventArgs(T param)
        {
            Param = param;
        }

        public T Param { get; set; }
    }

    public class TestVisual : Visual
    {
        public IVisual Child
        {
            get
            {
                return ((IVisual)this).VisualChildren.FirstOrDefault();
            }

            set
            {
                if (Child != null)
                {
                    VisualChildren.Remove(Child);
                }

                if (value != null)
                {
                    VisualChildren.Add(value);
                }
            }
        }

        public void AddChild(Visual v)
        {
            VisualChildren.Add(v);
        }

        public void AddChildren(IEnumerable<Visual> v)
        {
            VisualChildren.AddRange(v);
        }

        public void RemoveChild(Visual v)
        {
            VisualChildren.Remove(v);
        }

        public void ClearChildren()
        {
            VisualChildren.Clear();
        }
    }
}