テストを追加

OUCC · Apr 1, 2024 · ab04353 · ab04353
1 parent d0db7ad
commit ab04353
Show file tree

Hide file tree

Showing 8 changed files with 2,154 additions and 3 deletions.
diff --git a/KoeBook.Test/DiTestBase.cs b/KoeBook.Test/DiTestBase.cs
@@ -6,7 +6,7 @@ namespace KoeBook.Test;
 
 public class DiTestBase
 {
-    protected IHost Host { get; } = CreateDefaultBuilder().Build();
+    protected IHost Host { get; init; } = CreateDefaultBuilder().Build();
 
     protected T GetService<T>() where T : notnull
     {

diff --git a/KoeBook.Test/Epub/ScrapingNaroServiceTest.cs b/KoeBook.Test/Epub/ScrapingNaroServiceTest.cs
@@ -1,8 +1,10 @@
-using KoeBook.Core;
+using System.Text.Json;
+using AngleSharp;
+using KoeBook.Core;
 using KoeBook.Epub.Contracts.Services;
+using KoeBook.Epub.Models;
 using KoeBook.Epub.Services;
 using Microsoft.Extensions.DependencyInjection;
-using Microsoft.Extensions.Hosting;
 
 namespace KoeBook.Test.Epub;
 
@@ -18,6 +20,34 @@ public ScrapingNaroServiceTest()
             .Single();
     }
 
+    [Fact]
+    public async Task ReadPageAsync()
+    {
+        Directory.CreateDirectory("./tmp/img");
+        var config = Configuration.Default.WithDefaultLoader();
+        using var context = BrowsingContext.New(config);
+        using var doc = await context.OpenAsync(req => req.Content(File.ReadAllText("./TestData/Naro/n0000aa/1.html")));
+
+        var (chapterTitle, section) = await _scrapingNaroService.ReadPageAsync(doc, true, "./tmp/img", default);
+
+        Assert.Null(chapterTitle);
+        Assert.Equal("タイトル1", section.Title);
+        var elements = section.Elements;
+        Assert.Equal(6, elements.Count);
+        var text = Assert.IsType<Paragraph>(elements[0]);
+        Assert.Equal("名前は【<ruby><rb>佐久平</rb><rp>《</rp><rt>さくだいら</rt><rp>》</rp></ruby>　<ruby><rb>啓介</rb><rp>《</rp><rt>けいすけ</rt><rp>》</rp></ruby>】。", text.Text);
+        text = Assert.IsType<Paragraph>(elements[1]);
+        Assert.Equal("　テストテストテストテストテスト", text.Text);
+        text = Assert.IsType<Paragraph>(elements[2]);
+        Assert.Equal("「セリフセリフセリフセリフ！」", text.Text);
+        text = Assert.IsType<Paragraph>(elements[3]);
+        Assert.Equal("テスト", text.Text);
+        text = Assert.IsType<Paragraph>(elements[4]);
+        Assert.Equal("「インラインテスト」", text.Text);
+        text = Assert.IsType<Paragraph>(elements[5]);
+        Assert.Equal("テスト。ほげほげほげほげほげ。", text.Text);
+    }
+
     [Theory]
     [InlineData("https://ncode.syosetu.com/n0000a", "n0000a")]
     [InlineData("https://ncode.syosetu.com/n0000a/", "n0000a")]

diff --git a/KoeBook.Test/KoeBook.Test.csproj b/KoeBook.Test/KoeBook.Test.csproj
@@ -12,6 +12,7 @@
   <ItemGroup>
     <PackageReference Include="coverlet.collector" Version="6.0.0" />
     <PackageReference Include="Microsoft.Extensions.Hosting" Version="8.0.0" />
+    <PackageReference Include="Microsoft.Extensions.TimeProvider.Testing" Version="8.3.0" />
     <PackageReference Include="Microsoft.NET.Test.Sdk" Version="17.8.0" />
     <PackageReference Include="xunit" Version="2.5.3" />
     <PackageReference Include="xunit.runner.visualstudio" Version="2.5.3" />
@@ -29,4 +30,10 @@
   <ItemGroup>
     <Compile Include="../KoeBook/Startup.cs" />
   </ItemGroup>
+
+  <ItemGroup>
+    <None Update="TestData\**">
+      <CopyToOutputDirectory>Always</CopyToOutputDirectory>
+    </None>
+  </ItemGroup>
 </Project>
diff --git a/KoeBook.Test/MockHttpClientHandler.cs b/KoeBook.Test/MockHttpClientHandler.cs
@@ -0,0 +1,52 @@
+using System.Net.Http.Headers;
+using System.Text;
+
+namespace KoeBook.Test;
+
+internal class MockHttpClientHandler : HttpClientHandler
+{
+    protected override HttpResponseMessage Send(HttpRequestMessage request, CancellationToken cancellationToken)
+    {
+        return request.RequestUri?.GetLeftPart(UriPartial.Authority) switch
+        {
+            "https://ncode.syosetu.com" => ProcessRequest(request, cancellationToken),
+            _ => base.Send(request, cancellationToken),
+        };
+    }
+
+    protected override Task<HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken)
+    {
+        return request.RequestUri?.GetLeftPart(UriPartial.Authority) switch
+        {
+            "https://ncode.syosetu.com" => Task.FromResult(ProcessRequest(request, cancellationToken)),
+            _ => base.SendAsync(request, cancellationToken),
+        };
+    }
+
+    private HttpResponseMessage ProcessRequest(HttpRequestMessage requestMessage, CancellationToken cancellationToken)
+    {
+        if (requestMessage.Method != HttpMethod.Get || requestMessage.Headers.UserAgent.Count == 0)
+            throw new NotSupportedException();
+        var (filePath, mediaType, isString) = requestMessage.RequestUri?.ToString() switch
+        {
+            "https://ncode.syosetu.com/n0000aa" => ("./TestData/Naro/n0000aa.html", "text/html; charset=UTF-8", true),
+            "https://ncode.syosetu.com/n0000aa/1" => ("./TestData/Naro/n0000aa/1.html", "text/html; charset=UTF-8", true),
+            "https://ncode.syosetu.com/n0000aa/2" => ("./TestData/Naro/n0000aa/2.html", "text/html; charset=UTF-8", true),
+            "https://ncode.syosetu.com/n0000aa/3" => ("./TestData/Naro/n0000aa/3.html", "text/html; charset=UTF-8", true),
+            _ => throw new NotSupportedException()
+        };
+
+        return new HttpResponseMessage
+        {
+            Content = isString
+                ? new StringContent(File.ReadAllText(filePath), Encoding.UTF8, MediaTypeHeaderValue.Parse(mediaType))
+                : new ByteArrayContent(File.ReadAllBytes(filePath))
+                    {
+                        Headers =
+                        {
+                            ContentType = MediaTypeHeaderValue.Parse(mediaType),
+                        }
+                    }
+        };
+    }
+}