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

test/Npgsql.Tests/WriteBufferTests.cs
124 строки5 KB
Nino Floris
Merge pull request from GHSA-x9vc-6hfv-hg8c
09 май 2024, 14:20
09 май 2024, 14:20091655e
using System;
using System.IO;
using Npgsql.Internal;
using NUnit.Framework;

namespace Npgsql.Tests;

[FixtureLifeCycle(LifeCycle.InstancePerTestCase)] // Parallel access to a single buffer
class WriteBufferTests
{
    [Test]
    public void Buffered_full_buffer_no_flush()
    {
        WriteBuffer.WritePosition += WriteBuffer.WriteSpaceLeft - sizeof(int);
        var writer = WriteBuffer.GetWriter(null!, FlushMode.NonBlocking);
        Assert.That(writer.ShouldFlush(sizeof(int)), Is.False);

        Assert.DoesNotThrow(() =>
        {
            Span<byte> intBytes = stackalloc byte[4];
            writer.WriteBytes(intBytes);
        });
    }

    [Test]
    public void GetWriter_Full_Buffer()
    {
        WriteBuffer.WritePosition += WriteBuffer.WriteSpaceLeft;
        var writer = WriteBuffer.GetWriter(null!, FlushMode.Blocking);
        Assert.That(writer.ShouldFlush(sizeof(byte)), Is.True);
        writer.Flush();
        Assert.That(writer.ShouldFlush(sizeof(byte)), Is.False);
    }

    [Test, IssueLink("https://github.com/npgsql/npgsql/issues/1275")]
    public void Write_zero_characters()
    {
        // Fill up the buffer entirely
        WriteBuffer.WriteBytes(new byte[WriteBuffer.Size], 0, WriteBuffer.Size);
        Assert.That(WriteBuffer.WriteSpaceLeft, Is.Zero);

        int charsUsed;
        bool completed;
        WriteBuffer.WriteStringChunked("hello", 0, 5, true, out charsUsed, out completed);
        Assert.That(charsUsed, Is.Zero);
        Assert.That(completed, Is.False);
        WriteBuffer.WriteStringChunked("hello".ToCharArray(), 0, 5, true, out charsUsed, out completed);
        Assert.That(charsUsed, Is.Zero);
        Assert.That(completed, Is.False);
    }

    [Test, IssueLink("https://github.com/npgsql/npgsql/issues/2849")]
    public void Chunked_string_encoding_fits()
    {
        WriteBuffer.WriteBytes(new byte[WriteBuffer.Size - 1], 0, WriteBuffer.Size - 1);
        Assert.That(WriteBuffer.WriteSpaceLeft, Is.EqualTo(1));

        var charsUsed = 1;
        var completed = true;
        // This unicode character is three bytes when encoded in UTF8
        Assert.That(() => WriteBuffer.WriteStringChunked("\uD55C", 0, 1, true, out charsUsed, out completed), Throws.Nothing);
        Assert.That(charsUsed, Is.EqualTo(0));
        Assert.That(completed, Is.False);
    }

    [Test, IssueLink("https://github.com/npgsql/npgsql/issues/2849")]
    public void Chunked_byte_array_encoding_fits()
    {
        WriteBuffer.WriteBytes(new byte[WriteBuffer.Size - 1], 0, WriteBuffer.Size - 1);
        Assert.That(WriteBuffer.WriteSpaceLeft, Is.EqualTo(1));

        var charsUsed = 1;
        var completed = true;
        // This unicode character is three bytes when encoded in UTF8
        Assert.That(() => WriteBuffer.WriteStringChunked("\uD55C".ToCharArray(), 0, 1, true, out charsUsed, out completed), Throws.Nothing);
        Assert.That(charsUsed, Is.EqualTo(0));
        Assert.That(completed, Is.False);
    }

    [Test, IssueLink("https://github.com/npgsql/npgsql/issues/3733")]
    public void Chunked_string_encoding_fits_with_surrogates()
    {
        WriteBuffer.WriteBytes(new byte[WriteBuffer.Size - 1]);
        Assert.That(WriteBuffer.WriteSpaceLeft, Is.EqualTo(1));

        var charsUsed = 1;
        var completed = true;
        var cyclone = "🌀";

        Assert.That(() => WriteBuffer.WriteStringChunked(cyclone, 0, cyclone.Length, true, out charsUsed, out completed), Throws.Nothing);
        Assert.That(charsUsed, Is.EqualTo(0));
        Assert.That(completed, Is.False);
    }

    [Test, IssueLink("https://github.com/npgsql/npgsql/issues/3733")]
    public void Chunked_char_array_encoding_fits_with_surrogates()
    {
        WriteBuffer.WriteBytes(new byte[WriteBuffer.Size - 1]);
        Assert.That(WriteBuffer.WriteSpaceLeft, Is.EqualTo(1));

        var charsUsed = 1;
        var completed = true;
        var cyclone = "🌀";

        Assert.That(() => WriteBuffer.WriteStringChunked(cyclone.ToCharArray(), 0, cyclone.Length, true, out charsUsed, out completed), Throws.Nothing);
        Assert.That(charsUsed, Is.EqualTo(0));
        Assert.That(completed, Is.False);
    }

#pragma warning disable CS8625
    [SetUp]
    public void SetUp()
    {
        Underlying = new MemoryStream();
        WriteBuffer = new NpgsqlWriteBuffer(null, Underlying, null, NpgsqlReadBuffer.DefaultSize, NpgsqlWriteBuffer.UTF8Encoding);
        WriteBuffer.MessageLengthValidation = false;
    }
#pragma warning restore CS8625

    // ReSharper disable once InconsistentNaming
    NpgsqlWriteBuffer WriteBuffer = default!;
    // ReSharper disable once InconsistentNaming
    MemoryStream Underlying = default!;
}