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

test/Npgsql.Tests/LargeObjectTests.cs

56 строк2 KB

Shay Rojansky

Test tweaks

18 ноя 2024, 20:06

18 ноя 2024, 20:06d854e7b

using System.Linq;
using System.Text;
using NUnit.Framework;

namespace Npgsql.Tests;

public class LargeObjectTests : TestBase
{
    [Test]
    public void Test()
    {
        using var conn = OpenConnection();
        using var transaction = conn.BeginTransaction();
        var manager = new NpgsqlLargeObjectManager(conn);
        var oid = manager.Create();
        using (var stream = manager.OpenReadWrite(oid))
        {
            var buf = Encoding.UTF8.GetBytes("Hello");
            stream.Write(buf, 0, buf.Length);
            stream.Seek(0, System.IO.SeekOrigin.Begin);
            var buf2 = new byte[buf.Length];

#pragma warning disable CA2022 // Avoid inexact read with 'System.IO.Stream.ReadAsync(byte[], int, int)'
#if NET8_0_OR_GREATER
            stream.ReadExactly(buf2, 0, buf2.Length);
#else
            stream.Read(buf2, 0, buf2.Length);
#endif
#pragma warning restore CA2022

            Assert.That(buf.SequenceEqual(buf2));

            Assert.AreEqual(5, stream.Position);

            Assert.AreEqual(5, stream.Length);

            stream.Seek(-1, System.IO.SeekOrigin.Current);
            Assert.AreEqual((int)'o', stream.ReadByte());

            manager.MaxTransferBlockSize = 3;

            stream.Write(buf, 0, buf.Length);
            stream.Seek(-5, System.IO.SeekOrigin.End);
            var buf3 = new byte[100];
            Assert.AreEqual(5, stream.Read(buf3, 0, 100));
            Assert.That(buf.SequenceEqual(buf3.Take(5)));

            stream.SetLength(43);
            Assert.AreEqual(43, stream.Length);
        }

        manager.Unlink(oid);

        transaction.Rollback();
    }
}