diff --git a/spring-restdocs/src/main/java/org/springframework/restdocs/hypermedia/LinkSnippetResultHandler.java b/spring-restdocs/src/main/java/org/springframework/restdocs/hypermedia/LinkSnippetResultHandler.java index bc4441c1..d2c1f2c0 100644 --- a/spring-restdocs/src/main/java/org/springframework/restdocs/hypermedia/LinkSnippetResultHandler.java +++ b/spring-restdocs/src/main/java/org/springframework/restdocs/hypermedia/LinkSnippetResultHandler.java @@ -27,6 +27,8 @@ import java.util.Map.Entry; import java.util.Set; import org.springframework.restdocs.snippet.DocumentationWriter; +import org.springframework.restdocs.snippet.DocumentationWriter.TableAction; +import org.springframework.restdocs.snippet.DocumentationWriter.TableWriter; import org.springframework.restdocs.snippet.SnippetWritingResultHandler; import org.springframework.test.web.servlet.MvcResult; import org.springframework.util.Assert; @@ -100,16 +102,19 @@ public class LinkSnippetResultHandler extends SnippetWritingResultHandler { Assert.isTrue(actualRels.equals(expectedRels)); - writer.println("|==="); - writer.println("| Relation | Description"); + writer.table(new TableAction() { - for (Entry entry : this.descriptorsByRel.entrySet()) { - writer.println(); - writer.println("| " + entry.getKey()); - writer.println("| " + entry.getValue().getDescription()); - } + @Override + public void perform(TableWriter tableWriter) throws IOException { + tableWriter.headers("Relation", "Description"); + for (Entry entry : LinkSnippetResultHandler.this.descriptorsByRel + .entrySet()) { + tableWriter.row(entry.getKey(), entry.getValue().getDescription()); + } + } + + }); - writer.println("|==="); } } \ No newline at end of file