ET9_WebGL_Dev/CSharp/Test/UNetTest/UServiceTest.cs

using System;
using System.Threading;
using System.Threading.Tasks;
using Common.Helper;
using Common.Network;
using NUnit.Framework;
using UNet;

namespace UNetTest
{
	[TestFixture]
	public class UServiceTest
	{
		private const int echoTimes = 10000;
		private readonly Barrier barrier = new Barrier(2);

		private bool isClientStop;
		private bool isServerStop;

		private async void ClientEvent(IService clientService, string hostName, ushort port)
		{
			AChannel channel = await clientService.GetChannel(hostName, port);
			for (int i = 0; i < echoTimes; ++i)
			{
				channel.SendAsync("0123456789".ToByteArray());

				byte[] bytes = await channel.RecvAsync();
				CollectionAssert.AreEqual("9876543210".ToByteArray(), bytes);
			}
			this.barrier.RemoveParticipant();
		}

		private async void ServerEvent(IService service)
		{
			AChannel channel = await service.GetChannel();
			for (int i = 0; i < echoTimes; ++i)
			{
				byte[] bytes = await channel.RecvAsync();
				CollectionAssert.AreEqual("0123456789".ToByteArray(), bytes);
				Array.Reverse(bytes);
				channel.SendAsync(bytes);
			}
		}

		[Test]
		public void ClientSendToServer()
		{
			const string hostName = "127.0.0.1";
			const ushort port = 8889;
			using (IService clientService = new UService(hostName, 8888))
			{
				using (IService serverService = new UService(hostName, 8889))
				{
					Task task1 = Task.Factory.StartNew(() =>
					{
						while (!this.isClientStop)
						{
							clientService.Update();
						}
					}, TaskCreationOptions.LongRunning);

					Task task2 = Task.Factory.StartNew(() =>
					{
						while (!this.isServerStop)
						{
							serverService.Update();
						}
					}, TaskCreationOptions.LongRunning);

					// 往server host线程增加事件,accept
					serverService.Add(() => this.ServerEvent(serverService));

					Thread.Sleep(1000);

					// 往client host线程增加事件,client线程连接server
					clientService.Add(() => this.ClientEvent(clientService, hostName, port));
					this.barrier.SignalAndWait();

					serverService.Add(() => { this.isServerStop = true; });
					clientService.Add(() => { this.isClientStop = true; });
					Task.WaitAll(task1, task2);
				}
			}
		}
	}
}