Add note blocks and an integration test

1 files changed, 31 insertions, 0 deletions

diff --git a/src/test/kotlin/moe/nea/blog/gen/HtmlGenerationTest.kt b/src/test/kotlin/moe/nea/blog/gen/HtmlGenerationTest.kt
new file mode 100644
index 0000000..baee5d6
--- /dev/null
+++ b/src/test/kotlin/moe/nea/blog/gen/HtmlGenerationTest.kt
@@ -0,0 +1,31 @@
+package moe.nea.blog.gen
+
+import moe.nea.blog.md.MarkdownElement
+import moe.nea.blog.md.MarkdownParser
+import kotlin.test.assertEquals
+
+abstract class HtmlGenerationTest {
+
+	open fun registerGenerators(generator: MD2HtmlGenerator) {
+	}
+
+	open fun registerParsers(parser: MarkdownParser) {
+	}
+
+	fun assertGenerator(generatedHtml: String, markdown: MarkdownElement) {
+		val generator = MD2HtmlGenerator()
+		generator.registerDefaultGenerators()
+		registerGenerators(generator)
+		assertEquals(generatedHtml, generator.generateHtml(markdown).text)
+	}
+
+	fun assertGeneratorMD(generatedHtml: String, text: String) {
+		val generator = MD2HtmlGenerator()
+		generator.registerDefaultGenerators()
+		registerGenerators(generator)
+		val parser = MarkdownParser(text)
+		parser.addDefaultParsers()
+		registerParsers(parser)
+		assertEquals(generatedHtml, generator.generateHtml(parser.readDocument()).text)
+	}
+}
\ No newline at end of file