dev-npgsql/npgsql: Npgsql is the .NET data provider for PostgreSQL.

test/Npgsql.Tests/NpgsqlEventSourceTests.cs

53 строки2 KB

Shay Rojansky

Apply primary constructors to solution (#5908)

30 окт 2024, 18:30

30 окт 2024, 18:3040cc1a1

using System.Linq;
using System.Collections.Generic;
using System.Diagnostics.Tracing;
using NUnit.Framework;

namespace Npgsql.Tests;

[NonParallelizable] // Events
public class NpgsqlEventSourceTests : TestBase
{
    [Test]
    public void Command_start_stop()
    {
        using (var conn = OpenConnection())
        {
            // There is a new pool created, which sends a few queries to load pg types
            ClearEvents();
            conn.ExecuteScalar("SELECT 1");
        }

        var commandStart = _events.Single(e => e.EventId == NpgsqlEventSource.CommandStartId);
        Assert.That(commandStart.EventName, Is.EqualTo("CommandStart"));

        var commandStop = _events.Single(e => e.EventId == NpgsqlEventSource.CommandStopId);
        Assert.That(commandStop.EventName, Is.EqualTo("CommandStop"));
    }

    [OneTimeSetUp]
    public void EnableEventSource()
    {
        _listener = new TestEventListener(_events);
        _listener.EnableEvents(NpgsqlSqlEventSource.Log, EventLevel.Informational);
    }

    [OneTimeTearDown]
    public void DisableEventSource()
    {
        _listener.DisableEvents(NpgsqlSqlEventSource.Log);
        _listener.Dispose();
    }

    [SetUp]
    public void ClearEvents() => _events.Clear();

    TestEventListener _listener = null!;

    readonly List<EventWrittenEventArgs> _events = [];

    class TestEventListener(List<EventWrittenEventArgs> events) : EventListener
    {
        protected override void OnEventWritten(EventWrittenEventArgs eventData) => events.Add(eventData);
    }
}