diff options
| -rw-r--r-- | tests/images_tests.py | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/tests/images_tests.py b/tests/images_tests.py index 508d141..49784d3 100644 --- a/tests/images_tests.py +++ b/tests/images_tests.py @@ -25,3 +25,44 @@ def data_uri_encodes_images_in_base64(): assert_that(result, is_sequence( has_attrs(attributes={"src": "data:image/jpeg;base64,YWJj"}), )) + + +@istest +class ImgElementTests: + @istest + def test_when_element_does_not_have_alt_text_then_alt_attribute_is_not_set(self): + image_bytes = b"abc" + image = mammoth.documents.Image( + alt_text=None, + content_type="image/jpeg", + open=lambda: io.BytesIO(image_bytes), + ) + + @mammoth.images.img_element + def convert_image(image): + return {"src": "<src>"} + + result = convert_image(image) + + assert_that(result, is_sequence( + has_attrs(attributes={"src": "<src>"}), + )) + + @istest + def test_when_element_se_alt_text_then_alt_attribute_is_set(self): + image_bytes = b"abc" + image = mammoth.documents.Image( + alt_text="<alt>", + content_type="image/jpeg", + open=lambda: io.BytesIO(image_bytes), + ) + + @mammoth.images.img_element + def convert_image(image): + return {"src": "<src>"} + + result = convert_image(image) + + assert_that(result, is_sequence( + has_attrs(attributes={"alt": "<alt>", "src": "<src>"}), + )) |