test/Npgsql.Tests/GlobalTypeMapperTests.cs - main - dev-npgsql/npgsql

test/Npgsql.Tests/GlobalTypeMapperTests.cs
143 строки6 KB
Nino Floris
[Tests] Fully rework type mapping assertion helpers (#6314)
23 фев 2026, 04:10
23 фев 2026, 04:10594b5d3
using System;
using System.Data;
using System.Threading.Tasks;
using Npgsql.Internal;
using Npgsql.Internal.Postgres;
using NUnit.Framework;
using static Npgsql.Tests.TestUtil;

namespace Npgsql.Tests;

#pragma warning disable CS0618 // GlobalTypeMapper is obsolete

[NonParallelizable]
public class GlobalTypeMapperTests : TestBase
{
    [Test]
    public async Task MapEnum()
    {
        await using var adminConnection = await OpenConnectionAsync();
        var type = await GetTempTypeName(adminConnection);
        NpgsqlConnection.GlobalTypeMapper.MapEnum<Mood>(type);

        await using var dataSource1 = CreateDataSource();

        await using (var connection = await dataSource1.OpenConnectionAsync())
        {
            await connection.ExecuteNonQueryAsync($"CREATE TYPE {type} AS ENUM ('sad', 'ok', 'happy')");
            await connection.ReloadTypesAsync();

            await AssertType(connection, Mood.Happy, "happy", type, dataTypeInference: DataTypeInference.Nothing);
        }

        NpgsqlConnection.GlobalTypeMapper.UnmapEnum<Mood>(type);

        // Global mapping changes have no effect on already-built data sources
        await AssertType(dataSource1, Mood.Happy, "happy", type, dataTypeInference: DataTypeInference.Nothing);
        await AssertType(dataSource1, "happy", "happy",
            type, dataTypeInference: DataTypeInference.Mismatch,
            dbType: new(DbType.Object, DbType.String), valueTypeEqualsFieldType: false);

        // But they do affect new data sources
        await using var dataSource2 = CreateDataSource();
        Assert.ThrowsAsync<InvalidCastException>(() => AssertType(dataSource2, Mood.Happy, "happy",
            type, dataTypeInference: DataTypeInference.Nothing));
        await AssertType(dataSource2, "happy", "happy", "text", dbType: DbType.String);
    }

    [Test]
    public async Task MapEnum_NonGeneric()
    {
        await using var adminConnection = await OpenConnectionAsync();
        var type = await GetTempTypeName(adminConnection);
        NpgsqlConnection.GlobalTypeMapper.MapEnum(typeof(Mood), type);

        try
        {
            await using var dataSource1 = CreateDataSource();

            await using (var connection = await dataSource1.OpenConnectionAsync())
            {
                await connection.ExecuteNonQueryAsync($"CREATE TYPE {type} AS ENUM ('sad', 'ok', 'happy')");
                await connection.ReloadTypesAsync();

                await AssertType(connection, Mood.Happy, "happy", type, dataTypeInference: DataTypeInference.Nothing);
            }

            NpgsqlConnection.GlobalTypeMapper.UnmapEnum(typeof(Mood), type);

            // Global mapping changes have no effect on already-built data sources
            await AssertType(dataSource1, Mood.Happy, "happy", type, dataTypeInference: DataTypeInference.Nothing);
            await AssertType(dataSource1, "happy", "happy",
                type, dataTypeInference: DataTypeInference.Mismatch,
                dbType: new(DbType.Object, DbType.String), valueTypeEqualsFieldType: false);

            // But they do affect new data sources
            await using var dataSource2 = CreateDataSource();
            Assert.ThrowsAsync<InvalidCastException>(() => AssertType(dataSource2, Mood.Happy, "happy", type, dataTypeInference: DataTypeInference.Nothing));
            await AssertType(dataSource2, "happy", "happy", "text", dbType: DbType.String);
        }
        finally
        {
            NpgsqlConnection.GlobalTypeMapper.UnmapEnum<Mood>(type);
        }
    }

    [Test]
    public async Task Reset()
    {
        await using var adminConnection = await OpenConnectionAsync();
        var type = await GetTempTypeName(adminConnection);
        NpgsqlConnection.GlobalTypeMapper.MapEnum<Mood>(type);

        await using var dataSource1 = CreateDataSource();

        await using (var connection = await dataSource1.OpenConnectionAsync())
        {
            await connection.ExecuteNonQueryAsync($"CREATE TYPE {type} AS ENUM ('sad', 'ok', 'happy')");
            await connection.ReloadTypesAsync();

            await AssertType(connection, Mood.Happy, "happy", type, dataTypeInference: DataTypeInference.Nothing);
        }

        // A global mapping change has no effects on data sources which have already been built
        NpgsqlConnection.GlobalTypeMapper.Reset();

        // Global mapping changes have no effect on already-built data sources
        await AssertType(dataSource1, Mood.Happy, "happy", type, dataTypeInference: DataTypeInference.Nothing);
        await AssertType(dataSource1, "happy", "happy",
            type, dataTypeInference: DataTypeInference.Mismatch,
            dbType: new(DbType.Object, DbType.String), valueTypeEqualsFieldType: false);

        // But they do affect new data sources
        await using var dataSource2 = CreateDataSource();
        Assert.ThrowsAsync<InvalidCastException>(() => AssertType(dataSource2, Mood.Happy, "happy", type, dataTypeInference: DataTypeInference.Nothing));
        await AssertType(dataSource2, "happy", "happy",
            type, dataTypeInference: DataTypeInference.Mismatch,
            dbType: new(DbType.Object, DbType.String));
    }

    [Test]
    public void Reset_and_add_resolver()
    {
        NpgsqlConnection.GlobalTypeMapper.Reset();
        NpgsqlConnection.GlobalTypeMapper.AddTypeInfoResolverFactory(new DummyResolverFactory());
    }

    [TearDown]
    public void Teardown()
        => NpgsqlConnection.GlobalTypeMapper.Reset();

    enum Mood { Sad, Ok, Happy }

    class DummyResolverFactory : PgTypeInfoResolverFactory
    {
        public override IPgTypeInfoResolver CreateResolver() => new DummyResolver();
        public override IPgTypeInfoResolver? CreateArrayResolver() => null;

        class DummyResolver : IPgTypeInfoResolver
        {
            public PgTypeInfo? GetTypeInfo(Type? type, DataTypeName? dataTypeName, PgSerializerOptions options) => null;
        }
    }
}