using NUnit.Framework;
using UnityEngine;
using UnityEditor.Graphing;
using UnityEditor.ShaderGraph;

namespace UnityEditor.ShaderGraph.UnitTests
{
    [TestFixture]
    public class FixedSlotTests
    {
        private ShaderGraph.MaterialGraph m_Graph;
        private TestNode m_NodeA;
        private TestNode m_NodeB;

        class TestNode : AbstractMaterialNode
        {
            public const int V1Out = 0;
            public const int V2Out = 1;
            public const int V3Out = 2;
            public const int V4Out = 3;


            public const int V1In = 4;
            public const int V2In = 5;
            public const int V3In = 6;
            public const int V4In = 7;

            public TestNode()
            {
                AddSlot(new Vector1MaterialSlot(V1Out, "V1Out", "V1Out", SlotType.Output, 0));
                AddSlot(new Vector2MaterialSlot(V2Out, "V2Out", "V2Out", SlotType.Output, Vector4.zero));
                AddSlot(new Vector3MaterialSlot(V3Out, "V3Out", "V3Out", SlotType.Output, Vector4.zero));
                AddSlot(new Vector4MaterialSlot(V4Out, "V4Out", "V4Out", SlotType.Output, Vector4.zero));

                AddSlot(new Vector1MaterialSlot(V1In, "V1In", "V1In", SlotType.Input, 0));
                AddSlot(new Vector2MaterialSlot(V2In, "V2In", "V2In", SlotType.Input, Vector4.zero));
                AddSlot(new Vector3MaterialSlot(V3In, "V3In", "V3In", SlotType.Input, Vector4.zero));
                AddSlot(new Vector4MaterialSlot(V4In, "V4In", "V4In", SlotType.Input, Vector4.zero));
            }
        }

        [OneTimeSetUp]
        public void RunBeforeAnyTests()
        {
            Debug.unityLogger.logHandler = new ConsoleLogHandler();
        }

        [SetUp]
        public void TestSetUp()
        {
            m_Graph = new ShaderGraph.MaterialGraph();
            m_NodeA = new TestNode();
            m_NodeB = new TestNode();
            m_Graph.AddNode(m_NodeA);
            m_Graph.AddNode(m_NodeB);
        }

        [Test]
        public void ConnectV1ToV1Works()
        {
            m_Graph.Connect(m_NodeA.GetSlotReference(TestNode.V1Out), m_NodeB.GetSlotReference(TestNode.V1In));
            Assert.IsFalse(m_NodeB.hasError);
        }

        [Test]
        public void ConnectV1ToV2Works()
        {
            m_Graph.Connect(m_NodeA.GetSlotReference(TestNode.V1Out), m_NodeB.GetSlotReference(TestNode.V2In));
            Assert.IsFalse(m_NodeB.hasError);
        }

        [Test]
        public void ConnectV1ToV3Works()
        {
            m_Graph.Connect(m_NodeA.GetSlotReference(TestNode.V1Out), m_NodeB.GetSlotReference(TestNode.V3In));
            Assert.IsFalse(m_NodeB.hasError);
        }

        [Test]
        public void ConnectV1ToV4Works()
        {
            m_Graph.Connect(m_NodeA.GetSlotReference(TestNode.V1Out), m_NodeB.GetSlotReference(TestNode.V4In));
            Assert.IsFalse(m_NodeB.hasError);
        }

        [Test]
        public void ConnectV2ToV1Works()
        {
            m_Graph.Connect(m_NodeA.GetSlotReference(TestNode.V2Out), m_NodeB.GetSlotReference(TestNode.V1In));
            Assert.IsFalse(m_NodeB.hasError);
        }

        [Test]
        public void ConnectV2ToV2Works()
        {
            m_Graph.Connect(m_NodeA.GetSlotReference(TestNode.V2Out), m_NodeB.GetSlotReference(TestNode.V2In));
            Assert.IsFalse(m_NodeB.hasError);
        }

        [Test]
        public void ConnectV2ToV3Fails()
        {
            m_Graph.Connect(m_NodeA.GetSlotReference(TestNode.V2Out), m_NodeB.GetSlotReference(TestNode.V3In));
            Assert.IsTrue(m_NodeB.hasError);
        }

        [Test]
        public void ConnectV2ToV4Fails()
        {
            m_Graph.Connect(m_NodeA.GetSlotReference(TestNode.V2Out), m_NodeB.GetSlotReference(TestNode.V4In));
            Assert.IsTrue(m_NodeB.hasError);
        }

        [Test]
        public void ConnectV3ToV1Works()
        {
            m_Graph.Connect(m_NodeA.GetSlotReference(TestNode.V3Out), m_NodeB.GetSlotReference(TestNode.V1In));
            Assert.IsFalse(m_NodeB.hasError);
        }

        [Test]
        public void ConnectV3ToV2Works()
        {
            m_Graph.Connect(m_NodeA.GetSlotReference(TestNode.V3Out), m_NodeB.GetSlotReference(TestNode.V2In));
            Assert.IsFalse(m_NodeB.hasError);
        }

        [Test]
        public void ConnectV3ToV3Works()
        {
            m_Graph.Connect(m_NodeA.GetSlotReference(TestNode.V3Out), m_NodeB.GetSlotReference(TestNode.V3In));
            Assert.IsFalse(m_NodeB.hasError);
        }

        [Test]
        public void ConnectV3ToV4Fails()
        {
            m_Graph.Connect(m_NodeA.GetSlotReference(TestNode.V3Out), m_NodeB.GetSlotReference(TestNode.V4In));
            Assert.IsTrue(m_NodeB.hasError);
        }

        [Test]
        public void ConnectV4ToV1Works()
        {
            m_Graph.Connect(m_NodeA.GetSlotReference(TestNode.V4Out), m_NodeB.GetSlotReference(TestNode.V1In));
            Assert.IsFalse(m_NodeB.hasError);
        }

        [Test]
        public void ConnectV4ToV2Works()
        {
            m_Graph.Connect(m_NodeA.GetSlotReference(TestNode.V4Out), m_NodeB.GetSlotReference(TestNode.V2In));
            Assert.IsFalse(m_NodeB.hasError);
        }

        [Test]
        public void ConnectV4ToV3Works()
        {
            m_Graph.Connect(m_NodeA.GetSlotReference(TestNode.V4Out), m_NodeB.GetSlotReference(TestNode.V3In));
            Assert.IsFalse(m_NodeB.hasError);
        }

        [Test]
        public void ConnectV4ToV4Works()
        {
            m_Graph.Connect(m_NodeA.GetSlotReference(TestNode.V4Out), m_NodeB.GetSlotReference(TestNode.V4In));
            Assert.IsFalse(m_NodeB.hasError);
        }
    }
}