NatTypeTester/UnitTest/StunClien5389UDPTest.cs

using Dns.Net.Abstractions;
using Dns.Net.Clients;
using Microsoft.VisualStudio.TestTools.UnitTesting;
using Moq;
using STUN.Client;
using STUN.Enums;
using STUN.Messages;
using STUN.StunResult;
using System.Net;
using System.Threading;
using System.Threading.Tasks;

namespace UnitTest;

[TestClass]
public class StunClien5389UDPTest
{
	private readonly IDnsClient _dnsClient = new DefaultDnsClient();

	private const string Server = @"stun.syncthing.net";
	private const ushort Port = 3478;

	private static readonly IPEndPoint Any = new(IPAddress.Any, 0);
	private static readonly IPEndPoint LocalAddress1 = IPEndPoint.Parse(@"127.0.0.1:114");
	private static readonly IPEndPoint MappedAddress1 = IPEndPoint.Parse(@"1.1.1.1:114");
	private static readonly IPEndPoint MappedAddress2 = IPEndPoint.Parse(@"1.1.1.1:514");
	private static readonly IPEndPoint ServerAddress = IPEndPoint.Parse(@"2.2.2.2:1919");
	private static readonly IPEndPoint ChangedAddress1 = IPEndPoint.Parse(@"3.3.3.3:23333");
	private static readonly IPEndPoint ChangedAddress2 = IPEndPoint.Parse(@"2.2.2.2:810");
	private static readonly IPEndPoint ChangedAddress3 = IPEndPoint.Parse(@"3.3.3.3:1919");

	private static readonly StunMessage5389 DefaultStunMessage = new();

	[TestMethod]
	public async Task BindingTestSuccessAsync()
	{
		var ip = await _dnsClient.QueryAsync(Server);
		using var client = new StunClient5389UDP(new IPEndPoint(ip, Port), Any);

		var response = await client.BindingTestAsync();

		Assert.AreEqual(BindingTestResult.Success, response.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Unknown, response.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.Unknown, response.FilteringBehavior);
		Assert.IsNotNull(response.PublicEndPoint);
		Assert.IsNotNull(response.LocalEndPoint);
		Assert.IsNotNull(response.OtherEndPoint);
	}

	[TestMethod]
	public async Task BindingTestFailAsync()
	{
		var ip = IPAddress.Parse(@"1.1.1.1");
		using var client = new StunClient5389UDP(new IPEndPoint(ip, Port), Any);

		var response = await client.BindingTestAsync();

		Assert.AreEqual(BindingTestResult.Fail, response.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Unknown, response.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.Unknown, response.FilteringBehavior);
		Assert.IsNull(response.PublicEndPoint);
		Assert.IsNull(response.LocalEndPoint);
		Assert.IsNull(response.OtherEndPoint);
	}

	[TestMethod]
	public async Task MappingBehaviorTestFailAsync()
	{
		var mock = new Mock<StunClient5389UDP>(ServerAddress, Any, default);
		var client = mock.Object;

		var fail = new StunResult5389 { BindingTestResult = BindingTestResult.Fail };

		mock.Setup(x => x.BindingTestBaseAsync(It.IsAny<IPEndPoint>(), It.IsAny<CancellationToken>())).ReturnsAsync(fail);

		await client.MappingBehaviorTestAsync();

		Assert.AreEqual(BindingTestResult.Fail, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Unknown, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.Unknown, client.State.FilteringBehavior);
		Assert.IsNull(client.State.PublicEndPoint);
		Assert.IsNull(client.State.LocalEndPoint);
		Assert.IsNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task MappingBehaviorTestUnsupportedServerAsync()
	{
		var mock = new Mock<StunClient5389UDP>(ServerAddress, Any, default);
		var client = mock.Object;

		var r1 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1
		};
		mock.Setup(x => x.BindingTestBaseAsync(It.IsAny<IPEndPoint>(), It.IsAny<CancellationToken>())).ReturnsAsync(r1);
		await TestAsync();

		var r2 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress2
		};
		mock.Setup(x => x.BindingTestBaseAsync(It.IsAny<IPEndPoint>(), It.IsAny<CancellationToken>())).ReturnsAsync(r2);
		await TestAsync();

		var r3 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress3
		};
		mock.Setup(x => x.BindingTestBaseAsync(It.IsAny<IPEndPoint>(), It.IsAny<CancellationToken>())).ReturnsAsync(r3);
		await TestAsync();

		async Task TestAsync()
		{
			await client.MappingBehaviorTestAsync();

			Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
			Assert.AreEqual(MappingBehavior.UnsupportedServer, client.State.MappingBehavior);
			Assert.AreEqual(FilteringBehavior.Unknown, client.State.FilteringBehavior);
			Assert.IsNotNull(client.State.PublicEndPoint);
			Assert.IsNotNull(client.State.LocalEndPoint);
		}
	}

	[TestMethod]
	public async Task MappingBehaviorTestDirectAsync()
	{
		var mock = new Mock<StunClient5389UDP>(ServerAddress, Any, default);
		var client = mock.Object;

		var response = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = MappedAddress1,
			OtherEndPoint = ChangedAddress1
		};

		mock.Setup(x => x.BindingTestBaseAsync(It.IsAny<IPEndPoint>(), It.IsAny<CancellationToken>())).ReturnsAsync(response);

		await client.MappingBehaviorTestAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Direct, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.Unknown, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
		Assert.IsNotNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task MappingBehaviorTestEndpointIndependentAsync()
	{
		var mock = new Mock<StunClient5389UDP>(ServerAddress, Any, default);
		var client = mock.Object;

		var r1 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		mock.Setup(x => x.BindingTestBaseAsync(It.IsAny<IPEndPoint>(), It.IsAny<CancellationToken>())).ReturnsAsync(r1);
		await client.MappingBehaviorTestAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.EndpointIndependent, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.Unknown, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
		Assert.IsNotNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task MappingBehaviorTest2FailAsync()
	{
		var mock = new Mock<StunClient5389UDP>(ServerAddress, Any, default);
		var client = mock.Object;

		var r1 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		var r2 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Fail,
		};

		mock.Setup(x => x.BindingTestBaseAsync(It.Is<IPEndPoint>(p => Equals(p, ServerAddress)), It.IsAny<CancellationToken>())).ReturnsAsync(r1);
		mock.Setup(x => x.BindingTestBaseAsync(It.Is<IPEndPoint>(p => Equals(p, ChangedAddress3)), It.IsAny<CancellationToken>())).ReturnsAsync(r2);
		await client.MappingBehaviorTestAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Fail, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.Unknown, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
		Assert.IsNotNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task MappingBehaviorTestAddressDependentAsync()
	{
		var mock = new Mock<StunClient5389UDP>(ServerAddress, Any, default);
		var client = mock.Object;

		var r1 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		var r2 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress2,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		var r3 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress2,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		mock.Setup(x => x.BindingTestBaseAsync(It.Is<IPEndPoint>(p => Equals(p, ServerAddress)), It.IsAny<CancellationToken>())).ReturnsAsync(r1);
		mock.Setup(x => x.BindingTestBaseAsync(It.Is<IPEndPoint>(p => Equals(p, ChangedAddress3)), It.IsAny<CancellationToken>())).ReturnsAsync(r2);
		mock.Setup(x => x.BindingTestBaseAsync(It.Is<IPEndPoint>(p => Equals(p, ChangedAddress1)), It.IsAny<CancellationToken>())).ReturnsAsync(r3);

		await client.MappingBehaviorTestAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.AddressDependent, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.Unknown, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
		Assert.IsNotNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task MappingBehaviorTestAddressAndPortDependentAsync()
	{
		var mock = new Mock<StunClient5389UDP>(ServerAddress, Any, default);
		var client = mock.Object;

		var r1 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		var r2 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress2,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		var r3 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		mock.Setup(x => x.BindingTestBaseAsync(It.Is<IPEndPoint>(p => Equals(p, ServerAddress)), It.IsAny<CancellationToken>())).ReturnsAsync(r1);
		mock.Setup(x => x.BindingTestBaseAsync(It.Is<IPEndPoint>(p => Equals(p, ChangedAddress3)), It.IsAny<CancellationToken>())).ReturnsAsync(r2);
		mock.Setup(x => x.BindingTestBaseAsync(It.Is<IPEndPoint>(p => Equals(p, ChangedAddress1)), It.IsAny<CancellationToken>())).ReturnsAsync(r3);

		await client.MappingBehaviorTestAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.AddressAndPortDependent, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.Unknown, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
		Assert.IsNotNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task MappingBehaviorTest3FailAsync()
	{
		var mock = new Mock<StunClient5389UDP>(ServerAddress, Any, default);
		var client = mock.Object;

		var r1 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		var r2 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress2,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		var r3 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Fail
		};
		mock.Setup(x => x.BindingTestBaseAsync(It.Is<IPEndPoint>(p => Equals(p, ServerAddress)), It.IsAny<CancellationToken>())).ReturnsAsync(r1);
		mock.Setup(x => x.BindingTestBaseAsync(It.Is<IPEndPoint>(p => Equals(p, ChangedAddress3)), It.IsAny<CancellationToken>())).ReturnsAsync(r2);
		mock.Setup(x => x.BindingTestBaseAsync(It.Is<IPEndPoint>(p => Equals(p, ChangedAddress1)), It.IsAny<CancellationToken>())).ReturnsAsync(r3);

		await client.MappingBehaviorTestAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Fail, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.Unknown, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
		Assert.IsNotNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task FilteringBehaviorTestFailAsync()
	{
		var mock = new Mock<StunClient5389UDP>(ServerAddress, Any, default);
		var client = mock.Object;

		var fail = new StunResult5389 { BindingTestResult = BindingTestResult.Fail };

		mock.Setup(x => x.BindingTestBaseAsync(It.IsAny<IPEndPoint>(), It.IsAny<CancellationToken>())).ReturnsAsync(fail);

		await client.FilteringBehaviorTestAsync();

		Assert.AreEqual(BindingTestResult.Fail, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Unknown, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.Unknown, client.State.FilteringBehavior);
		Assert.IsNull(client.State.PublicEndPoint);
		Assert.IsNull(client.State.LocalEndPoint);
		Assert.IsNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task FilteringBehaviorTestUnsupportedServerAsync()
	{
		var mock = new Mock<StunClient5389UDP>(ServerAddress, Any, default);
		var client = mock.Object;

		var r1 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1
		};
		mock.Setup(x => x.BindingTestBaseAsync(It.IsAny<IPEndPoint>(), It.IsAny<CancellationToken>())).ReturnsAsync(r1);
		await TestAsync();

		var r2 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress2
		};
		mock.Setup(x => x.BindingTestBaseAsync(It.IsAny<IPEndPoint>(), It.IsAny<CancellationToken>())).ReturnsAsync(r2);
		await TestAsync();

		var r3 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress3
		};
		mock.Setup(x => x.BindingTestBaseAsync(It.IsAny<IPEndPoint>(), It.IsAny<CancellationToken>())).ReturnsAsync(r3);
		await TestAsync();

		async Task TestAsync()
		{
			await client.FilteringBehaviorTestAsync();

			Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
			Assert.AreEqual(MappingBehavior.Unknown, client.State.MappingBehavior);
			Assert.AreEqual(FilteringBehavior.UnsupportedServer, client.State.FilteringBehavior);
			Assert.IsNotNull(client.State.PublicEndPoint);
			Assert.IsNotNull(client.State.LocalEndPoint);
		}
	}

	[TestMethod]
	public async Task FilteringBehaviorTestEndpointIndependentAsync()
	{
		var mock = new Mock<StunClient5389UDP>(ServerAddress, Any, default);
		var client = mock.Object;

		var r1 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		var r2 = new StunResponse(DefaultStunMessage, ChangedAddress1, LocalAddress1.Address);
		mock.Setup(x => x.BindingTestBaseAsync(It.IsAny<IPEndPoint>(), It.IsAny<CancellationToken>())).ReturnsAsync(r1);
		mock.Setup(x => x.FilteringBehaviorTest2Async(It.IsAny<CancellationToken>())).ReturnsAsync(r2);

		await client.FilteringBehaviorTestAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Unknown, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.EndpointIndependent, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
		Assert.IsNotNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task FilteringBehaviorTest2UnsupportedServerAsync()
	{
		var mock = new Mock<StunClient5389UDP>(ServerAddress, Any, default);
		var client = mock.Object;

		var r1 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		var r2 = new StunResponse(DefaultStunMessage, ServerAddress, LocalAddress1.Address);
		mock.Setup(x => x.BindingTestBaseAsync(It.IsAny<IPEndPoint>(), It.IsAny<CancellationToken>())).ReturnsAsync(r1);
		mock.Setup(x => x.FilteringBehaviorTest2Async(It.IsAny<CancellationToken>())).ReturnsAsync(r2);

		await client.FilteringBehaviorTestAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Unknown, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.UnsupportedServer, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
		Assert.IsNotNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task FilteringBehaviorTestAddressAndPortDependentAsync()
	{
		var mock = new Mock<StunClient5389UDP>(ServerAddress, Any, default);
		var client = mock.Object;

		var r1 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		mock.Setup(x => x.BindingTestBaseAsync(It.IsAny<IPEndPoint>(), It.IsAny<CancellationToken>())).ReturnsAsync(r1);
		mock.Setup(x => x.FilteringBehaviorTest2Async(It.IsAny<CancellationToken>())).ReturnsAsync(default(StunResponse?));
		mock.Setup(x => x.FilteringBehaviorTest3Async(It.IsAny<CancellationToken>())).ReturnsAsync(default(StunResponse?));

		await client.FilteringBehaviorTestAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Unknown, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.AddressAndPortDependent, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
		Assert.IsNotNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task FilteringBehaviorTestAddressDependentAsync()
	{
		var mock = new Mock<StunClient5389UDP>(ServerAddress, Any, default);
		var client = mock.Object;

		var r1 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		var r3 = new StunResponse(DefaultStunMessage, ChangedAddress2, LocalAddress1.Address);
		mock.Setup(x => x.BindingTestBaseAsync(It.IsAny<IPEndPoint>(), It.IsAny<CancellationToken>())).ReturnsAsync(r1);
		mock.Setup(x => x.FilteringBehaviorTest2Async(It.IsAny<CancellationToken>())).ReturnsAsync(default(StunResponse?));
		mock.Setup(x => x.FilteringBehaviorTest3Async(It.IsAny<CancellationToken>())).ReturnsAsync(r3);

		await client.FilteringBehaviorTestAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Unknown, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.AddressDependent, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
		Assert.IsNotNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task FilteringBehaviorTest3UnsupportedServerAsync()
	{
		var mock = new Mock<StunClient5389UDP>(ServerAddress, Any, default);
		var client = mock.Object;

		var r1 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		var r3 = new StunResponse(DefaultStunMessage, ServerAddress, LocalAddress1.Address);
		mock.Setup(x => x.BindingTestBaseAsync(It.IsAny<IPEndPoint>(), It.IsAny<CancellationToken>())).ReturnsAsync(r1);
		mock.Setup(x => x.FilteringBehaviorTest2Async(It.IsAny<CancellationToken>())).ReturnsAsync(default(StunResponse?));
		mock.Setup(x => x.FilteringBehaviorTest3Async(It.IsAny<CancellationToken>())).ReturnsAsync(r3);

		await client.FilteringBehaviorTestAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Unknown, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.UnsupportedServer, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
		Assert.IsNotNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task QueryFailTestAsync()
	{
		var mock = new Mock<StunClient5389UDP>(ServerAddress, Any, default);
		var client = mock.Object;

		var fail = new StunResult5389 { BindingTestResult = BindingTestResult.Fail };

		mock.Setup(x => x.BindingTestBaseAsync(It.IsAny<IPEndPoint>(), It.IsAny<CancellationToken>())).ReturnsAsync(fail);

		await client.QueryAsync();

		Assert.AreEqual(BindingTestResult.Fail, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Unknown, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.Unknown, client.State.FilteringBehavior);
		Assert.IsNull(client.State.PublicEndPoint);
		Assert.IsNull(client.State.LocalEndPoint);
		Assert.IsNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task QueryUnsupportedServerTestAsync()
	{
		var mock = new Mock<StunClient5389UDP>(ServerAddress, Any, default);
		var client = mock.Object;

		var r1 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ServerAddress
		};
		mock.Setup(x => x.BindingTestBaseAsync(It.IsAny<IPEndPoint>(), It.IsAny<CancellationToken>())).ReturnsAsync(r1);

		await client.QueryAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Unknown, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.UnsupportedServer, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
	}

	[TestMethod]
	public async Task QueryMappingBehaviorDirectTestAsync()
	{
		var mock = new Mock<StunClient5389UDP>(ServerAddress, Any, default);
		var client = mock.Object;

		var r1 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = MappedAddress1,
			OtherEndPoint = ChangedAddress1
		};
		mock.Setup(x => x.BindingTestBaseAsync(It.Is<IPEndPoint>(p => Equals(p, ServerAddress)), It.IsAny<CancellationToken>())).ReturnsAsync(r1);
		mock.Setup(x => x.FilteringBehaviorTest2Async(It.IsAny<CancellationToken>())).ReturnsAsync(default(StunResponse?));
		mock.Setup(x => x.FilteringBehaviorTest3Async(It.IsAny<CancellationToken>())).ReturnsAsync(default(StunResponse?));

		await client.QueryAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.Direct, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.AddressAndPortDependent, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
		Assert.IsNotNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task QueryMappingBehaviorEndpointIndependentTestAsync()
	{
		var mock = new Mock<StunClient5389UDP>(ServerAddress, Any, default);
		var client = mock.Object;

		var r1 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		mock.Setup(x => x.BindingTestBaseAsync(It.IsAny<IPEndPoint>(), It.IsAny<CancellationToken>())).ReturnsAsync(r1);
		mock.Setup(x => x.FilteringBehaviorTest2Async(It.IsAny<CancellationToken>())).ReturnsAsync(default(StunResponse?));
		mock.Setup(x => x.FilteringBehaviorTest3Async(It.IsAny<CancellationToken>())).ReturnsAsync(default(StunResponse?));

		await client.QueryAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.EndpointIndependent, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.AddressAndPortDependent, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
		Assert.IsNotNull(client.State.OtherEndPoint);
	}

	[TestMethod]
	public async Task QueryMappingBehaviorAddressAndPortDependentTestAsync()
	{
		var mock = new Mock<StunClient5389UDP>(ServerAddress, Any, default);
		var client = mock.Object;

		var r1 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		var r2 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress2,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		var r3 = new StunResult5389
		{
			BindingTestResult = BindingTestResult.Success,
			PublicEndPoint = MappedAddress1,
			LocalEndPoint = LocalAddress1,
			OtherEndPoint = ChangedAddress1
		};
		mock.Setup(x => x.BindingTestBaseAsync(It.Is<IPEndPoint>(p => Equals(p, ServerAddress)), It.IsAny<CancellationToken>())).ReturnsAsync(r1);
		mock.Setup(x => x.BindingTestBaseAsync(It.Is<IPEndPoint>(p => Equals(p, ChangedAddress3)), It.IsAny<CancellationToken>())).ReturnsAsync(r2);
		mock.Setup(x => x.BindingTestBaseAsync(It.Is<IPEndPoint>(p => Equals(p, ChangedAddress1)), It.IsAny<CancellationToken>())).ReturnsAsync(r3);
		mock.Setup(x => x.FilteringBehaviorTest2Async(It.IsAny<CancellationToken>())).ReturnsAsync(default(StunResponse?));
		mock.Setup(x => x.FilteringBehaviorTest3Async(It.IsAny<CancellationToken>())).ReturnsAsync(default(StunResponse?));

		await client.QueryAsync();

		Assert.AreEqual(BindingTestResult.Success, client.State.BindingTestResult);
		Assert.AreEqual(MappingBehavior.AddressAndPortDependent, client.State.MappingBehavior);
		Assert.AreEqual(FilteringBehavior.AddressAndPortDependent, client.State.FilteringBehavior);
		Assert.IsNotNull(client.State.PublicEndPoint);
		Assert.IsNotNull(client.State.LocalEndPoint);
		Assert.IsNotNull(client.State.OtherEndPoint);
	}
}