1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
"""Step implementations for hyperlink-related features."""
from __future__ import annotations
from typing import Dict, Tuple
from behave import given, then
from behave.runner import Context
from docx import Document
from helpers import test_docx
# given ===================================================
@given("a hyperlink")
def given_a_hyperlink(context: Context):
document = Document(test_docx("par-hyperlinks"))
context.hyperlink = document.paragraphs[1].hyperlinks[0]
@given("a hyperlink having a URI fragment")
def given_a_hyperlink_having_a_uri_fragment(context: Context):
document = Document(test_docx("par-hlink-frags"))
context.hyperlink = document.paragraphs[1].hyperlinks[0]
@given("a hyperlink having address {address} and fragment {fragment}")
def given_a_hyperlink_having_address_and_fragment(context: Context, address: str, fragment: str):
paragraph_idxs: Dict[Tuple[str, str], int] = {
("''", "linkedBookmark"): 1,
("https://foo.com", "''"): 2,
("https://foo.com?q=bar", "''"): 3,
("http://foo.com/", "intro"): 4,
("https://foo.com?q=bar#baz", "''"): 5,
("court-exif.jpg", "''"): 7,
}
paragraph_idx = paragraph_idxs[(address, fragment)]
document = Document(test_docx("par-hlink-frags"))
paragraph = document.paragraphs[paragraph_idx]
context.hyperlink = paragraph.hyperlinks[0]
@given("a hyperlink having {zero_or_more} rendered page breaks")
def given_a_hyperlink_having_rendered_page_breaks(context: Context, zero_or_more: str):
paragraph_idx = {
"no": 1,
"one": 2,
}[zero_or_more]
document = Document(test_docx("par-hyperlinks"))
paragraph = document.paragraphs[paragraph_idx]
context.hyperlink = paragraph.hyperlinks[0]
@given("a hyperlink having {one_or_more} runs")
def given_a_hyperlink_having_one_or_more_runs(context: Context, one_or_more: str):
paragraph_idx, hyperlink_idx = {
"one": (1, 0),
"two": (2, 1),
}[one_or_more]
document = Document(test_docx("par-hyperlinks"))
paragraph = document.paragraphs[paragraph_idx]
context.hyperlink = paragraph.hyperlinks[hyperlink_idx]
# then =====================================================
@then("hyperlink.address is the URL of the hyperlink")
def then_hyperlink_address_is_the_URL_of_the_hyperlink(context: Context):
actual_value = context.hyperlink.address
expected_value = "http://yahoo.com/"
assert actual_value == expected_value, f"expected: {expected_value}, got: {actual_value}"
@then("hyperlink.contains_page_break is {value}")
def then_hyperlink_contains_page_break_is_value(context: Context, value: str):
actual_value = context.hyperlink.contains_page_break
expected_value = {"True": True, "False": False}[value]
assert actual_value == expected_value, f"expected: {expected_value}, got: {actual_value}"
@then("hyperlink.fragment is the URI fragment of the hyperlink")
def then_hyperlink_fragment_is_the_URI_fragment_of_the_hyperlink(context: Context):
actual_value = context.hyperlink.fragment
expected_value = "linkedBookmark"
assert actual_value == expected_value, f"expected: {expected_value}, got: {actual_value}"
@then("hyperlink.runs contains only Run instances")
def then_hyperlink_runs_contains_only_Run_instances(context: Context):
actual_value = [type(item).__name__ for item in context.hyperlink.runs]
expected_value = ["Run" for _ in context.hyperlink.runs]
assert actual_value == expected_value, f"expected: {expected_value}, got: {actual_value}"
@then("hyperlink.runs has length {value}")
def then_hyperlink_runs_has_length(context: Context, value: str):
actual_value = len(context.hyperlink.runs)
expected_value = int(value)
assert actual_value == expected_value, f"expected: {expected_value}, got: {actual_value}"
@then("hyperlink.text is the visible text of the hyperlink")
def then_hyperlink_text_is_the_visible_text_of_the_hyperlink(context: Context):
actual_value = context.hyperlink.text
expected_value = "awesome hyperlink"
assert actual_value == expected_value, f"expected: {expected_value}, got: {actual_value}"
@then("hyperlink.url is {value}")
def then_hyperlink_url_is_value(context: Context, value: str):
actual_value = context.hyperlink.url
expected_value = "" if value == "''" else value
assert actual_value == expected_value, f"expected: {expected_value}, got: {actual_value}"
|
