fix unit tests

Author: Mpdreamz

tests/Elastic.Markdown.Tests/CodeBlocks/CallOutTests.cs

@@ -118,21 +118,17 @@ public void ParsesMagicCallOuts() => Block!.CallOuts
 
 	[Fact]
 	public void RendersExpectedHtml() =>
-		Html.ReplaceLineEndings().Should().Contain("""
-		                      <div class="highlight-csharp notranslate">
-		                      	<div class="highlight">
-		                      		<pre><code class="language-csharp">var x = 1;<span style="display: inline-block; width: 1ch"></span><span class="code-callout" data-index="1"></span>
-		                      var y = x - 2;
-		                      var z = y - 2;<span style="display: inline-block; width: 1ch"></span><span class="code-callout" data-index="2"></span>
-		                      </code></pre>
-		                      	</div>
-		                      </div>
-		                      <p><strong>OUTPUT:</strong></p>
-		                      <ol class="code-callouts">
-		                      <li>Marking the first callout</li>
-		                      <li>Marking the second callout</li>
-		                      </ol>
-		                      """.ReplaceLineEndings());
+
+		Html.ReplaceLineEndings()
+			.Should().Contain(
+		// language=html
+		"""
+		<p><strong>OUTPUT:</strong></p>
+		<ol class="code-callouts">
+		<li>Marking the first callout</li>
+		<li>Marking the second callout</li>
+		</ol>
+		""".ReplaceLineEndings());
 
 
 	[Fact]

tests/Elastic.Markdown.Tests/Directives/MermaidTests.cs renamed to tests/Elastic.Markdown.Tests/CodeBlocks/MermaidTests.cs

@@ -1,14 +1,17 @@
 // Licensed to Elasticsearch B.V under one or more agreements.
 // Elasticsearch B.V licenses this file to you under the Apache 2.0 License.
 // See the LICENSE file in the project root for more information
+
+using Elastic.Markdown.Myst.CodeBlocks;
 using Elastic.Markdown.Myst.Directives;
+using Elastic.Markdown.Tests.Inline;
 using FluentAssertions;
 
-namespace Elastic.Markdown.Tests.Directives;
+namespace Elastic.Markdown.Tests.CodeBlocks;
 
-public class MermaidBlockTests(ITestOutputHelper output) : DirectiveTest<UnsupportedDirectiveBlock>(output,
+public class MermaidBlockTests(ITestOutputHelper output) : BlockTest<EnhancedCodeBlock>(output,
 """
-:::{mermaid} as
+```mermaid
 flowchart LR
   A[Jupyter Notebook] --> C
   B[MyST Markdown] --> C
@@ -19,14 +22,14 @@ B[MyST Markdown] --> C
   D --> H[React]
   D --> I[HTML]
   D <--> J[JATS]
-:::
+```
 """
 )
 {
 	[Fact]
 	public void ParsesBlock() => Block.Should().NotBeNull();
 
-	// should still attempt to render contents as markdown
+	// should still attempt to render contents as Markdown
 	[Fact]
 	public void IncludesRawFlowChart() =>
 		Html.Should().Contain("D --&gt; I[HTML]");