69 lines
2.2 KiB
C#
69 lines
2.2 KiB
C#
using ImageCatalog_2.Services;
|
|
using ImageCatalog_2.Models;
|
|
using Shouldly;
|
|
using System.Text;
|
|
|
|
namespace MaddoShared.Tests;
|
|
|
|
[TestClass]
|
|
public sealed class AiExtractionServiceCsvTests
|
|
{
|
|
[TestMethod]
|
|
public void WriteCsvOutput_UsesLegacyCompatibleQuotingAndSkipsEmptyResults()
|
|
{
|
|
using var tempDir = new TempDirectory();
|
|
var csvPath = Path.Combine(tempDir.Path, "ocr.csv");
|
|
|
|
AiExtractionService.WriteCsvOutput(
|
|
csvPath,
|
|
[
|
|
new AiResultItem { Path = @"C:\images\IMG_7146.JPG", Text = "43" },
|
|
new AiResultItem { Path = @"C:\images\IMG_7207.JPG", Text = "43,84,61" },
|
|
new AiResultItem { Path = @"C:\images\IMG_7208.JPG", Text = string.Empty },
|
|
new AiResultItem { Path = @"C:\images\IMG_7209.JPG", Text = " " },
|
|
new AiResultItem { Path = @"C:\images\IMG_7210.JPG", Text = "a\"b,c" }
|
|
]);
|
|
|
|
var csv = File.ReadAllText(csvPath, Encoding.UTF8);
|
|
|
|
csv.ShouldBe("filename,text\r"
|
|
+ "IMG_7146.JPG,43\r"
|
|
+ "IMG_7207.JPG,\"43,84,61\"\r"
|
|
+ "IMG_7210.JPG,\"a\"\"b,c\"\r");
|
|
}
|
|
|
|
[TestMethod]
|
|
public void WriteCsvOutput_WritesUtf8WithoutBom()
|
|
{
|
|
using var tempDir = new TempDirectory();
|
|
var csvPath = Path.Combine(tempDir.Path, "ocr.csv");
|
|
|
|
AiExtractionService.WriteCsvOutput(
|
|
csvPath,
|
|
[new AiResultItem { Path = @"C:\images\IMG_7146.JPG", Text = "43" }]);
|
|
|
|
var bytes = File.ReadAllBytes(csvPath);
|
|
|
|
bytes[.."filename,text\r".Length].ShouldBe(Encoding.ASCII.GetBytes("filename,text\r"));
|
|
bytes[..3].ShouldNotBe(new byte[] { 0xEF, 0xBB, 0xBF });
|
|
}
|
|
|
|
private sealed class TempDirectory : IDisposable
|
|
{
|
|
public TempDirectory()
|
|
{
|
|
Path = System.IO.Path.Combine(System.IO.Path.GetTempPath(), System.IO.Path.GetRandomFileName());
|
|
Directory.CreateDirectory(Path);
|
|
}
|
|
|
|
public string Path { get; }
|
|
|
|
public void Dispose()
|
|
{
|
|
if (Directory.Exists(Path))
|
|
{
|
|
Directory.Delete(Path, recursive: true);
|
|
}
|
|
}
|
|
}
|
|
}
|