Skip to content

Commit d5a03da

Browse files
ondrej-111scanny
authored andcommitted
hdr: add DocumentPart.drop_header_part()
1 parent 383ebd1 commit d5a03da

File tree

2 files changed

+12
-1
lines changed

2 files changed

+12
-1
lines changed

docx/parts/document.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ def document(self):
4141

4242
def drop_header_part(self, rId):
4343
"""Remove related header part identified by *rId*."""
44-
raise NotImplementedError
44+
self.drop_rel(rId)
4545

4646
def get_or_add_image(self, image_descriptor):
4747
"""

tests/parts/test_document.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,13 @@
3030

3131
class DescribeDocumentPart(object):
3232

33+
def it_can_drop_a_specified_header_part(self, drop_rel_):
34+
document_part = DocumentPart(None, None, None, None)
35+
36+
document_part.drop_header_part("rId42")
37+
38+
drop_rel_.assert_called_once_with(document_part, "rId42")
39+
3340
def it_can_save_the_package_to_a_file(self, save_fixture):
3441
document, file_ = save_fixture
3542
document.save(file_)
@@ -262,6 +269,10 @@ def styles_fixture(self, _styles_part_prop_, styles_part_, styles_):
262269
def core_properties_(self, request):
263270
return instance_mock(request, CoreProperties)
264271

272+
@pytest.fixture
273+
def drop_rel_(self, request):
274+
return method_mock(request, DocumentPart, "drop_rel", autospec=True)
275+
265276
@pytest.fixture
266277
def get_or_add_image_(self, request):
267278
return method_mock(request, DocumentPart, 'get_or_add_image')

0 commit comments

Comments
 (0)