using Microsoft.VisualStudio.TestTools.UnitTesting;
using SystemExtensions.Collections;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace SystemExtensions.Collections.Tests
{
    [TestClass()]
    public class AVLTreeTests
    {
        private static int IntegerComparison(int x, int y)
        {
            if (x == y)
            {
                return 0;
            }
            else if (x > y)
            {
                return 1;
            }
            else
            {
                return -1;
            }
        }
        private AVLTree<int> avlTree = new AVLTree<int>(new Comparison<int>(IntegerComparison));
        [TestMethod()]
        public void AVLTreeTest()
        {
            avlTree = new AVLTree<int>(IntegerComparison);
        }

        [TestMethod()]
        public void AddTest()
        {
            try
            {
                for (int i = 0; i < 100; i++)
                {
                    avlTree.Add(i * 20 - (50 + i));
                }
            }
            catch (Exception e)
            {
                Assert.Fail(e.Message + "\n" + e.StackTrace);
            }
            if (avlTree.Count != 100)
            {
                Assert.Fail();
            }
        }

        [TestMethod()]
        public void ContainsTest()
        {
            for (int i = 0; i < 100; i++)
            {
                avlTree.Add(i);
            }

            if (!avlTree.Contains(50))
            {
                Assert.Fail();
            }
        }

        [TestMethod()]
        public void RemoveTest()
        {
            for (int i = 0; i < 100; i++)
            {
                avlTree.Add(i);
            }

            avlTree.Remove(50);

            if (avlTree.Contains(50))
            {
                Assert.Fail();
            }
        }

        [TestMethod()]
        public void ClearTest()
        {
            for (int i = 0; i < 100; i++)
            {
                avlTree.Add(i);
            }
            avlTree.Clear();
            if(avlTree.Count > 0)
            {
                Assert.Fail();
            }
        }

        [TestMethod()]
        public void CopyToTest()
        {
            for (int i = 0; i < 100; i++)
            {
                avlTree.Add(i);
            }

            int[] array = new int[100];
            avlTree.CopyTo(array, 0);
        }

        [TestMethod()]
        public void GetEnumeratorTest()
        {
            for (int i = 0; i < 100; i++)
            {
                avlTree.Add(i);
            }
            int count = 0;
            foreach(int value in avlTree)
            {
                count++;
                System.Diagnostics.Debug.WriteLine(value);
            }

            if(count != avlTree.Count)
            {
                Assert.Fail();
            }
        }

        [TestMethod()]
        public void ToArrayTest()
        {
            for (int i = 0; i < 100; i++)
            {
                avlTree.Add(i);
            }

            int[] array = avlTree.ToArray();
        }
    }
}