From 976dd00365cf3151ee17970b0e44acd06f6975ab Mon Sep 17 00:00:00 2001 From: Andy Wilkinson Date: Tue, 24 Mar 2015 17:31:04 +0000 Subject: [PATCH] Use TableWriter in LinkSnippetResultHandler Closes gh-40 --- .../hypermedia/LinkSnippetResultHandler.java | 21 ++++++++++++------- 1 file changed, 13 insertions(+), 8 deletions(-) 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