diff options
Diffstat (limited to 'src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java')
| -rw-r--r-- | src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java b/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java index a3595ae..ea53458 100644 --- a/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java +++ b/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java @@ -25,6 +25,9 @@ public class DocumentElementMakers { private static final ArgumentKey<Integer> COLSPAN = new ArgumentKey<>("colspan"); private static final ArgumentKey<Integer> ROWSPAN = new ArgumentKey<>("rowspan"); private static final ArgumentKey<List<Comment>> COMMENTS = new ArgumentKey<>("comments"); + private static final ArgumentKey<Optional<String>> HREF = new ArgumentKey<>("href"); + private static final ArgumentKey<Optional<String>> ANCHOR = new ArgumentKey<>("anchor"); + private static final ArgumentKey<Optional<String>> TARGET_FRAME = new ArgumentKey<>("targetFrame"); public static Argument<Optional<Style>> withStyle(Style style) { return arg(STYLE, Optional.of(style)); @@ -130,4 +133,26 @@ public class DocumentElementMakers { public static Comment comment(String commentId, List<DocumentElement> body) { return new Comment(commentId, body, Optional.empty(), Optional.empty()); } + + public static Hyperlink hyperlink(Object... args) { + Arguments arguments = new Arguments(args); + return new Hyperlink( + arguments.get(HREF, Optional.empty()), + arguments.get(ANCHOR, Optional.empty()), + arguments.get(TARGET_FRAME, Optional.empty()), + arguments.get(CHILDREN, list()) + ); + } + + public static Argument<Optional<String>> withHref(String href) { + return arg(HREF, Optional.of(href)); + } + + public static Argument<Optional<String>> withAnchor(String anchor) { + return arg(ANCHOR, Optional.of(anchor)); + } + + public static Argument<Optional<String>> withTargetFrame(String targetFrame) { + return arg(TARGET_FRAME, Optional.of(targetFrame)); + } } |
