diff --git a/shared/src/main/scala/scala/xml/dtd/Decl.scala b/shared/src/main/scala/scala/xml/dtd/Decl.scala index 9ac9761ea..5bf37403a 100644 --- a/shared/src/main/scala/scala/xml/dtd/Decl.scala +++ b/shared/src/main/scala/scala/xml/dtd/Decl.scala @@ -26,6 +26,7 @@ import Utility.sbToString sealed abstract class Decl sealed abstract class MarkupDecl extends Decl { + override def toString(): String = sbToString(buildString) def buildString(sb: StringBuilder): StringBuilder } diff --git a/shared/src/test/scala/scala/xml/dtd/DeclTest.scala b/shared/src/test/scala/scala/xml/dtd/DeclTest.scala new file mode 100644 index 000000000..a41542897 --- /dev/null +++ b/shared/src/test/scala/scala/xml/dtd/DeclTest.scala @@ -0,0 +1,42 @@ +package scala.xml +package dtd + +import org.junit.Test +import org.junit.Assert.assertEquals + +class DeclTest { + + @Test + def elemDeclToString: Unit = { + assertEquals( + "", + ElemDecl("x", PCDATA).toString + ) + } + + @Test + def attListDeclToString: Unit = { + + val expected = + """|""".stripMargin + + val actual = AttListDecl("x", + List( + AttrDecl("y", "CDATA", REQUIRED), + AttrDecl("z", "CDATA", REQUIRED) + ) + ).toString + + assertEquals(expected, actual) + } + + @Test + def parsedEntityDeclToString: Unit = { + assertEquals( + """""", + ParsedEntityDecl("foo", ExtDef(SystemID("bar"))).toString + ) + } +}