using System;
using System.Net.Sockets;
using System.Threading;
using Common.Helper;
using Common.Logger;
using Microsoft.VisualStudio.TestTools.UnitTesting;
using TNet;

namespace TNetTest
{
    [TestClass]
    public class TcpAcceptorTest
    {
        private const ushort port = 11111;
        private int count;
        private readonly Barrier barrier = new Barrier(2);
        private readonly object lockObject = new object();

        [TestMethod]
        public void AcceptAsync()
        {
            var thread1 = new Thread(this.Server);
            thread1.Start();

            Thread.Sleep(2);

            for (int i = 0; i < 99; ++i)
            {
                var thread = new Thread(this.Client);
                thread.Start();
            }
            this.barrier.SignalAndWait();
        }

        private async void Client()
        {
            using (var tcpClient = new TcpClient(AddressFamily.InterNetwork))
            {
                await tcpClient.ConnectAsync("127.0.0.1", port);
                using (NetworkStream ns = tcpClient.GetStream())
                {
                    try
                    {
                        var bytes = "tanghai".ToByteArray();
                        for (int i = 0; i < 100000; ++i)
                        {
                            await ns.WriteAsync(bytes, 0, bytes.Length);
                            int n = await ns.ReadAsync(bytes, 0, bytes.Length);
                            Assert.AreEqual(7, n);
                        }
                    }
                    catch (Exception e)
                    {
                        Log.Debug(e.ToString());
                    }
                }
            }
            this.barrier.RemoveParticipants(1);
        }

        private async void Server()
        {
            using (var tcpAcceptor = new TcpAcceptor("127.0.0.1", port))
            {
                while (this.count != 99)
                {
                    Log.Debug("start server response");
                    NetworkStream ns = await tcpAcceptor.AcceptAsync();
                    // 这里可能已经不在Server函数线程了
                    Log.Debug("server response");
                    this.Response(ns);
                }
            }
        }

        private async void Response(NetworkStream ns)
        {
            try
            {
                var bytes = new byte[1000];
                for (int i = 0; i < 100000; ++i)
                {
                    int n = await ns.ReadAsync(bytes, 0, 100);
                    await ns.WriteAsync(bytes, 0, n);
                }
                lock (this.lockObject)
                {
                    ++this.count;
                }
            }
            catch (Exception e)
            {
                Log.Debug(e.ToString());
            }
        }
    }
}