Spring/spring-restdocs
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use TableWriter in LinkSnippetResultHandler

Browse Source 
Closes gh-40
This commit is contained in:
Andy Wilkinson
2015-03-24 17:31:04 +00:00
parent 0d2d9cce73
commit 976dd00365
1 changed files with 13 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
spring-restdocs/src/main/java/org/springframework/restdocs/hypermedia/LinkSnippetResultHandler.java
View File

@@ -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<String, LinkDescriptor> 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<String, LinkDescriptor> entry : LinkSnippetResultHandler.this.descriptorsByRel
.entrySet()) {
tableWriter.row(entry.getKey(), entry.getValue().getDescription());
}
}
});
writer.println("|===");
}
}