52 lines
1.9 KiB
C#
52 lines
1.9 KiB
C#
using NATS.Server.JetStream.Storage;
|
|
|
|
namespace NATS.Server.JetStream.Tests.JetStream.Storage;
|
|
|
|
public sealed class FileStoreOptimizationGuardTests
|
|
{
|
|
[Fact]
|
|
public async Task Snapshot_round_trip_preserves_headers_and_payload_separately()
|
|
{
|
|
var srcDir = Directory.CreateTempSubdirectory();
|
|
var dstDir = Directory.CreateTempSubdirectory();
|
|
|
|
await using var src = new FileStore(new FileStoreOptions { Directory = srcDir.FullName });
|
|
var hdr = "NATS/1.0\r\nX-Test: two\r\n\r\n"u8.ToArray();
|
|
var msg = "payload-two"u8.ToArray();
|
|
|
|
var (seq, _) = src.StoreMsg("events.a", hdr, msg, 0L);
|
|
var snapshot = await src.CreateSnapshotAsync(default);
|
|
|
|
await using var dst = new FileStore(new FileStoreOptions { Directory = dstDir.FullName });
|
|
await dst.RestoreSnapshotAsync(snapshot, default);
|
|
|
|
var loaded = dst.LoadMsg(seq, null);
|
|
loaded.Header.ShouldNotBeNull();
|
|
loaded.Header.ShouldBe(hdr);
|
|
loaded.Data.ShouldNotBeNull();
|
|
loaded.Data.ShouldBe(msg);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task PurgeEx_updates_last_by_subject_after_recovery()
|
|
{
|
|
var dir = Directory.CreateTempSubdirectory();
|
|
|
|
await using (var store = new FileStore(new FileStoreOptions { Directory = dir.FullName }))
|
|
{
|
|
store.StoreMsg("events.a", null, "one"u8.ToArray(), 0L);
|
|
store.StoreMsg("events.a", null, "two"u8.ToArray(), 0L);
|
|
store.StoreMsg("events.b", null, "other"u8.ToArray(), 0L);
|
|
store.PurgeEx("events.a", 0, 1);
|
|
await store.FlushAllPending();
|
|
}
|
|
|
|
await using var recovered = new FileStore(new FileStoreOptions { Directory = dir.FullName });
|
|
var last = await recovered.LoadLastBySubjectAsync("events.a", default);
|
|
|
|
last.ShouldNotBeNull();
|
|
last.Sequence.ShouldBe(2UL);
|
|
last.Payload.ToArray().ShouldBe("two"u8.ToArray());
|
|
}
|
|
}
|