cesnimda
40 lines
1.9 KiB
C#
40 lines
1.9 KiB
C#
using JobTrackerApi.Models;
|
|
using JobTrackerApi.Services;
|
|
using JobTrackerApi.Tests.TestSupport;
|
|
using Microsoft.Extensions.Configuration;
|
|
using Microsoft.Extensions.Hosting;
|
|
using Moq;
|
|
using Xunit;
|
|
|
|
namespace JobTrackerApi.Tests;
|
|
|
|
public class CvImportRetentionTests
|
|
{
|
|
[Theory]
|
|
[InlineData("pending_review", 8, true)]
|
|
[InlineData("cancelled", 2, true)]
|
|
[InlineData("failed", 2, true)]
|
|
[InlineData("pending_review", 2, false)]
|
|
public async Task Original_sources_have_bounded_retention_and_recent_reviews_survive(string status, int days, bool expires)
|
|
{
|
|
var root = Path.Combine(Path.GetTempPath(), "cv-import-retention-" + Guid.NewGuid().ToString("N"));
|
|
var environment = Mock.Of<IHostEnvironment>(e => e.ContentRootPath == root);
|
|
var paths = new AppPaths(new ConfigurationBuilder().AddInMemoryCollection(new Dictionary<string, string?> { ["Data:Root"] = root }).Build(), environment);
|
|
try
|
|
{
|
|
await using var db = TestHostFactory.CreateInMemoryDb();
|
|
var file = Path.Combine(paths.CvArtifactsRoot, "synthetic.pdf");
|
|
await File.WriteAllTextAsync(file, "Synthetic source");
|
|
var now = DateTimeOffset.UtcNow;
|
|
var run = new CvExtractionRun { OwnerUserId = "user-1", ParserVersion = "docling-v1", Trigger = "document_import", Status = status, StartedAtUtc = now.AddDays(-days), RawExtractedText = "Source", DocumentJson = "{}", StructuredProfileJson = "{}", Artifact = new() { OwnerUserId = "user-1", StoragePath = file } };
|
|
db.CvExtractionRuns.Add(run);
|
|
await db.SaveChangesAsync();
|
|
await CvImportRetentionHostedService.SweepAsync(db, paths, now, default);
|
|
Assert.Equal(!expires, File.Exists(file));
|
|
Assert.Equal(expires, run.DocumentJson is null);
|
|
Assert.Equal(expires && status == "pending_review" ? "expired" : status, run.Status);
|
|
}
|
|
finally { Directory.Delete(root, true); }
|
|
}
|
|
}
|