SocialPub/PrivaPub.Tests/Infrastructure/GeoLocatorTests.cs

using Microsoft.Extensions.Logging.Abstractions;

using PrivaPub.Infrastructure.Geo;
using PrivaPub.Infrastructure.Http;
using PrivaPub.Infrastructure.Statistics;
using PrivaPub.Tests.Support;

using System.Net;

namespace PrivaPub.Tests.Infrastructure
{
	public class GeoLocatorTests
	{
		static DbIpLocator Locator(string directory) =>
			new(new StaticOptions<StatisticsOptions>(new StatisticsOptions { GeoDirectory = directory }), NullLogger<DbIpLocator>.Instance);

		[Fact]
		public void Without_databases_nothing_is_located()
		{
			using var locator = Locator(Path.Combine(Path.GetTempPath(), $"no-geo-{Guid.NewGuid():N}"));

			Assert.Null(locator.Locate(IPAddress.Parse("1.1.1.1")));
			Assert.Null(locator.Source);
		}

		[Fact]
		public void A_private_address_is_never_looked_up()
		{
			using var locator = Locator(Path.GetTempPath());

			Assert.Null(locator.Locate(IPAddress.Parse("10.1.2.3")));
			Assert.Null(locator.Locate(IPAddress.Loopback));
			Assert.Null(locator.Locate(default));
		}

		[Fact]
		public void The_real_databases_give_a_country_a_city_and_a_network()
		{
			var directory = Environment.GetEnvironmentVariable("PRIVAPUB_TEST_GEO");
			Assert.SkipWhen(string.IsNullOrEmpty(directory), "set PRIVAPUB_TEST_GEO to a directory holding the DB-IP Lite city and ASN files");
			using var locator = Locator(directory);

			var fix = locator.Locate(IPAddress.Parse("1.1.1.1"));

			Assert.NotNull(fix);
			Assert.Equal(2, fix.Country.Length);
			Assert.Equal(13335, fix.Asn);
			Assert.Equal("Cloudflare", CdnCatalog.OfAsn(fix.Asn)?.Name);
			Assert.NotNull(fix.AsnOrg);
			Assert.NotNull(fix.Latitude);
			Assert.Equal(Math.Round(fix.Latitude.Value, 1), fix.Latitude);
			Assert.StartsWith("DB-IP Lite ", locator.Source);
		}

		[Fact]
		public void A_connection_remembers_its_address_as_ipv4()
		{
			var connected = new ConnectedAddresses();

			connected.Remember("Social.Example", IPAddress.Parse("::ffff:203.0.113.7"));

			Assert.Equal(IPAddress.Parse("203.0.113.7"), connected.Of("social.example"));
			Assert.Null(connected.Of("other.example"));
			Assert.Equal("Fastly", CdnCatalog.OfAsn(54113)?.Name);
			Assert.Null(CdnCatalog.OfAsn(64496));
			Assert.Null(CdnCatalog.OfAsn(default));
		}
	}
}